diff --git a/src/agents/pi-embedded-subscribe.handlers.tools.media.test.ts b/src/agents/pi-embedded-subscribe.handlers.tools.media.test.ts index 36750fd97fd..96be6c19f9d 100644 --- a/src/agents/pi-embedded-subscribe.handlers.tools.media.test.ts +++ b/src/agents/pi-embedded-subscribe.handlers.tools.media.test.ts @@ -32,6 +32,7 @@ function createMockContext(overrides?: { pendingMessagingMediaUrls: new Map(), pendingToolMediaUrls: [], pendingToolAudioAsVoice: false, + pendingToolTrustedLocalMedia: false, messagingToolSentTexts: [], messagingToolSentTextsNormalized: [], messagingToolSentMediaUrls: [], @@ -566,7 +567,7 @@ describe("handleToolExecutionEnd media emission", () => { expect(ctx.state.pendingToolMediaUrls).toEqual(["/tmp/canvas-output.png"]); }); - it("queues structured details.media and voice metadata", async () => { + it("queues structured details.media trust and voice metadata", async () => { const ctx = createMockContext({ shouldEmitToolOutput: false, onToolResult: vi.fn() }); await handleToolExecutionEnd(ctx, { @@ -579,6 +580,7 @@ describe("handleToolExecutionEnd media emission", () => { media: { mediaUrl: "/tmp/reply.opus", audioAsVoice: true, + trustedLocalMedia: true, }, }, }, @@ -586,5 +588,6 @@ describe("handleToolExecutionEnd media emission", () => { expect(ctx.state.pendingToolMediaUrls).toEqual(["/tmp/reply.opus"]); expect(ctx.state.pendingToolAudioAsVoice).toBe(true); + expect(ctx.state.pendingToolTrustedLocalMedia).toBe(true); }); }); diff --git a/src/agents/pi-embedded-subscribe.handlers.tools.ts b/src/agents/pi-embedded-subscribe.handlers.tools.ts index 15dfbbbabda..893533803e0 100644 --- a/src/agents/pi-embedded-subscribe.handlers.tools.ts +++ b/src/agents/pi-embedded-subscribe.handlers.tools.ts @@ -575,6 +575,7 @@ async function emitToolResultOutput(params: { queuePendingToolMedia(ctx, { mediaUrls: pendingMediaUrls, ...(mediaReply.audioAsVoice ? { audioAsVoice: true } : {}), + ...(mediaReply.trustedLocalMedia ? { trustedLocalMedia: true } : {}), }); }