@anuma/sdk 3.13.0 → 3.13.2

package/dist/react/index.cjs

@@ -18890,6 +18890,31 @@ function createMemoryEngineTool(storageCtx, embeddingOptions, searchOptions, cal
 // src/react/useChatStorage.ts
 init_constants();
 
+// src/lib/memoryVault/embeddingCacheRegistry.ts
+init_useEncryption();
+init_lruCache();
+var registry = /* @__PURE__ */ new WeakMap();
+function buildKey(walletAddress, model) {
+  return JSON.stringify([walletAddress ?? null, model]);
+}
+function getVaultEmbeddingCache(database, walletAddress, model) {
+  let perDb = registry.get(database);
+  if (!perDb) {
+    perDb = /* @__PURE__ */ new Map();
+    registry.set(database, perDb);
+  }
+  const key = buildKey(walletAddress, model);
+  let cache = perDb.get(key);
+  if (!cache) {
+    cache = createVaultEmbeddingCache();
+    perDb.set(key, cache);
+  }
+  return cache;
+}
+onClearAllEncryptionState(() => {
+  registry = /* @__PURE__ */ new WeakMap();
+});
+
 // src/lib/memoryVault/index.ts
 init_searchTool();
 
@@ -19411,8 +19436,8 @@ var ZipProcessor = class {
    * Set the processor registry for handling nested files
    * This must be called before processing if you want nested file support
    */
-  setRegistry(registry) {
-    this.registry = registry;
+  setRegistry(registry2) {
+    this.registry = registry2;
   }
   async process(file) {
     try {
@@ -19651,15 +19676,15 @@ var ProcessorRegistry = class {
 init_logger();
 var MAX_TOTAL_IMAGES = 20;
 function createDefaultRegistry() {
-  const registry = new ProcessorRegistry();
-  registry.register(new PdfProcessor());
-  registry.register(new ExcelProcessor());
-  registry.register(new WordProcessor());
-  registry.register(new TextProcessor());
+  const registry2 = new ProcessorRegistry();
+  registry2.register(new PdfProcessor());
+  registry2.register(new ExcelProcessor());
+  registry2.register(new WordProcessor());
+  registry2.register(new TextProcessor());
   const zipProcessor = new ZipProcessor();
-  zipProcessor.setRegistry(registry);
-  registry.register(zipProcessor);
-  return registry;
+  zipProcessor.setRegistry(registry2);
+  registry2.register(zipProcessor);
+  return registry2;
 }
 var cachedDefaultRegistry = null;
 function getDefaultRegistry() {
@@ -19672,10 +19697,10 @@ function isSupportedFile(file) {
   return getDefaultRegistry().isSupported(file);
 }
 function getSupportedFileTypes() {
-  const registry = getDefaultRegistry();
+  const registry2 = getDefaultRegistry();
   return {
-    mimeTypes: registry.getSupportedMimeTypes(),
-    extensions: registry.getSupportedExtensions()
+    mimeTypes: registry2.getSupportedMimeTypes(),
+    extensions: registry2.getSupportedExtensions()
   };
 }
 function formatExtractedContent(fileName, content, format) {
@@ -19731,16 +19756,16 @@ async function preprocessFiles(files, options = {}) {
       metadata: { processedCount: 0, skippedCount: 0, errorCount: 0 }
     };
   }
-  let registry;
+  let registry2;
   if (processors === void 0) {
-    registry = getDefaultRegistry();
+    registry2 = getDefaultRegistry();
   } else {
-    registry = new ProcessorRegistry();
+    registry2 = new ProcessorRegistry();
     processors.forEach((p) => {
       if (p instanceof ZipProcessor) {
-        p.setRegistry(registry);
+        p.setRegistry(registry2);
       }
-      registry.register(p);
+      registry2.register(p);
     });
   }
   const extractedTexts = [];
@@ -19759,7 +19784,7 @@ async function preprocessFiles(files, options = {}) {
       skippedCount++;
       continue;
     }
-    const processor = registry.findProcessor(file);
+    const processor = registry2.findProcessor(file);
     if (!processor) {
       skippedCount++;
       continue;
@@ -21412,6 +21437,10 @@ function useChatStorage(options) {
     }),
     [database, vaultMemoryCollection, walletAddress, signMessage, embeddedWalletSigner]
   );
+  const vaultEmbeddingCache = (0, import_react5.useMemo)(
+    () => getVaultEmbeddingCache(database, walletAddress, embeddingModel),
+    [database, walletAddress, embeddingModel]
+  );
   const entityCollection = (0, import_react5.useMemo)(() => database.get("entity"), [database]);
   const memoryEntityCollection = (0, import_react5.useMemo)(
     () => database.get("memory_entity"),
@@ -21637,10 +21666,10 @@ function useChatStorage(options) {
         vaultCtx,
         options2,
         embOpts,
-        embOpts ? vaultEmbeddingCacheRef.current : void 0
+        embOpts ? vaultEmbeddingCache : void 0
       );
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const getVaultMemories = (0, import_react5.useCallback)(
     (options2) => {
@@ -21655,7 +21684,7 @@ function useChatStorage(options) {
         eagerEmbedContent(
           content,
           vaultEmbeddingOptions,
-          vaultEmbeddingCacheRef.current,
+          vaultEmbeddingCache,
           vaultCtx,
           result.uniqueId
         ).catch((err) => {
@@ -21664,7 +21693,7 @@ function useChatStorage(options) {
       }
       return result;
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const updateVaultMemory = (0, import_react5.useCallback)(
     async (id, content, scope) => {
@@ -21672,55 +21701,46 @@ function useChatStorage(options) {
       const result = await updateVaultMemoryOp(vaultCtx, id, { content, scope, embedding: null });
       if (result && getToken) {
         if (existing) {
-          vaultEmbeddingCacheRef.current.delete(id);
+          vaultEmbeddingCache.delete(id);
         }
-        eagerEmbedContent(
-          content,
-          vaultEmbeddingOptions,
-          vaultEmbeddingCacheRef.current,
-          vaultCtx,
-          id
-        ).catch((err) => {
-          getLogger().warn("[useChatStorage] Failed to eagerly embed updated vault memory:", err);
-        });
+        eagerEmbedContent(content, vaultEmbeddingOptions, vaultEmbeddingCache, vaultCtx, id).catch(
+          (err) => {
+            getLogger().warn("[useChatStorage] Failed to eagerly embed updated vault memory:", err);
+          }
+        );
       }
       return result;
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const deleteVaultMemory = (0, import_react5.useCallback)(
     async (id) => {
       const existing = await getVaultMemoryOp(vaultCtx, id);
       const result = await deleteVaultMemoryOp(vaultCtx, id);
       if (result && existing) {
-        vaultEmbeddingCacheRef.current.delete(id);
+        vaultEmbeddingCache.delete(id);
       }
       return result;
     },
-    [vaultCtx]
+    [vaultCtx, vaultEmbeddingCache]
   );
-  const vaultEmbeddingCacheRef = (0, import_react5.useRef)(createVaultEmbeddingCache());
   const chunkVectorCacheRef = (0, import_react5.useRef)(createChunkVectorCache());
   const clientToolEmbeddingsCacheRef = (0, import_react5.useRef)(/* @__PURE__ */ new Map());
   (0, import_react5.useEffect)(() => {
     if (!getToken) return;
     void (async () => {
       try {
-        await preEmbedVaultMemories(
-          vaultCtx,
-          vaultEmbeddingOptions,
-          vaultEmbeddingCacheRef.current
-        );
+        await preEmbedVaultMemories(vaultCtx, vaultEmbeddingOptions, vaultEmbeddingCache);
       } catch {
       }
     })();
-  }, [vaultCtx, getToken, vaultEmbeddingOptions]);
+  }, [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]);
   (0, import_react5.useEffect)(() => {
     return onClearAllEncryptionState(() => {
-      vaultEmbeddingCacheRef.current.clear();
+      vaultEmbeddingCache.clear();
       chunkVectorCacheRef.current.clear();
     });
-  }, []);
+  }, [vaultEmbeddingCache]);
   const createMemoryVaultSearchTool2 = (0, import_react5.useCallback)(
     (searchOptions) => {
       if (!getToken) {
@@ -21729,11 +21749,11 @@ function useChatStorage(options) {
       return createMemoryVaultSearchTool(
         vaultCtx,
         vaultEmbeddingOptions,
-        vaultEmbeddingCacheRef.current,
+        vaultEmbeddingCache,
         searchOptions
       );
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const createRecallTool2 = (0, import_react5.useCallback)(
     (toolOptions, callbacks) => {
@@ -21746,7 +21766,7 @@ function useChatStorage(options) {
           vaultCtx,
           storageCtx,
           embeddingOptions: vaultEmbeddingOptions,
-          vaultCache: vaultEmbeddingCacheRef.current,
+          vaultCache: vaultEmbeddingCache,
           chunkCache: chunkVectorCacheRef.current,
           // Graph lane fires when entityCtx is present and the query
           // contains extractable entities. Empty memory_entity (e.g.
@@ -21758,7 +21778,15 @@ function useChatStorage(options) {
         callbacks
       );
     },
-    [vaultCtx, storageCtx, entityCtx, getToken, vaultEmbeddingOptions, currentConversationId]
+    [
+      vaultCtx,
+      storageCtx,
+      entityCtx,
+      getToken,
+      vaultEmbeddingOptions,
+      vaultEmbeddingCache,
+      currentConversationId
+    ]
   );
   const searchVaultMemoriesFn = (0, import_react5.useCallback)(
     async (query, searchOptions) => {
@@ -21767,11 +21795,11 @@ function useChatStorage(options) {
         query,
         vaultCtx,
         vaultEmbeddingOptions,
-        vaultEmbeddingCacheRef.current,
+        vaultEmbeddingCache,
         searchOptions
       );
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const recallFn = (0, import_react5.useCallback)(
     async (query, options2) => {
@@ -21794,7 +21822,7 @@ function useChatStorage(options) {
           // createRecallTool / searchVaultMemoriesFn. Without this, recall()
           // would embed the raw query (often containing the user's PII).
           embeddingOptions: vaultEmbeddingOptions,
-          vaultCache: vaultEmbeddingCacheRef.current,
+          vaultCache: vaultEmbeddingCache,
           chunkCache: chunkVectorCacheRef.current,
           // Graph lane fires only when entityCtx is present and the query
           // has extractable entities; empty memory_entity is a graceful
@@ -21804,7 +21832,15 @@ function useChatStorage(options) {
         resolvedOptions
       );
     },
-    [vaultCtx, storageCtx, entityCtx, getToken, currentConversationId, vaultEmbeddingOptions]
+    [
+      vaultCtx,
+      storageCtx,
+      entityCtx,
+      getToken,
+      currentConversationId,
+      vaultEmbeddingOptions,
+      vaultEmbeddingCache
+    ]
   );
   const {
     isLoading,
@@ -22274,6 +22310,45 @@ function useChatStorage(options) {
         }
       }
       const { callRedactor, callPiiRedaction, maskForCall } = resolvePiiForCall(convId);
+      const isServerToolsFunction = typeof serverToolsFilter === "function";
+      const needsEmbeddings = isServerToolsFunction || !!clientToolsFilter || !!clientTools?.length;
+      const embeddingsPromise = needsEmbeddings && getToken ? (async () => {
+        try {
+          const embeddingOptions = { getToken, baseUrl, model: embeddingModel };
+          if (shouldChunkMessage(contentForStorage, DEFAULT_CHUNK_SIZE)) {
+            const textChunks = chunkText(contentForStorage);
+            const chunkTexts = textChunks.map((c) => maskForCall(c.text));
+            return {
+              embeddings: await generateEmbeddings(chunkTexts, embeddingOptions),
+              failed: false
+            };
+          }
+          if (contentForStorage.length >= MIN_CONTENT_LENGTH_FOR_TOOLS) {
+            return {
+              embeddings: await generateEmbedding(
+                maskForCall(contentForStorage),
+                embeddingOptions
+              ),
+              failed: false
+            };
+          }
+          return { failed: false };
+        } catch {
+          return { failed: true };
+        }
+      })() : null;
+      const serverToolsPromise = getToken && !(Array.isArray(serverToolsFilter) && serverToolsFilter.length === 0) ? (async () => {
+        try {
+          return await getServerTools({
+            baseUrl,
+            cacheExpirationMs: serverToolsConfig?.cacheExpirationMs,
+            getToken,
+            cache: serverToolsConfig?.cache
+          });
+        } catch {
+          return null;
+        }
+      })() : null;
       let messagesToSend;
       const storedMessages = await getMessagesOp(storageCtx, convId);
       const knownToolCallEventIds = /* @__PURE__ */ new Set();
@@ -22453,44 +22528,27 @@ function useChatStorage(options) {
       let filteredServerTools = [];
       let userMessageEmbeddings;
       let userMessageEmbeddingsFailed = false;
-      const isServerToolsFunction = typeof serverToolsFilter === "function";
-      const needsEmbeddings = isServerToolsFunction || !!clientToolsFilter || !!clientTools?.length;
-      if (needsEmbeddings && getToken) {
-        try {
-          const embeddingOptions = { getToken, baseUrl, model: embeddingModel };
-          if (shouldChunkMessage(contentForStorage, DEFAULT_CHUNK_SIZE)) {
-            const textChunks = chunkText(contentForStorage);
-            const chunkTexts = textChunks.map((c) => maskForCall(c.text));
-            userMessageEmbeddings = await generateEmbeddings(chunkTexts, embeddingOptions);
-          } else if (contentForStorage.length >= MIN_CONTENT_LENGTH_FOR_TOOLS) {
-            userMessageEmbeddings = await generateEmbedding(
-              maskForCall(contentForStorage),
-              embeddingOptions
-            );
-          }
-        } catch {
-          userMessageEmbeddingsFailed = true;
-        }
-      }
-      if (getToken && !(Array.isArray(serverToolsFilter) && serverToolsFilter.length === 0)) {
-        try {
-          const allServerTools = await getServerTools({
-            baseUrl,
-            cacheExpirationMs: serverToolsConfig?.cacheExpirationMs,
-            getToken,
-            cache: serverToolsConfig?.cache
-          });
-          if (serverToolsConfig?.deferLoading?.enabled && effectiveApiType === "responses") {
-            filteredServerTools = allServerTools;
-          } else if (isServerToolsFunction) {
-            if (userMessageEmbeddings) {
-              const toolNames = serverToolsFilter(userMessageEmbeddings, allServerTools);
-              filteredServerTools = filterServerTools(allServerTools, toolNames);
+      if (embeddingsPromise) {
+        const embeddingsResult = await embeddingsPromise;
+        userMessageEmbeddings = embeddingsResult.embeddings;
+        userMessageEmbeddingsFailed = embeddingsResult.failed;
+      }
+      if (serverToolsPromise) {
+        const allServerTools = await serverToolsPromise;
+        if (allServerTools) {
+          try {
+            if (serverToolsConfig?.deferLoading?.enabled && effectiveApiType === "responses") {
+              filteredServerTools = allServerTools;
+            } else if (isServerToolsFunction) {
+              if (userMessageEmbeddings) {
+                const toolNames = serverToolsFilter(userMessageEmbeddings, allServerTools);
+                filteredServerTools = filterServerTools(allServerTools, toolNames);
+              }
+            } else {
+              filteredServerTools = filterServerTools(allServerTools, serverToolsFilter);
             }
-          } else {
-            filteredServerTools = filterServerTools(allServerTools, serverToolsFilter);
+          } catch {
           }
-        } catch {
         }
       }
       let filteredClientTools = clientTools;
@@ -22892,7 +22950,7 @@ Based on these results, continue with the task.`;
     createRecallTool: createRecallTool2,
     recall: recallFn,
     searchVaultMemories: searchVaultMemoriesFn,
-    vaultEmbeddingCache: vaultEmbeddingCacheRef.current,
+    vaultEmbeddingCache,
     getVaultMemories,
     createVaultMemory,
     updateVaultMemory,

package/dist/react/index.d.mts

@@ -9752,6 +9752,13 @@ interface UseChatStorageResult extends BaseUseChatStorageResult {
     /**
      * The shared vault embedding cache. Use this to eagerly embed content
      * when saving vault memories (via eagerEmbedContent).
+     *
+     * Shared per `(database, walletAddress, embeddingModel)` rather than owned by
+     * this hook instance: every `useChatStorage` on the same three gets this exact
+     * object, and it outlives their unmounts. Writes and evictions are therefore
+     * visible to all of them — which is the point (a memory deleted through one
+     * hook stops being served by the others) but does mean `clear()` clears for
+     * everyone.
      */
     vaultEmbeddingCache: VaultEmbeddingCache;
     /**

package/dist/react/index.d.ts

@@ -9752,6 +9752,13 @@ interface UseChatStorageResult extends BaseUseChatStorageResult {
     /**
      * The shared vault embedding cache. Use this to eagerly embed content
      * when saving vault memories (via eagerEmbedContent).
+     *
+     * Shared per `(database, walletAddress, embeddingModel)` rather than owned by
+     * this hook instance: every `useChatStorage` on the same three gets this exact
+     * object, and it outlives their unmounts. Writes and evictions are therefore
+     * visible to all of them — which is the point (a memory deleted through one
+     * hook stops being served by the others) but does mean `clear()` clears for
+     * everyone.
      */
     vaultEmbeddingCache: VaultEmbeddingCache;
     /**