from crewai import Crew, Process
from langchain_openai import ChatOpenAI

from agents import get_researcher_agent, get_author_agent
from tasks import get_researcher_task, get_author_task

def get_crew(model):
    return Crew(
        agents=[get_researcher_agent(model), get_author_agent(model)],
        tasks=[get_researcher_task(), get_author_task()],
        manager_llm=ChatOpenAI(model=model, temperature=0.7),
        process=Process.sequential,
        verbose=True
    )