@agent-native/core 0.77.24 → 0.78.1

package/corpus/templates/assets/actions/generate-asset.ts

@@ -1,4 +1,5 @@
 import { defineAction, embedApp } from "@agent-native/core";
+import type { ActionRunContext } from "@agent-native/core/action";
 import { z } from "zod";
 
 import {
@@ -116,16 +117,19 @@ const action = defineAction({
       view: "picker",
     };
   },
-  run: async (args) => {
+  run: async (args, context?: ActionRunContext) => {
     if (args.mediaType === "video") {
-      const picker = (await openAssetPicker.run({
-        mediaType: "video",
-        prompt: args.prompt,
-        libraryId: args.libraryId,
-        libraryHint: args.libraryHint,
-        count: args.count,
-        callerAppId: args.callerAppId,
-      })) as Record<string, unknown>;
+      const picker = (await openAssetPicker.run(
+        {
+          mediaType: "video",
+          prompt: args.prompt,
+          libraryId: args.libraryId,
+          libraryHint: args.libraryHint,
+          count: args.count,
+          callerAppId: args.callerAppId,
+        },
+        context,
+      )) as Record<string, unknown>;
       return {
         ...picker,
         generated: false,
@@ -134,20 +138,23 @@ const action = defineAction({
       };
     }
 
-    const picker = (await openAssetPicker.run({
-      mediaType: "image",
-      prompt: args.prompt,
-      libraryId: args.libraryId,
-      libraryHint: args.libraryHint,
-      aspectRatio: args.aspectRatio,
-      presetId: args.presetId,
-      count: args.count,
-      autoGenerate: true,
-      tier: args.tier,
-      styleStrength: args.styleStrength,
-      includeLogo: args.includeLogo,
-      callerAppId: args.callerAppId,
-    })) as Record<string, unknown>;
+    const picker = (await openAssetPicker.run(
+      {
+        mediaType: "image",
+        prompt: args.prompt,
+        libraryId: args.libraryId,
+        libraryHint: args.libraryHint,
+        aspectRatio: args.aspectRatio,
+        presetId: args.presetId,
+        count: args.count,
+        autoGenerate: true,
+        tier: args.tier,
+        styleStrength: args.styleStrength,
+        includeLogo: args.includeLogo,
+        callerAppId: args.callerAppId,
+      },
+      context,
+    )) as Record<string, unknown>;
     return {
       ...picker,
       generated: false,

package/corpus/templates/assets/actions/generate-image-batch.ts

@@ -1,4 +1,5 @@
 import { defineAction } from "@agent-native/core";
+import type { ActionRunContext } from "@agent-native/core/action";
 import { assertAccess } from "@agent-native/core/sharing";
 import { eq } from "drizzle-orm";
 import { nanoid } from "nanoid";
@@ -17,13 +18,20 @@ import {
   STYLE_STRENGTHS,
 } from "../shared/api.js";
 import { requireGenerationSessionInLibrary } from "./_helpers.js";
+import { readImageModelDefault } from "./_image-model-default.js";
 import generateImage from "./generate-image.js";
+import { upsertVariantSlot } from "./variant-slots.js";
 
 export default defineAction({
   description:
-    "Generate several brand-consistent images in parallel from one library. This is synchronous for images: one call waits for every slot and returns final image artifacts. Use this for slide decks, landing pages, and multi-slot design work. Do not call get-generation-run or refresh-generation-run after a normal image batch result.",
+    "Generate several brand-consistent images in parallel from one brand kit/library. Use @brand-kit mentions as libraryId and @preset mentions as presetId when present. This is synchronous for images: one call waits for every slot and returns final image artifacts. Use this for slide decks, landing pages, and multi-slot design work. Do not call get-generation-run or refresh-generation-run after a normal image batch result.",
   schema: z.object({
-    libraryId: z.string(),
+    libraryId: z
+      .string()
+      .optional()
+      .describe(
+        "Brand kit/library ID. Pass the refId from a brand-kit @mention, or choose a kit from view-screen/list-libraries.",
+      ),
     collectionId: z.string().optional(),
     presetId: z.string().optional(),
     sessionId: z.string().optional(),
@@ -45,6 +53,12 @@ export default defineAction({
       )
       .min(1)
       .max(12),
+    variantScopeId: z
+      .string()
+      .optional()
+      .describe(
+        "Internal UI state scope for live candidate slots. Usually omitted; embedded picker UIs pass a browser-tab scope.",
+      ),
     model: z.enum(IMAGE_MODELS).optional(),
     tier: z.enum(IMAGE_QUALITY_TIERS).optional(),
     intent: z.enum(GENERATION_INTENTS).default("generate"),
@@ -60,41 +74,74 @@ export default defineAction({
       ),
   }),
   parallelSafe: true,
-  run: async ({ slots, ...base }) => {
+  run: async ({ slots, ...inputBase }, context?: ActionRunContext) => {
+    const imageModelDefault = await readImageModelDefault();
+    const libraryId = inputBase.libraryId;
+    if (!libraryId) {
+      throw new Error(
+        "No brand kit selected. Tag a brand kit with @ or pass libraryId.",
+      );
+    }
+    const base = {
+      ...inputBase,
+      libraryId,
+      model: inputBase.model ?? imageModelDefault,
+    };
     await assertAccess("asset-library", base.libraryId, "editor");
     if (base.sessionId) {
       await requireGenerationSessionInLibrary(base.sessionId, base.libraryId);
     }
-    const limit = pLimit(4);
     const variantBatchId = nanoid();
+    await Promise.all(
+      slots.map((slot, index) =>
+        upsertVariantSlot({
+          runId: `pending-${variantBatchId}-${index + 1}`,
+          batchId: variantBatchId,
+          libraryId: base.libraryId,
+          collectionId: base.collectionId ?? null,
+          presetId: base.presetId ?? null,
+          sessionId: base.sessionId ?? null,
+          threadId: context?.threadId ?? null,
+          variantScopeId: base.variantScopeId ?? null,
+          prompt: slot.prompt,
+          slotId: slot.slotId,
+          status: "pending",
+        }),
+      ),
+    );
+    const limit = pLimit(4);
     const results = await Promise.allSettled(
       slots.map((slot) =>
         limit(() =>
-          generateImage.run({
-            libraryId: base.libraryId,
-            collectionId: base.collectionId,
-            presetId: base.presetId,
-            sessionId: base.sessionId,
-            prompt: slot.prompt,
-            aspectRatio: slot.aspectRatio,
-            imageSize: slot.imageSize,
-            model: base.model,
-            tier: base.tier,
-            intent: slot.intent ?? base.intent,
-            styleStrength: slot.styleStrength ?? base.styleStrength,
-            categories: slot.categories,
-            referenceAssetIds: slot.referenceAssetIds,
-            includeLogo: base.includeLogo,
-            groundingMode: base.groundingMode,
-            slotId: slot.slotId,
-            variantBatchId,
-            dismissible: slot.dismissible,
-            sourceAssetId: slot.sourceAssetId,
-            subjectAssetId: slot.subjectAssetId,
-            source: base.source,
-            callerAppId: base.callerAppId,
-            activateSessionAsset: false,
-          }),
+          generateImage.run(
+            {
+              libraryId: base.libraryId,
+              collectionId: base.collectionId,
+              presetId: base.presetId,
+              sessionId: base.sessionId,
+              prompt: slot.prompt,
+              aspectRatio: slot.aspectRatio,
+              imageSize: slot.imageSize,
+              model: base.model,
+              tier: base.tier,
+              intent: slot.intent ?? base.intent,
+              styleStrength: slot.styleStrength ?? base.styleStrength,
+              categories: slot.categories,
+              referenceAssetIds: slot.referenceAssetIds,
+              includeLogo: base.includeLogo,
+              groundingMode: base.groundingMode,
+              slotId: slot.slotId,
+              variantBatchId,
+              variantScopeId: base.variantScopeId,
+              dismissible: slot.dismissible,
+              sourceAssetId: slot.sourceAssetId,
+              subjectAssetId: slot.subjectAssetId,
+              source: base.source,
+              callerAppId: base.callerAppId,
+              activateSessionAsset: false,
+            },
+            context,
+          ),
         ),
       ),
     );

package/corpus/templates/assets/actions/generate-image.ts

@@ -1,6 +1,6 @@
 import { defineAction } from "@agent-native/core";
+import type { ActionRunContext } from "@agent-native/core/action";
 import {
-  readAppState,
   writeAppState,
   deleteAppState,
 } from "@agent-native/core/application-state";
@@ -43,6 +43,7 @@ import {
   requireGenerationSessionInLibrary,
   serializeAsset,
 } from "./_helpers.js";
+import { readImageModelDefault } from "./_image-model-default.js";
 import { upsertVariantSlot, wasVariantSlotDismissed } from "./variant-slots.js";
 
 function resolveModelForTier(
@@ -57,33 +58,16 @@ function resolveModelForTier(
     : "gemini-3.1-flash-image";
 }
 
-/**
- * The user's default image model, chosen from the composer's model picker and
- * persisted in per-user application state. Used as a fallback when no explicit
- * model, tier, or preset model is supplied. Returns undefined when unset or
- * invalid so the hardcoded default still applies.
- */
-async function readUserDefaultImageModel(): Promise<ImageModel | undefined> {
-  try {
-    const stored = await readAppState("imageGenerationModel");
-    const model = stored?.model;
-    if (
-      typeof model === "string" &&
-      (IMAGE_MODELS as readonly string[]).includes(model)
-    ) {
-      return model as ImageModel;
-    }
-  } catch {
-    // No request context or read failure — fall back to defaults below.
-  }
-  return undefined;
-}
-
 export default defineAction({
   description:
-    "Generate one brand-consistent image from a library. This is synchronous for images and returns the final asset with preview/download/embed URLs. Use generate-image-batch for multiple independent slots; do not poll image runs after this action returns.",
+    "Generate one brand-consistent image from a brand kit/library. This is synchronous for images and returns the final asset with preview/download/embed URLs. Use @brand-kit mentions as libraryId and @preset mentions as presetId when present. Use generate-image-batch for multiple independent slots; do not poll image runs after this action returns.",
   schema: z.object({
-    libraryId: z.string(),
+    libraryId: z
+      .string()
+      .optional()
+      .describe(
+        "Brand kit/library ID. Pass the refId from a brand-kit @mention, or choose a kit from view-screen/list-libraries.",
+      ),
     collectionId: z.string().optional(),
     presetId: z.string().optional(),
     sessionId: z.string().optional(),
@@ -104,6 +88,12 @@ export default defineAction({
     includeLogo: z.coerce.boolean().default(false),
     slotId: z.string().optional(),
     variantBatchId: z.string().optional(),
+    variantScopeId: z
+      .string()
+      .optional()
+      .describe(
+        "Internal UI state scope for live candidate slots. Usually omitted; embedded picker UIs pass a browser-tab scope.",
+      ),
     dismissible: z.coerce
       .boolean()
       .default(true)
@@ -135,7 +125,18 @@ export default defineAction({
       ),
   }),
   parallelSafe: true,
-  run: async (args) => {
+  run: async (input, context?: ActionRunContext) => {
+    const imageModelDefault = await readImageModelDefault();
+    const libraryId = input.libraryId;
+    if (!libraryId) {
+      throw new Error(
+        "No brand kit selected. Tag a brand kit with @ or pass libraryId.",
+      );
+    }
+    const args = {
+      ...input,
+      libraryId,
+    };
     await assertAccess("asset-library", args.libraryId, "editor");
     const db = getDb();
     const [library] = await db
@@ -248,9 +249,9 @@ export default defineAction({
       preset?.category ??
       collection?.category) as ImageCategory | undefined;
     const resolvedModel = (args.model ??
+      imageModelDefault ??
       resolveModelForTier(resolvedTier, category) ??
       preset?.model ??
-      (await readUserDefaultImageModel()) ??
       "gemini-3.1-flash-image") as (typeof IMAGE_MODELS)[number];
     const resolvedCategories =
       args.categories ??
@@ -303,6 +304,8 @@ export default defineAction({
     });
     const runId = nanoid();
     const now = nowIso();
+    const slotId = args.slotId ?? runId;
+    const variantScopeId = args.variantScopeId ?? context?.threadId ?? null;
     // Capture identity at insert time so the org-admin audit log can filter
     // by owner / org without re-resolving who triggered the run later.
     const ownerEmail = getRequestUserEmail() ?? null;
@@ -342,11 +345,12 @@ export default defineAction({
       sessionId: session?.id ?? null,
       customInstructions: library.customInstructions ?? "",
     };
-    const slotId = args.slotId ?? runId;
     const dismissibleSlot = args.dismissible !== false && Boolean(slotId);
     const baseMetadata = {
       slotId,
       variantBatchId: args.variantBatchId ?? null,
+      threadId: context?.threadId ?? null,
+      variantScopeId,
       dismissible: dismissibleSlot,
       sourceAssetId: args.sourceAssetId,
       subjectAssetId: args.subjectAssetId,
@@ -389,6 +393,8 @@ export default defineAction({
       collectionId: resolvedCollectionId ?? null,
       presetId: preset?.id ?? null,
       sessionId: session?.id ?? null,
+      threadId: context?.threadId ?? null,
+      variantScopeId,
       prompt: args.prompt,
       slotId,
       status: "pending",
@@ -430,7 +436,10 @@ export default defineAction({
       }
       if (
         dismissibleSlot &&
-        (await wasVariantSlotDismissed(args.libraryId, slotId))
+        (await wasVariantSlotDismissed(args.libraryId, slotId, {
+          threadId: context?.threadId ?? null,
+          variantScopeId,
+        }))
       ) {
         await db
           .update(schema.assetGenerationRuns)
@@ -539,6 +548,8 @@ export default defineAction({
         collectionId: resolvedCollectionId ?? null,
         presetId: preset?.id ?? null,
         sessionId: session?.id ?? null,
+        threadId: context?.threadId ?? null,
+        variantScopeId,
         prompt: args.prompt,
         slotId,
         status: "ready",
@@ -570,7 +581,10 @@ export default defineAction({
         .where(eq(schema.assetGenerationRuns.id, runId));
       if (
         dismissibleSlot &&
-        (await wasVariantSlotDismissed(args.libraryId, slotId))
+        (await wasVariantSlotDismissed(args.libraryId, slotId, {
+          threadId: context?.threadId ?? null,
+          variantScopeId,
+        }))
       ) {
         throw err;
       }
@@ -581,6 +595,8 @@ export default defineAction({
         collectionId: resolvedCollectionId ?? null,
         presetId: preset?.id ?? null,
         sessionId: session?.id ?? null,
+        threadId: context?.threadId ?? null,
+        variantScopeId,
         prompt: args.prompt,
         slotId,
         status: "failed",

package/corpus/templates/assets/actions/generate-video.ts

@@ -32,9 +32,14 @@ import { serializeAsset, serializeGenerationRun } from "./_helpers.js";
 
 export default defineAction({
   description:
-    "Start an async Veo video generation run from an asset library. Poll the returned run with refresh-generation-run until it completes and creates a video asset.",
+    "Start an async Veo video generation run from a brand kit/library. Use a media-type @mention with refId video to choose this instead of image generation, and use a brand-kit @mention as libraryId. Poll the returned run with refresh-generation-run until it completes and creates a video asset.",
   schema: z.object({
-    libraryId: z.string(),
+    libraryId: z
+      .string()
+      .optional()
+      .describe(
+        "Brand kit/library ID. Pass the refId from a brand-kit @mention, or choose a kit from view-screen/list-libraries.",
+      ),
     folderId: z.string().min(1).nullable().optional(),
     collectionId: z.string().optional(),
     prompt: z.string().min(1),
@@ -66,7 +71,17 @@ export default defineAction({
     callerAppId: z.string().optional(),
     waitForCompletion: z.coerce.boolean().default(false),
   }),
-  run: async (args) => {
+  run: async (input) => {
+    const libraryId = input.libraryId;
+    if (!libraryId) {
+      throw new Error(
+        "No brand kit selected. Tag a brand kit with @ or pass libraryId.",
+      );
+    }
+    const args = {
+      ...input,
+      libraryId,
+    };
     await assertAccess("asset-library", args.libraryId, "editor");
     const db = getDb();
     const [library] = await db

package/corpus/templates/assets/actions/list-assets.ts

@@ -1,5 +1,6 @@
 import { defineAction } from "@agent-native/core";
-import { and, desc, eq, isNull } from "drizzle-orm";
+import { accessFilter } from "@agent-native/core/sharing";
+import { and, desc, eq, inArray, isNull } from "drizzle-orm";
 import { z } from "zod";
 
 import { getDb, schema } from "../server/db/index.js";
@@ -17,9 +18,9 @@ import {
 
 export default defineAction({
   description:
-    "List DAM assets in a library, optionally filtered by folder, collection, media type, status, role, category, or text query.",
+    "List DAM assets in one library, or across all accessible libraries when libraryId is omitted. Optionally filter by folder, collection, media type, status, role, category, or text query.",
   schema: z.object({
-    libraryId: z.string(),
+    libraryId: z.string().optional(),
     collectionId: z.string().optional(),
     folderId: z.string().nullable().optional(),
     mediaType: z.enum(ASSET_MEDIA_TYPES).optional(),
@@ -51,8 +52,32 @@ export default defineAction({
     includeCandidates,
     candidateRunIds,
   }) => {
-    await requireLibrary(libraryId);
-    const filters = [eq(schema.assets.libraryId, libraryId)];
+    const db = getDb();
+    const libraryRows = libraryId
+      ? [await requireLibrary(libraryId)]
+      : await db
+          .select({
+            id: schema.assetLibraries.id,
+            title: schema.assetLibraries.title,
+          })
+          .from(schema.assetLibraries)
+          .where(
+            and(
+              accessFilter(schema.assetLibraries, schema.assetLibraryShares),
+              isNull(schema.assetLibraries.archivedAt),
+            ),
+          );
+    const libraryIds = libraryRows.map((library) => library.id);
+    if (!libraryIds.length) return { count: 0, assets: [] };
+
+    const libraryTitleById = new Map(
+      libraryRows.map((library) => [library.id, library.title]),
+    );
+    const filters = [
+      libraryId
+        ? eq(schema.assets.libraryId, libraryId)
+        : inArray(schema.assets.libraryId, libraryIds),
+    ];
     if (collectionId)
       filters.push(eq(schema.assets.collectionId, collectionId));
     if (folderId !== undefined) {
@@ -67,7 +92,6 @@ export default defineAction({
     if (role) filters.push(eq(schema.assets.role, role));
     const normalizedQuery = query?.trim().toLowerCase();
     const candidateRunIdSet = new Set(candidateRunIds ?? []);
-    const db = getDb();
     const [rows, lineageRows] = await Promise.all([
       db
         .select()
@@ -89,7 +113,11 @@ export default defineAction({
           createdAt: schema.assets.createdAt,
         })
         .from(schema.assets)
-        .where(eq(schema.assets.libraryId, libraryId)),
+        .where(
+          libraryId
+            ? eq(schema.assets.libraryId, libraryId)
+            : inArray(schema.assets.libraryId, libraryIds),
+        ),
     ]);
     const lineageById = buildAssetLineage(lineageRows);
     const assets = rows
@@ -111,7 +139,10 @@ export default defineAction({
         );
       })
       .filter((asset) => assetMatchesSearch(asset, normalizedQuery, category))
-      .map((asset) => serializeAsset(asset, lineageById.get(asset.id) ?? null));
+      .map((asset) => ({
+        ...serializeAsset(asset, lineageById.get(asset.id) ?? null),
+        libraryTitle: libraryTitleById.get(asset.libraryId) ?? null,
+      }));
     return { count: assets.length, assets };
   },
 });

package/corpus/templates/assets/actions/list-libraries.ts

@@ -67,10 +67,11 @@ export default defineAction({
     "List asset libraries accessible to the current user, including counts and preview thumbnails.",
   schema: z.object({
     compact: z.coerce.boolean().optional(),
+    includeFolders: z.coerce.boolean().optional(),
   }),
   http: { method: "GET" },
   readOnly: true,
-  run: async ({ compact }) => {
+  run: async ({ compact, includeFolders }) => {
     const db = getDb();
     const rows = await db
       .select()
@@ -93,8 +94,23 @@ export default defineAction({
             ),
           )
       : [];
+    const folders =
+      includeFolders && rows.length
+        ? await db
+            .select()
+            .from(schema.assetFolders)
+            .where(
+              inArray(
+                schema.assetFolders.libraryId,
+                rows.map((row) => row.id),
+              ),
+            )
+        : [];
     const libraries = rows.map((row) => {
       const libAssets = assets.filter((asset) => asset.libraryId === row.id);
+      const libFolders = includeFolders
+        ? folders.filter((folder) => folder.libraryId === row.id)
+        : undefined;
       const visibleAssets = libAssets.filter((asset) =>
         shouldIncludeAssetInLibraryResults(asset),
       );
@@ -106,7 +122,12 @@ export default defineAction({
       const previewAssets = uniqueAssets([cover, ...imageAssets]).slice(0, 4);
       const base = serializeLibrary(row);
       return compact
-        ? { id: base.id, title: base.title, description: base.description }
+        ? {
+            id: base.id,
+            title: base.title,
+            description: base.description,
+            ...(includeFolders ? { folders: libFolders } : {}),
+          }
         : {
             ...base,
             referenceCount: visibleAssets.filter((asset) =>
@@ -122,6 +143,7 @@ export default defineAction({
             ).length,
             coverAsset: cover ? serializeAsset(cover) : null,
             previewAssets: previewAssets.map(serializeAsset),
+            ...(includeFolders ? { folders: libFolders } : {}),
           };
     });
     return { count: libraries.length, libraries };

package/corpus/templates/assets/actions/navigate.ts

@@ -4,7 +4,7 @@ import { z } from "zod";
 
 export default defineAction({
   description:
-    "Navigate the Assets UI. Views (internal keys, with the surface they open): create, picker (the image Library browser), libraries (the Brand Kits list), library (a single Brand Kit), asset, generation-session, generation-run, extensions, audit, settings. Use threadId to open a specific create/chat thread; use libraryId, assetId, sessionId, runId, or extensionId where appropriate.",
+    "Navigate the Assets UI. Views (internal keys, with the surface they open): create, picker (the embedded image Library picker), libraries (the unified Library workspace), library (a single Brand Kit inside Library), asset, generation-session, generation-run, extensions, audit, settings. Use threadId to open a specific create/chat thread; use libraryId, assetId, sessionId, runId, or extensionId where appropriate.",
   schema: z.object({
     view: z
       .enum([

package/corpus/templates/assets/actions/open-asset-picker.ts

@@ -1,4 +1,7 @@
 import { defineAction, embedApp } from "@agent-native/core";
+import type { ActionRunContext } from "@agent-native/core/action";
+import { writeAppState } from "@agent-native/core/application-state";
+import { getRequestRunContext } from "@agent-native/core/server/request-context";
 import { z } from "zod";
 
 import { IMAGE_QUALITY_TIERS, STYLE_STRENGTHS } from "../shared/api.js";
@@ -77,6 +80,7 @@ const schema = z.object({
 });
 
 type OpenAssetPickerArgs = z.infer<typeof schema>;
+const SAFE_BROWSER_TAB_ID_RE = /^[A-Za-z0-9_-]{1,96}$/;
 
 type ActionWithToolParameters = {
   tool: { parameters?: { properties?: Record<string, any> } };
@@ -109,6 +113,7 @@ const FALLBACK_INSTRUCTIONS =
 
 function pickerPath(args: Partial<OpenAssetPickerArgs>): string {
   const params = new URLSearchParams();
+  params.set("__an_picker", "1");
   params.set("mediaType", args.mediaType ?? "image");
   if (args.prompt?.trim()) params.set("prompt", args.prompt.trim());
   if (args.query?.trim()) params.set("q", args.query.trim());
@@ -134,6 +139,21 @@ function pickerPath(args: Partial<OpenAssetPickerArgs>): string {
   return `/library?${params.toString()}`;
 }
 
+function navigateCommandKey(): string | null {
+  const browserTabId = getRequestRunContext()?.browserTabId?.trim();
+  if (browserTabId && SAFE_BROWSER_TAB_ID_RE.test(browserTabId)) {
+    return `navigate:${browserTabId}`;
+  }
+  return null;
+}
+
+function shouldWriteNavigateCommand(
+  context: ActionRunContext | undefined,
+  commandKey: string | null,
+): commandKey is string {
+  return Boolean(commandKey) && context?.caller !== "mcp";
+}
+
 const action = defineAction({
   description:
     'Open the image Library picker inline so a person can browse, search, generate, and select an image or video asset. When the user asks to create a specific image and choose the best one, pass prompt, autoGenerate: true, and count: 3 so the Library opens with generated candidates. If the host can only open a browser link (e.g. a CLI or code editor), surface that link: after the user picks, the page auto-copies a short paste-back summary — or the user can simply tell you which candidate they want (e.g. "use image A"). Use search-assets, generate-image, generate-video, and export-asset for unattended flows.',
@@ -173,8 +193,25 @@ const action = defineAction({
       view: "picker",
     };
   },
-  run: async (args) => {
+  run: async (args, context) => {
     const path = pickerPath(args);
+    const commandKey = navigateCommandKey();
+    if (shouldWriteNavigateCommand(context, commandKey)) {
+      const command = {
+        view: "picker" as const,
+        mediaType: args.mediaType,
+        path,
+        libraryId: args.libraryId ?? null,
+        query: args.query ?? null,
+        prompt: args.prompt ?? null,
+        aspectRatio: args.aspectRatio ?? null,
+        presetId: args.presetId ?? null,
+        _writeId: `open-asset-picker-${Date.now()}-${Math.random()
+          .toString(36)
+          .slice(2, 8)}`,
+      };
+      await writeAppState(commandKey, command);
+    }
     return {
       app: "assets",
       view: "picker",

package/corpus/templates/assets/actions/refresh-generation-run.ts

@@ -36,6 +36,14 @@ async function syncImageVariantSlot(
     typeof metadata.variantBatchId === "string" && metadata.variantBatchId
       ? metadata.variantBatchId
       : null;
+  const threadId =
+    typeof metadata.threadId === "string" && metadata.threadId
+      ? metadata.threadId
+      : null;
+  const variantScopeId =
+    typeof metadata.variantScopeId === "string" && metadata.variantScopeId
+      ? metadata.variantScopeId
+      : null;
   const serialized = options.asset ? serializeAsset(options.asset) : null;
 
   await upsertVariantSlot({
@@ -45,6 +53,8 @@ async function syncImageVariantSlot(
     collectionId: run.collectionId ?? null,
     presetId: run.presetId ?? null,
     sessionId: run.sessionId ?? null,
+    threadId,
+    variantScopeId,
     prompt: run.prompt,
     slotId,
     status,