Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -491,6 +491,44 @@ class AITools():
             print(f"🤖 AITools: algebraic_notation_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
+    @tool("Final Answer Tool")
+    def final_answer_tool(question: str, answer: str) -> str:
+        """Given a question and initial answer, get the final answer.
+    
+        Args:
+            question (str): The question
+            answer (str): The initial answer
+            
+        Returns:
+            str: Final answer
+            
+        Raises:
+            RuntimeError: If processing fails
+        """    
+        print(f"🤖 AITools: get_final_answer: question={question}, answer={answer}")
+
+        try:
+            client = AITools._get_client()
+        
+            prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
+
+            response = client.models.generate_content(
+                model=LLM_FINAL_ANSWER, 
+                contents=[prompt],
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_FINAL_ANSWER
+                    )
+                )
+            )
+    
+            result = response.text.strip()
+            print(f"🤖 AITools: get_final_answer: result={result}")
+            return result
+        except Exception as e:
+            print(f"🤖 AITools: get_final_answer: result={str(e)}")
+            raise RuntimeError(f"Processing failed: {str(e)}")
+            
     def get_final_answer(question, answer):
         print(f"🤖 AITools: get_final_answer: question={question}, answer={answer}")