diff --git a/swarms/structs/__init__.py b/swarms/structs/__init__.py index 85aa3a3f..95fa8e8e 100644 --- a/swarms/structs/__init__.py +++ b/swarms/structs/__init__.py @@ -98,8 +98,6 @@ try: from swarms.tools.mcp_unified_client import ( MCPUnifiedClient, UnifiedTransportConfig, - call_tool_streaming_sync, - execute_tool_call_streaming_unified, create_auto_config, create_http_config, create_streamable_http_config, @@ -107,8 +105,19 @@ try: create_sse_config, ) MCP_STREAMING_AVAILABLE = True + MCP_IMPORTS = [ + "MCPUnifiedClient", + "UnifiedTransportConfig", + "create_auto_config", + "create_http_config", + "create_streamable_http_config", + "create_stdio_config", + "create_sse_config", + "MCP_STREAMING_AVAILABLE", + ] except ImportError: MCP_STREAMING_AVAILABLE = False + MCP_IMPORTS = [] __all__ = [ "Agent", @@ -187,15 +196,4 @@ __all__ = [ "HierarchicalSwarm", "HeavySwarm", "CronJob", - # MCP Streaming Support - "MCPUnifiedClient", - "UnifiedTransportConfig", - "call_tool_streaming_sync", - "execute_tool_call_streaming_unified", - "create_auto_config", - "create_http_config", - "create_streamable_http_config", - "create_stdio_config", - "create_sse_config", - "MCP_STREAMING_AVAILABLE", -] +] + MCP_IMPORTS