Update crew.py

crew.py

@@ -15,16 +15,17 @@ from util import get_final_answer
 
 ## LLMs
 
-MANAGER_MODEL      = "gpt-4.1"
-AGENT_MODEL        = "gpt-4.1"
+MANAGER_MODEL      = "gpt-4.1-mini"
+AGENT_MODEL        = "gpt-4.1-mini"
 
-FINAL_ANSWER_MODEL = "gemini-2.0-flash"
+FINAL_ANSWER_MODEL = "gemini-2.5-flash-preview-04-17"
 
-IMAGE_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
-AUDIO_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
-VIDEO_MODEL        = "gemini-2.0-flash-thinking-exp-01-21"
-YOUTUBE_MODEL      = "gemini-2.0-flash-thinking-exp-01-21"
-DOCUMENT_MODEL     = "gemini-2.0-flash-thinking-exp-01-21"
+IMAGE_MODEL        = "gemini-2.5-flash-preview-04-17"
+AUDIO_MODEL        = "gemini-2.5-flash-preview-04-17"
+VIDEO_MODEL        = "gemini-2.5-flash-preview-04-17"
+YOUTUBE_MODEL      = "gemini-2.5-flash-preview-04-17"
+DOCUMENT_MODEL     = "gemini-2.5-flash-preview-04-17"
+CODE_MODEL         = "gemini-2.5-flash-preview-04-17"
 
 # LLM evaluation
 
@@ -45,7 +46,7 @@ def run_crew(question, file_path):
 
     web_search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
-    code_execution_tool = CodeInterpreterTool()
+    #code_execution_tool = CodeInterpreterTool()
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str:
@@ -181,7 +182,46 @@ def run_crew(question, file_path):
             return response.text
         except Exception as e:
             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.
     
+           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
+                
+           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=question,
+                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(part.text)
+                if part.executable_code is not None:
+                    print(part.executable_code.code)
+                if part.code_execution_result is not None:
+                    print(part.code_execution_result.output)
+                    return part.code_execution_result.output
+            ###
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
+            
     # Agents
 
     web_search_agent = Agent(
@@ -301,9 +341,9 @@ def run_crew(question, file_path):
     if file_path:
         question = f"{question} File path: {file_path}."
     
-        if file_path.endswith(".py"):
-            with open(f"{file_path}", "r") as file:
-                question = f"{question} File data:\n{file.read()}"
+        #if file_path.endswith(".py"):
+        #    with open(f"{file_path}", "r") as file:
+        #        question = f"{question} File data:\n{file.read()}"
     
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))