Update agent.py

agent.py

@@ -1,3 +1,4 @@
+# agent_v24.py
 import os
 import re
 import requests
@@ -6,15 +7,6 @@ import io
 import pandas as pd
 from openai import OpenAI
 from word2number import w2n
-from difflib import get_close_matches
-
-KNOWN_INGREDIENTS = {
-    'salt', 'sugar', 'water', 'vanilla extract', 'lemon juice', 'cornstarch', 'granulated sugar', 'ripe strawberries'
-}
-
-KNOWN_VEGETABLES = {
-    'acorns', 'broccoli', 'celery', 'green beans', 'lettuce', 'sweet potatoes', 'peanuts'
-}
 
 class GaiaAgent:
     def __init__(self):
@@ -35,23 +27,12 @@ class GaiaAgent:
                 return match.group(1) if match else text
 
         if "algebraic notation" in question.lower():
-            match = re.search(r"\b(Qd1\+?|Nf3\+?|[KQBNR]?[a-h]?[1-8]?x?[a-h][1-8][+#]?)\b", text)
+            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 "commutative" in question.lower():
-            return "a, b, d, e"
-
-        if "vegetables" in question.lower():
-            return ", ".join(sorted(KNOWN_VEGETABLES))
-
-        if "ingredients" in question.lower():
-            found = set()
-            for word in text.lower().split(','):
-                word = word.strip()
-                match = get_close_matches(word, KNOWN_INGREDIENTS, n=1, cutoff=0.6)
-                if match:
-                    found.add(match[0])
-            return ", ".join(sorted(found))
+        if "comma separated list" in question.lower():
+            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 question:
             match = re.search(r"\$?([0-9]+(?:\.[0-9]{1,2})?)", text)
@@ -66,8 +47,6 @@ class GaiaAgent:
             return ", ".join(str(n) for n in nums)
 
         if "at bats" in question.lower():
-            if "Mickey Rivers" in text:
-                return "565"
             match = re.search(r"(\d{3,4})", text)
             return match.group(1) if match else text
 
@@ -76,23 +55,12 @@ class GaiaAgent:
             return match.group(1) if match else text
 
         if "first name" in question.lower():
-            if "Malko" in question:
-                return "Uroš"
             return text.split()[0]
 
         if "NASA award number" in question:
             match = re.search(r"(80NSSC[0-9A-Z]{6,7})", text)
             return match.group(1) if match else text
 
-        if "who did the actor" in question.lower():
-            return "Cezary"
-
-        if "equine veterinarian" in question.lower():
-            return "Strasinger"
-
-        if "youtube.com" in question.lower():
-            return "3"
-
         return text
 
     def fetch_file(self, task_id):