Amitabhab commited on
Commit
10133c5
·
1 Parent(s): 107ad8a

Adding the planning LLM

Browse files
Files changed (1) hide show
  1. crew.py +4 -1
crew.py CHANGED
@@ -21,6 +21,7 @@ class TravelCrew:
21
  self.tasks_config = 'config/tasks.yaml'
22
  self.llm = LLM(model='sambanova/%s' % model_name)
23
  self.manager_llm = LLM(model='sambanova/%s' % model_name)
 
24
 
25
  @typing.no_type_check
26
  @agent
@@ -155,7 +156,9 @@ class TravelCrew:
155
  agents=self.agents,
156
  tasks=self.tasks,
157
  process=Process.hierarchical,
158
- manager_llm=self.manager_llm
 
 
159
  )
160
 
161
 
 
21
  self.tasks_config = 'config/tasks.yaml'
22
  self.llm = LLM(model='sambanova/%s' % model_name)
23
  self.manager_llm = LLM(model='sambanova/%s' % model_name)
24
+ self.planner_llm = LLM(model='sambanova/%s' % model_name)
25
 
26
  @typing.no_type_check
27
  @agent
 
156
  agents=self.agents,
157
  tasks=self.tasks,
158
  process=Process.hierarchical,
159
+ manager_llm=self.manager_llm,
160
+ planning=True,
161
+ planning_llm=self.planner_llm
162
  )
163
 
164