[STRUCTS][Artifact][cleanup]

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "5.4.2"
+version = "5.4.3"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

swarms/structs/__init__.py

@@ -1,13 +1,4 @@
 from swarms.schemas.plan import Plan
-from swarms.schemas.schemas import (
-    Artifact,
-    ArtifactUpload,
-    StepInput,
-    StepOutput,
-    StepRequestBody,
-    TaskInput,
-    TaskRequestBody,
-)
 from swarms.schemas.step import Step
 from swarms.structs.agent import Agent
 from swarms.structs.agent_job import AgentJob
@@ -96,7 +87,6 @@ from swarms.structs.graph_workflow import (
     Edge,
 )
 
-
 __all__ = [
     "Agent",
     "AgentJob",
@@ -110,26 +100,43 @@ __all__ = [
     "ConcurrentWorkflow",
     "Conversation",
     "GroupChat",
+    "HiearchicalSwarm",
     "MajorityVoting",
     "majority_voting",
     "most_frequent",
     "parse_code_completion",
     "Message",
+    "MessagePool",
     "MultiAgentCollaboration",
     "MultiProcessWorkflow",
     "MultiThreadedWorkflow",
-    "Plan",
+    "SwarmNetwork",
+    "AgentRearrange",
+    "rearrange",
     "RecursiveWorkflow",
-    "Artifact",
-    "ArtifactUpload",
-    "StepInput",
-    "StepOutput",
-    "StepRequestBody",
-    "TaskInput",
-    "TaskRequestBody",
+    "RoundRobinSwarm",
     "SequentialWorkflow",
+    "Task",
+    "TaskQueueBase",
+    "synchronized_queue",
+    "detect_markdown",
+    "distribute_tasks",
+    "extract_key_from_json",
+    "extract_tokens_from_text",
+    "find_agent_by_id",
+    "find_token_in_text",
+    "parse_tasks",
+    "YamlModel",
+    "create_yaml_schema_from_dict",
+    "get_type_name",
+    "pydantic_type_to_yaml_schema",
+    "MixtureOfAgents",
+    "GraphWorkflow",
+    "Node",
+    "NodeType",
+    "Edge",
+    "Plan",
     "Step",
-    "SwarmNetwork",
     "broadcast",
     "circular_swarm",
     "exponential_swarm",
@@ -148,29 +155,4 @@ __all__ = [
     "sigmoid_swarm",
     "staircase_swarm",
     "star_swarm",
-    "Task",
-    "TaskQueueBase",
-    "synchronized_queue",
-    "detect_markdown",
-    "distribute_tasks",
-    "extract_key_from_json",
-    "extract_tokens_from_text",
-    "find_agent_by_id",
-    "find_token_in_text",
-    "parse_tasks",
-    "AgentRearrange",
-    "get_type_name",
-    "create_yaml_schema_from_dict",
-    "pydantic_type_to_yaml_schema",
-    "YamlModel",
-    "MessagePool",
-    "rearrange",
-    "RoundRobinSwarm",
-    "HiearchicalSwarm",
-    # "AgentLoadBalancer",
-    "MixtureOfAgents",
-    "GraphWorkflow",
-    "Node",
-    "NodeType",
-    "Edge",
-]
+]