Update __init__.py

pull/1005/merge^2
CI-DEV 2 months ago committed by GitHub
parent 4515203865
commit baa2e5d99d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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",
]

Loading…
Cancel
Save