diff --git a/pyproject.toml b/pyproject.toml index 5d9da9e4..d798d986 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -78,7 +78,7 @@ litellm = "*" torch = "*" httpx = "*" langchain-openai = "^0.3.14" -fastmcp = "^2.2.0" +fastmcp = {version = ">=2.0", extras = ["sse"]} [tool.poetry.scripts] swarms = "swarms.cli.main:main" diff --git a/swarms/tools/mcp_integration.py b/swarms/tools/mcp_integration.py index e1fa6679..516a9d36 100644 --- a/swarms/tools/mcp_integration.py +++ b/swarms/tools/mcp_integration.py @@ -4,7 +4,8 @@ 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.servers import fast_sse_client as sse_client +from fastmcp.client.sse import sse_client +from fastmcp.client.stdio import stdio_client, StdioServerParameters from loguru import logger import abc