fix(mcp): update imports and fix server transport args for MCP integration

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

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"

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