File size: 1,073 Bytes
d6269a1
93fc605
f15d7ef
93fc605
c25771b
5ffb8f8
b3d6c85
6edf3f2
f15d7ef
82601b2
0262a8b
c25771b
b3d6c85
 
c25771b
5ffb8f8
b3d6c85
43948f6
6edf3f2
 
fd3a4f4
82601b2
0262a8b
c25771b
b3d6c85
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from crewai import Task

from agents import get_researcher_agent, get_writer_agent

def get_research_task(model, temperature, verbose):
    return Task(
        description=(
            "1. Search the web for content on topic '{topic}'. Prioritize scientific sources.\n"
            "2. Scrape the 10 most relevant web sites for content."
        ),
        expected_output="Content on topic: {topic}.",
        agent=get_researcher_agent(model, temperature, verbose),
    )

def get_write_task(model, temperature, verbose):
    return Task(
        description=(
            "1. Use the context to write an in-depth article on topic '{topic}'. "
            "The output must be in markdown format (omit the triple backticks).\n"
            "2. At the beginning of the article, add the current date and author: Multi-Agent AI System.\n"
            "3. At the end of the article, add a references section with links to relevant content."
        ),
        expected_output="An article on topic {topic}.",
        agent=get_writer_agent(model, temperature, verbose),
    )