Update crew.py

crew.py

@@ -7,6 +7,7 @@ import os
 import pandas as pd
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
+from crewai_tools import CodeInterpreterTool
 from google import genai
 from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
@@ -179,34 +180,7 @@ def run_crew(question, file_path):
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
-    @tool("Code Generation Tool")
-    def code_generation_tool(question: str) -> str:
-        """Given a question, generate code to answer the question.
-
-           Args:
-               question (str): Question to answer
-                
-           Returns:
-               str: Answer to the question
-               
-           Raises:
-               RuntimeError: If processing fails"""
-        try:
-            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-
-            response = client.models.generate_content(
-                model=CODE_GENERATION_MODEL,
-                contents=[question],
-                config=types.GenerateContentConfig(
-                    tools=[types.Tool(code_execution=types.ToolCodeExecution)]
-                ),
-            )
-            
-            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)}")
+    code_generation_tool = CodeInterpreterTool()
             
     @tool("Code Execution Tool")
     def code_execution_tool(question: str, file_path: str) -> str:
@@ -356,15 +330,18 @@ def run_crew(question, file_path):
 
     # Process
 
-    if file_path:
-        ext = os.path.splitext(file_path)[1].lower()
+    #if file_path:
+    #    ext = os.path.splitext(file_path)[1].lower()
 
-        if ext in (".csv", ".xls", ".xlsx", ".json", ".jsonl"):
-            df = read_file(file_path)
-            question = f"{question} File {file_path} data: {df.to_string()}"
-        else:
-            question = f"{question} File path: {file_path}."
+    #    if ext in (".csv", ".xls", ".xlsx", ".json", ".jsonl"):
+    #        df = read_file(file_path)
+    #        question = f"{question} File {file_path} data: {df.to_string()}"
+    #    else:
+    #        question = f"{question} File path: {file_path}."
 
+    if file_path:
+        question = f"{question} File path: {file_path}."
+    
     print("###")
     print(question)
     print("###")