diff --git a/swarms/structs/__init__.py b/swarms/structs/__init__.py index 08d4e889..98402831 100644 --- a/swarms/structs/__init__.py +++ b/swarms/structs/__init__.py @@ -108,9 +108,26 @@ from swarms.structs.swarming_architectures import ( star_swarm, ) from swarms.structs.hierarchical_structured_communication_framework import ( - HierarchicalStructuredCommunicationFramework as HierarchicalStructuredCommSwarm, + HierarchicalStructuredCommunicationFramework, + HierarchicalStructuredCommunicationGenerator, + HierarchicalStructuredCommunicationEvaluator, + HierarchicalStructuredCommunicationRefiner, + HierarchicalStructuredCommunicationSupervisor, + StructuredMessage, + HierarchicalOrder, + EvaluationResult, + StructuredMessageSchema, + EvaluationResultSchema, + GeneratorResponseSchema, + EvaluatorResponseSchema, + RefinerResponseSchema, + CommunicationType, + AgentRole, ) +# Convenience alias(fixes old code if any was left out in the wild) +HierarchicalStructuredCommunicationSwarm = HierarchicalStructuredCommunicationFramework + __all__ = [ "Agent", "BaseStructure", @@ -189,7 +206,21 @@ __all__ = [ "HierarchicalSwarm", "HeavySwarm", "CronJob", - "HierarchicalStructuredCommSwarm", + "HierarchicalStructuredCommunicationSwarm", + "HierarchicalStructuredCommunicationGenerator", + "HierarchicalStructuredCommunicationEvaluator", + "HierarchicalStructuredCommunicationRefiner", + "HierarchicalStructuredCommunicationSupervisor", + "StructuredMessage", + "HierarchicalOrder", + "EvaluationResult", + "StructuredMessageSchema", + "EvaluationResultSchema", + "GeneratorResponseSchema", + "EvaluatorResponseSchema", + "RefinerResponseSchema", + "CommunicationType", + "AgentRole", # Stopping conditions "check_done", "check_finished",