Add an example of MCP

Author: RussellLuo

examples/mcp/README.md

@@ -0,0 +1,38 @@
+# Model Context Protocol
+
+This example demonstrates how to implement an agent that can use tools provided by MCP ([Model Context Protocol][1]) servers.
+
+
+## Prerequisites
+
+- Install `coagent` (see [Installation](../../README.md#installation)).
+- Start a NATS server (see [Distributed](../../README.md#distributed)).
+
+
+## Quick Start
+
+First start the MCP server in one terminal:
+
+```bash
+python examples/mcp/server.py
+```
+
+Then start the MCP agent in another terminal:
+
+```bash
+export AZURE_MODEL=csg-gpt4
+export AZURE_API_BASE=https://opencsg-us.openai.azure.com
+export AZURE_API_VERSION=2024-02-15-preview
+export AZURE_API_KEY=<YOUR API KEY>
+
+python examples/mcp/agent.py
+```
+
+Finally, communicate with the agent using the `coagent` CLI:
+
+```bash
+coagent mcp -H type:ChatMessage -F .content.content -d '{"role":"user","content":"What is the weather like in Beijing"}'
+```
+
+
+[1]: https://modelcontextprotocol.io/

examples/mcp/agent.py

@@ -0,0 +1,26 @@
+import asyncio
+
+from coagent.agents import MCPAgent
+from coagent.core import AgentSpec, idle_loop, new, set_stderr_logger
+from coagent.runtimes import NATSRuntime
+
+
+mcp = AgentSpec(
+    "mcp",
+    new(
+        MCPAgent,
+        system="""You are an agent who can use tools.""",
+        mcp_server_base_url="http://localhost:8080",
+    ),
+)
+
+
+async def main():
+    async with NATSRuntime.from_servers() as runtime:
+        await runtime.register(mcp)
+        await idle_loop()
+
+
+if __name__ == "__main__":
+    set_stderr_logger("TRACE")
+    asyncio.run(main())

examples/mcp/server.py

@@ -0,0 +1,13 @@
+from mcp.server.fastmcp import FastMCP
+
+mcp = FastMCP("Weather", port=8080)
+
+
+@mcp.tool()
+def query_weather(city: str) -> str:
+    """Query the weather in the given city."""
+    return f"The weather in {city} is sunny."
+
+
+if __name__ == "__main__":
+    mcp.run(transport="sse")