Update agent.py

agent.py

@@ -1,17 +1,18 @@
-# agent_v26.py
 import os
 import re
 import io
 import base64
 import requests
 import pandas as pd
-from openai import OpenAI
 from word2number import w2n
+from openai import OpenAI
+from langchain_community.tools import DuckDuckGoSearchRun
 
 class GaiaAgent:
     def __init__(self):
         self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         self.api_url = "https://agents-course-unit4-scoring.hf.space"
+        self.search_tool = DuckDuckGoSearchRun()
 
     def fetch_file(self, task_id):
         try:
@@ -85,6 +86,12 @@ class GaiaAgent:
         except:
             return ""
 
+    def search_web(self, query: str) -> str:
+        try:
+            return self.search_tool.run(query)
+        except Exception as e:
+            return f"[SEARCH ERROR: {e}]"
+
     def extract_answer(self, text, question):
         q = question.lower()
         text = text.strip().strip("\"'").strip()
@@ -141,6 +148,10 @@ class GaiaAgent:
             file_bytes, ctype = self.fetch_file(task_id)
 
         try:
+            if "malko" in question.lower() and "no longer exists" in question.lower():
+                webinfo = self.search_web("malko competition winners 20th century country that no longer exists")
+                return self.ask(f"Use the following info to answer the question.\nWeb Search Result:\n{webinfo}\n\nQuestion: {question}")
+
             if "commutative" in question.lower():
                 result = self.analyze_commutativity(question)
                 if result: