Update app.py

app.py

@@ -5,7 +5,7 @@ import inspect
 import pandas as pd
 
 # smolagents imports
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
+from smolagents import CodeAgent, InferenceClientModel, tool
 import re
 from typing import Optional, Union, Any
 import json
@@ -235,18 +235,19 @@ class GAIAAgent:
     def __init__(self):
         print("GAIAAgent initializing with smolagents...")
         
-        # Initialize the model (using HuggingFace free inference API)
+        # Initialize the model (using HuggingFace Inference Client)
         try:
-            self.model = HfApiModel()
-            print("✅ Model initialized successfully")
+            # Use a powerful model for better reasoning on GAIA benchmark
+            model_id = "meta-llama/Llama-3.3-70B-Instruct"
+            self.model = InferenceClientModel(model_id=model_id)
+            print(f"✅ Model initialized successfully: {model_id}")
         except Exception as e:
             print(f"❌ Error initializing model: {e}")
-            # Fallback to a basic model
-            self.model = HfApiModel()
+            # Fallback to default model
+            self.model = InferenceClientModel()
         
-        # Initialize tools
-        self.tools = [
-            DuckDuckGoSearchTool(),
+        # Initialize tools (custom tools + base tools from smolagents)
+        self.custom_tools = [
             visit_webpage,
             calculate_math,
             analyze_data,
@@ -257,8 +258,9 @@ class GAIAAgent:
         # Create the CodeAgent with enhanced capabilities
         try:
             self.agent = CodeAgent(
-                tools=self.tools,
+                tools=self.custom_tools,
                 model=self.model,
+                add_base_tools=True,  # Adds DuckDuckGoSearchTool and other base tools
                 additional_authorized_imports=[
                     'requests', 'bs4', 'json', 'csv', 'math', 'statistics', 
                     're', 'urllib.parse', 'base64', 'datetime', 'calendar'
@@ -266,7 +268,7 @@ class GAIAAgent:
                 max_steps=10,  # Allow multiple reasoning steps
                 verbosity_level=1  # Reduce verbosity for cleaner output
             )
-            print("✅ GAIA Agent initialized successfully with enhanced tools")
+            print("✅ GAIA Agent initialized successfully with enhanced tools and base toolkit")
         except Exception as e:
             print(f"❌ Error initializing agent: {e}")
             raise e
@@ -483,12 +485,14 @@ with gr.Blocks(title="GAIA Agent Evaluation") as demo:
         **Enhanced Agent for GAIA Benchmark Certification**
 
         This enhanced agent uses Hugging Face's **smolagents** framework with multiple specialized tools:
-        - 🔍 **Web Search**: DuckDuckGoSearchTool for finding information
+        - 🔍 **Web Search**: DuckDuckGoSearchTool (from base toolkit) for finding information
+        - 🐍 **Python Interpreter**: Code execution capabilities (from base toolkit)
         - 🌐 **Web Scraping**: Custom webpage visitor for content extraction  
         - 🧮 **Mathematics**: Advanced calculation capabilities
         - 📊 **Data Analysis**: Statistical analysis of numerical data
         - 🔢 **Number Extraction**: Intelligent number parsing from text
         - 📝 **Text Analysis**: Counting and text processing utilities
+        - 🤖 **LLM Model**: Llama-3.3-70B-Instruct for advanced reasoning
 
         **Instructions:**
         1. 🔄 **Clone this space** and customize the agent as needed