from crewai import Task

from agents import get_researcher_agent, get_author_agent

def get_researcher_task(model):
    return Task(
        description=(
            "1. Search the web for content on topic {topic}.\n"
            "2. Scrape the 3 most relevant web sites for content."
        ),
        expected_output="Content on topic {topic}.",
        agent=get_researcher_agent(model),
    )

def get_author_task(model):
    return Task(
        description=(
            "1. Use the context to write a LinkedIn post on topic {topic}.\n"
            "2. Structure the post to include an introduction, a body, and a conclusion.\n"
            "3. In the introduction, use 3 emojis.\n"
            "4. After the conclusion, add 3 hashtags."
        ),
        expected_output="A LinkedIn post on topic {topic}.",
        agent=get_author_agent(model),
    )