Create mcp/mcp_client.py

agents/mcp/mcp_client.py

@@ -0,0 +1,57 @@
+import asyncio
+from mcp.client.sse import sse_client
+from mcp.client.session import ClientSession, DEFAULT_CLIENT_INFO
+import anyio
+
+MCP_SSE_URL = "https://bstraehle-mcp.hf.space/gradio_api/mcp/sse"
+MCP_TOOL    = "mcp_chess_analysis"
+
+async def call_mcp_tool_async(mcp_url: str, tool_name: str, arguments: dict, timeout: int = 30) -> str:
+    """Call an MCP tool asynchronously.
+    
+    Args:
+        mcp_url (str): The MCP SSE URL
+        tool_name (str): The name of the MCP tool to call
+        arguments (dict): The arguments to pass to the tool
+        timeout (int): Timeout in seconds (default: 30)
+        
+    Returns:
+        str: The tool result as a string
+        
+    Raises:
+        Exception: If the MCP call fails
+    """
+    try:
+        with anyio.fail_after(timeout):
+            async with sse_client(mcp_url) as (read, write):
+                async with ClientSession(read, write, client_info=DEFAULT_CLIENT_INFO) as session:
+                    await session.initialize()
+                    result = await session.call_tool(tool_name, arguments=arguments)
+                    
+                    if result.content:
+                        text_parts = [
+                            item.text for item in result.content 
+                            if hasattr(item, 'text')
+                        ]
+                        return " ".join(text_parts)
+                    return str(result)
+    except Exception as e:
+        raise
+
+
+def call_mcp_tool(mcp_url: str, tool_name: str, arguments: dict, timeout: int = 30) -> str:
+    """Call an MCP tool synchronously.
+    
+    Args:
+        mcp_url (str): The MCP SSE URL
+        tool_name (str): The name of the MCP tool to call
+        arguments (dict): The arguments to pass to the tool
+        timeout (int): Timeout in seconds (default: 30)
+        
+    Returns:
+        str: The tool result as a string
+        
+    Raises:
+        Exception: If the MCP call fails
+    """
+    return asyncio.run(call_mcp_tool_async(mcp_url, tool_name, arguments, timeout))