Update LG_agent.py

LG_agent.py

@@ -27,24 +27,29 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 def extract_gaia_answer(text: str) -> str:
+    """
+    Extracts just the final answer in raw form, stripping explanation and prefixes like:
+    - 'The answer is: ...'
+    - 'Answer: ...'
+    - Or just the raw line if short and valid.
+    """
     patterns = [
-        r"The answer is: (.+)",
-        r"Answer: (.+)",
-        r"^([a-z0-9,\s]+)$",
+        r"The answer is:\s*(.+)",
+        r"Answer:\s*(.+)",
+        r"^([a-z0-9\s,\-]+)$",  # simple raw line (numbers, text)
     ]
     for pattern in patterns:
         match = re.search(pattern, text.strip(), re.IGNORECASE | re.MULTILINE)
         if match:
-            answer = match.group(1).strip().lower()
-            if any(kw in answer for kw in ["incomplete", "missing", "clarify", "need more"]):
-                return match.group(1).strip()
+            return match.group(1).strip().lower()
 
-    # Add fallback keyword suggestion for vague answers
-    if "provide" in text.lower() and "ingredient" in text.lower():
-        return "The question is incomplete — please provide more information."
+    # Fallback: return first short line if it's probably the answer
+    lines = [l.strip() for l in text.strip().splitlines() if l.strip()]
+    if lines and len(lines[0]) < 80:
+        return lines[0].strip().lower()
 
-    # Fallback
-    return text.strip()
+    # Final fallback: return full text, lowercase
+    return text.strip().lower()
 
 # 3. Assistant node