From 2f47623f5d970383742c7a28c46e82363b415cd1 Mon Sep 17 00:00:00 2001
From: Kye Gomez <swarms_wd@Kyes-MacBook-Pro.local>
Date: Sat, 27 Jul 2024 18:01:22 -0400
Subject: [PATCH] [STRUCTS][Artifact][cleanup]

---
 pyproject.toml             |  2 +-
 swarms/structs/__init__.py | 72 ++++++++++++++------------------------
 2 files changed, 28 insertions(+), 46 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index eb0c1e0d..9ad22d31 100644
--- a/pyproject.toml
+++ b/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>"]
diff --git a/swarms/structs/__init__.py b/swarms/structs/__init__.py
index af603803..600e23ae 100644
--- a/swarms/structs/__init__.py
+++ b/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",
-]
+]
\ No newline at end of file