From 1ded4c672a0ad23f2c6a761c190c06ede2f51f25 Mon Sep 17 00:00:00 2001
From: vignesh07 <vigneshnatarajan92@gmail.com>
Date: Fri, 20 Feb 2026 20:17:45 -0800
Subject: [PATCH] test(memory): fix TS types after vitest/ts updates

---
 src/memory/manager.async-search.test.ts | 14 +++++++-------
 src/memory/qmd-manager.test.ts          |  2 +-
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/memory/manager.async-search.test.ts b/src/memory/manager.async-search.test.ts
index 2d366c6e9ee..30ac2dc07d0 100644
--- a/src/memory/manager.async-search.test.ts
+++ b/src/memory/manager.async-search.test.ts
@@ -7,14 +7,14 @@ import { getMemorySearchManager, type MemoryIndexManager } from "./index.js";
 import { createOpenAIEmbeddingProviderMock } from "./test-embeddings-mock.js";
 import { createMemoryManagerOrThrow } from "./test-manager.js";
 
-const embedBatch = vi.fn(async () => []);
-const embedQuery = vi.fn(async () => [0.2, 0.2, 0.2]);
+const embedBatch = vi.fn(async (_input: string[]) => [] as number[][]);
+const embedQuery = vi.fn(async (_input: string) => [0.2, 0.2, 0.2] as number[]);
 
 vi.mock("./embeddings.js", () => ({
-  createEmbeddingProvider: async () =>
+  createEmbeddingProvider: async (_options: unknown) =>
     createOpenAIEmbeddingProviderMock({
-      embedQuery,
-      embedBatch,
+      embedQuery: embedQuery as unknown as (input: string) => Promise<number[]>,
+      embedBatch: embedBatch as unknown as (input: string[]) => Promise<number[][]>,
     }),
 }));
 
@@ -78,7 +78,7 @@ describe("memory search async sync", () => {
 
   it("waits for in-flight search sync during close", async () => {
     const cfg = buildConfig();
-    let releaseSync: (() => void) | null = null;
+    let releaseSync!: (value?: void) => void;
     const syncGate = new Promise<void>((resolve) => {
       releaseSync = resolve;
     });
@@ -98,7 +98,7 @@ describe("memory search async sync", () => {
     await Promise.resolve();
     expect(closed).toBe(false);
 
-    releaseSync?.();
+    releaseSync();
     await closePromise;
     manager = null;
 
diff --git a/src/memory/qmd-manager.test.ts b/src/memory/qmd-manager.test.ts
index 90b0ae11d97..64c6a78c700 100644
--- a/src/memory/qmd-manager.test.ts
+++ b/src/memory/qmd-manager.test.ts
@@ -1303,7 +1303,7 @@ describe("QmdMemoryManager", () => {
     };
     inner.db = {
       prepare: () => ({
-        all: () => {
+        get: () => {
           throw new Error("SQLITE_BUSY: database is locked");
         },
       }),