Update agent.py

agent.py

@@ -1,72 +1,91 @@
-"""
-GAIA Agent - Main Agent Implementation
-A sophisticated agent for solving GAIA benchmark questions using multiple tools.
-"""
-
-import os
-from smolagents import CodeAgent, LiteLLMModel
-from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
-from code_interpreter import CodeInterpreterTool
-from image_processing import ImageProcessingTool
-
-# Get API key from environment
-openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
-if not openrouter_api_key:
-    print("⚠️  WARNING: OPENROUTER_API_KEY not found!")
-    print("Get FREE API key: https://openrouter.ai/keys")
-    openrouter_api_key = "dummy"
-
-# Initialize the model
-model = LiteLLMModel(
-    model_id="openrouter/meta-llama/llama-3.3-70b-instruct:free",
-    api_key=openrouter_api_key,
-    temperature=0.5,
-)
-
-# Initialize tools
-search_tool = DuckDuckGoSearchTool()
-weather_info_tool = WeatherInfoTool()
-hub_stats_tool = HubStatsTool()
-code_interpreter_tool = CodeInterpreterTool()
-image_processing_tool = ImageProcessingTool()
-
-# Create GAIA Agent with all tools
-gaia_agent = CodeAgent(
-    tools=[
-        search_tool,
-        weather_info_tool,
-        hub_stats_tool,
-        code_interpreter_tool,
-        image_processing_tool,
-    ],
-    model=model,
-    add_base_tools=True,
-    planning_interval=3,
-)
-
-def run_agent(question: str) -> str:
-    """
-    Run the GAIA agent on a question and return the answer.
-    
-    Args:
-        question: The GAIA benchmark question to answer
-        
-    Returns:
-        The agent's answer as a string
-    """
-    try:
-        response = gaia_agent.run(question)
-        return response
-    except Exception as e:
-        return f"Error processing question: {str(e)}"
-
-if __name__ == "__main__":
-    # Example usage
-    test_question = "What is the capital of France?"
-    print("=" * 80)
-    print("GAIA Agent Test")
-    print("=" * 80)
-    print(f"Question: {test_question}")
-    answer = run_agent(test_question)
-    print(f"Answer: {answer}")
-
+"""
+GAIA Agent - Main Agent Implementation
+A sophisticated agent for solving GAIA benchmark questions using multiple tools.
+"""
+
+import os
+from smolagents import CodeAgent, LiteLLMModel
+from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
+from code_interpreter import CodeInterpreterTool
+from image_processing import ImageProcessingTool
+
+# Get API key from environment
+openrouter_api_key = os.environ.get("OPENROUTER_API_KEY")
+if not openrouter_api_key or openrouter_api_key == "dummy":
+    raise ValueError(
+        "⚠️  OPENROUTER_API_KEY not found or invalid!\n"
+        "Please set OPENROUTER_API_KEY environment variable.\n"
+        "Get FREE API key: https://openrouter.ai/keys\n"
+        "In Hugging Face Space: Settings → Repository secrets → Add secret"
+    )
+
+# Initialize the model
+model = LiteLLMModel(
+    model_id="openrouter/meta-llama/llama-3.3-70b-instruct:free",
+    api_key=openrouter_api_key,
+    temperature=0.5,
+)
+
+# Initialize tools
+search_tool = DuckDuckGoSearchTool()
+weather_info_tool = WeatherInfoTool()
+hub_stats_tool = HubStatsTool()
+code_interpreter_tool = CodeInterpreterTool()
+image_processing_tool = ImageProcessingTool()
+
+# Create GAIA Agent with all tools
+gaia_agent = CodeAgent(
+    tools=[
+        search_tool,
+        weather_info_tool,
+        hub_stats_tool,
+        code_interpreter_tool,
+        image_processing_tool,
+    ],
+    model=model,
+    add_base_tools=True,
+    planning_interval=3,
+)
+
+def run_agent(question: str) -> str:
+    """
+    Run the GAIA agent on a question and return the answer.
+    
+    Args:
+        question: The GAIA benchmark question to answer
+        
+    Returns:
+        The agent's answer as a string
+    """
+    try:
+        if not question or not question.strip():
+            return "Error: Empty question provided"
+        
+        response = gaia_agent.run(question)
+        
+        # Ensure we return a string
+        if response is None:
+            return "Agent returned None"
+        
+        response_str = str(response).strip()
+        if not response_str:
+            return "Agent returned empty response"
+        
+        return response_str
+    except Exception as e:
+        error_msg = f"Error processing question: {str(e)}"
+        print(error_msg)
+        import traceback
+        traceback.print_exc()
+        return error_msg
+
+if __name__ == "__main__":
+    # Example usage
+    test_question = "What is the capital of France?"
+    print("=" * 80)
+    print("GAIA Agent Test")
+    print("=" * 80)
+    print(f"Question: {test_question}")
+    answer = run_agent(test_question)
+    print(f"Answer: {answer}")
+