Skip to content

Commit 87b0000

Browse files
committed
Update planner_agent.py
1 parent fe8fe5b commit 87b0000

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/backend/kernel_agents/planner_agent.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ async def handle_input_task(self, input_task: InputTask) -> str:
181181
user_id=self._user_id,
182182
plan_id=plan.id,
183183
content=f"Generated a plan with {len(steps)} steps. Click the blue check box beside each step to complete it, click the x to remove this step.",
184-
source="PlannerAgent",
184+
source=AgentType.PLANNER.value,
185185
step_id="",
186186
)
187187
)
@@ -193,7 +193,7 @@ async def handle_input_task(self, input_task: InputTask) -> str:
193193
"user_id": self._user_id,
194194
"plan_id": plan.id,
195195
"content": f"Generated a plan with {len(steps)} steps. Click the blue check box beside each step to complete it, click the x to remove this step.",
196-
"source": "PlannerAgent",
196+
"source": AgentType.PLANNER.value,
197197
},
198198
)
199199

@@ -208,7 +208,7 @@ async def handle_input_task(self, input_task: InputTask) -> str:
208208
user_id=self._user_id,
209209
plan_id=plan.id,
210210
content=f"I require additional information before we can proceed: {plan.human_clarification_request}",
211-
source="PlannerAgent",
211+
source=AgentType.PLANNER.value,
212212
step_id="",
213213
)
214214
)
@@ -220,7 +220,7 @@ async def handle_input_task(self, input_task: InputTask) -> str:
220220
"user_id": self._user_id,
221221
"plan_id": plan.id,
222222
"content": f"I require additional information before we can proceed: {plan.human_clarification_request}",
223-
"source": "PlannerAgent",
223+
"source": AgentType.PLANNER.value,
224224
},
225225
)
226226

@@ -275,7 +275,7 @@ async def handle_plan_clarification(self, kernel_arguments: KernelArguments) ->
275275
user_id=self._user_id,
276276
plan_id="",
277277
content="Thanks. The plan has been updated.",
278-
source="PlannerAgent",
278+
source=AgentType.PLANNER.value,
279279
step_id="",
280280
)
281281
)
@@ -286,7 +286,7 @@ async def handle_plan_clarification(self, kernel_arguments: KernelArguments) ->
286286
"session_id": session_id,
287287
"user_id": self._user_id,
288288
"content": "Thanks. The plan has been updated.",
289-
"source": "PlannerAgent",
289+
"source": AgentType.PLANNER.value,
290290
},
291291
)
292292

@@ -508,7 +508,7 @@ async def _create_structured_plan(
508508
"user_id": self._user_id,
509509
"error": str(e),
510510
"description": input_task.description,
511-
"source": "PlannerAgent",
511+
"source": AgentType.PLANNER.value,
512512
},
513513
)
514514
except Exception as event_error:

0 commit comments

Comments
 (0)