@@ -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