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

@@ -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"})

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"

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