bstraehle commited on
Commit
9b47a3c
·
verified ·
1 Parent(s): ae56f5f

Update tasks.py

Browse files
Files changed (1) hide show
  1. tasks.py +3 -3
tasks.py CHANGED
@@ -1,6 +1,6 @@
1
  from crewai import Task
2
 
3
- from agents import get_researcher_agent, get_writer_agent
4
 
5
  def get_research_task(model, verbose):
6
  return Task(
@@ -12,7 +12,7 @@ def get_research_task(model, verbose):
12
  agent=get_researcher_agent(model, verbose),
13
  )
14
 
15
- def get_write_task(model, verbose):
16
  return Task(
17
  description=(
18
  "1. Use the context to write a long and in-depth article on topic: {topic}.\n"
@@ -22,5 +22,5 @@ def get_write_task(model, verbose):
22
  "5. Include inline references to the links, for example: [1]."
23
  ),
24
  expected_output="An article on topic {topic}.",
25
- agent=get_writer_agent(model, verbose),
26
  )
 
1
  from crewai import Task
2
 
3
+ from agents import get_researcher_agent, get_author_agent
4
 
5
  def get_research_task(model, verbose):
6
  return Task(
 
12
  agent=get_researcher_agent(model, verbose),
13
  )
14
 
15
+ def get_author_task(model, verbose):
16
  return Task(
17
  description=(
18
  "1. Use the context to write a long and in-depth article on topic: {topic}.\n"
 
22
  "5. Include inline references to the links, for example: [1]."
23
  ),
24
  expected_output="An article on topic {topic}.",
25
+ agent=get_author_agent(model, verbose),
26
  )