Update agent.py

agent.py

@@ -7,22 +7,14 @@ from llama_index.core.agent.react import ReActAgent
 from llama_index.core.tools import FunctionTool
 from duckduckgo_search import DDGS
 
-# --- Custom DuckDuckGo Tool ---
-class DuckDuckGoSearchTool:
-    def __init__(self):
-        self.metadata = {
-            "name": "duckduckgo_search",
-            "description": "Search the web via DuckDuckGo."
-        }
-    def __call__(self, query: str) -> str:
-        try:
-            with DDGS() as ddg:
-                results = ddg.text(query=query, region="wt-wt", max_results=3)
-                return "\n".join(r.get('body','') for r in results if r.get('body'))
-        except Exception as e:
-            return f"ERROR: {e}"
+def duckduckgo_search(query: str) -> str:
+    try:
+        with DDGS() as ddg:
+            results = ddg.text(query=query, region="wt-wt", max_results=3)
+            return "\n".join(r.get('body', '') for r in results if r.get('body'))
+    except Exception as e:
+        return f"ERROR: {e}"
 
-# --- Tools ---
 def eval_python_code(code: str) -> str:
     try:
         return str(eval(code, {"__builtins__": {}}))
@@ -40,7 +32,6 @@ def format_gaia_answer(answer: str, question: str = "") -> str:
         ans = ans[1:-1]
     if not re.match(r'^[A-Za-z]+\.$', ans):
         ans = re.sub(r'\.$', '', ans)
-
     if question:
         if re.search(r'how many|number of|at bats|total sales|albums|output.*python|highest number', question, re.I):
             m = re.search(r'(\$?\d[\d,\.]*)', ans)
@@ -74,11 +65,10 @@ def format_gaia_answer(answer: str, question: str = "") -> str:
             return ', '.join(items)
     return ans.strip().rstrip('.')
 
-# --- LLM setup ---
-llm = OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY"))
+llm = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))  # no model= param
 
 tools = [
-    FunctionTool.from_defaults(DuckDuckGoSearchTool(), name="duckduckgo_search", description="Web search tool"),
+    FunctionTool.from_defaults(duckduckgo_search, name="duckduckgo_search", description="Web search tool"),
     FunctionTool.from_defaults(eval_python_code, name="python_eval", description="Evaluate Python code"),
     FunctionTool.from_defaults(format_gaia_answer, name="format_gaia_answer", description="GAIA output formatting")
 ]