[CLEANUP]

kyegomez · Apr 10, 2024 · 864fc69 · 864fc69
1 parent b36d0cf
commit 864fc69
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/neo_sapiens/__init__.py b/neo_sapiens/__init__.py
@@ -0,0 +1,25 @@
+from neo_sapiens.few_shot_prompts import (
+    data,
+    data1,
+    data2,
+    data3,
+    orchestrator_prompt_agent,
+)
+from neo_sapiens.hass_schema import (
+    AgentSchema,
+    ToolSchema,
+    parse_json_from_input,
+    create_agents,
+)
+
+__all__ = [
+    "data",
+    "data1",
+    "data2",
+    "data3",
+    "orchestrator_prompt_agent",
+    "AgentSchema",
+    "ToolSchema",
+    "parse_json_from_input",
+    "create_agents",
+]
diff --git a/few_shot_prompts.py → neo_sapiens/few_shot_prompts.py b/few_shot_prompts.py → neo_sapiens/few_shot_prompts.py
diff --git a/hass_schema.py → neo_sapiens/hass_schema.py b/hass_schema.py → neo_sapiens/hass_schema.py
@@ -7,7 +7,7 @@
 from pydantic import BaseModel, Field
 from swarms import Agent, Anthropic, SwarmNetwork
 
-from few_shot_prompts import (
+from neo_sapiens.few_shot_prompts import (
     data,
     data1,
     data2,