Update crew.py

crew.py

@@ -39,11 +39,10 @@ def run_crew(question, file_name):
 
     # Agents
 
-    image_search_agent = Agent(
-        role="Image Search Agent",
-        #goal="Search an image for question \"{topic}\". ",
-        goal="{topic} ",
-        backstory="As an expert image search assistant, you search an image for the question. ",
+    image_analysis_agent = Agent(
+        role="Image Analysis Agent",
+        goal="Analyze image to help answer question \"{topic}\". ",
+        backstory="As an expert image analysis assistant, you analyze the image to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         #max_iter=1,
@@ -53,9 +52,8 @@ def run_crew(question, file_name):
     
     python_coding_agent = Agent(
         role="Python Coding Agent",
-        #goal="Write and execute Python code to solve question \"{topic}\". ",
-        goal="{topic} ",
-        backstory="As an expert Python coding assistant, you write and execute Python for the question. ",
+        goal="Write and/or execute Python code to help answer question \"{topic}\". ",
+        backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=1,
@@ -63,64 +61,62 @@ def run_crew(question, file_name):
         verbose=False,
     )
    
-    web_search_agent = Agent(
-        role="Web Search Agent",
-        goal="Search the web for question \"{topic}\" and scrape the most relevant web page. ",
-        backstory="As an expert web search assistant, you search the web for the question and scrape the most relevant web page. ",
+    video_analysis_agent = Agent(
+        role="Vidoe Analysis Agent",
+        goal="Analyze video to help answer question \"{topic}\". ",
+        backstory="As an expert video analysis assistant, you analyze the video to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=1,
-        tools=[web_search_tool, web_rag_tool],
+        tools=[youtube_search_tool],
         verbose=False
     )
 
-    youtube_search_agent = Agent(
-        role="YouTube Search Agent",
-        goal="Search a YouTube video for question \"{topic}\". ",
-        backstory="As an expert YouTube search assistant, you search a YouTube video for the question. ",
+    web_search_agent = Agent(
+        role="Web Search Agent",
+        goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
+        backstory="As an expert web search assistant, you search the web to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=1,
-        tools=[youtube_search_tool],
+        tools=[web_search_tool, web_rag_tool],
         verbose=False
     )
 
     # Tasks
 
-    image_search_task = Task(
-        agent=image_search_agent,
-        #description="Search an image for question \"{topic}\". ",
-        description="{topic} ",
-        expected_output="Content to help answer the image related question. "
+    image_analysis_task = Task(
+        agent=image_analysis_agent,
+        description="Analyze image to help answer question \"{topic}\". ",
+        expected_output="Content to help answer the question. "
     )
     
     python_coding_task = Task(
         agent=python_coding_agent,
-        #description="Write and execute Python code to solve question \"{topic}\". ",
-        description="{topic} ",
-        expected_output="Content to help answer the Phyton coding related question. "
+        description="Write and/or execute Python code to help answer question \"{topic}\". ",
+        expected_output="Content to help answer the question. "
     )
 
-    web_search_task = Task(
-        agent=web_search_agent,
-        description="Search the web for question \"{topic}\" and scrape the most relevant web page. ",
-        expected_output="Content to help answer the web related question. "
+    video_analysis_task = Task(
+        agent=video_analysis_agent,
+        description="Analyze video to help answer question \"{topic}\". ",
+        expected_output="Content to help answer the question. "
     )
     
-    youtube_search_task = Task(
-        agent=youtube_search_agent,
-        description="Search a YouTube video for question \"{topic}\". ",
-        expected_output="Content to help answer the YouTube related question. "
-    )
+    web_search_task = Task(
+        agent=web_search_agent,
+        description="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
+        expected_output="Content to help answer the question. "
+    )    
 
     # Crew
     
     crew = Crew(
-        agents=[image_search_agent, python_coding_agent, web_search_agent, youtube_search_agent],
+        agents=[image_analysis_agent, python_coding_agent, web_search_agent, video_analysis_agent],
         planning=True,
         planning_llm=CREW_MODEL,
         process=Process.sequential,
-        tasks=[image_search_task, python_coding_task, web_search_task, youtube_search_task],
+        tasks=[image_analysis_task, python_coding_task, web_search_task, video_analysis_task],
         verbose=False
     )