from crewai import Crew, Process
from langchain_openai import ChatOpenAI

from agents import get_content_planner_agent, get_content_writer_agent, get_editor_agent
from tasks import get_content_planning_task, get_content_writing_task, get_editing_task

def get_crew():
    return Crew(
        agents=[get_content_planner_agent(), 
                get_content_writer_agent(), 
                get_editor_agent()],
        
        tasks=[get_content_planning_task(), 
               get_content_writing_task(), 
               get_editing_task()],
        
        manager_llm=ChatOpenAI(model="gpt-3.5-turbo", 
                               temperature=0.7),
        process=Process.hierarchical,
        verbose=True
    )