samples: add simple remote mcp agent

Author: cristipufu

samples/simple-remote-mcp-agent/.env.example

@@ -0,0 +1,2 @@
+OPENAI_API_KEY=xxx
+UIPATH_MCP_SERVER_URL=xxx

samples/simple-remote-mcp-agent/agent.mermaid

@@ -0,0 +1,13 @@
+flowchart TD
+    step__done["_done"]:::stepStyle
+    step_process_query["process_query"]:::stepStyle
+    event_UserQueryEvent([<p>UserQueryEvent</p>]):::defaultEventStyle
+    event_AgentResponseEvent([<p>AgentResponseEvent</p>]):::stopEventStyle
+    event_AgentResponseEvent --> step__done
+    step_process_query --> event_AgentResponseEvent
+    event_UserQueryEvent --> step_process_query
+    classDef stepStyle fill:#f2f0ff,line-height:1.2
+    classDef externalStyle fill:#f2f0ff,line-height:1.2
+    classDef defaultEventStyle fill-opacity:0
+    classDef stopEventStyle fill:#bfb6fc
+    classDef inputRequiredStyle fill:#f2f0ff,line-height:1.2

samples/simple-remote-mcp-agent/llama_index.json

@@ -0,0 +1,7 @@
+{
+  "dependencies": ["."],
+  "workflows": {
+    "agent": "main.py:workflow"
+  },
+  "env": ".env"
+}

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

@@ -0,0 +1,66 @@
+import os
+
+from llama_index.core.agent.workflow import FunctionAgent
+from llama_index.core.workflow import (
+    StartEvent,
+    StopEvent,
+    Workflow,
+    step,
+)
+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
+
+
+# Define events
+class UserQueryEvent(StartEvent):
+    """Event representing a user query to the MCP agent."""
+
+    query: str
+
+
+class AgentResponseEvent(StopEvent):
+    """Event representing the agent's response."""
+
+    response: str
+
+
+# Define the workflow
+class MCPAgentWorkflow(Workflow):
+    """Workflow that uses MCP tools to respond to user queries."""
+
+    @step
+    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(
+            url=os.getenv("UIPATH_MCP_SERVER_URL"),
+            headers={"Authorization": f"Bearer {os.getenv('UIPATH_ACCESS_TOKEN')}"},
+            timeout=60,
+        ) as (read, write):
+            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()
+
+                # Initialize the agent
+                self.agent = FunctionAgent(
+                    name="UiPath MCP Agent",
+                    description="An agent that can interact with MCP tools",
+                    llm=OpenAI(model="gpt-4o"),
+                    tools=tools,
+                    system_prompt="""You are a helpful assistant.
+                    You have access to various tools through MCP (Model Context Protocol).
+                    Use these tools to help users with their tasks.
+
+                    Always be helpful and provide clear, actionable responses.""",
+                )
+
+                response = await self.agent.run(user_msg=ev.query)
+
+        return AgentResponseEvent(response=str(response))
+
+
+workflow = MCPAgentWorkflow(timeout=300, verbose=True)

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

@@ -0,0 +1,12 @@
+[project]
+name = "llama-simple-remote-mcp-agent"
+version = "0.0.6"
+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"
+]

samples/simple-remote-mcp-agent/uipath.json

@@ -0,0 +1,37 @@
+{
+  "entryPoints": [
+    {
+      "filePath": "agent",
+      "uniqueId": "f967e1ad-bab9-4fcc-a335-e82cbe7c4ef1",
+      "type": "agent",
+      "input": {
+        "type": "object",
+        "properties": {
+          "query": {
+            "title": "Query",
+            "type": "string"
+          }
+        },
+        "required": [
+          "query"
+        ]
+      },
+      "output": {
+        "type": "object",
+        "properties": {
+          "response": {
+            "title": "Response",
+            "type": "string"
+          }
+        },
+        "required": [
+          "response"
+        ]
+      }
+    }
+  ],
+  "bindings": {
+    "version": "2.0",
+    "resources": []
+  }
+}