Update LG_agent.py

LG_agent.py

@@ -27,32 +27,23 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 def extract_gaia_answer(text: str) -> str:
-    """
-    Extract only the final answer from the LLM output using common patterns like:
-    - 'The answer is: ...'
-    - 'Answer: ...'
-    - bullet points
-    - Or just return raw answer if short and plain
-    
-    Strips everything else to conform with GAIA expectations.
-    """
-    # Common patterns GAIA outputs follow
     patterns = [
         r"The answer is: (.+)",
         r"Answer: (.+)",
-        r"^([a-z0-9,\s]+)$",  # raw answer line
+        r"^([a-z0-9,\s]+)$",
     ]
     for pattern in patterns:
         match = re.search(pattern, text.strip(), re.IGNORECASE | re.MULTILINE)
         if match:
-            return match.group(1).strip()
-    
-    # Fallback: if it's a single short line, return it
-    lines = text.strip().splitlines()
-    if len(lines) == 1 and len(lines[0]) < 100:
-        return lines[0].strip()
-    
-    # Fallback: return full (stripped) content
+            answer = match.group(1).strip().lower()
+            if any(kw in answer for kw in ["incomplete", "missing", "clarify", "need more"]):
+                return match.group(1).strip()
+
+    # 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 text.strip()
 
 # 3. Assistant node