12

frontend/src/store/builderStore.ts

@@ -57,6 +57,7 @@ interface BuilderState {
   activeBlueprint?: SiteBlueprint;
   pages: PageBlueprint[];
   selectedPageIds: number[];
+  structureTaskId?: string | null;
   generationProgress?: {
     pagesQueued: number;
     taskIds: number[];
@@ -96,6 +97,7 @@ export const useBuilderStore = create<BuilderState>((set, get) => ({
   metadataError: undefined,
   pages: [],
   selectedPageIds: [],
+  structureTaskId: null,
 
   setField: (key, value) =>
     set((state) => ({
@@ -284,6 +286,10 @@ export const useBuilderStore = create<BuilderState>((set, get) => ({
           },
         );
 
+        if (generation?.task_id) {
+          set({ structureTaskId: generation.task_id });
+        }
+
         if (generation?.structure) {
           lastStructure = generation.structure;
         }