first agent version

app.py

@@ -1,10 +1,11 @@
 import os
 import gradio as gr
+from griffe import LogLevel
 import requests
 import inspect
 import pandas as pd
 
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, FinalAnswerTool
+from smolagents import CodeAgent, OpenAIServerModel, WebSearchTool, FinalAnswerTool, VisitWebpageTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -25,21 +26,18 @@ SYSTEM_PROMPT = """
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        
+        print("Billy Agent initialized.")
         model = OpenAIServerModel(
-            model_id="gemini-2.0-flash-lite",
+            model_id='gemini-2.5-flash-lite',
             # Google Gemini OpenAI-compatible API base URL
             api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
             api_key=os.getenv("GEMINI_API_KEY"),
         )
-        
-        print("Billy Agent initialized.")
         self.billy = CodeAgent(
             model=model,
-            tools=[DuckDuckGoSearchTool(), FinalAnswerTool()],
+            tools=[WebSearchTool(), FinalAnswerTool(), VisitWebpageTool()],
             instructions=SYSTEM_PROMPT,
-            planning_interval=3,
-            max_steps=5,
+            max_steps=10,
             add_base_tools=True
         )
     def __call__(self, question: str) -> str: