Use virtual infrastructure when returning blueprint

mitdbg · Jan 27, 2025 · 9ff3510 · 9ff3510
1 parent 1cfa247
commit 9ff3510
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/brad/ui/manager_impl.py b/src/brad/ui/manager_impl.py
@@ -239,13 +239,15 @@ async def get_predicted_changes(args: PredictedChangesArgs) -> DisplayableBluepr
     """
     assert manager is not None
     assert manager.planner is not None
+    assert manager.vdbe_mgr is not None
     result = await manager.planner.run_replan_direct(
         intensity_multipliers=(args.t_multiplier, args.a_multiplier)
     )
     if result is None:
         raise HTTPException(500, "Failed to run a replan.")
     blueprint, _ = result
-    return DisplayableBlueprint.from_blueprint(blueprint)
+    virtual_infra = manager.vdbe_mgr.infra()
+    return DisplayableBlueprint.from_blueprint(blueprint, virtual_infra)
 
 
 @app.post("/api/1/vdbe")