Merge pull request #159 from ks6088ts-labs/feature/issue-158_mcp-servers

add a custome mcp server to tool call agent

Author: ks6088ts

.vscode/mcp.json

@@ -1,5 +1,9 @@
 {
   "servers": {
+    "math_server": {
+      "command": "uv",
+      "args": ["run", "python", "template_langgraph/mcps/math_server.py"]
+    },
     "playwright": {
       "command": "npx",
       "args": ["-y", "@playwright/mcp@latest", "--caps=vision"]

Makefile

@@ -14,7 +14,7 @@ info: ## show information
 
 .PHONY: install-deps-dev
 install-deps-dev: ## install dependencies for development
-	uv sync --all-extras
+	uv sync --all-groups
 	uv run pre-commit install
 
 .PHONY: install-deps

pyproject.toml

@@ -47,9 +47,7 @@ docs = [
     "mkdocs-material>=9.6.14",
     "mkdocs-static-i18n>=1.3.0",
 ]
-
-[tool.uv]
-dev-dependencies = [
+dev = [
     "coverage>=7.8.0",
     "jupyterlab>=4.4.2",
     "langgraph-cli[inmem]>=0.3.6",

template_langgraph/mcps/__init__.py

@@ -0,0 +1,25 @@
+from mcp.server.fastmcp import FastMCP
+
+from template_langgraph.loggers import get_logger
+
+logger = get_logger(__name__)
+logger.setLevel("DEBUG")
+
+mcp = FastMCP("Math")
+
+
+@mcp.tool()
+def add(a: int, b: int) -> int:
+    """Add two numbers"""
+    return a + b
+
+
+@mcp.tool()
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers"""
+    return a * b
+
+
+if __name__ == "__main__":
+    logger.info("Starting Math MCP server...")
+    mcp.run(transport="stdio")

template_langgraph/mcps/math_server.py

@@ -31,6 +31,7 @@ def __init__(
         if settings is None:
             settings = get_mcp_settings()
         self.settings = settings
+        self.client = None
 
     def get_tools(self) -> list[BaseTool]:
         if self.settings.mcp_config_path == "":
@@ -39,6 +40,6 @@ def get_tools(self) -> list[BaseTool]:
             config = json.load(f)
             for _, value in config["servers"].items():
                 value["transport"] = "stdio"
-            client = MultiServerMCPClient(config["servers"])
-            tools = asyncio.run(client.get_tools())
-            return tools
+            self.client = MultiServerMCPClient(config["servers"])
+            self.tools = asyncio.run(self.client.get_tools())
+            return self.tools