refactor(agents): Agent file on agents root

raestrada · Oct 28, 2024 · ef2ef09 · ef2ef09
1 parent a970f73
commit ef2ef09
Show file tree

Hide file tree

Showing 11 changed files with 11 additions and 403 deletions.
diff --git a/storycraftr/agent/retrieval.py b/storycraftr/agent/retrieval.py
@@ -1,4 +1,4 @@
-from storycraftr.agent.story.agents import create_message, get_thread
+from storycraftr.agent.agents import create_message, get_thread
 from rich.console import Console
 from rich.progress import Progress
 
@@ -130,7 +130,7 @@ def handle_failed_prompt(assistant, original_prompt: str) -> str:
 
 # Example usage
 if __name__ == "__main__":
-    from storycraftr.agent.story.agents import create_or_get_assistant
+    from storycraftr.agent.agents import create_or_get_assistant
 
     # Initialize or retrieve the assistant
     assistant = create_or_get_assistant(book_path=None)