Spaces:
Sleeping
Sleeping
File size: 709 Bytes
93e5d93 282b4c0 074a596 767f98e e8f9166 767f98e a5f9d4e 9d278b1 e8f9166 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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, temperature=temperature),
process=Process.sequential,
verbose=verbose
) |