Update cool_agent.py

cool_agent.py

@@ -85,6 +85,7 @@ def create_agent(model_id="gpt-4o-mini"):
 
     text_limit = 100000
     browser = SimpleTextBrowser(**BROWSER_CONFIG)
+    text_inspection_tool = TextInspectorTool(model, text_limit)
     WEB_TOOLS = [
         LinkupSearchTool(),
         # VisitTool(browser),
@@ -93,12 +94,12 @@ def create_agent(model_id="gpt-4o-mini"):
         # FinderTool(browser),
         # FindNextTool(browser),
         # ArchiveSearchTool(browser),
-        TextInspectorTool(model, text_limit),
+        text_inspection_tool,
     ]
     search_agent = ToolCallingAgent(
         model=model,
         tools=WEB_TOOLS,
-        max_steps=2,#20,
+        max_steps=5,#20,
         verbosity_level=1,#2,
         planning_interval=4,
         name="search_agent",
@@ -128,11 +129,11 @@ def create_agent(model_id="gpt-4o-mini"):
     manager_agent = CodeAgent(
         model=model,
         tools=[visualizer, TextInspectorTool(model, text_limit)],
-        max_steps=2, #12,
+        max_steps=5, #12,
         verbosity_level=1, #2,
         additional_authorized_imports=AUTHORIZED_IMPORTS,
         planning_interval=4,
         managed_agents=[search_agent],
     )
 
-    return manager_agent
+    return {"agent": manager_agent, "text_inspection_tool": text_inspection_tool, "visualizer": visualizer}