Merge pull request #23 from leonvanbokhorst/smol-agent

Enhance: Update quick smol agent config and prompt
leonvanbokhorst · Jan 7, 2025 · 2667e23 · 2667e23
2 parents 479e21d + 9b9127b
commit 2667e23
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/smol_agents/quick_smol.py b/src/smol_agents/quick_smol.py
@@ -7,13 +7,13 @@ def run_story_generation():
     agent = CodeAgent(
         tools=[DuckDuckGoSearchTool()],
         model=model,
-        #additional_authorized_imports=["requests", "random"],
-        planning_interval=1,
-        max_iterations=5,
+        additional_authorized_imports=["math"],
+        planning_interval=3,
+        max_iterations=20,
     )
 
     result = agent.run(
-        """Find a way to escape from the matrix."""
+        """How many apples go in a tiny bushel?"""
     )
 
     print("\nFinal Story:")