diff --git a/swarms/schemas/__init__.py b/swarms/schemas/__init__.py index 7eb2ff5d..9d8dfa29 100644 --- a/swarms/schemas/__init__.py +++ b/swarms/schemas/__init__.py @@ -3,10 +3,16 @@ from swarms.schemas.mcp_schemas import ( MCPConnection, MultipleMCPConnections, ) +from swarms.schemas.hierarchical_structured_communication_schemas import ( + HierarchicalStructuredCommConnection, + MultipleHierarchicalStructuredCommConnections, +) __all__ = [ "Step", "ManySteps", "MCPConnection", "MultipleMCPConnections", + "HierarchicalStructuredCommConnection", + "MultipleHierarchicalStructuredCommConnections", ]