Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -17,9 +17,10 @@ from agents.models.llms import (
     LLM_CODE_GENERATION,
     LLM_CODE_EXECUTION,
     LLM_WEB_BROWSER,
+    LLM_FINAL_ANSWER,
     THINKING_LEVEL_TOOLS
 )
-from agents.models.prompts import PROMPT_IMG_TO_FEN
+from agents.models.prompts import PROMPT_IMG_TO_FEN, PROMPT_FINAL_ANSWER
 from crewai.tools import tool
 from crewai_tools import StagehandTool
 from google import genai
@@ -440,4 +441,21 @@ class AITools():
             
             return response.text
         except Exception as e:
-            raise RuntimeError(f"Processing failed: {str(e)}")
+            raise RuntimeError(f"Processing failed: {str(e)}")
+
+    def get_final_answer(question, answer):
+        client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+    
+        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
+                )
+            )
+        )
+        
+        return response.text