Update utils/mistral.py

utils/mistral.py

@@ -65,12 +65,13 @@ 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):
-        response += message.choices[0].delta.content
+    #response = ""
+    #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True):#, temperature=0.35):
+    #    response += message.choices[0].delta.content
         
     data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False)#, temperature=0.35)
     print("This is without stream data ",data.choices[0].message.content)
+    response = data.choices[0].message.content
     
     try:
         clean_response = Data_Cleaner(response)
@@ -104,11 +105,12 @@ 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):
-        response += message.choices[0].delta.content
+    #response = ""
+    #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=4096, stream=True):#, temperature=0.35):
+    #    response += message.choices[0].delta.content
     data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False)#, temperature=0.35)
     print("This is without stream data ",data.choices[0].message.content)
+    response = data.choices[0].message.content
     
     try:
         clean_response = Data_Cleaner(response)
@@ -143,12 +145,13 @@ 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):
-        response += message.choices[0].delta.content
+    #response = ""
+    #for message in client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=True):#, temperature=0.35):
+    #    response += message.choices[0].delta.content
 
     data = client.chat_completion(messages=[system_role, user_prompt], max_tokens=3000, stream=False)#, temperature=0.35)
     print("This is without stream data ",data.choices[0].message.content)
+    response = data.choices[0].message.content
     
     # Handle cases where the response might have formatting issues
     try: