From 9a682a2aed2b50c27dc079d9f028e53ff9f3cde4 Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Sun, 20 Apr 2025 10:40:24 +0000
Subject: [PATCH] fix(mcp): update imports and fix server transport args for
 MCP integration

---
 examples/mcp_example/mock_math_server.py | 2 +-
 pyproject.toml                           | 3 ++-
 swarms/tools/mcp_integration.py          | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

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