diff --git a/packages/memory-host-sdk/src/host/internal.test.ts b/packages/memory-host-sdk/src/host/internal.test.ts index 926632ca2d7..3759d963f87 100644 --- a/packages/memory-host-sdk/src/host/internal.test.ts +++ b/packages/memory-host-sdk/src/host/internal.test.ts @@ -17,6 +17,9 @@ import { } from "./multimodal.js"; type FileEntry = NonNullable>>; +type MultimodalIndexingChunk = NonNullable< + Awaited> +>; let sharedTempRoot = ""; let sharedTempId = 0; @@ -41,12 +44,23 @@ function setupTempDirLifecycle(prefix: string): () => string { } function expectFileEntry(entry: Awaited>): FileEntry { + expect(entry).toBeTruthy(); if (!entry) { throw new Error("Expected file entry to be built"); } return entry; } +function expectMultimodalIndexingChunk( + built: Awaited>, +): MultimodalIndexingChunk { + expect(built).toBeTruthy(); + if (!built) { + throw new Error("Expected multimodal indexing chunk to be built"); + } + return built; +} + const multimodal: MemoryMultimodalSettings = { enabled: true, modalities: ["image", "audio"], @@ -118,8 +132,8 @@ describe("memory host SDK package internals", () => { fsSync.writeFileSync(imagePath, Buffer.from("png")); const entry = expectFileEntry(await buildFileEntry(imagePath, tmpDir, multimodal)); - const built = await buildMultimodalChunkForIndexing(entry); - expect(built?.chunk.embeddingInput?.parts).toEqual([ + const built = expectMultimodalIndexingChunk(await buildMultimodalChunkForIndexing(entry)); + expect(built.chunk.embeddingInput.parts).toEqual([ { type: "text", text: "Image file: diagram.png" }, expect.objectContaining({ type: "inline-data", mimeType: "image/png" }), ]);