test

app.py

@@ -3,6 +3,8 @@ from openai import OpenAI
 import os
 from mcp import ClientSession, StdioServerParameters
 from mcp.client.stdio import stdio_client
+import asyncio
+from contextlib import AsyncExitStack
 
 cle_api = os.environ.get("CLE_API_MISTRAL")
 
@@ -30,33 +32,55 @@ def chatbot(message, history):
     return history, history
 
 
-def connect(self) -> str:
-    
-    server_params = StdioServerParameters(
-        command='python',
-        args=['mcp_server.py'],
-        env={"PYTHONIOENCODING": "utf-8", "PYTHONUNBUFFERED": "1"}
-    )
-    
-    stdio_transport = self.exit_stack.enter_async_context(stdio_client(server_params))
-    self.stdio, self.write = stdio_transport
-    
-    self.session = self.exit_stack.enter_async_context(ClientSession(self.stdio, self.write))
-    self.session.initialize()
+loop = asyncio.new_event_loop()
+asyncio.set_event_loop(loop)
+
+class MCPClientWrapper:
+    def __init__(self):
+        self.session = None
+        self.exit_stack = None
+        self.tools = []
     
-    response = self.session.list_tools()
-    self.tools = [{ 
-        "name": tool.name,
-        "description": tool.description,
-        "input_schema": tool.inputSchema
-    } for tool in response.tools]
+    def connect(self, server_path: str) -> str:
+        return loop.run_until_complete(self._connect(server_path))
     
-    tool_names = [tool["name"] for tool in self.tools]
-    return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
+    async def _connect(self, server_path: str) -> str:
+        if self.exit_stack:
+            await self.exit_stack.aclose()
+        
+        self.exit_stack = AsyncExitStack()
+        
+        is_python = server_path.endswith('.py')
+        command = "python" if is_python else "node"
+        
+        server_params = StdioServerParameters(
+            command=command,
+            args=[server_path],
+            env={"PYTHONIOENCODING": "utf-8", "PYTHONUNBUFFERED": "1"}
+        )
+        
+        stdio_transport = await self.exit_stack.enter_async_context(stdio_client(server_params))
+        self.stdio, self.write = stdio_transport
+        
+        self.session = await self.exit_stack.enter_async_context(ClientSession(self.stdio, self.write))
+        await self.session.initialize()
+        
+        response = await self.session.list_tools()
+        self.tools = [{ 
+            "name": tool.name,
+            "description": tool.description,
+            "input_schema": tool.inputSchema
+        } for tool in response.tools]
+        
+        tool_names = [tool["name"] for tool in self.tools]
+        return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
+
+client = MCPClientWrapper()
     
 with gr.Blocks() as demo:
 
-    print(connect())
+    client.connect("mcp_server.py")
+    gr.Markdown(f"Connected to MCP server. Available tools: {', '.join([tool['name'] for tool in client.tools])}")
 
     chatbot_ui = gr.Chatbot(label="ChatBot")
     msg = gr.Textbox(placeholder="Écrivez un message...")