From 9cd53bb15a9e805ba7e27f62e5895806d8099ecf Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Sun, 20 Apr 2025 09:50:47 +0000
Subject: [PATCH] fix(mcp): update FastMCP imports and server configuration for
 compatibility

---
 examples/mcp_example/mock_math_server.py |  2 +-
 swarms/structs/agent.py                  | 10 ++--------
 swarms/tools/mcp_integration.py          |  2 +-
 3 files changed, 4 insertions(+), 10 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/swarms/structs/agent.py b/swarms/structs/agent.py
index 7a723f9b..a10af6e0 100644
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@@ -47,12 +47,6 @@ from swarms.structs.safe_loading import (
 )
 from swarms.telemetry.main import log_agent_data
 from swarms.tools.base_tool import BaseTool
-
-# from swarms.tools.mcp_integration import (
-#     MCPServerSseParams,
-#     batch_mcp_flow,
-#     mcp_flow_get_tool_schema,
-# )
 from swarms.tools.tool_parse_exec import parse_and_execute_json
 from swarms.utils.any_to_str import any_to_str
 from swarms.utils.data_to_text import data_to_text
@@ -838,7 +832,7 @@ class Agent:
             # If no components available, fall back to task
             if not components and task:
                 logger.warning(
-                    "No agent details found. Using task as fallback for prompt generation."
+                    "No agent details found. Usingtask as fallback for prompt generation."
                 )
                 self.system_prompt = auto_generate_prompt(
                     task, self.llm
@@ -1869,7 +1863,7 @@ class Agent:
     # Response Filtering
     def add_response_filter(self, filter_word: str) -> None:
         """
-        Add a response filter to filter out certain words from the response
+        Add a response filter tofilter out certain words from the response
 
         Example:
         agent.add_response_filter("Trump")
diff --git a/swarms/tools/mcp_integration.py b/swarms/tools/mcp_integration.py
index dad88d75..e1fa6679 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 fastmcp import FastClientSession as ClientSession
+from fastmcp import Client as ClientSession
 from fastmcp.servers import fast_sse_client as sse_client
 
 from loguru import logger