Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -425,22 +425,19 @@ class AITools():
         """
         try:
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-    
-            file = client.files.upload(file=file_path)
-    
+        
+            prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
+        
             response = client.models.generate_content(
-                model=LLM_CODE_EXECUTION,
-                contents=[file, question],
+                model=LLM_FINAL_ANSWER, 
+                contents=[prompt],
                 config=types.GenerateContentConfig(
-                    tools=[types.Tool(code_execution=types.ToolCodeExecution)],
                     thinking_config=types.ThinkingConfig(
-                        thinking_level=THINKING_LEVEL_TOOLS
+                        thinking_level=THINKING_LEVEL_FINAL_ANSWER
                     )
-                ),
+                )
             )
             
-            for part in response.candidates[0].content.parts:
-                if part.code_execution_result is not None:
-                    return part.code_execution_result.output
+            return response.text
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")