From 95d8c4e46017e62188790de269497d3c9de5ec9a Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Fri, 15 Aug 2025 00:00:36 +0300
Subject: [PATCH] Update mcp_unified_client.py

---
 swarms/tools/mcp_unified_client.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/swarms/tools/mcp_unified_client.py b/swarms/tools/mcp_unified_client.py
index 37d9a6fd..bc4c35fa 100644
--- a/swarms/tools/mcp_unified_client.py
+++ b/swarms/tools/mcp_unified_client.py
@@ -763,6 +763,27 @@ async def example_unified_usage():
         logger.error(f"Error getting tools: {e}")
 
 
+# Export constants for availability checking
+MCP_STREAMING_AVAILABLE = MCP_AVAILABLE and STREAMABLE_HTTP_AVAILABLE
+
+# Export all public functions and classes
+__all__ = [
+    "MCPUnifiedClient",
+    "UnifiedTransportConfig", 
+    "create_auto_config",
+    "create_http_config",
+    "create_streamable_http_config",
+    "create_stdio_config",
+    "create_sse_config",
+    "MCP_STREAMING_AVAILABLE",
+    "STREAMABLE_HTTP_AVAILABLE",
+    "HTTPX_AVAILABLE",
+    "MCP_AVAILABLE",
+    "call_tool_streaming_sync",
+    "execute_tool_call_streaming_unified",
+]
+
+
 if __name__ == "__main__":
     # Run example
     asyncio.run(example_unified_usage())