Update crew.py

crew.py

@@ -47,7 +47,7 @@ def run_crew(question, file_path):
 
     @tool("Web Search Tool")
     def web_search_tool(question: str) -> str:
-        """Search the web to answer a question.
+        """Given a question, search the web to answer the question.
     
            Args:
                question (str): Question to answer
@@ -74,7 +74,7 @@ def run_crew(question, file_path):
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str:
-        """Answer a question about an image file.
+        """Given a question and image file, analyze the image to answer the question.
     
            Args:
                question (str): Question about an image file
@@ -101,7 +101,7 @@ def run_crew(question, file_path):
 
     @tool("Audio Analysis Tool")
     def audio_analysis_tool(question: str, file_path: str) -> str:
-        """Answer a question about an audio file.
+        """Given a question and audio file, analyze the audio to answer the question.
     
            Args:
                question (str): Question about an audio file
@@ -128,7 +128,7 @@ def run_crew(question, file_path):
 
     @tool("Video Analysis Tool")
     def video_analysis_tool(question: str, file_path: str) -> str:
-        """Answer a question about a video file.
+        """Given a question and video file, analyze the video to answer the question.
     
            Args:
                question (str): Question about a video file
@@ -155,7 +155,7 @@ def run_crew(question, file_path):
             
     @tool("YouTube Analysis Tool")
     def youtube_analysis_tool(question: str, url: str) -> str:
-        """Answer a question about a YouTube video.
+        """Given a question and YouTube URL, analyze the video to answer the question.
     
            Args:
                question (str): Question about a YouTube video
@@ -180,11 +180,12 @@ def run_crew(question, file_path):
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Code Generation Tool")
-    def code_generation_tool(question: str) -> str:
-        """Given a question with json data, generate code to answer the question.
+    def code_generation_tool(question: str, json_data: str) -> str:
+        """Given a question and JSON data, generate and execute code to answer the question.
 
            Args:
                question (str): Question to answer
+               json_data (str): The JSON data
 
            Returns:
                str: Answer to the question
@@ -196,7 +197,7 @@ def run_crew(question, file_path):
 
             response = client.models.generate_content(
                 model=CODE_GENERATION_MODEL,
-                contents=[question],
+                contents=[f"{question}\n{json_data}"],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
                 ),
@@ -210,11 +211,11 @@ def run_crew(question, file_path):
             
     @tool("Code Execution Tool")
     def code_execution_tool(question: str, file_path: str) -> str:
-        """Given a question and .py Python file, execute the file to answer the question.
+        """Given a question and Python file, execute the file to answer the question.
     
            Args:
                question (str): Question to answer
-               file_path (str): The .py Python file path
+               file_path (str): The Python file path
                 
            Returns:
                str: Answer to the question
@@ -244,8 +245,8 @@ def run_crew(question, file_path):
 
     web_search_agent = Agent(
         role="Web Search Agent",
-        goal="Search the web to help answer the question: {question}",
-        backstory="As an expert web search assistant, you search the web to help answer the question.",
+        goal="Given a question, search the web and answer the question: {question}",
+        backstory="As an expert web search assistant, you search the web to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -255,8 +256,8 @@ def run_crew(question, file_path):
 
     image_analysis_agent = Agent(
         role="Image Analysis Agent",
-        goal="Analyze image file to help answer the question: {question}",
-        backstory="As an expert image analysis assistant, you analyze the image file to help answer the question.",
+        goal="Given a question and image file, analyze the image and answer the question: {question}",
+        backstory="As an expert image analysis assistant, you analyze the image to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -266,8 +267,8 @@ def run_crew(question, file_path):
 
     audio_analysis_agent = Agent(
         role="Audio Analysis Agent",
-        goal="Analyze audio file to help answer the question: {question}",
-        backstory="As an expert audio analysis assistant, you analyze the audio file to help answer the question.",
+        goal="Given a question and audio file, analyze the audio and answer the question: {question}",
+        backstory="As an expert audio analysis assistant, you analyze the audio to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -277,8 +278,8 @@ def run_crew(question, file_path):
 
     video_analysis_agent = Agent(
         role="Video Analysis Agent",
-        goal="Analyze video file to help answer the question: {question}",
-        backstory="As an expert video analysis assistant, you analyze the video file to help answer the question.",
+        goal="Given a question and video file, analyze the video and answer the question: {question}",
+        backstory="As an expert video analysis assistant, you analyze the video file to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -288,8 +289,8 @@ def run_crew(question, file_path):
     
     youtube_analysis_agent = Agent(
         role="YouTube Analysis Agent",
-        goal="Analyze YouTube video to help answer the question: {question}",
-        backstory="As an expert YouTube analysis assistant, you analyze the video to help answer the question.",
+        goal="Given a question and YouTube video, analyze the video and answer the question: {question}",
+        backstory="As an expert YouTube analysis assistant, you analyze the video to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
@@ -299,8 +300,8 @@ def run_crew(question, file_path):
 
     code_generation_agent = Agent(
         role="Code Generation Agent",
-        goal="Given a question with json data, generate code to help answer the question: {question}",
-        backstory="As an expert Python code generation assistant, you generate code to help answer the question.",
+        goal="Given a question and JSON data, generate and execute code to answer the question: {question}",
+        backstory="As an expert Python code generation assistant, you generate and execute code to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,
@@ -310,8 +311,8 @@ def run_crew(question, file_path):
     
     code_execution_agent = Agent(
         role="Code Execution Agent",
-        goal="Given a question and .py Python file, execute the file to help answer the question: {question}",
-        backstory="As an expert Python code execution assistant, you execute the code file to help answer the question.",
+        goal="Given a question and Python file, execute the file to answer the question: {question}",
+        backstory="As an expert Python code execution assistant, you execute code to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,
@@ -321,21 +322,19 @@ def run_crew(question, file_path):
 
     manager_agent = Agent(
         role="Manager Agent",
-        goal="Try to answer the following question. If needed, delegate to one or more of your coworkers for help. "
-             "If there is no good coworker, delegate to the Code Generation Agent to implement a tool for the task. "
-             "Question: {question}",
+        goal="Answer the following question. If needed, delegate to one of your coworkers. Question: {question}",
         backstory="As an expert manager assistant, you answer the question.",
         allow_delegation=True,
         llm=MANAGER_MODEL,
         max_iter=5,
-        verbose=True
+        verbose=False
     )
 
     # Task
 
     manager_task = Task(
         agent=manager_agent,
-        description="Try to answer the following question. If needed, delegate to one or more of your coworkers for help. Question: {question}",
+        description="Answer the following question. If needed, delegate to one of your coworkers. Question: {question}",
         expected_output="The answer to the question."
     )
     
@@ -360,7 +359,7 @@ def run_crew(question, file_path):
         file_data = read_file(file_path)
 
         if file_data:
-            question = f"{question} File data: {file_data}" # sandbox contraints
+            question = f"{question} JSON data: {file_data}" # sandbox contraints
         else:
             question = f"{question} File path: {file_path}."