Spaces:
Running
Running
Adding the planning LLM
Browse files
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 |
|