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