fix: imports

HRI-EU · Nov 1, 2024 · 02edc70 · 02edc70
1 parent da26535
commit 02edc70
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/tulip_agent/agents/dfs_tulip_agent.py b/src/tulip_agent/agents/dfs_tulip_agent.py
@@ -47,7 +47,8 @@
     TREE_TULIP_SYSTEM_PROMPT,
     TREE_TULIP_TASK_PROMPT,
 )
-from tulip_agent.task_tree import Task, Tool
+from tulip_agent.task import Task
+from tulip_agent.tool import Tool
 from tulip_agent.tool_library import ToolLibrary
 
 from .tulip_agent import TulipAgent

diff --git a/tests/test_task_tree.py b/tests/test_task_tree.py
@@ -29,7 +29,8 @@
 #
 import unittest
 
-from tulip_agent.task_tree import Task, Tool
+from tulip_agent.task import Task
+from tulip_agent.tool import Tool
 
 
 class TestCore(unittest.TestCase):