Update crew.py

crew.py

@@ -1,10 +1,6 @@
 import os
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
-#from crewai_tools import (
-#    SerperDevTool,
-#    WebsiteSearchTool
-#)
 from google import genai
 from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
@@ -44,9 +40,6 @@ CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 def run_crew(question, file_path):
     # Tools
 
-    #web_search_tool = SerperDevTool()
-    #web_rag_tool = WebsiteSearchTool()
-
     @tool("Web Search Tool")
     def web_search_tool(question: str) -> str:
         """Search the web to answer a question.
@@ -60,15 +53,6 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            #client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-            
-            #response = client.models.generate_content(
-            #    model=IMAGE_MODEL, 
-            #    contents=[file, question]
-            #)
-          
-            #return response.text
-            ###
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
             
             response = client.models.generate_content(
@@ -80,7 +64,6 @@ def run_crew(question, file_path):
             )
 
             return response.text
-            ###
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
@@ -194,7 +177,7 @@ def run_crew(question, file_path):
     @tool("Document Analysis Tool")
     def document_analysis_tool(question: str, file_path: str) -> str:
         """Answer a question about a document file. Supported document types include:
-           .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js
+           .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py
     
            Args:
                question (str): Question about a document file
@@ -341,7 +324,7 @@ def run_crew(question, file_path):
 
     document_analysis_agent = Agent(
         role="Document Analysis Agent",
-        goal="Analyze document of type .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js to help answer question \"{question}\"",
+        goal="Analyze document of type .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py to help answer question \"{question}\"",
         backstory="As an expert document analysis assistant, you analyze the document to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
@@ -400,7 +383,7 @@ def run_crew(question, file_path):
                 audio_analysis_agent, 
                 video_analysis_agent, 
                 youtube_analysis_agent, 
-                document_analysis_agent,
+                #document_analysis_agent,
                 code_generation_agent,
                 code_execution_agent],
         manager_agent=manager_agent,
@@ -413,10 +396,6 @@ 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()}"
-    
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))