Merge pull request #52 from UiPath/fix/update_remote_mcp_sample

Author: cristipufu

samples/simple-remote-mcp-agent/main.py

@@ -10,7 +10,7 @@
 from llama_index.llms.openai import OpenAI
 from llama_index.tools.mcp import McpToolSpec
 from mcp import ClientSession
-from mcp.client.sse import sse_client
+from mcp.client.streamable_http import streamablehttp_client
 
 
 # Define events
@@ -35,16 +35,31 @@ async def process_query(self, ev: UserQueryEvent) -> AgentResponseEvent:
         """Process the user query using the MCP-enabled agent."""
 
         # Initialize MCP client and tools
-        async with sse_client(
+        async with streamablehttp_client(
             url=os.getenv("UIPATH_MCP_SERVER_URL"),
             headers={"Authorization": f"Bearer {os.getenv('UIPATH_ACCESS_TOKEN')}"},
             timeout=60,
-        ) as (read, write):
+        ) as (read, write, _get_session_id_callback):
             async with ClientSession(read, write) as client_session:
                 await client_session.initialize()
                 mcp_tool_spec = McpToolSpec(client=client_session)
                 tools = await mcp_tool_spec.to_tool_list_async()
 
+                # Print name and input schema for each tool
+                print(f"Available MCP tools: {len(tools)}")
+                for i, tool in enumerate(tools, 1):
+                    print(f"{i}. {tool.metadata.name}")
+                    if tool.metadata.fn_schema:
+                        fn_schema = tool.metadata.fn_schema
+                        if fn_schema.model_fields:
+                            params = list(fn_schema.model_fields.keys())
+                        else:
+                            params = []
+                        print(f"   Parameters: {params}")
+                    else:
+                        print("   Parameters: None")
+                print("-" * 40)
+
                 # Initialize the agent
                 self.agent = FunctionAgent(
                     name="UiPath MCP Agent",

samples/simple-remote-mcp-agent/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "llama-simple-remote-mcp-agent"
-version = "0.0.6"
+version = "0.0.1"
 description = "UiPath LlamaIndex Remote MCP Agent"
 authors = [{ name = "John Doe", email = "john.doe@myemail.com" }]
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath-llamaindex>=0.0.22",
-    "llama-index-llms-openai>=0.4.4",
-    "llama-index-tools-mcp>=0.2.5"
+    "uipath-llamaindex>=0.0.30",
+    "llama-index-llms-openai>=0.4.7",
+    "llama-index-tools-mcp>=0.2.6"
 ]