Update test_agent.py

test_agent.py

@@ -4,24 +4,20 @@ import pathlib
 import dotenv
 from smolagents import CodeAgent, OpenAIServerModel
 
-# Load environment
 if pathlib.Path(".env").exists():
     dotenv.load_dotenv(".env")
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", None)
 
 if not OPENAI_API_KEY:
-    print("Please set OPENAI_API_KEY in your .env file")
     exit(1)
 
-# Create model
 model = OpenAIServerModel(
     model_id="gpt-4o-mini",
     api_key=OPENAI_API_KEY,
 )
 
 def test_simple_calculation():
-    """Test that calc agent can solve a simple math problem and call final_answer()"""
     from prompt import calc_agent_prompt
     
     calc_agent = CodeAgent(
@@ -46,7 +42,6 @@ def test_simple_calculation():
         return False
 
 def test_simple_search():
-    """Test that web search agent can answer a simple question and call final_answer()"""
     from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool
 
     from prompt import web_search_agent_prompt