Spaces:
Running
Running
Update tasks.py
Browse files
tasks.py
CHANGED
@@ -2,17 +2,17 @@ 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_write_task(model):
|
16 |
return Task(
|
17 |
description=(
|
18 |
"1. Use the context to write a 2000-word article on topic: {topic}.\n"
|
@@ -20,5 +20,5 @@ def get_write_task(model):
|
|
20 |
"3. At the end of the article, add a references section with research papers."
|
21 |
),
|
22 |
expected_output="An article on topic {topic}.",
|
23 |
-
agent=get_writer_agent(model),
|
24 |
)
|
|
|
2 |
|
3 |
from agents import get_researcher_agent, get_writer_agent
|
4 |
|
5 |
+
def get_research_task(model, temperature, verbose):
|
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, temperature, verbose),
|
13 |
)
|
14 |
|
15 |
+
def get_write_task(model, temperature, verbose):
|
16 |
return Task(
|
17 |
description=(
|
18 |
"1. Use the context to write a 2000-word article on topic: {topic}.\n"
|
|
|
20 |
"3. At the end of the article, add a references section with research papers."
|
21 |
),
|
22 |
expected_output="An article on topic {topic}.",
|
23 |
+
agent=get_writer_agent(model, temperature, verbose),
|
24 |
)
|