bstraehle commited on
Commit
23b92b8
·
verified ·
1 Parent(s): 1cbe64f

Create crew.py

Browse files
Files changed (1) hide show
  1. crew.py +16 -0
crew.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from crewai import Crew, Process
2
+ from langchain_openai import ChatOpenAI
3
+
4
+ from agents import get_researcher_agent, get_writer_agent
5
+ from tasks import get_research_task, get_write_task
6
+
7
+ def get_crew(llm_manager, llm_agents, verbose):
8
+ return Crew(
9
+ agents=[get_researcher_agent(llm_agents, verbose),
10
+ get_writer_agent(llm_agents, verbose)],
11
+ tasks=[get_research_task(llm_agents, verbose),
12
+ get_write_task(llm_agents, verbose)],
13
+ manager_llm=ChatOpenAI(model=llm_manager),
14
+ process=Process.sequential,
15
+ verbose=verbose
16
+ )