Skip to content

Commit b78999d

Browse files
committed
Clean up Agent names
1 parent 57181f6 commit b78999d

File tree

13 files changed

+442
-324
lines changed

13 files changed

+442
-324
lines changed

src/backend/app_kernel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
242242
agents = await get_agents(human_feedback.session_id, user_id)
243243

244244
# Send the feedback to the human agent
245-
human_agent = agents["HumanAgent"]
245+
human_agent = agents[AgentType.HUMAN.value]
246246

247247
# Convert feedback to JSON for the kernel function
248248
human_feedback_json = human_feedback.json()
@@ -325,7 +325,7 @@ async def human_clarification_endpoint(
325325
agents = await get_agents(human_clarification.session_id, user_id)
326326

327327
# Send the clarification to the planner agent
328-
planner_agent = agents["PlannerAgent"]
328+
planner_agent = agents[AgentType.PLANNER.value]
329329

330330
# Convert clarification to JSON for proper processing
331331
human_clarification_json = human_clarification.json()
@@ -419,7 +419,7 @@ async def approve_step_endpoint(
419419
human_feedback_json = human_feedback.json()
420420

421421
# First process with HumanAgent to update step status
422-
human_agent = agents["HumanAgent"]
422+
human_agent = agents[AgentType.HUMAN.value]
423423
await human_agent.handle_human_feedback(
424424
KernelArguments(human_feedback_json=human_feedback_json)
425425
)

0 commit comments

Comments
 (0)