From eb5eca7c80a0ac95a70cd459822dce35c2c49b46 Mon Sep 17 00:00:00 2001
From: harshalmore31 <harshalmore2468@gmail.com>
Date: Sat, 8 Feb 2025 00:57:56 +0530
Subject: [PATCH] Updated __init__.py

---
 swarms/schemas/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/swarms/schemas/__init__.py b/swarms/schemas/__init__.py
index f81ae400..b193483f 100644
--- a/swarms/schemas/__init__.py
+++ b/swarms/schemas/__init__.py
@@ -1,10 +1,12 @@
 from swarms.schemas.agent_step_schemas import Step, ManySteps
-
 from swarms.schemas.agent_input_schema import AgentSchema
-
+from swarms.schemas.base_swarm_schemas import BaseSwarmSchema
+from swarms.schemas.output_schemas import OutputSchema
 
 __all__ = [
     "Step",
     "ManySteps",
     "AgentSchema",
+    "BaseSwarmSchema",
+    "OutputSchema",
 ]