From 0d8c717cef36bf14f7eaf774bb8bc1da92f77819 Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Fri, 15 Aug 2025 15:49:58 +0300
Subject: [PATCH] Update __init__.py

---
 swarms/structs/__init__.py | 26 ++++++++++++--------------
 1 file changed, 12 insertions(+), 14 deletions(-)

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