Update cool_agent.py

cool_agent.py

@@ -138,4 +138,4 @@ def create_agent(model_id="gpt-4o-mini"):
         managed_agents=[search_agent],
     )
 
-    return {"agent": manager_agent, "text_inspection_tool": text_inspection_tool, "visualizer": visualizer}
+    return {"agent": manager_agent, "text_inspection_tool": text_inspection_tool, "visualizer": visualizer, "model": model}