diff --git a/swarms/structs/__init__.py b/swarms/structs/__init__.py new file mode 100644 index 00000000..8287a960 --- /dev/null +++ b/swarms/structs/__init__.py @@ -0,0 +1,64 @@ +""" This module lists all the data structures used in the swarms package.""" +from swarms.structs.agent import Agent +from swarms.structs.autoscaler import AutoScaler +from swarms.structs.base import BaseStructure +from swarms.structs.base_swarm import AbstractSwarm +from swarms.structs.base_workflow import BaseWorkflow +from swarms.structs.concurrent_workflow import ConcurrentWorkflow +from swarms.structs.conversation import Conversation +from swarms.structs.groupchat import GroupChat, GroupChatManager +from swarms.structs.model_parallizer import ModelParallelizer +from swarms.structs.multi_agent_collab import MultiAgentCollaboration +from swarms.structs.nonlinear_workflow import NonlinearWorkflow +from swarms.structs.recursive_workflow import RecursiveWorkflow +from swarms.structs.schemas import ( + Artifact, + ArtifactUpload, + StepInput, + TaskInput, +) +from swarms.structs.sequential_workflow import SequentialWorkflow +from swarms.structs.swarm_net import SwarmNetwork +from swarms.structs.utils import ( + distribute_tasks, + extract_key_from_json, + extract_tokens_from_text, + find_agent_by_id, + find_token_in_text, + parse_tasks, +) + +__all__ = [ + "Agent", + "SequentialWorkflow", + "AutoScaler", + "Conversation", + "TaskInput", + "Artifact", + "ArtifactUpload", + "StepInput", + "SwarmNetwork", + "ModelParallelizer", + "MultiAgentCollaboration", + "AbstractSwarm", + "GroupChat", + "GroupChatManager", + "parse_tasks", + "find_agent_by_id", + "distribute_tasks", + "find_token_in_text", + "extract_key_from_json", + "extract_tokens_from_text", + "ConcurrentWorkflow", + "RecursiveWorkflow", + "NonlinearWorkflow", +<<<<<<< HEAD + "BaseStruct", # from swarms/structs/base.py +======= + "BaseWorkflow", +<<<<<<< HEAD +>>>>>>> db2dbf3 ([CODE QUALITY]) +======= + "BaseStructure" +>>>>>>> a9b3d7d ([CLEANUP][__init__]) +]