Update main.py

main.py

@@ -19,6 +19,8 @@ from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_pinecone import PineconeVectorStore
 
+from openai import OpenAI
+
 from offres_emploi import Api
 from offres_emploi.utils import dt_to_str_iso
 from langchain_community.chat_message_histories import ChatMessageHistory
@@ -170,7 +172,7 @@ def Connexion_Mistral():
     #llm = HuggingFaceEndpoint(
     #    repo_id=repo_id, max_new_tokens=3000, temperature=0.5, top_p=0.7, task="text2text-generation", streaming=True
     #)
-    llm = ChatOpenAI(model_name="mistralai/Mistral-Small-3.1-24B-Instruct-2503", base_url=os.environ['BASEURL_RENNES_API_KEY'], api_key=os.environ['ENDPOINT_RENNES_API_KEY'])
+    llm = OpenAI(model="mistralai/Mistral-Small-3.1-24B-Instruct-2503", base_url=os.environ['BASEURL_RENNES_API_KEY'], api_key=os.environ['ENDPOINT_RENNES_API_KEY'])
 
     return llm