diff --git a/examples/mcp_example/mock_math_server.py b/examples/mcp_example/mock_math_server.py index f8d85e3c..3d54f21e 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", transport_kwargs={"port": 8000, "host": "0.0.0.0"}) + mcp.run(transport="sse", port=8000, host="0.0.0.0") diff --git a/pyproject.toml b/pyproject.toml index cc2f7f5a..b8915ba7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -79,7 +79,8 @@ torch = "*" httpx = "*" langchain-openai = "^0.3.14" fastmcp = {version = ">=2.0", extras = ["sse"]} -mcp = ">=1.6.0" +mcp = ">=0.3.0" +typing-extensions = "^4.13.2" [tool.poetry.scripts] swarms = "swarms.cli.main:main" diff --git a/swarms/tools/mcp_integration.py b/swarms/tools/mcp_integration.py index 242a9ecd..1df684ad 100644 --- a/swarms/tools/mcp_integration.py +++ b/swarms/tools/mcp_integration.py @@ -3,7 +3,7 @@ from __future__ import annotations from typing import Any, Dict, List, Optional from typing_extensions import NotRequired, TypedDict from contextlib import AbstractAsyncContextManager -from mcp import Client as ClientSession +from mcp import ClientSession from mcp.client.sse import sse_client from mcp.client.stdio import stdio_client, StdioServerParameters