@anuma/sdk 7.0.4 → 7.0.6

package/dist/react/index.mjs

@@ -4537,6 +4537,22 @@ var postApiV1WalletsBinding = (options) => (options.client ?? client).post({
 var postApiV1WalletsBindingNonce = (options) => (options?.client ?? client).post({ url: "/api/v1/wallets/binding/nonce", ...options });
 var deleteApiV1WalletsBindingByAddress = (options) => (options.client ?? client).delete({ url: "/api/v1/wallets/binding/{address}", ...options });
 
+// src/lib/memoryEngine/embeddingCacheState.ts
+var EMBEDDING_CACHE_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/embedding-cache/v1");
+var cacheGlobal = globalThis;
+var fallbackState = {
+  scopedViews: /* @__PURE__ */ new WeakMap(),
+  inFlightByCache: /* @__PURE__ */ new WeakMap(),
+  cachedEmbeddingModels: /* @__PURE__ */ new WeakMap()
+};
+function getEmbeddingCacheState() {
+  try {
+    return cacheGlobal[EMBEDDING_CACHE_STATE_KEY] ?? (cacheGlobal[EMBEDDING_CACHE_STATE_KEY] = fallbackState);
+  } catch {
+    return fallbackState;
+  }
+}
+
 // src/lib/memoryEngine/generate.ts
 var EMBED_MAX_ATTEMPTS = 4;
 var DEFAULT_EMBEDDING_REQUEST_TIMEOUT_MS = 15e3;
@@ -4628,15 +4644,18 @@ async function generateEmbedding(text14, options) {
   if (total === void 0) return embedOne(text14, options, void 0);
   return withDeadline((signal) => embedOne(text14, options, signal), total, "embedding");
 }
-var inFlightByCache = /* @__PURE__ */ new WeakMap();
+var { inFlightByCache, cachedEmbeddingModels } = getEmbeddingCacheState();
 async function embedOne(text14, options, outer) {
   const { cache } = options;
   if (!cache) return Array.from(await requestEmbedding(text14, options, outer));
+  const model = options.model ?? DEFAULT_API_EMBEDDING_MODEL;
   const cached = cache.get(text14);
-  if (cached) return Array.from(cached);
+  const cachedModel = cached && cachedEmbeddingModels.get(cached);
+  if (cached && (cachedModel === void 0 || cachedModel === model)) return Array.from(cached);
   const inFlight = inFlightByCache.get(cache) ?? /* @__PURE__ */ new Map();
   inFlightByCache.set(cache, inFlight);
-  const pending = inFlight.get(text14);
+  const requestKey = JSON.stringify([model, text14]);
+  const pending = inFlight.get(requestKey);
   if (pending) {
     try {
       return Array.from(await pending);
@@ -4645,13 +4664,14 @@ async function embedOne(text14, options, outer) {
     }
   }
   const request = requestEmbedding(text14, options, outer);
-  inFlight.set(text14, request);
+  inFlight.set(requestKey, request);
   try {
     const embedding2 = await request;
+    cachedEmbeddingModels.set(embedding2, model);
     cache.set(text14, embedding2);
     return Array.from(embedding2);
   } finally {
-    if (inFlight.get(text14) === request) inFlight.delete(text14);
+    if (inFlight.get(requestKey) === request) inFlight.delete(requestKey);
   }
 }
 async function requestEmbedding(text14, options, outer) {
@@ -4734,13 +4754,15 @@ async function embedMany(texts, options, outer) {
   const { baseUrl = BASE_URL, model, batchSize, cache } = options;
   const timeoutMs = options.timeoutMs ?? DEFAULT_EMBEDDING_REQUEST_TIMEOUT_MS;
   const chunkSize = batchSize ?? DEFAULT_EMBEDDING_BATCH_SIZE;
+  const embeddingModel = model ?? DEFAULT_API_EMBEDDING_MODEL;
   const results = new Array(texts.length).fill(null);
   const uncachedIndices = [];
   const uncachedTexts = [];
   for (let i = 0; i < texts.length; i++) {
     if (cache) {
       const cached = cache.get(texts[i]);
-      if (cached) {
+      const cachedModel = cached && cachedEmbeddingModels.get(cached);
+      if (cached && (cachedModel === void 0 || cachedModel === embeddingModel)) {
         results[i] = Array.from(cached);
         continue;
       }
@@ -4756,7 +4778,6 @@ async function embedMany(texts, options, outer) {
     ...taskTypeHeader(options.taskType)
   };
   if (outer?.aborted) throw outer.reason;
-  const embeddingModel = model ?? DEFAULT_API_EMBEDDING_MODEL;
   let newEmbeddings;
   if (uncachedTexts.length <= chunkSize) {
     newEmbeddings = await generateEmbeddingsBatch(
@@ -4804,6 +4825,7 @@ async function embedMany(texts, options, outer) {
     const f32Embedding = Float32Array.from(newEmbeddings[i]);
     results[uncachedIndices[i]] = Array.from(f32Embedding);
     if (cache) {
+      cachedEmbeddingModels.set(f32Embedding, embeddingModel);
       cache.set(uncachedTexts[i], f32Embedding);
     }
   }
@@ -19448,7 +19470,7 @@ var MaskScopedEmbeddingCache = class extends Map {
     return this;
   }
 };
-var scopedViews = /* @__PURE__ */ new WeakMap();
+var { scopedViews } = getEmbeddingCacheState();
 function maskScopedEmbeddingCache(cache, masked) {
   const views = scopedViews.get(cache) ?? {};
   scopedViews.set(cache, views);
@@ -19870,44 +19892,48 @@ async function recall(query, ctx, options = {}) {
       embeddingOptions,
       ctx.vaultCache,
       searchOptions
-    )))(query, {
-      limit: types.includes("chunk") ? Math.max(limit * 2, 16) : limit,
-      minSimilarity: vaultMinScore,
-      useFusion: true,
-      rerank: flags.rerank,
-      ...options.rerankTopN !== void 0 && { rerankTopN: options.rerankTopN },
-      ...options.ceWeight !== void 0 && { ceWeight: options.ceWeight },
-      ...options.rerankLoadTimeoutMs !== void 0 && {
-        rerankLoadTimeoutMs: options.rerankLoadTimeoutMs
-      },
-      ...options.recencyAlpha !== void 0 && { recencyAlpha: options.recencyAlpha },
-      ...options.recency && { recency: options.recency },
-      ...options.mmr !== void 0 && { mmr: options.mmr },
-      ...options.supersessionBoost !== void 0 && {
-        supersessionBoost: options.supersessionBoost
-      },
-      ...options.supersessionWindow !== void 0 && {
-        supersessionWindow: options.supersessionWindow
-      },
-      ...options.proofCountAlpha !== void 0 && {
-        proofCountAlpha: options.proofCountAlpha
-      },
-      ...options.bm25AdmissionDivisor !== void 0 && {
-        bm25AdmissionDivisor: options.bm25AdmissionDivisor
+    )))(
+      query,
+      {
+        limit: types.includes("chunk") ? Math.max(limit * 2, 16) : limit,
+        minSimilarity: vaultMinScore,
+        useFusion: true,
+        rerank: flags.rerank,
+        ...options.rerankTopN !== void 0 && { rerankTopN: options.rerankTopN },
+        ...options.ceWeight !== void 0 && { ceWeight: options.ceWeight },
+        ...options.rerankLoadTimeoutMs !== void 0 && {
+          rerankLoadTimeoutMs: options.rerankLoadTimeoutMs
+        },
+        ...options.recencyAlpha !== void 0 && { recencyAlpha: options.recencyAlpha },
+        ...options.recency && { recency: options.recency },
+        ...options.mmr !== void 0 && { mmr: options.mmr },
+        ...options.supersessionBoost !== void 0 && {
+          supersessionBoost: options.supersessionBoost
+        },
+        ...options.supersessionWindow !== void 0 && {
+          supersessionWindow: options.supersessionWindow
+        },
+        ...options.proofCountAlpha !== void 0 && {
+          proofCountAlpha: options.proofCountAlpha
+        },
+        ...options.bm25AdmissionDivisor !== void 0 && {
+          bm25AdmissionDivisor: options.bm25AdmissionDivisor
+        },
+        ...options.rrfK !== void 0 && { rrfK: options.rrfK },
+        ...options.decryptLast !== void 0 && { decryptLast: options.decryptLast },
+        ...subQueries && { subQueries },
+        ...options.scopes && { scopes: options.scopes },
+        ...options.folderId !== void 0 && { folderId: options.folderId },
+        ...options.factTypes?.length && { factTypes: options.factTypes },
+        ...options.memoryIds !== void 0 && { memoryIds: options.memoryIds },
+        ...options.factTypeWeights && { factTypeWeights: options.factTypeWeights },
+        ...entityRanking.length > 0 && { entityRanking },
+        ...temporalRanking.length > 0 && { temporalRanking },
+        ...needsChunkEmbedding && { queryEmbedding: queryEmbedding ?? [] },
+        queryEmbedTotalTimeoutMs
       },
-      ...options.rrfK !== void 0 && { rrfK: options.rrfK },
-      ...options.decryptLast !== void 0 && { decryptLast: options.decryptLast },
-      ...subQueries && { subQueries },
-      ...options.scopes && { scopes: options.scopes },
-      ...options.folderId !== void 0 && { folderId: options.folderId },
-      ...options.factTypes?.length && { factTypes: options.factTypes },
-      ...options.memoryIds !== void 0 && { memoryIds: options.memoryIds },
-      ...options.factTypeWeights && { factTypeWeights: options.factTypeWeights },
-      ...entityRanking.length > 0 && { entityRanking },
-      ...temporalRanking.length > 0 && { temporalRanking },
-      ...needsChunkEmbedding && { queryEmbedding: queryEmbedding ?? [] },
-      queryEmbedTotalTimeoutMs
-    });
+      options.embeddingCache
+    );
     factResults.push(
       ...dedupeBy(
         results,
@@ -21046,10 +21072,19 @@ function createLocalMemoryStore(options) {
       retainOptions
     ),
     factSource: {
-      search: (query, searchOptions) => searchVaultMemoriesWithSize(
+      search: (query, searchOptions, embeddingCache) => searchVaultMemoriesWithSize(
         query,
         vaultCtx,
-        { ...options.embeddingOptions, taskType: "memory_recall_embed" },
+        {
+          ...options.embeddingOptions,
+          taskType: "memory_recall_embed",
+          ...embeddingCache && !options.embeddingOptions.cache && {
+            cache: maskScopedEmbeddingCache(
+              embeddingCache,
+              Boolean(options.embeddingOptions.maskInput)
+            )
+          }
+        },
         vaultCache,
         searchOptions
       ),
@@ -21679,7 +21714,7 @@ function stored(row) {
 function createRemoteMemoryPipeline(options) {
   const { persistence, embeddingOptions } = options;
   const model = embeddingOptions.model ?? DEFAULT_API_EMBEDDING_MODEL;
-  const operation = (mode) => {
+  const operation = (mode, queryOptions = embeddingOptions) => {
     const rejectUndecryptable = (failed) => {
       if (!failed.length) return;
       if (mode === "retain")
@@ -21710,7 +21745,7 @@ function createRemoteMemoryPipeline(options) {
       const start = Date.now();
       const budgetMs = search.queryEmbedTotalTimeoutMs ?? 8e3;
       const embed = (texts) => generateEmbeddings(texts, {
-        ...embeddingOptions,
+        ...queryOptions,
         taskType: mode === "retain" ? "memory_retain_embed" : "memory_recall_embed",
         totalTimeoutMs: budgetMs
       });
@@ -21730,7 +21765,11 @@ function createRemoteMemoryPipeline(options) {
         }
       };
       const [primary, facetEmbeddings] = await Promise.all([
-        search.queryEmbedding ?? embed([query]).then((vectors2) => vectors2[0]),
+        search.queryEmbedding ?? (mode === "recall" && queryOptions.cache ? generateEmbedding(query, {
+          ...queryOptions,
+          taskType: "memory_recall_embed",
+          totalTimeoutMs: budgetMs
+        }) : embed([query]).then((vectors2) => vectors2[0])),
         embedFacets()
       ]);
       const queryEmbedding = storedVector(primary);
@@ -21954,15 +21993,17 @@ function createRemoteMemoryPipeline(options) {
   const factSource = {
     graphRanking: options.graphRanking,
     temporalRanking: options.temporalRanking,
-    search: async (text14, searchOptions) => {
-      const { prepare } = operation("recall");
+    search: async (text14, searchOptions, embeddingCache) => {
+      const queryOptions = {
+        ...embeddingOptions,
+        taskType: "memory_recall_embed",
+        ...embeddingCache && !embeddingOptions.cache && {
+          cache: maskScopedEmbeddingCache(embeddingCache, Boolean(embeddingOptions.maskInput))
+        }
+      };
+      const { prepare } = operation("recall", queryOptions);
       const prepared = await prepare(text14, searchOptions);
-      const ranked = await rankPreparedVaultCandidates(
-        text14,
-        prepared,
-        { ...embeddingOptions, taskType: "memory_recall_embed" },
-        searchOptions
-      );
+      const ranked = await rankPreparedVaultCandidates(text14, prepared, queryOptions, searchOptions);
       return {
         ...ranked,
         rankedOnCosine: prepared.memories.length > 0 && !prepared.embeddingsUnavailable,
@@ -29478,6 +29519,7 @@ function useChat(options) {
 
 // src/react/useChatStorage.ts
 import { useCallback as useCallback5, useEffect as useEffect4, useMemo as useMemo4, useRef as useRef4, useState as useState5 } from "react";
+import { v7 as uuidv77 } from "uuid";
 
 // src/lib/chat/assembleMessages.ts
 function assembleMessagesWithHistory(historyMessages, callerMessages, summarySystemMessage) {
@@ -31233,11 +31275,15 @@ function useChatStorage(options) {
   });
   const createConversation = useCallback5(
     async (opts) => {
+      const conversationOpts = {
+        ...opts,
+        conversationId: opts?.conversationId || generateConversationId()
+      };
       const { result, queued, queueId } = await writeOrQueue(
         "createConversation",
-        { conversationId: opts?.conversationId, title: opts?.title, projectId: opts?.projectId },
-        () => createConversationOp(storageCtx, opts, defaultConversationTitle),
-        () => makeSyntheticStoredConversation(opts, defaultConversationTitle)
+        conversationOpts,
+        () => createConversationOp(storageCtx, conversationOpts, defaultConversationTitle),
+        () => makeSyntheticStoredConversation(conversationOpts, defaultConversationTitle)
       );
       if (queued) {
         syntheticConvIdsRef.current.add(result.conversationId);
@@ -31925,6 +31971,7 @@ function useChatStorage(options) {
         );
       }
       const userMsgOpts = {
+        uniqueId: `msg_${uuidv77()}`,
         conversationId: convId,
         role: "user",
         content: contentForStorage,
@@ -32252,7 +32299,7 @@ function useChatStorage(options) {
         thinking: thinkingContent,
         parentMessageId: storedUserMessage.uniqueId,
         toolCallEvents: currentTurnToolCallEvents && currentTurnToolCallEvents.length > 0 ? currentTurnToolCallEvents : void 0,
-        uniqueId: assistantUniqueId
+        uniqueId: assistantUniqueId || `msg_${uuidv77()}`
       };
       let storedAssistantMessage;
       let assistantMsgQueueId;
@@ -32299,6 +32346,7 @@ function useChatStorage(options) {
       let storedToolResultsMessage;
       if (autoToolResults && autoToolResults.length > 0) {
         const toolResultsOpts = {
+          uniqueId: `msg_${uuidv77()}`,
           conversationId: convId,
           role: "user",
           content: buildToolResultContent(autoToolResults),

package/dist/server/index.cjs

@@ -20138,6 +20138,24 @@ var postApiV1Embeddings = (options) => (options.client ?? client).post({
 // src/lib/memoryEngine/generate.ts
 init_clientConfig();
 init_taskType();
+
+// src/lib/memoryEngine/embeddingCacheState.ts
+var EMBEDDING_CACHE_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/embedding-cache/v1");
+var cacheGlobal = globalThis;
+var fallbackState = {
+  scopedViews: /* @__PURE__ */ new WeakMap(),
+  inFlightByCache: /* @__PURE__ */ new WeakMap(),
+  cachedEmbeddingModels: /* @__PURE__ */ new WeakMap()
+};
+function getEmbeddingCacheState() {
+  try {
+    return cacheGlobal[EMBEDDING_CACHE_STATE_KEY] ?? (cacheGlobal[EMBEDDING_CACHE_STATE_KEY] = fallbackState);
+  } catch {
+    return fallbackState;
+  }
+}
+
+// src/lib/memoryEngine/generate.ts
 var EMBED_MAX_ATTEMPTS = 4;
 var DEFAULT_EMBEDDING_REQUEST_TIMEOUT_MS = 15e3;
 var DEFAULT_EMBEDDING_TOKEN_TIMEOUT_MS = 1e4;
@@ -20228,15 +20246,18 @@ async function generateEmbedding(text12, options) {
   if (total === void 0) return embedOne(text12, options, void 0);
   return withDeadline((signal) => embedOne(text12, options, signal), total, "embedding");
 }
-var inFlightByCache = /* @__PURE__ */ new WeakMap();
+var { inFlightByCache, cachedEmbeddingModels } = getEmbeddingCacheState();
 async function embedOne(text12, options, outer) {
   const { cache } = options;
   if (!cache) return Array.from(await requestEmbedding(text12, options, outer));
+  const model = options.model ?? DEFAULT_API_EMBEDDING_MODEL;
   const cached = cache.get(text12);
-  if (cached) return Array.from(cached);
+  const cachedModel = cached && cachedEmbeddingModels.get(cached);
+  if (cached && (cachedModel === void 0 || cachedModel === model)) return Array.from(cached);
   const inFlight = inFlightByCache.get(cache) ?? /* @__PURE__ */ new Map();
   inFlightByCache.set(cache, inFlight);
-  const pending = inFlight.get(text12);
+  const requestKey = JSON.stringify([model, text12]);
+  const pending = inFlight.get(requestKey);
   if (pending) {
     try {
       return Array.from(await pending);
@@ -20245,13 +20266,14 @@ async function embedOne(text12, options, outer) {
     }
   }
   const request = requestEmbedding(text12, options, outer);
-  inFlight.set(text12, request);
+  inFlight.set(requestKey, request);
   try {
     const embedding = await request;
+    cachedEmbeddingModels.set(embedding, model);
     cache.set(text12, embedding);
     return Array.from(embedding);
   } finally {
-    if (inFlight.get(text12) === request) inFlight.delete(text12);
+    if (inFlight.get(requestKey) === request) inFlight.delete(requestKey);
   }
 }
 async function requestEmbedding(text12, options, outer) {
@@ -20334,13 +20356,15 @@ async function embedMany(texts, options, outer) {
   const { baseUrl = BASE_URL, model, batchSize, cache } = options;
   const timeoutMs = options.timeoutMs ?? DEFAULT_EMBEDDING_REQUEST_TIMEOUT_MS;
   const chunkSize = batchSize ?? DEFAULT_EMBEDDING_BATCH_SIZE;
+  const embeddingModel = model ?? DEFAULT_API_EMBEDDING_MODEL;
   const results = new Array(texts.length).fill(null);
   const uncachedIndices = [];
   const uncachedTexts = [];
   for (let i = 0; i < texts.length; i++) {
     if (cache) {
       const cached = cache.get(texts[i]);
-      if (cached) {
+      const cachedModel = cached && cachedEmbeddingModels.get(cached);
+      if (cached && (cachedModel === void 0 || cachedModel === embeddingModel)) {
         results[i] = Array.from(cached);
         continue;
       }
@@ -20356,7 +20380,6 @@ async function embedMany(texts, options, outer) {
     ...taskTypeHeader(options.taskType)
   };
   if (outer?.aborted) throw outer.reason;
-  const embeddingModel = model ?? DEFAULT_API_EMBEDDING_MODEL;
   let newEmbeddings;
   if (uncachedTexts.length <= chunkSize) {
     newEmbeddings = await generateEmbeddingsBatch(
@@ -20404,6 +20427,7 @@ async function embedMany(texts, options, outer) {
     const f32Embedding = Float32Array.from(newEmbeddings[i]);
     results[uncachedIndices[i]] = Array.from(f32Embedding);
     if (cache) {
+      cachedEmbeddingModels.set(f32Embedding, embeddingModel);
       cache.set(uncachedTexts[i], f32Embedding);
     }
   }
@@ -34246,7 +34270,7 @@ var MaskScopedEmbeddingCache = class extends Map {
     return this;
   }
 };
-var scopedViews = /* @__PURE__ */ new WeakMap();
+var { scopedViews } = getEmbeddingCacheState();
 function maskScopedEmbeddingCache(cache, masked) {
   const views = scopedViews.get(cache) ?? {};
   scopedViews.set(cache, views);
@@ -34671,44 +34695,48 @@ async function recall(query, ctx, options = {}) {
       embeddingOptions,
       ctx.vaultCache,
       searchOptions
-    )))(query, {
-      limit: types.includes("chunk") ? Math.max(limit * 2, 16) : limit,
-      minSimilarity: vaultMinScore,
-      useFusion: true,
-      rerank: flags.rerank,
-      ...options.rerankTopN !== void 0 && { rerankTopN: options.rerankTopN },
-      ...options.ceWeight !== void 0 && { ceWeight: options.ceWeight },
-      ...options.rerankLoadTimeoutMs !== void 0 && {
-        rerankLoadTimeoutMs: options.rerankLoadTimeoutMs
-      },
-      ...options.recencyAlpha !== void 0 && { recencyAlpha: options.recencyAlpha },
-      ...options.recency && { recency: options.recency },
-      ...options.mmr !== void 0 && { mmr: options.mmr },
-      ...options.supersessionBoost !== void 0 && {
-        supersessionBoost: options.supersessionBoost
-      },
-      ...options.supersessionWindow !== void 0 && {
-        supersessionWindow: options.supersessionWindow
-      },
-      ...options.proofCountAlpha !== void 0 && {
-        proofCountAlpha: options.proofCountAlpha
-      },
-      ...options.bm25AdmissionDivisor !== void 0 && {
-        bm25AdmissionDivisor: options.bm25AdmissionDivisor
+    )))(
+      query,
+      {
+        limit: types.includes("chunk") ? Math.max(limit * 2, 16) : limit,
+        minSimilarity: vaultMinScore,
+        useFusion: true,
+        rerank: flags.rerank,
+        ...options.rerankTopN !== void 0 && { rerankTopN: options.rerankTopN },
+        ...options.ceWeight !== void 0 && { ceWeight: options.ceWeight },
+        ...options.rerankLoadTimeoutMs !== void 0 && {
+          rerankLoadTimeoutMs: options.rerankLoadTimeoutMs
+        },
+        ...options.recencyAlpha !== void 0 && { recencyAlpha: options.recencyAlpha },
+        ...options.recency && { recency: options.recency },
+        ...options.mmr !== void 0 && { mmr: options.mmr },
+        ...options.supersessionBoost !== void 0 && {
+          supersessionBoost: options.supersessionBoost
+        },
+        ...options.supersessionWindow !== void 0 && {
+          supersessionWindow: options.supersessionWindow
+        },
+        ...options.proofCountAlpha !== void 0 && {
+          proofCountAlpha: options.proofCountAlpha
+        },
+        ...options.bm25AdmissionDivisor !== void 0 && {
+          bm25AdmissionDivisor: options.bm25AdmissionDivisor
+        },
+        ...options.rrfK !== void 0 && { rrfK: options.rrfK },
+        ...options.decryptLast !== void 0 && { decryptLast: options.decryptLast },
+        ...subQueries && { subQueries },
+        ...options.scopes && { scopes: options.scopes },
+        ...options.folderId !== void 0 && { folderId: options.folderId },
+        ...options.factTypes?.length && { factTypes: options.factTypes },
+        ...options.memoryIds !== void 0 && { memoryIds: options.memoryIds },
+        ...options.factTypeWeights && { factTypeWeights: options.factTypeWeights },
+        ...entityRanking.length > 0 && { entityRanking },
+        ...temporalRanking.length > 0 && { temporalRanking },
+        ...needsChunkEmbedding && { queryEmbedding: queryEmbedding ?? [] },
+        queryEmbedTotalTimeoutMs
       },
-      ...options.rrfK !== void 0 && { rrfK: options.rrfK },
-      ...options.decryptLast !== void 0 && { decryptLast: options.decryptLast },
-      ...subQueries && { subQueries },
-      ...options.scopes && { scopes: options.scopes },
-      ...options.folderId !== void 0 && { folderId: options.folderId },
-      ...options.factTypes?.length && { factTypes: options.factTypes },
-      ...options.memoryIds !== void 0 && { memoryIds: options.memoryIds },
-      ...options.factTypeWeights && { factTypeWeights: options.factTypeWeights },
-      ...entityRanking.length > 0 && { entityRanking },
-      ...temporalRanking.length > 0 && { temporalRanking },
-      ...needsChunkEmbedding && { queryEmbedding: queryEmbedding ?? [] },
-      queryEmbedTotalTimeoutMs
-    });
+      options.embeddingCache
+    );
     factResults.push(
       ...dedupeBy(
         results,
@@ -35855,10 +35883,19 @@ function createLocalMemoryStore(options) {
       retainOptions
     ),
     factSource: {
-      search: (query, searchOptions) => searchVaultMemoriesWithSize(
+      search: (query, searchOptions, embeddingCache) => searchVaultMemoriesWithSize(
         query,
         vaultCtx,
-        { ...options.embeddingOptions, taskType: "memory_recall_embed" },
+        {
+          ...options.embeddingOptions,
+          taskType: "memory_recall_embed",
+          ...embeddingCache && !options.embeddingOptions.cache && {
+            cache: maskScopedEmbeddingCache(
+              embeddingCache,
+              Boolean(options.embeddingOptions.maskInput)
+            )
+          }
+        },
         vaultCache,
         searchOptions
       ),

package/dist/server/index.d.mts

@@ -4190,7 +4190,7 @@ interface RecallOptions {
 }
 /** A fact source supplies bounded candidates; recall owns ranking and result handling. */
 interface RecallFactSource {
-    search: (query: string, options?: MemoryVaultSearchOptions) => ReturnType<typeof searchVaultMemoriesWithSize>;
+    search: (query: string, options?: MemoryVaultSearchOptions, embeddingCache?: Map<string, Float32Array>) => ReturnType<typeof searchVaultMemoriesWithSize>;
     graphRanking: (query: string, traverse: boolean, options: RecallOptions) => Promise<string[]>;
     temporalRanking: (query: string, now?: number) => Promise<string[]>;
 }

package/dist/server/index.d.ts

@@ -4190,7 +4190,7 @@ interface RecallOptions {
 }
 /** A fact source supplies bounded candidates; recall owns ranking and result handling. */
 interface RecallFactSource {
-    search: (query: string, options?: MemoryVaultSearchOptions) => ReturnType<typeof searchVaultMemoriesWithSize>;
+    search: (query: string, options?: MemoryVaultSearchOptions, embeddingCache?: Map<string, Float32Array>) => ReturnType<typeof searchVaultMemoriesWithSize>;
     graphRanking: (query: string, traverse: boolean, options: RecallOptions) => Promise<string[]>;
     temporalRanking: (query: string, now?: number) => Promise<string[]>;
 }