refactor

test_agent.py

@@ -1,9 +1,3 @@
-#!/usr/bin/env python3
-
-"""
-Simple test script to verify that our agents work correctly with final_answer()
-"""
-
 import os
 import pathlib
 
@@ -22,7 +16,7 @@ if not OPENAI_API_KEY:
 
 # Create model
 model = OpenAIServerModel(
-    model_id="gpt-4o-mini",  # Using a more cost-effective model for testing
+    model_id="gpt-4o-mini",
     api_key=OPENAI_API_KEY,
 )
 
@@ -34,7 +28,7 @@ def test_simple_calculation():
         model=model,
         prompt_templates=calc_agent_prompt,
         additional_authorized_imports=["pandas", "numpy"],
-        tools=[]  # No special tools needed for basic calculations
+        tools=[]
     )
     
     question = "What is the result of 5 + 3 + 1294.678?"