bstraehle's picture
Update crew.py
7eedd14 verified
raw
history blame
720 Bytes
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
)