Update crew.py

crew.py

@@ -2,13 +2,11 @@ import os
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
 from crewai_tools import (
-    CodeInterpreterTool,
     SerperDevTool,
     WebsiteSearchTool
 )
 from google import genai
 from google.genai import types
-#from openai import OpenAI
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
@@ -46,7 +44,6 @@ def run_crew(question, file_path):
 
     web_search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
-    #code_execution_tool = CodeInterpreterTool()
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str:
@@ -184,44 +181,33 @@ def run_crew(question, file_path):
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Code Execution Tool")
-    def code_execution_tool(question: str, file_path: str) -> str:
-        """Answer a question about a Python code file.
+    def code_execution_tool(file_path: str) -> str:
+        """Execute a Python code file and return the result.
     
            Args:
-               question (str): Question about a Pyton code file
                file_path (str): The Python code file path
                 
            Returns:
-               str: Answer to the question about the Python code file
+               str: The code execution result
                 
            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=CODE_MODEL,
-                contents=[file, question],
+                contents=[file],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
                 ),
             )
             
             for part in response.candidates[0].content.parts:
-                if part.text is not None:
-                    print("###")
-                    print(part.text)
-                if part.executable_code is not None:
-                    print("###")
-                    print(part.executable_code.code)
                 if part.code_execution_result is not None:
-                    print("###")
-                    print(part.code_execution_result.output)
                     return part.code_execution_result.output
-            ###
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
             
@@ -295,8 +281,8 @@ def run_crew(question, file_path):
     
     code_execution_agent = Agent(
         role="Code Execution Agent",
-        goal="Write and/or execute Python code to help answer question \"{question}\"",
-        backstory="As an expert code execution assistant, you write and/or execute Python code to help answer the question.",
+        goal="Execute Python code file to help answer question \"{question}\"",
+        backstory="As an expert Python code execution assistant, you execute the code file to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,