Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -384,6 +384,42 @@ class AITools():
         Returns:
             str: Answer to the question
             
+        Raises:
+            RuntimeError: If processing fails
+        """
+        try:
+            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+    
+            file = client.files.upload(file=file_path)
+    
+            response = client.models.generate_content(
+                model=LLM_CODE_EXECUTION,
+                contents=[file, question],
+                config=types.GenerateContentConfig(
+                    tools=[types.Tool(code_execution=types.ToolCodeExecution)],
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_TOOLS
+                    )
+                ),
+            )
+            
+            for part in response.candidates[0].content.parts:
+                if part.code_execution_result is not None:
+                    return part.code_execution_result.output
+        except Exception as 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, generate the final answer.
+    
+        Args:
+            question (str): Question to answer
+            answer (str): The initial answer
+            
+        Returns:
+            str: Final answer to the question
+            
         Raises:
             RuntimeError: If processing fails
         """