from textwrap import dedent
from crewai import Task
from agents import topic_researcher_agent, writer_agent

topic_research_task = Task(
    description=dedent(
        """
        Get valuable and high quality web information about the comparison between Llama 2 and Llama 3
        """
    ),
    expected_output=dedent(
        """
        Your task is to gather high quality information about the comparison between Llama 2 and Llama 3
        """
    ),
    agent=topic_researcher_agent,
)

write_task = Task(
    description=dedent(
        """
        Create an article that summarises the difference between Llama 2 and Llama 3
        """
    ),
    expected_output=dedent(
        """
        A high-quality and engaging article comparing Llama 2 and Llama 3.
        """
    ),
    agent=writer_agent,
)