bstraehle commited on
Commit
fd399dc
·
verified ·
1 Parent(s): c70f203

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +18 -23
crew.py CHANGED
@@ -35,7 +35,7 @@ def run_crew(question, file_name):
35
  python_coding_tool = CodeInterpreterTool()
36
  web_search_tool = SerperDevTool()
37
  web_rag_tool = WebsiteSearchTool()
38
- video_analysis_tool = YoutubeVideoSearchTool()
39
 
40
  # Agents
41
 
@@ -56,37 +56,37 @@ def run_crew(question, file_name):
56
  backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
57
  allow_delegation=False,
58
  llm=AGENT_MODEL,
59
- max_iter=3,
60
  tools=[python_coding_tool],
61
  verbose=False
62
  )
63
 
64
- video_analysis_agent = Agent(
65
- role="Video Analysis Agent",
66
- goal="Analyze video to help answer question \"{topic}\". ",
67
- backstory="As an expert video analysis assistant, you analyze the video to help answer the question. ",
68
  allow_delegation=False,
69
  llm=AGENT_MODEL,
70
  max_iter=3,
71
- tools=[video_analysis_tool],
72
  verbose=False
73
  )
74
 
75
- web_search_agent = Agent(
76
- role="Web Search Agent",
77
- goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
78
- backstory="As an expert web search assistant, you search the web to help answer the question. ",
79
  allow_delegation=False,
80
  llm=AGENT_MODEL,
81
  max_iter=3,
82
- tools=[web_search_tool, web_rag_tool],
83
  verbose=False
84
  )
85
 
86
  manager_agent = Agent(
87
  role="Manager Agent",
88
- goal="Try to answer the following question. If needed, delegate to **one** of your coworkers, image_analysis_agent, python_coding_agent, video_analysis_agent, or web_search_agent for help. "
89
- "If there is no good coworker, delegate to the python_coding_agent to implement a tool for the task. "
90
  "Question: \"{topic}\" ",
91
  backstory="As an expert manager assistant, you answer the question. ",
92
  allow_delegation=True,
@@ -99,14 +99,14 @@ def run_crew(question, file_name):
99
 
100
  manager_task = Task(
101
  agent=manager_agent,
102
- description="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Video Analysis Agent, or Web Search Agent for help. Question: \"{topic}\" ",
103
  expected_output="The answer to the question. "
104
  )
105
 
106
  # Crew
107
 
108
  crew = Crew(
109
- agents=[image_analysis_agent, python_coding_agent, video_analysis_agent, web_search_agent],
110
  manager_agent=manager_agent,
111
  tasks=[manager_task],
112
  verbose=True
@@ -114,16 +114,11 @@ def run_crew(question, file_name):
114
 
115
  print(crew.agents)
116
 
117
- ###
118
- #if file_name:
119
- # question = f"{question} File name: data/{file_name}."
120
- ###
121
-
122
  if file_name.endswith(".py"):
123
  with open(f"data/{file_name}", "r") as file:
124
- file_content = file.read()
125
 
126
- question = f"{question} File content: {file_content}"
127
  else:
128
  question = f"{question} File name: data/{file_name}."
129
 
 
35
  python_coding_tool = CodeInterpreterTool()
36
  web_search_tool = SerperDevTool()
37
  web_rag_tool = WebsiteSearchTool()
38
+ youtube_analysis_tool = YoutubeVideoSearchTool()
39
 
40
  # Agents
41
 
 
56
  backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
57
  allow_delegation=False,
58
  llm=AGENT_MODEL,
59
+ max_iter=5,
60
  tools=[python_coding_tool],
61
  verbose=False
62
  )
63
 
64
+ web_search_agent = Agent(
65
+ role="Web Search Agent",
66
+ goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
67
+ backstory="As an expert web search assistant, you search the web to help answer the question. ",
68
  allow_delegation=False,
69
  llm=AGENT_MODEL,
70
  max_iter=3,
71
+ tools=[web_search_tool, web_rag_tool],
72
  verbose=False
73
  )
74
 
75
+ youtube_analysis_agent = Agent(
76
+ role="YouTube Analysis Agent",
77
+ goal="Analyze YouTube video to help answer question \"{topic}\". ",
78
+ backstory="As an expert YouTube video analysis assistant, you analyze the video to help answer the question. ",
79
  allow_delegation=False,
80
  llm=AGENT_MODEL,
81
  max_iter=3,
82
+ tools=[youtube_analysis_tool],
83
  verbose=False
84
  )
85
 
86
  manager_agent = Agent(
87
  role="Manager Agent",
88
+ goal="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Web Search Agent, or YouTube Analysis Agent for help. "
89
+ "If there is no good coworker, delegate to the Python Coding Agent to implement a tool for the task. "
90
  "Question: \"{topic}\" ",
91
  backstory="As an expert manager assistant, you answer the question. ",
92
  allow_delegation=True,
 
99
 
100
  manager_task = Task(
101
  agent=manager_agent,
102
+ description="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Web Search Agent, or YouTube Analysis Agent for help. Question: \"{topic}\" ",
103
  expected_output="The answer to the question. "
104
  )
105
 
106
  # Crew
107
 
108
  crew = Crew(
109
+ agents=[image_analysis_agent, python_coding_agent, web_search_agent, youtube_analysis_agent],
110
  manager_agent=manager_agent,
111
  tasks=[manager_task],
112
  verbose=True
 
114
 
115
  print(crew.agents)
116
 
 
 
 
 
 
117
  if file_name.endswith(".py"):
118
  with open(f"data/{file_name}", "r") as file:
119
+ file_data = file.read()
120
 
121
+ question = f"{question} File data:\n{file_data}"
122
  else:
123
  question = f"{question} File name: data/{file_name}."
124