Update agent.py

agent.py

@@ -401,8 +401,9 @@ class EsiAgent:
         last_msg = final_state["messages"][-1].content
 
         # If you need to strip off "FINAL ANSWER: " prefix, do it here:
-        if last_msg.startswith("FINAL ANSWER:"):
-            return last_msg[len("FINAL ANSWER:"):].strip()
+        idx = last_msg.find("FINAL ANSWER:")
+        if idx >= 0:
+          return last_msg[idx+len("FINAL ANSWER:"):].strip()
         return last_msg.strip()
 
 if __name__ == "__main__":