from crewai import Task

from agents import get_researcher_agent, get_writer_agent

def get_research_task(model):
    return Task(
        description=(
            "1. Prioritizing research papers, search the web for content on topic: {topic}.\n"
            "2. Scrape the 10 most relevant web sites for content."
        ),
        expected_output="Content on topic: {topic}.",
        agent=get_researcher_agent(model),
    )

def get_write_task(model):
    return Task(
        description=(
            "Use the context to write a 2000-word article on topic {topic}, including a reference section with research papers. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o."
        ),
        expected_output="An article on topic {topic}.",
        agent=get_writer_agent(model),
    )