parsing final answer

custom_agent.py

@@ -182,8 +182,20 @@ class CustomAgent:
                 }
             )
             result = answer["messages"][-1].content
-            print(f"CustomAgent returning answer: {result}")
-            return result
+
+            if "FINAL ANSWER: " in result:
+                final_answer_start = result.find("FINAL ANSWER: ") + len(
+                    "FINAL ANSWER: "
+                )
+                extracted_answer = result[final_answer_start:].strip()
+                print(f"CustomAgent extracted answer: {extracted_answer}")
+                return extracted_answer
+            else:
+                print(
+                    f"CustomAgent returning full answer (no FINAL ANSWER found): {result}"
+                )
+                return result
+
         except Exception as e:
             print(f"Error in CustomAgent: {e}")
             return f"Error: {e}"
@@ -194,3 +206,9 @@ if __name__ == "__main__":
     agent(
         "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     )
+    agent(
+        "How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?"
+    )
+    agent(
+        "In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?"
+    )