File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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 = []
You can’t perform that action at this time.
0 commit comments