from crewai import Crew, Process
from langchain_openai import ChatOpenAI

from agents import get_researcher_agent, get_writer_agent
from tasks import get_research_task, get_write_task

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