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())