Update agent.py

agent.py

@@ -1,3 +1,80 @@
+Agent V31
+1
+2
+3
+4
+5
+6
+7
+8
+9
+10
+11
+12
+13
+14
+15
+16
+17
+18
+19
+20
+21
+22
+23
+24
+25
+26
+27
+28
+29
+30
+31
+32
+33
+34
+35
+36
+37
+38
+39
+40
+41
+42
+43
+44
+45
+46
+47
+48
+49
+50
+51
+52
+53
+54
+55
+56
+57
+58
+59
+60
+61
+62
+63
+64
+65
+66
+67
+68
+69
+70
+71
+72
+73
+74
+75
+76
 # agent_v43.py — Najlepsze cechy z V18–V34: precyzja, retry fallback, stabilność
 import os
 import re
@@ -74,81 +151,4 @@ class GaiaAgent:
                 return "[NO FOOD SALES DATA]"
             return content.decode("utf-8", errors="ignore")[:3000]
         except Exception as e:
-            return f"[FILE ERROR: {e}]"
 
-    def extract_commutativity_set(self, table_txt):
-        try:
-            lines = table_txt.splitlines()
-            S, table = [], {}
-            for l in lines:
-                if l.startswith("|*"):
-                    S = l.strip().split("|")[2:]
-                elif l.startswith("|"):
-                    parts = l.strip().split("|")[1:-1]
-                    table[parts[0]] = parts[1:]
-            fail = set()
-            for x in S:
-                for y in S:
-                    if table[x][S.index(y)] != table[y][S.index(x)]:
-                        fail |= {x, y}
-            return ", ".join(sorted(fail))
-        except:
-            return "[COMMUTATIVE ERROR]"
-
-    def extract_ingredients(self, text):
-        try:
-            candidates = re.findall(r"[a-zA-Z]+(?:\s[a-zA-Z]+)?", text)
-            blocked = {"add", "combine", "cook", "stir", "remove", "cool", "mixture", "saucepan", "until", "heat", "dash"}
-            clean = [c.lower() for c in candidates if c.lower() not in blocked and len(c.split()) <= 3]
-            return ", ".join(sorted(set(clean)))
-        except:
-            return text[:100]
-
-    def format_answer(self, answer, question):
-        q = question.lower()
-        raw = answer.strip().strip("\"'")
-        if "commutative" in q:
-            return self.extract_commutativity_set(question)
-        if "ingredient" in q:
-            return self.extract_ingredients(raw)
-        if "algebraic notation" in q:
-            m = re.search(r"[KQBNR]?[a-h]?[1-8]?x?[a-h][1-8][+#]?", raw)
-            return m.group(0) if m else raw
-        if "usd" in q:
-            m = re.search(r"\$?\d+(\.\d{2})", raw)
-            return f"${m.group()}" if m else "$0.00"
-        if "award number" in q:
-            m = re.search(r"80NSSC[0-9A-Z]+", raw)
-            return m.group(0) if m else raw
-        if "first name" in q:
-            return raw.split()[0]
-        try:
-            return str(w2n.word_to_num(raw))
-        except:
-            m = re.search(r"\d+", raw)
-            return m.group(0) if m else raw
-
-    def retry_fallback(self, question):
-        try:
-            prompt = f"""Answer concisely and factually:
-Question: {question}"""
-            return self.ask(prompt)
-        except:
-            return "[RETRY FAILED]"
-
-    def __call__(self, question, task_id=None):
-        try:
-            content, ctype = self.fetch_file(task_id) if task_id else (None, None)
-            context = self.handle_file(content, ctype, question) if content else self.search_context(question)
-            raw = self.ask(f"Use this context to answer:
-{context}
-
-Question:
-{question}
-Answer:")
-            if not raw or "[ERROR" in raw or "step execution failed" in raw:
-                retry = self.retry_fallback(question)
-                return self.format_answer(retry, question)
-            return self.format_answer(raw, question)
-        except Exception as e:
-            return f"[AGENT ERROR: {e}]"