Update agent.py

agent.py

@@ -215,6 +215,9 @@ def generate_answer(question: str) -> str:
 
             # Extract only the answer part after 'FINAL ANSWER:'
             match = re.search(r"FINAL ANSWER:\s*(.+)", raw_text, re.IGNORECASE)
-            return match.group(1).strip() if match else raw_text
+            if match:
+                return match.group(1).strip()
+            else:
+                return raw_text
 
     return "I'm not sure."