Skip to content

Commit 665b530

Browse files
committed
Update app_kernel.py
1 parent 6f76caa commit 665b530

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/backend/app_kernel.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -548,15 +548,16 @@ async def get_plans(
548548
)
549549
raise HTTPException(status_code=404, detail="Plan not found")
550550

551-
steps = await memory_store.get_steps_for_plan(plan.id, session_id)
551+
# Use get_steps_by_plan to match the original implementation
552+
steps = await memory_store.get_steps_by_plan(plan_id=plan.id)
552553
plan_with_steps = PlanWithSteps(**plan.model_dump(), steps=steps)
553554
plan_with_steps.update_step_counts()
554555
return [plan_with_steps]
555556

556557
all_plans = await memory_store.get_all_plans()
557558
# Fetch steps for all plans concurrently
558559
steps_for_all_plans = await asyncio.gather(
559-
*[memory_store.get_steps_for_plan(plan.id, plan.session_id) for plan in all_plans]
560+
*[memory_store.get_steps_by_plan(plan_id=plan.id) for plan in all_plans]
560561
)
561562
# Create list of PlanWithSteps and update step counts
562563
list_of_plans_with_steps = []

0 commit comments

Comments
 (0)