Update agent.py

agent.py

@@ -27,8 +27,8 @@ class GaiaAgent:
         response = self.client.chat.completions.create(
             model=model,
             messages=[
-                {"role": "system", "content": "You are a precise assistant. Only return the final answer. Do not guess. Use reasoning and tools when needed."},
-                {"role": "user", "content": prompt.strip() + "\nFinal Answer:"}
+                {"role": "system", "content": "You are a precise assistant. Answer concisely and factually. Do not guess."},
+                {"role": "user", "content": prompt.strip() + "\nAnswer:"}
             ],
             temperature=0.0,
         )
@@ -60,32 +60,13 @@ class GaiaAgent:
         try:
             df = pd.read_excel(io.BytesIO(file_bytes), engine="openpyxl")
             if 'category' in df.columns and 'sales' in df.columns:
-                food_df = df[df['category'].str.lower() == 'food']
+                food_df = df[df['category'].str.lower().str.contains("food")]
                 total = food_df['sales'].sum()
                 return f"${total:.2f}"
             return "$0.00"
         except Exception:
             return "$0.00"
 
-    def analyze_commutativity(self, question):
-        try:
-            rows = re.findall(r"\|([a-e])\|([a-e\|]+)\|", question)
-            table = {}
-            for row in rows:
-                key, values = row
-                table[key] = values.strip('|').split('|')
-            elements = list(table.keys())
-            non_comm = set()
-            for i, x in enumerate(elements):
-                for j, y in enumerate(elements):
-                    a = table[x][j]
-                    b = table[y][i]
-                    if a != b:
-                        non_comm.update([x, y])
-            return ", ".join(sorted(non_comm))
-        except:
-            return ""
-
     def search_web(self, query: str) -> str:
         try:
             return self.search_tool.run(query)
@@ -107,10 +88,6 @@ class GaiaAgent:
             match = re.search(r"\b([KQBNR]?[a-h]?[1-8]?x?[a-h][1-8][+#]?)\b", text)
             return match.group(1) if match else text
 
-        if "comma separated list" in q:
-            words = re.findall(r"[a-zA-Z][a-zA-Z ]+[a-zA-Z]", text)
-            return ", ".join(sorted(set(w.strip().lower() for w in words)))
-
         if "usd with two decimal places" in q:
             match = re.search(r"\$?([0-9]+(?:\.[0-9]{1,2})?)", text)
             return f"${float(match.group(1)):.2f}" if match else "$0.00"
@@ -148,14 +125,15 @@ 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 "youtube.com" in question.lower():
+                video_id_match = re.search(r"v=([\w-]+)", question)
+                if video_id_match:
+                    search = self.search_web(f"summary or transcript of YouTube video {video_id_match.group(1)}")
+                    return self.ask(f"Based on this video content:\n{search}\n\n{question}")
 
-            if "commutative" in question.lower():
-                result = self.analyze_commutativity(question)
-                if result:
-                    return result
+            if "malko competition" in question.lower() and "no longer exists" in question.lower():
+                webinfo = self.search_web("malko competition winners 20th century nationality country that no longer exists")
+                return self.ask(f"Based on this info:\n{webinfo}\n\n{question}")
 
             if file_bytes and "image" in ctype:
                 raw = self.ask_image(file_bytes, question)
@@ -174,4 +152,4 @@ class GaiaAgent:
         except Exception as e:
             return f"[ERROR: {e}]"
 
-        return self.extract_answer(raw, question)
+        return self.extract_answer(raw, question)