from crewai import Crew, Process
from langchain_openai import ChatOpenAI

from agents import get_researcher_agent, get_author_agent, get_editor_agent
from tasks import get_research_task, get_author_task, get_edit_task

def get_crew(llm_manager, llm_agents, verbose):
    return Crew(
        agents=[get_researcher_agent(llm_agents, verbose), 
                get_author_agent(llm_agents, verbose),
                get_editor_agent(llm_agents, verbose)],
        tasks=[get_research_task(llm_agents, verbose), 
               get_author_task(llm_agents, verbose),
               get_edit_task(llm_agents, verbose)],
        manager_llm=ChatOpenAI(model=llm_manager),
        process=Process.sequential,
        verbose=verbose
    )