Sborole commited on
Commit
40f7c1a
·
verified ·
1 Parent(s): 6138bdf

Update tools/FinalAnswerTool.py

Browse files
Files changed (1) hide show
  1. tools/FinalAnswerTool.py +5 -0
tools/FinalAnswerTool.py CHANGED
@@ -16,5 +16,10 @@ class FinalAnswerTool(Tool):
16
  output_type = "string"
17
 
18
  def forward(self, answer: str) -> str:
 
 
 
 
 
19
  final_answer = f"FINAL ANSWER: {answer}"
20
  return final_answer
 
16
  output_type = "string"
17
 
18
  def forward(self, answer: str) -> str:
19
+ answer = re.sub(r'\\boxed\{([^}]+)\}', r'\1', answer) # \boxed{2} -> 2
20
+ answer = re.sub(r'[\$\{\}\\]', '', answer) # Remove other LaTeX symbols
21
+ answer = re.sub(r'[^A-Za-z0-9,]', ' ', answer) # Keep letters, digits, commas
22
+ if not answer:
23
+ return "NA"
24
  final_answer = f"FINAL ANSWER: {answer}"
25
  return final_answer