Correct parsing of final answer

langchain_agent.py

@@ -1,4 +1,5 @@
 import os
+import re
 from dotenv import load_dotenv
 from langchain.agents import create_agent
 from langgraph.checkpoint.memory import InMemorySaver
@@ -8,6 +9,7 @@ from langchain.agents.middleware import (
     ToolCallLimitMiddleware,
 )
 from langchain.messages import HumanMessage, AIMessage
+from smolagents.local_python_executor import fix_final_answer_code
 
 load_dotenv()
 
@@ -39,6 +41,23 @@ If you are asked for a comma separated list, apply the above rules depending of
             {"messages": [HumanMessage(content=question)]},
             {"configurable": {"thread_id": "1"}, "recursion_limit": 50},
         )
-        answer = response["messages"][-1].text()
-        print(f"Agent returning answer: {answer}")
-        return answer
+        answer = response["messages"][-1].text
+        final_answer = self.extract_final_answer(answer)
+        print(f"Agent returning answer: {final_answer}")
+        return final_answer
+
+    def extract_final_answer(self, text):
+        """
+        Extracts the text following 'FINAL ANSWER:' using regex.
+        """
+        # re.DOTALL allows '.' to match newline characters
+        pattern = re.compile(r"FINAL ANSWER:\s*(.*)", re.DOTALL)
+
+        match = pattern.search(text)
+
+        if match:
+            # .group(1) retrieves the content of the first capturing group
+            # .strip() removes leading/trailing whitespace
+            return match.group(1).strip()
+        else:
+            return "No final answer provided, here is the complete text : " + text