bstraehle commited on
Commit
c971397
1 Parent(s): 4d658da

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +4 -9
crew.py CHANGED
@@ -4,16 +4,11 @@ from langchain_openai import ChatOpenAI
4
  from agents import get_researcher_agent, get_author_agent
5
  from tasks import get_researcher_task, get_author_task
6
 
7
- def get_crew():
8
  return Crew(
9
- agents=[get_researcher_agent(),
10
- get_author_agent()],
11
-
12
- tasks=[get_researcher_task(),
13
- get_author_task()],
14
-
15
- manager_llm=ChatOpenAI(model="gpt-4o",
16
- temperature=0.7),
17
  process=Process.sequential,
18
  verbose=True
19
  )
 
4
  from agents import get_researcher_agent, get_author_agent
5
  from tasks import get_researcher_task, get_author_task
6
 
7
+ def get_crew(model):
8
  return Crew(
9
+ agents=[get_researcher_agent(), get_author_agent()],
10
+ tasks=[get_researcher_task(), get_author_task()],
11
+ manager_llm=ChatOpenAI(model=model, temperature=0.7),
 
 
 
 
 
12
  process=Process.sequential,
13
  verbose=True
14
  )