rollback mistral

app.py

@@ -1,22 +1,20 @@
 import gradio as gr
+from openai import OpenAI
 import os
-import asyncio
-from anthropic import Anthropic
 from mcp import ClientSession, StdioServerParameters
 from mcp.client.stdio import stdio_client
+import asyncio
 from contextlib import AsyncExitStack
 
-# --- Clé API Anthropique ---
-cle_api = os.environ.get("CLE_API_ANTHROPIC")
+cle_api = os.environ.get("CLE_API_MISTRAL")
+
+# Initialisation du client Mistral (API compatible OpenAI)
+clientLLM = OpenAI(api_key=cle_api, base_url="https://api.mistral.ai/v1")
 
-# Initialisation du client Anthropic
-clientLLM = Anthropic(api_key=cle_api)
 
-# --- Event Loop ---
 loop = asyncio.new_event_loop()
 asyncio.set_event_loop(loop)
 
-# --- Wrapper MCP ---
 class MCPClientWrapper:
     def __init__(self):
         self.session = None
@@ -57,15 +55,13 @@ class MCPClientWrapper:
         tool_names = [tool["name"] for tool in self.tools]
         return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
 
-# --- Connexion au serveur MCP ---
 clientMCP = MCPClientWrapper()
 clientMCP.connect("mcp_server.py")
 print(clientMCP.tools)
 
-
-# --- Fonction chatbot reliée à Claude ---
+# Chatbot : simple écho Fonction chatbot reliée à Mistral
 def chatbot(message, history):
-    # Préparer l’historique pour Anthropique
+    # Préparer l’historique dans le format de Mistral
     messages = []
     for user_msg, bot_msg in history:
         messages.append({"role": "user", "content": user_msg})
@@ -73,26 +69,17 @@ def chatbot(message, history):
     
     messages.append({"role": "user", "content": message})
 
-    # Appel API Anthropique
-    response = clientLLM.messages.create(
-        model="claude-3-5-sonnet-20240620",   # tu peux changer pour claude-3-opus ou autre
-        max_tokens=500,
+    # Appel API Mistral
+    response = clientLLM.chat.completions.create(
+        model="mistral-small-latest", 
         messages=messages,
-        tools=clientMCP.tools if clientMCP.tools else None
+        tools=clientMCP.tools
     )
 
-    # Récupération de la réponse
-    bot_reply = ""
-    if response.content:
-        for block in response.content:
-            if block.type == "text":
-                bot_reply += block.text.strip()
-    
+    bot_reply = response.choices[0].message.content.strip()
     history.append(("Vous: " + message, "Bot: " + bot_reply))
     return history, history
     
-
-# --- UI Gradio ---
 with gr.Blocks() as demo:
 
     chatbot_ui = gr.Chatbot(label="ChatBot")
@@ -100,4 +87,4 @@ with gr.Blocks() as demo:
 
     msg.submit(chatbot, [msg, chatbot_ui], [chatbot_ui, chatbot_ui])
 
-demo.launch(debug=True)
+demo.launch(debug=True)