Update agent.py

agent.py

@@ -1,12 +1,12 @@
-# agent_v25.py
+# agent_v26.py
 import os
 import re
 import io
 import base64
 import requests
 import pandas as pd
-from word2number import w2n
 from openai import OpenAI
+from word2number import w2n
 
 class GaiaAgent:
     def __init__(self):
@@ -26,7 +26,7 @@ class GaiaAgent:
         response = self.client.chat.completions.create(
             model=model,
             messages=[
-                {"role": "system", "content": "You are a precise assistant. Return only the final answer. Do not explain."},
+                {"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:"}
             ],
             temperature=0.0,
@@ -66,6 +66,25 @@ class GaiaAgent:
         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 extract_answer(self, text, question):
         q = question.lower()
         text = text.strip().strip("\"'").strip()
@@ -81,14 +100,9 @@ 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 "ingredients" in q or "comma separated list" in q:
-            items = re.findall(r"[a-zA-Z]+(?: [a-zA-Z]+)?", text)
-            return ", ".join(sorted(set(i.lower() for i in items)))
-
-        if "vegetables" in q:
-            veggies = ['acorns', 'broccoli', 'celery', 'green beans', 'lettuce', 'peanuts', 'sweet potatoes']
-            found = [v for v in veggies if v in text.lower()]
-            return ", ".join(sorted(found))
+        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)
@@ -127,6 +141,11 @@ class GaiaAgent:
             file_bytes, ctype = self.fetch_file(task_id)
 
         try:
+            if "commutative" in question.lower():
+                result = self.analyze_commutativity(question)
+                if result:
+                    return result
+
             if file_bytes and "image" in ctype:
                 raw = self.ask_image(file_bytes, question)
             elif file_bytes and ("audio" in ctype or task_id.endswith(".mp3")):