update demo, formatting

Author: Levivus

examples/rc_basics/mcp_demo.py

@@ -1,12 +1,11 @@
 import streamlit as st
 import asyncio
 import requestcompletion as rc
-from requestcompletion.nodes.library.mcp_tool import async_from_mcp_server
+from requestcompletion.nodes.library.mcp_tool import from_mcp_server
 from requestcompletion.rc_mcp.main import MCPHttpParams
 
 # MCP server URL
 urls = [
-    "https://mcp.paypal.com/sse",
     "https://remote.mcpservers.org/fetch/mcp",
 ]
 
@@ -15,11 +14,8 @@
 
     # Initialize tools in session state
     async def get_node():
-        all_tools = set()
-        tasks = [async_from_mcp_server(MCPHttpParams(url=url)) for url in urls]
-        results = await asyncio.gather(*tasks)
-        for tools in results:
-            all_tools.update(tools)
+        servers = [from_mcp_server(MCPHttpParams(url=url)) for url in urls]
+        all_tools = set(*[server.tools for server in servers])
 
         return rc.library.tool_call_llm(
             connected_nodes=all_tools,

src/requestcompletion/nodes/library/mcp_tool.py

@@ -1,10 +1,7 @@
-from typing import Type
 
 from mcp import StdioServerParameters
-import asyncio
 
-from ...rc_mcp.main import MCPAsyncClient, MCPHttpParams, from_mcp, MCPServer
-from ...nodes.nodes import Node
+from ...rc_mcp.main import MCPHttpParams, MCPServer
 
 
 def from_mcp_server(

src/requestcompletion/rc_mcp/main.py

@@ -4,7 +4,7 @@
 from datetime import timedelta
 from typing import Any, Dict
 
-from typing_extensions import Self, Type
+from typing_extensions import Self
 
 from mcp import ClientSession, StdioServerParameters
 from mcp.client.stdio import stdio_client
@@ -139,22 +139,18 @@ def __exit__(self, exc_type, exc, tb):
         self.close()
 
     def _thread_main(self):
-        print("MCP server thread is starting...")
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
         self._loop = loop
 
         self._shutdown_event = asyncio.Event()
 
-        print("Setting up MCP server in thread...")
         self._loop.run_until_complete(self.setup())
         self._ready_event.set()
 
-        print("MCP server setup complete, waiting for shutdown...")
         loop.run_until_complete(self._shutdown_event.wait())
 
         self._loop.close()
-        print("MCP server thread has been closed.")
 
     async def setup(self):
         """
@@ -200,15 +196,16 @@ def __init__(self, **kwargs):
             self.kwargs = kwargs
 
         async def invoke(self):
-            print("Invoke!")
             try:
                 future = asyncio.run_coroutine_threadsafe(
                     client.call_tool(tool.name, self.kwargs), loop
                 )
                 result = future.result(timeout=30)
                 return result.content if hasattr(result, "content") else result
             except Exception as e:
-                raise RuntimeError(f"Tool invocation failed: {type(e).__name__}: {str(e)}") from e
+                raise RuntimeError(
+                    f"Tool invocation failed: {type(e).__name__}: {str(e)}"
+                ) from e
 
         @classmethod
         def pretty_name(cls):