Spaces:
Running
Running
Update tasks.py
Browse files
tasks.py
CHANGED
@@ -1,23 +1,22 @@
|
|
1 |
from crewai import Task
|
2 |
|
3 |
-
from agents import get_researcher_agent,
|
4 |
|
5 |
def get_research_task(model):
|
6 |
return Task(
|
7 |
description=(
|
8 |
-
"1. Search the web for content on topic {topic}.\n"
|
9 |
-
"2. Scrape the 10 most relevant web sites for content
|
10 |
),
|
11 |
expected_output="Content on topic {topic}.",
|
12 |
agent=get_researcher_agent(model),
|
13 |
)
|
14 |
|
15 |
-
def
|
16 |
return Task(
|
17 |
description=(
|
18 |
-
"
|
19 |
-
"2. If applicable, add a references section with research papers."
|
20 |
),
|
21 |
-
expected_output="A
|
22 |
-
agent=
|
23 |
)
|
|
|
1 |
from crewai import Task
|
2 |
|
3 |
+
from agents import get_researcher_agent, get_writer_agent
|
4 |
|
5 |
def get_research_task(model):
|
6 |
return Task(
|
7 |
description=(
|
8 |
+
"1. Search the web for content on topic {topic}, prioritizing research papers.\n"
|
9 |
+
"2. Scrape the 10 most relevant web sites for content."
|
10 |
),
|
11 |
expected_output="Content on topic {topic}.",
|
12 |
agent=get_researcher_agent(model),
|
13 |
)
|
14 |
|
15 |
+
def get_writer_task(model):
|
16 |
return Task(
|
17 |
description=(
|
18 |
+
"Use the context to write a 2000-word article on topic {topic}, including a reference section with research papers."
|
|
|
19 |
),
|
20 |
+
expected_output="A 2000-word article on topic {topic}.",
|
21 |
+
agent=get_writer_agent(model),
|
22 |
)
|