From baa2e5d99dfe877a8b8277a83703c230ec5a75bf Mon Sep 17 00:00:00 2001 From: CI-DEV <154627941+IlumCI@users.noreply.github.com> Date: Mon, 4 Aug 2025 20:25:31 +0300 Subject: [PATCH] Update __init__.py --- swarms/structs/__init__.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) 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", ]