From d7e8849bd978841dfaf7ad596008ffeae294800c Mon Sep 17 00:00:00 2001 From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com> Date: Sun, 20 Apr 2025 10:38:13 +0000 Subject: [PATCH] fix(mcp): update imports to use mcp package and fix FastMCP server transport in mock_math_server.py --- examples/mcp_example/mock_math_server.py | 2 +- pyproject.toml | 1 + swarms/tools/mcp_integration.py | 6 +++--- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/examples/mcp_example/mock_math_server.py b/examples/mcp_example/mock_math_server.py index 3d54f21e..f8d85e3c 100644 --- a/examples/mcp_example/mock_math_server.py +++ b/examples/mcp_example/mock_math_server.py @@ -22,4 +22,4 @@ def divide(a: int, b: int) -> float: if __name__ == "__main__": print("Starting Mock Math Server on port 8000...") - mcp.run(transport="sse", port=8000, host="0.0.0.0") + mcp.run(transport="sse", transport_kwargs={"port": 8000, "host": "0.0.0.0"}) diff --git a/pyproject.toml b/pyproject.toml index d798d986..cc2f7f5a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -79,6 +79,7 @@ torch = "*" httpx = "*" langchain-openai = "^0.3.14" fastmcp = {version = ">=2.0", extras = ["sse"]} +mcp = ">=1.6.0" [tool.poetry.scripts] swarms = "swarms.cli.main:main" diff --git a/swarms/tools/mcp_integration.py b/swarms/tools/mcp_integration.py index 516a9d36..242a9ecd 100644 --- a/swarms/tools/mcp_integration.py +++ b/swarms/tools/mcp_integration.py @@ -3,9 +3,9 @@ from __future__ import annotations from typing import Any, Dict, List, Optional from typing_extensions import NotRequired, TypedDict from contextlib import AbstractAsyncContextManager -from fastmcp import Client as ClientSession -from fastmcp.client.sse import sse_client -from fastmcp.client.stdio import stdio_client, StdioServerParameters +from mcp import Client as ClientSession +from mcp.client.sse import sse_client +from mcp.client.stdio import stdio_client, StdioServerParameters from loguru import logger import abc