bstraehle commited on
Commit
7da0809
·
verified ·
1 Parent(s): 15b8627

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +26 -2
crew.py CHANGED
@@ -27,10 +27,14 @@ tracer_provider = register(
27
  CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
28
 
29
  def run_crew(question):
 
 
30
  code_interpreter_tool = CodeInterpreterTool()
31
  web_search_tool = SerperDevTool()
32
  web_rag_tool = WebsiteSearchTool()
33
  youtube_search_tool = YoutubeVideoSearchTool()
 
 
34
 
35
  web_search_agent = Agent(
36
  role="Web Search Agent",
@@ -51,6 +55,18 @@ def run_crew(question):
51
  tools=[youtube_search_tool],
52
  verbose=True
53
  )
 
 
 
 
 
 
 
 
 
 
 
 
54
 
55
  web_search = Task(
56
  agent=web_search_agent,
@@ -63,13 +79,21 @@ def run_crew(question):
63
  description="Search a YouTube video for question \"{topic}\"",
64
  expected_output="Content to help answer the YouTube related question."
65
  )
 
 
 
 
 
 
 
 
66
 
67
  crew = Crew(
68
- agents=[web_search_agent, youtube_search_agent],
69
  #planning=True,
70
  #planning_llm=PLANNING_MODEL,
71
  process=Process.sequential,
72
- tasks=[web_search, youtube_search],
73
  verbose=True
74
  )
75
 
 
27
  CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
28
 
29
  def run_crew(question):
30
+ # Tools
31
+
32
  code_interpreter_tool = CodeInterpreterTool()
33
  web_search_tool = SerperDevTool()
34
  web_rag_tool = WebsiteSearchTool()
35
  youtube_search_tool = YoutubeVideoSearchTool()
36
+
37
+ # Agents
38
 
39
  web_search_agent = Agent(
40
  role="Web Search Agent",
 
55
  tools=[youtube_search_tool],
56
  verbose=True
57
  )
58
+
59
+ python_coding_agent = Agent(
60
+ role="Python Coding Agent",
61
+ goal="Write and execute Python code to solve question \"{topic}\".",
62
+ backstory="As an expert Python coding assistant, you write and execute Python for the question.",
63
+ allow_delegation=False,
64
+ max_iter=1,
65
+ tools=[code_interpreter_tool],
66
+ verbose=True,
67
+ )
68
+
69
+ # Tasks
70
 
71
  web_search = Task(
72
  agent=web_search_agent,
 
79
  description="Search a YouTube video for question \"{topic}\"",
80
  expected_output="Content to help answer the YouTube related question."
81
  )
82
+
83
+ coding_task = Task(
84
+ agent=python_coding_agent,
85
+ description="Write and execute Python code to solve question \"{topic}\".",
86
+ expected_output="Content to help answer the Phyton coding related question."
87
+ )
88
+
89
+ # Crew
90
 
91
  crew = Crew(
92
+ agents=[python_coding_agent, web_search_agent, youtube_search_agent],
93
  #planning=True,
94
  #planning_llm=PLANNING_MODEL,
95
  process=Process.sequential,
96
+ tasks=[coding_task, web_search, youtube_search],
97
  verbose=True
98
  )
99