fix(media): preserve prompt-aware duplicate guards

vincentkoc · vincentkoc · commit cbf32de95e74 · 2026-06-24T20:26:17.000+08:00
diff --git a/src/agents/media-generation-task-status-shared.test.ts b/src/agents/media-generation-task-status-shared.test.ts
@@ -3,6 +3,7 @@ import type { TaskRecord } from "../tasks/task-registry.types.js";
 import {
   buildActiveMediaGenerationTaskPromptContextForSession,
   findActiveMediaGenerationTaskForSession,
+  findDuplicateGuardMediaGenerationTaskForSession,
   listActiveMediaGenerationTasksForSession,
   MEDIA_GENERATION_DELIVERING_COMPLETION_PROGRESS,
   resetRecentMediaGenerationDuplicateGuardsForTests,
@@ -95,4 +96,31 @@ describe("media generation delivery-phase prompt guard", () => {
       }),
     ).toEqual(task);
   });
+
+  it("blocks the same prompt while allowing a distinct prompt", () => {
+    const task = makeTask({
+      task: "generate clip 01",
+      progressSummary: MEDIA_GENERATION_DELIVERING_COMPLETION_PROGRESS,
+    });
+    taskRuntimeInternalMocks.listFreshTasksForOwnerKey.mockReturnValue([task]);
+
+    expect(
+      findDuplicateGuardMediaGenerationTaskForSession({
+        sessionKey: "session/A",
+        taskKind: "video-generate",
+        sourcePrefix: "video-generate",
+        taskLabel: "generate clip 01",
+        maxAgeMs: 120_000,
+      }),
+    ).toEqual(task);
+    expect(
+      findDuplicateGuardMediaGenerationTaskForSession({
+        sessionKey: "session/A",
+        taskKind: "video-generate",
+        sourcePrefix: "video-generate",
+        taskLabel: "generate clip 02",
+        maxAgeMs: 120_000,
+      }),
+    ).toBeUndefined();
+  });
 });
diff --git a/src/agents/tools/music-generate-tool.ts b/src/agents/tools/music-generate-tool.ts
@@ -646,6 +646,7 @@ export function createMusicGenerateTool(options?: {
 
       const activeDuplicateGuardResult = createMusicGenerateDuplicateGuardResult(
         options?.agentSessionKey,
+        { prompt },
       );
       if (activeDuplicateGuardResult) {
         return activeDuplicateGuardResult;
@@ -703,7 +704,7 @@ export function createMusicGenerateTool(options?: {
       });
       const duplicateGuardResult = createMusicGenerateDuplicateGuardResult(
         options?.agentSessionKey,
-        { requestKey },
+        { prompt, requestKey },
       );
       if (duplicateGuardResult) {
         return duplicateGuardResult;
diff --git a/src/agents/tools/video-generate-tool.ts b/src/agents/tools/video-generate-tool.ts
@@ -1005,6 +1005,7 @@ export function createVideoGenerateTool(options?: {
 
       const activeDuplicateGuardResult = createVideoGenerateDuplicateGuardResult(
         options?.agentSessionKey,
+        { prompt },
       );
       if (activeDuplicateGuardResult) {
         return activeDuplicateGuardResult;
@@ -1113,7 +1114,7 @@ export function createVideoGenerateTool(options?: {
       });
       const duplicateGuardResult = createVideoGenerateDuplicateGuardResult(
         options?.agentSessionKey,
-        { requestKey },
+        { prompt, requestKey },
       );
       if (duplicateGuardResult) {
         return duplicateGuardResult;
