Update utils/mistral.py

utils/mistral.py

@@ -46,7 +46,7 @@ def Model_ProfessionalDetails_Output(resume, client):
     }
     user_prompt = {
     "role": "user",
-    "content": f'''Act as a resume parser for the following text given in text: {resume}
+    "content": f'''<s>[INST] Act as a resume parser for the following text given in text: {resume}
     Extract the text in the following output JSON string as:
     {{
         "professional": {{
@@ -61,6 +61,7 @@ def Model_ProfessionalDetails_Output(resume, client):
         }}
     }}
     output:
+    [/INST]</s>
     ''' 
     }
 
@@ -85,7 +86,7 @@ def Model_EducationalDetails_Output(resume, client):
     }
     user_prompt = {
     "role": "user",
-    "content": f'''Act as a resume parser for the following text given in text: {resume}
+    "content": f'''<s>[INST] Act as a resume parser for the following text given in text: {resume}
     Extract the text in the following output JSON string as:
     {{
         "educational": {{                        
@@ -96,6 +97,7 @@ def Model_EducationalDetails_Output(resume, client):
         }}
     }}
     output:
+    [/INST]</s>
     ''' 
     }
 
@@ -119,7 +121,7 @@ def Model_PersonalDetails_Output(resume, client):
     }
     user_prompt = {
     "role": "user",
-    "content": f'''Act as a resume parser for the following text given in text: {resume}
+    "content": f'''<s>[INST] Act as a resume parser for the following text given in text: {resume}
     Extract the text in the following output JSON string as:
     {{
         "personal": {{
@@ -131,6 +133,7 @@ def Model_PersonalDetails_Output(resume, client):
         }}
     }} 
     output:
+    [/INST]</s>
     '''
     }