Update app.py

app.py

@@ -428,20 +428,26 @@ def ask_gpt(query, apikey,history,ui_session_id):
 def gpt_answer(api_key, query, model="gpt-3.5-turbo-1106", system_prompt="Use the provided References to answer the user Question. If the provided document do not contain the elements to answer the user question, just say 'No information.'."):
     if 'gpt' in model:
         client = OpenAI( api_key=api_key)
-
+    
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": query},
+        
+            ],
+            model=model,
+        )
+        return chat_completion.choices[0].message.content
+        
     if 'mistral' in model:
         client = MistralClient(api_key=api_key)
-    
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": query},
-    
-        ],
-        model=model,
-    )
-    return chat_completion.choices[0].message.content
-
+        chat_response = client.chat(
+            model=model,
+            messages=[
+                ChatMessage(role="system", content=system_prompt),
+                ChatMessage(role="user", content=query)],
+        )
+        return chat_response.choices[0].message.content
 
 
 def add_line_breaks(input_string, line_length=100):