diff --git a/swarms/structs/__init__.py b/swarms/structs/__init__.py index 0241a2c1..85aa3a3f 100644 --- a/swarms/structs/__init__.py +++ b/swarms/structs/__init__.py @@ -93,6 +93,23 @@ from swarms.structs.swarming_architectures import ( star_swarm, ) +# MCP Streaming Support +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, + create_stdio_config, + create_sse_config, + ) + MCP_STREAMING_AVAILABLE = True +except ImportError: + MCP_STREAMING_AVAILABLE = False + __all__ = [ "Agent", "BaseStructure", @@ -170,4 +187,15 @@ __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", ]