@anuma/sdk 3.13.0 → 3.13.2

package/dist/react/index.mjs

@@ -9646,6 +9646,29 @@ function createMemoryEngineTool(storageCtx, embeddingOptions, searchOptions, cal
   };
 }
 
+// src/lib/memoryVault/embeddingCacheRegistry.ts
+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/tool.ts
 var MANUAL_FACT_TYPES = [
   "identity",
@@ -10150,8 +10173,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 {
@@ -10389,15 +10412,15 @@ var ProcessorRegistry = class {
 // src/lib/processors/preprocessor.ts
 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() {
@@ -10410,10 +10433,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) {
@@ -10469,16 +10492,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 = [];
@@ -10497,7 +10520,7 @@ async function preprocessFiles(files, options = {}) {
       skippedCount++;
       continue;
     }
-    const processor = registry.findProcessor(file);
+    const processor = registry2.findProcessor(file);
     if (!processor) {
       skippedCount++;
       continue;
@@ -12139,6 +12162,10 @@ function useChatStorage(options) {
     }),
     [database, vaultMemoryCollection, walletAddress, signMessage, embeddedWalletSigner]
   );
+  const vaultEmbeddingCache = useMemo3(
+    () => getVaultEmbeddingCache(database, walletAddress, embeddingModel),
+    [database, walletAddress, embeddingModel]
+  );
   const entityCollection = useMemo3(() => database.get("entity"), [database]);
   const memoryEntityCollection = useMemo3(
     () => database.get("memory_entity"),
@@ -12364,10 +12391,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 = useCallback4(
     (options2) => {
@@ -12382,7 +12409,7 @@ function useChatStorage(options) {
         eagerEmbedContent(
           content,
           vaultEmbeddingOptions,
-          vaultEmbeddingCacheRef.current,
+          vaultEmbeddingCache,
           vaultCtx,
           result.uniqueId
         ).catch((err) => {
@@ -12391,7 +12418,7 @@ function useChatStorage(options) {
       }
       return result;
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const updateVaultMemory = useCallback4(
     async (id, content, scope) => {
@@ -12399,55 +12426,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 = useCallback4(
     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 = useRef4(createVaultEmbeddingCache());
   const chunkVectorCacheRef = useRef4(createChunkVectorCache());
   const clientToolEmbeddingsCacheRef = useRef4(/* @__PURE__ */ new Map());
   useEffect3(() => {
     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]);
   useEffect3(() => {
     return onClearAllEncryptionState(() => {
-      vaultEmbeddingCacheRef.current.clear();
+      vaultEmbeddingCache.clear();
       chunkVectorCacheRef.current.clear();
     });
-  }, []);
+  }, [vaultEmbeddingCache]);
   const createMemoryVaultSearchTool2 = useCallback4(
     (searchOptions) => {
       if (!getToken) {
@@ -12456,11 +12474,11 @@ function useChatStorage(options) {
       return createMemoryVaultSearchTool(
         vaultCtx,
         vaultEmbeddingOptions,
-        vaultEmbeddingCacheRef.current,
+        vaultEmbeddingCache,
         searchOptions
       );
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const createRecallTool2 = useCallback4(
     (toolOptions, callbacks) => {
@@ -12473,7 +12491,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.
@@ -12485,7 +12503,15 @@ function useChatStorage(options) {
         callbacks
       );
     },
-    [vaultCtx, storageCtx, entityCtx, getToken, vaultEmbeddingOptions, currentConversationId]
+    [
+      vaultCtx,
+      storageCtx,
+      entityCtx,
+      getToken,
+      vaultEmbeddingOptions,
+      vaultEmbeddingCache,
+      currentConversationId
+    ]
   );
   const searchVaultMemoriesFn = useCallback4(
     async (query, searchOptions) => {
@@ -12494,11 +12520,11 @@ function useChatStorage(options) {
         query,
         vaultCtx,
         vaultEmbeddingOptions,
-        vaultEmbeddingCacheRef.current,
+        vaultEmbeddingCache,
         searchOptions
       );
     },
-    [vaultCtx, getToken, vaultEmbeddingOptions]
+    [vaultCtx, getToken, vaultEmbeddingOptions, vaultEmbeddingCache]
   );
   const recallFn = useCallback4(
     async (query, options2) => {
@@ -12521,7 +12547,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
@@ -12531,7 +12557,15 @@ function useChatStorage(options) {
         resolvedOptions
       );
     },
-    [vaultCtx, storageCtx, entityCtx, getToken, currentConversationId, vaultEmbeddingOptions]
+    [
+      vaultCtx,
+      storageCtx,
+      entityCtx,
+      getToken,
+      currentConversationId,
+      vaultEmbeddingOptions,
+      vaultEmbeddingCache
+    ]
   );
   const {
     isLoading,
@@ -13001,6 +13035,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();
@@ -13180,44 +13253,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;
@@ -13619,7 +13675,7 @@ Based on these results, continue with the task.`;
     createRecallTool: createRecallTool2,
     recall: recallFn,
     searchVaultMemories: searchVaultMemoriesFn,
-    vaultEmbeddingCache: vaultEmbeddingCacheRef.current,
+    vaultEmbeddingCache,
     getVaultMemories,
     createVaultMemory,
     updateVaultMemory,

package/dist/server/{chunk-ME6ZQN52.mjs → chunk-WZE6ZVKE.mjs}

@@ -20310,7 +20310,7 @@ function createMemoryVaultSearchTool(vaultCtx, embeddingOptions, cache, searchOp
             legacy.map((r) => ({ id: r.uniqueId, content: r.content, score: r.similarity }))
           );
         }
-        const { recall: recall2 } = await import("./recall-RMCSM6L6.mjs");
+        const { recall: recall2 } = await import("./recall-YZWIV6IH.mjs");
         let recallDegraded = [];
         const result = await recall2(
           query,
@@ -21357,14 +21357,14 @@ export {
   embedAllMessages,
   chunkAndEmbedMessage,
   chunkAndEmbedAllMessages,
+  DEFAULT_VAULT_CACHE_SIZE,
+  LRUCache,
+  createVaultEmbeddingCache,
   RECALL_TOOL_NAME,
   RECALL_MAX_LIMIT,
   EMBEDDINGS_DEGRADED_EMPTY,
   RerankerUnavailableError,
   isRerankerAvailable,
-  DEFAULT_VAULT_CACHE_SIZE,
-  LRUCache,
-  createVaultEmbeddingCache,
   MAX_HOPS,
   ENTITY_FANOUT,
   NODE_BUDGET,

package/dist/server/index.cjs

@@ -19576,6 +19576,44 @@ var init_embeddings = __esm({
   }
 });
 
+// src/lib/memoryVault/lruCache.ts
+function createVaultEmbeddingCache(maxSize = DEFAULT_VAULT_CACHE_SIZE) {
+  return new LRUCache(maxSize);
+}
+var DEFAULT_VAULT_CACHE_SIZE, LRUCache;
+var init_lruCache = __esm({
+  "src/lib/memoryVault/lruCache.ts"() {
+    "use strict";
+    DEFAULT_VAULT_CACHE_SIZE = 5e3;
+    LRUCache = class extends Map {
+      constructor(maxSize) {
+        super();
+        this.maxSize = maxSize;
+      }
+      get(key) {
+        if (!super.has(key)) return void 0;
+        const value = super.get(key);
+        super.delete(key);
+        super.set(key, value);
+        return value;
+      }
+      set(key, value) {
+        if (super.has(key)) {
+          super.delete(key);
+        }
+        super.set(key, value);
+        if (this.size > this.maxSize) {
+          const oldest = this.keys().next().value;
+          if (oldest !== void 0) {
+            super.delete(oldest);
+          }
+        }
+        return this;
+      }
+    };
+  }
+});
+
 // src/lib/memory/mmr.ts
 function applyMMR(candidates, k, lambda = 0.5) {
   if (candidates.length === 0 || k <= 0) return [];
@@ -20355,44 +20393,6 @@ Output strict JSON. No prose.`;
   }
 });
 
-// src/lib/memoryVault/lruCache.ts
-function createVaultEmbeddingCache(maxSize = DEFAULT_VAULT_CACHE_SIZE) {
-  return new LRUCache(maxSize);
-}
-var DEFAULT_VAULT_CACHE_SIZE, LRUCache;
-var init_lruCache = __esm({
-  "src/lib/memoryVault/lruCache.ts"() {
-    "use strict";
-    DEFAULT_VAULT_CACHE_SIZE = 5e3;
-    LRUCache = class extends Map {
-      constructor(maxSize) {
-        super();
-        this.maxSize = maxSize;
-      }
-      get(key) {
-        if (!super.has(key)) return void 0;
-        const value = super.get(key);
-        super.delete(key);
-        super.set(key, value);
-        return value;
-      }
-      set(key, value) {
-        if (super.has(key)) {
-          super.delete(key);
-        }
-        super.set(key, value);
-        if (this.size > this.maxSize) {
-          const oldest = this.keys().next().value;
-          if (oldest !== void 0) {
-            super.delete(oldest);
-          }
-        }
-        return this;
-      }
-    };
-  }
-});
-
 // src/lib/memory/queryEntities.ts
 function hasMeaningfulCandidate(candidates) {
   for (const candidate of candidates) {

package/dist/server/index.mjs

@@ -133,7 +133,7 @@ import {
   updateVaultMemoryOp,
   upsertMessageOp,
   vaultMemoryToStored
-} from "./chunk-ME6ZQN52.mjs";
+} from "./chunk-WZE6ZVKE.mjs";
 import {
   createStreamingDeAnonymizer,
   notifyConsolidationFallback,

package/dist/server/{recall-RMCSM6L6.mjs → recall-YZWIV6IH.mjs}

@@ -1,6 +1,6 @@
 import {
   recall
-} from "./chunk-ME6ZQN52.mjs";
+} from "./chunk-WZE6ZVKE.mjs";
 import "./chunk-OBDRKQQI.mjs";
 import "./chunk-3RPGXXPZ.mjs";
 export {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anuma/sdk",
-  "version": "3.13.0",
+  "version": "3.13.2",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -136,7 +136,8 @@
     "lint:fix": "eslint . --fix",
     "format": "prettier --write .",
     "format:check": "prettier --check .",
-    "typecheck": "tsc --noEmit",
+    "typecheck": "tsc --noEmit && pnpm run typecheck:test",
+    "typecheck:test": "tsc -p tsconfig.test.json",
     "knip": "knip",
     "check:export-consumers": "node scripts/assert-export-consumers.mjs",
     "check-exports": "publint && attw --pack --ignore-rules no-resolution false-cjs && node scripts/assert-no-client-self-import.mjs",