Update utils/mistral.py

utils/mistral.py

@@ -66,8 +66,11 @@ def Model_ProfessionalDetails_Output(resume, client):
     }
 
     response = ""
-    for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True, temperature=0.35, return_full_text=False):
+    for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True, temperature=0.35, response_format={"type": "json"}):
         response += message.choices[0].delta.content
+        
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35, response_format={"type": "json"})
+    print("This is without stream data ",data)
     
     try:
         clean_response = Data_Cleaner(response)
@@ -102,8 +105,10 @@ def Model_EducationalDetails_Output(resume, client):
     }
 
     response = ""
-    for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True, temperature=0.35, return_full_text=False):
+    for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True, temperature=0.35, response_format={"type": "json"}):
         response += message.choices[0].delta.content
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35, response_format={"type": "json"})
+    print("This is without stream data ",data)
     
     try:
         clean_response = Data_Cleaner(response)
@@ -139,15 +144,12 @@ def Model_PersonalDetails_Output(resume, client):
 
     # Response
     response = ""
-    for message in client.chat_completion(
-        messages=[system_role, user_prompt],
-        max_tokens=3000,
-        stream=True,
-        temperature=0.35,
-        return_full_text=False
-    ):
+    for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=True, temperature=0.35, response_format={"type": "json"}):
         response += message.choices[0].delta.content
 
+    data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False, temperature=0.35, response_format={"type": "json"})
+    print("This is without stream data ",data)
+    
     # Handle cases where the response might have formatting issues
     try:
         #print('The Og response:-->',response)