diff --git a/examples/mcp/servers/okx_crypto_server.py b/examples/mcp/servers/okx_crypto_server.py index a7e3247c..f082c353 100644 --- a/examples/mcp/servers/okx_crypto_server.py +++ b/examples/mcp/servers/okx_crypto_server.py @@ -117,4 +117,4 @@ def get_okx_crypto_volume(symbol: str) -> str: if __name__ == "__main__": # Run the server on port 8000 (you can change this to any available port) - mcp.run(transport="sse") + mcp.run(transport="streamable-http") diff --git a/swarms/tools/mcp_client_tools.py b/swarms/tools/mcp_client_tools.py index e70d0e45..bd42f8cf 100644 --- a/swarms/tools/mcp_client_tools.py +++ b/swarms/tools/mcp_client_tools.py @@ -11,7 +11,6 @@ from typing import Any, Dict, List, Literal, Optional, Union from litellm.types.utils import ChatCompletionMessageToolCall from loguru import logger from mcp import ClientSession -from mcp.client.sse import sse_client try: from mcp.client.streamable_http import streamablehttp_client @@ -315,7 +314,7 @@ def get_mcp_client(transport, url, headers=None, timeout=5, **kwargs): url, headers=headers, timeout=timeout, **kwargs ) else: - return sse_client( + return streamablehttp_client( url, headers=headers, timeout=timeout, **kwargs )