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(model, max_tokens, temperature, verbose):
    return Crew(
        agents=[get_researcher_agent(model, max_tokens, temperature, verbose), 
                get_writer_agent(model, max_tokens, temperature, verbose)],
        tasks=[get_research_task(model, max_tokens, temperature, verbose), 
               get_write_task(model, max_tokens, temperature, verbose)],
        manager_llm=ChatOpenAI(model=model, max_tokens=max_tokens, temperature=temperature),
        process=Process.sequential,
        verbose=verbose
    )