Update cool_agent.py

cool_agent.py

@@ -47,6 +47,7 @@ AUTHORIZED_IMPORTS = [
     "datetime",
     "fractions",
     "csv",
+    "wiki",
 ]
 
 # load_dotenv(override=True)
@@ -127,8 +128,8 @@ def create_agent(model_id="gpt-4o-mini"):
     manager_agent = CodeAgent(
         model=model,
         tools=[visualizer, TextInspectorTool(model, text_limit)],
-        max_steps=12,
-        verbosity_level=2,
+        max_steps=2, #12,
+        verbosity_level=1, #2,
         additional_authorized_imports=AUTHORIZED_IMPORTS,
         planning_interval=4,
         managed_agents=[search_agent],