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 current content on {topic}.\n"
            "2. Scrape a maximum of 3 web sites for content.\n"
            "3. Attribute credit to the web site."
        ),
        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 social media post on {topic}.\n"
            "2. Ensure the post is structured with an engaging introduction with 3 emojis, an insightful body, and a summarizing conclusion.\n"
            "3. Proofread for grammatical errors."
        ),
        expected_output="A LinkedIn social media post on topic: {topic}.",
        agent=get_author_agent(model),
    )