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