Update tools/FinalAnswerTool.py

tools/FinalAnswerTool.py

@@ -4,22 +4,25 @@ from src.utils import InputTokenRateLimiter
 from smolagents.tools import Tool
 from litellm import completion
 import os
+import re
 
 settings = Settings()
 print(settings.llm_model_id)
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides the exact, final answer to the given question."
+    description = "Provides the exact, few comma separated  words or a single final answer to the given question."
     inputs = {
         "answer": {"type": "string", "description": "The final, correctly formatted answer string."},
     }
     output_type = "string"
 
     def forward(self, answer: str) -> str:
-        answer = re.sub(r'\\boxed\{([^}]+)\}', r'\1', answer)  # \boxed{2} -> 2
+        match = re.search(r'boxed\{([^}]+)\}', answer)
+        if match:
+            answer = match.group(1)
         answer = re.sub(r'[\$\{\}\\]', '', answer)              # Remove other LaTeX symbols
         answer = re.sub(r'[^A-Za-z0-9,]', ' ', answer)         # Keep letters, digits, commas
         if not answer:
             return "NA"
-        final_answer = f"FINAL ANSWER: {answer}"
+        final_answer = f"{str(answer)}"
         return final_answer