@anuma/sdk 7.6.1 → 8.0.1

package/dist/expo/index.mjs

@@ -2420,16 +2420,19 @@ async function updateMessageEmbeddingOp(ctx, uniqueId, vector, embeddingModel) {
   } catch {
     return null;
   }
+  const address = ctx.walletAddress;
   await ctx.database.write(async () => {
+    const key = await resolveMessageEmbeddingKey(address, message.content);
+    const storedVector2 = await serializeMessageEmbedding(key, vector);
     await message.update((msg) => {
-      msg._setRaw("vector", JSON.stringify(vector));
+      msg._setRaw("vector", storedVector2);
       if (msg.origin === "chunks_discarded" && vector.length > 0) {
         msg._setRaw("origin", "message");
       }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
-  return messageToStored(message, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+  return messageToStored(message, ctx.walletAddress);
 }
 async function updateMessageChunksOp(ctx, uniqueId, chunks3, embeddingModel) {
   let message;
@@ -2439,16 +2442,123 @@ async function updateMessageChunksOp(ctx, uniqueId, chunks3, embeddingModel) {
     return null;
   }
   const storedChunks = chunks3.map(({ text: _text, ...rest }) => rest);
+  const address = ctx.walletAddress;
   await ctx.database.write(async () => {
+    const key = await resolveMessageEmbeddingKey(address, message.content);
+    const serializedChunks = await serializeMessageEmbedding(key, storedChunks);
     await message.update((msg) => {
-      msg._setRaw("chunks", JSON.stringify(storedChunks));
+      msg._setRaw("chunks", serializedChunks);
       if (msg.origin === "chunks_discarded" && storedChunks.length > 0 && storedChunks.every((chunk) => chunk.vector.length > 0)) {
         msg._setRaw("origin", "message");
       }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
-  return messageToStored(message, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+  return messageToStored(message, ctx.walletAddress);
+}
+async function resolveMessageEmbeddingKey(address, content) {
+  if (!address || !isEncrypted(content)) return null;
+  const key = await getEncryptionKey(address, "v3");
+  const contentKey = content.startsWith("enc:v2:") ? await getEncryptionKey(address, "v2") : key;
+  try {
+    await decryptDataWithKey(content.slice("enc:vN:".length), contentKey);
+  } catch {
+    throw new Error("Cannot decrypt the message with the cached wallet key");
+  }
+  return key;
+}
+async function serializeMessageEmbedding(key, value) {
+  const json4 = JSON.stringify(value);
+  return key ? `enc:v3:${await encryptDataWithKey(json4, key)}` : json4;
+}
+async function decryptMessageEmbeddingFields(raw, walletAddress) {
+  const result = { ...raw };
+  for (const column of ["vector", "chunks"]) {
+    const value = raw[column];
+    if (typeof value !== "string" || !value.startsWith("enc:")) continue;
+    const decrypted = walletAddress ? await decryptField(value, walletAddress) : value;
+    result[column] = decrypted.startsWith("enc:") ? void 0 : decrypted;
+  }
+  return result;
+}
+async function sealLegacyMessageEmbeddingsOp(ctx) {
+  if (!ctx.walletAddress) return 0;
+  const candidateIds = await ctx.messagesCollection.query(
+    Q.where("content", Q.like("enc:%")),
+    Q.or(
+      ...["vector", "chunks"].map(
+        (column) => Q.and(
+          Q.where(column, Q.notEq(null)),
+          Q.where(column, Q.notEq("")),
+          Q.where(column, Q.notLike("enc:%"))
+        )
+      )
+    )
+  ).fetchIds();
+  let sealed = 0;
+  let undecryptable = 0;
+  for (let start = 0; start < candidateIds.length; start += 50) {
+    const pageIds = candidateIds.slice(start, start + 50);
+    await ctx.database.write(async () => {
+      const pageMessages = await ctx.messagesCollection.query(Q.where("id", Q.oneOf(pageIds))).fetch();
+      for (const message of pageMessages) {
+        if (!isEncrypted(message.content)) continue;
+        const content = await decryptFieldDetailed(message.content, ctx.walletAddress);
+        if (content.status === "key_missing") {
+          throw new Error("Cannot decrypt the message with the cached wallet key");
+        }
+        if (content.status !== "ok") {
+          undecryptable++;
+          continue;
+        }
+        const plaintext = content.value;
+        const key = await resolveMessageEmbeddingKey(ctx.walletAddress, message.content);
+        const updates = {};
+        for (const column of ["vector", "chunks"]) {
+          const value = message._getRaw(column);
+          if (typeof value !== "string" || value.startsWith("enc:") || !value) continue;
+          let parsed;
+          try {
+            parsed = JSON.parse(value);
+          } catch {
+            updates[column] = null;
+            continue;
+          }
+          if (!Array.isArray(parsed)) {
+            updates[column] = null;
+            continue;
+          }
+          if (column === "chunks") {
+            const chunks3 = parsed;
+            if (chunks3.some(
+              (chunk) => !chunk || !Number.isInteger(chunk.startOffset) || !Number.isInteger(chunk.endOffset) || chunk.startOffset < 0 || chunk.endOffset <= chunk.startOffset || chunk.endOffset > plaintext.length
+            )) {
+              updates.chunks = null;
+              updates.vector = null;
+              updates.origin = message.origin === "tool_result" ? "tool_result" : "chunks_discarded";
+              break;
+            }
+            parsed = chunks3.map(({ text: _text, ...rest }) => rest);
+          }
+          updates[column] = await serializeMessageEmbedding(key, parsed);
+        }
+        if (Object.keys(updates).length === 0) continue;
+        const updatedAt = Number(message._getRaw("updated_at"));
+        const discarded = updates.origin !== void 0;
+        await message.update((msg) => {
+          for (const [column, value] of Object.entries(updates)) msg._setRaw(column, value);
+          msg._setRaw("updated_at", discarded ? Math.max(Date.now(), updatedAt + 1) : updatedAt);
+        });
+        sealed++;
+      }
+    });
+  }
+  if (undecryptable > 0) {
+    getLogger().warn(
+      `Skipped ${undecryptable} message embeddings that the cached wallet key cannot decrypt`
+    );
+  }
+  return sealed;
 }
 var RAW_REPAIR_PAGE_SIZE = 100;
 function chunksWithoutText(raw) {
@@ -26391,6 +26501,8 @@ function useChatStorage(options) {
       embeddedWalletSigner
     ]
   );
+  const storageCtxRef = useRef4(storageCtx);
+  storageCtxRef.current = storageCtx;
   const vaultMemoryCollection = useMemo(
     () => database.get("memory_vault"),
     [database]
@@ -26494,24 +26606,12 @@ function useChatStorage(options) {
     refreshQueueStatus();
     return queueManager.onQueueChange(walletAddress, refreshQueueStatus);
   }, [walletAddress, refreshQueueStatus]);
-  useEffect4(() => {
-    if (!walletAddress || !enableQueue || !autoFlushOnKeyAvailable || !signMessage) return;
-    return onKeyAvailable(walletAddress, () => {
-      flushQueue().catch((err) => {
-        getLogger().warn("[useChatStorage] Auto-flush failed:", err);
-      });
-    });
-  }, [walletAddress, enableQueue, autoFlushOnKeyAvailable, signMessage, flushQueue]);
   useEffect4(() => {
     if (!getWalletAddress || walletAddress) return;
     const poller = new WalletPoller();
     return poller.startPolling(getWalletAddress, () => {
     });
   }, [getWalletAddress, walletAddress]);
-  const isEncryptionReady = useCallback4(() => {
-    if (!walletAddress || !signMessage) return true;
-    return hasEncryptionKey(walletAddress);
-  }, [walletAddress, signMessage]);
   const pendingOpsRef = useRef4([]);
   const syntheticConvIdsRef = useRef4(/* @__PURE__ */ new Set());
   useEffect4(() => {
@@ -26524,34 +26624,49 @@ function useChatStorage(options) {
     pendingOpsRef.current = [];
     refreshQueueStatus();
   }, [walletAddress, enableQueue, refreshQueueStatus]);
+  useEffect4(() => {
+    if (!walletAddress || !enableQueue || !autoFlushOnKeyAvailable || !signMessage) return;
+    return onKeyAvailable(walletAddress, () => {
+      flushQueue().catch((err) => {
+        getLogger().warn("[useChatStorage] Auto-flush failed:", err);
+      });
+    });
+  }, [walletAddress, enableQueue, autoFlushOnKeyAvailable, signMessage, flushQueue]);
   const writeOrQueue = useCallback4(
     async (opType, payload, directWrite, makeSynthetic, dependencies = []) => {
-      if (isEncryptionReady()) {
-        const result2 = await directWrite();
+      const currentStorageCtx = storageCtx.walletAddress ? storageCtx : storageCtxRef.current;
+      const currentWalletAddress = currentStorageCtx.walletAddress;
+      if (enableQueue && signMessage && !currentWalletAddress && getWalletAddress) {
+        pendingOpsRef.current.push({ type: opType, payload, dependencies });
+        return { result: makeSynthetic(), queued: true };
+      }
+      if (!currentWalletAddress || !currentStorageCtx.signMessage || hasEncryptionKey(currentWalletAddress)) {
+        const result2 = await directWrite(currentStorageCtx);
         return { result: result2, queued: false };
       }
       if (!enableQueue) {
-        const result2 = await directWrite();
+        const result2 = await directWrite(currentStorageCtx);
         return { result: result2, queued: false };
       }
-      if (!walletAddress && getWalletAddress) {
-        pendingOpsRef.current.push({ type: opType, payload, dependencies });
-        return { result: makeSynthetic(), queued: true };
-      }
-      if (walletAddress) {
-        const queueId = queueManager.queueOperation(walletAddress, opType, payload, dependencies);
+      if (currentWalletAddress) {
+        const queueId = queueManager.queueOperation(
+          currentWalletAddress,
+          opType,
+          payload,
+          dependencies
+        );
         if (queueId === null) {
           getLogger().warn("[useChatStorage] Queue full, falling back to direct write");
-          const result2 = await directWrite();
+          const result2 = await directWrite(currentStorageCtx);
           return { result: result2, queued: false };
         }
-        refreshQueueStatus();
+        setQueueStatus(queueManager.getStatus(currentWalletAddress));
         return { result: makeSynthetic(), queued: true, queueId };
       }
-      const result = await directWrite();
+      const result = await directWrite(currentStorageCtx);
       return { result, queued: false };
     },
-    [isEncryptionReady, enableQueue, walletAddress, getWalletAddress, refreshQueueStatus]
+    [storageCtx, enableQueue, signMessage, getWalletAddress]
   );
   const embedMessageAsync = useCallback4(
     async (message, mask = maskForEmbedding) => {
@@ -26825,7 +26940,7 @@ function useChatStorage(options) {
       const { result, queued } = await writeOrQueue(
         "createConversation",
         conversationOpts,
-        () => createConversationOp(storageCtx, conversationOpts, defaultConversationTitle),
+        (ctx) => createConversationOp(ctx, conversationOpts, defaultConversationTitle),
         () => makeSyntheticStoredConversation(conversationOpts, defaultConversationTitle)
       );
       if (queued) {
@@ -26834,7 +26949,7 @@ function useChatStorage(options) {
       setCurrentConversationId(result.conversationId);
       return result;
     },
-    [storageCtx, defaultConversationTitle, writeOrQueue]
+    [defaultConversationTitle, writeOrQueue]
   );
   const getConversation = useCallback4(
     async (id) => {
@@ -26850,24 +26965,24 @@ function useChatStorage(options) {
       const { result } = await writeOrQueue(
         "updateConversationTitle",
         { conversationId: id, title },
-        () => updateConversationTitleOp(storageCtx, id, title),
+        (ctx) => updateConversationTitleOp(ctx, id, title),
         () => true
       );
       return result;
     },
-    [storageCtx, writeOrQueue]
+    [writeOrQueue]
   );
   const updateConversationPinned = useCallback4(
     async (id, pinned) => {
       const { result } = await writeOrQueue(
         "updateConversationPinned",
         { conversationId: id, pinned },
-        () => updateConversationPinnedOp(storageCtx, id, pinned),
+        (ctx) => updateConversationPinnedOp(ctx, id, pinned),
         () => true
       );
       return result;
     },
-    [storageCtx, writeOrQueue]
+    [writeOrQueue]
   );
   const deleteConversation = useCallback4(
     async (id) => {
@@ -27085,7 +27200,7 @@ function useChatStorage(options) {
       const { result } = await writeOrQueue(
         "createMessage",
         rowOpts,
-        () => upsertMessageOp(storageCtx, rowOpts),
+        (currentStorageCtx) => upsertMessageOp(currentStorageCtx, rowOpts),
         () => makeSyntheticStoredMessage(rowOpts),
         ctx.userMessageQueueId ? [ctx.userMessageQueueId] : []
       );
@@ -27547,7 +27662,7 @@ function useChatStorage(options) {
         const userMsgResult = await writeOrQueue(
           "createMessage",
           userMsgOpts,
-          () => createMessageOp(storageCtx, userMsgOpts),
+          (ctx) => createMessageOp(ctx, userMsgOpts),
           () => makeSyntheticStoredMessage(userMsgOpts)
         );
         storedUserMessage = userMsgResult.result;
@@ -27896,7 +28011,7 @@ function useChatStorage(options) {
         const assistantMsgResult = await writeOrQueue(
           "createMessage",
           assistantMsgOpts,
-          () => createMessageOp(storageCtx, assistantMsgOpts),
+          (ctx) => createMessageOp(ctx, assistantMsgOpts),
           () => makeSyntheticStoredMessage(assistantMsgOpts),
           userMsgQueueId ? [userMsgQueueId] : []
         );
@@ -27928,7 +28043,7 @@ function useChatStorage(options) {
           const toolResultsWrite = await writeOrQueue(
             "createMessage",
             toolResultsOpts,
-            () => createMessageOp(storageCtx, toolResultsOpts),
+            (ctx) => createMessageOp(ctx, toolResultsOpts),
             () => makeSyntheticStoredMessage(toolResultsOpts),
             assistantMsgQueueId ? [assistantMsgQueueId] : []
           );
@@ -27971,7 +28086,6 @@ function useChatStorage(options) {
       walletAddress,
       signMessage,
       embeddedWalletSigner,
-      isEncryptionReady,
       enableQueue,
       getWalletAddress,
       refreshQueueStatus,
@@ -28289,6 +28403,7 @@ export {
   createWebSearchPreProcessor,
   decodeChunkVector,
   decryptConversationTitle,
+  decryptMessageEmbeddingFields,
   deleteAllVaultMemoriesForUserOp,
   deleteMessageOp,
   deleteVaultFolderOp,
@@ -28369,6 +28484,7 @@ export {
   sdkMigrations,
   sdkModelClasses,
   sdkSchema,
+  sealLegacyMessageEmbeddingsOp,
   searchVaultMemories,
   seedEncryptionKeys,
   setLogger,

package/dist/react/index.cjs

@@ -1564,6 +1564,7 @@ __export(index_exports, {
   decryptDataBytes: () => decryptDataBytes,
   decryptDataBytesFromBytes: () => decryptDataBytesFromBytes,
   decryptDataWithKey: () => decryptDataWithKey,
+  decryptMessageEmbeddingFields: () => decryptMessageEmbeddingFields,
   defaultServerToolsFilter: () => defaultServerToolsFilter,
   deleteAllAppFilesOp: () => deleteAllAppFilesOp,
   deleteAllVaultMemoriesForUserOp: () => deleteAllVaultMemoriesForUserOp,
@@ -1744,6 +1745,7 @@ __export(index_exports, {
   sdkMigrations: () => sdkMigrations,
   sdkModelClasses: () => sdkModelClasses,
   sdkSchema: () => sdkSchema,
+  sealLegacyMessageEmbeddingsOp: () => sealLegacyMessageEmbeddingsOp,
   searchChunksOp: () => searchChunksOp,
   searchMediaOp: () => searchMediaOp,
   searchMessagesOp: () => searchMessagesOp,
@@ -4714,16 +4716,19 @@ async function updateMessageEmbeddingOp(ctx, uniqueId, vector, embeddingModel) {
   } catch {
     return null;
   }
+  const address = ctx.walletAddress;
   await ctx.database.write(async () => {
+    const key = await resolveMessageEmbeddingKey(address, message.content);
+    const storedVector2 = await serializeMessageEmbedding(key, vector);
     await message.update((msg) => {
-      msg._setRaw("vector", JSON.stringify(vector));
+      msg._setRaw("vector", storedVector2);
       if (msg.origin === "chunks_discarded" && vector.length > 0) {
         msg._setRaw("origin", "message");
       }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
-  return messageToStored(message, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+  return messageToStored(message, ctx.walletAddress);
 }
 async function updateMessageChunksOp(ctx, uniqueId, chunks3, embeddingModel) {
   let message;
@@ -4733,16 +4738,123 @@ async function updateMessageChunksOp(ctx, uniqueId, chunks3, embeddingModel) {
     return null;
   }
   const storedChunks = chunks3.map(({ text: _text, ...rest }) => rest);
+  const address = ctx.walletAddress;
   await ctx.database.write(async () => {
+    const key = await resolveMessageEmbeddingKey(address, message.content);
+    const serializedChunks = await serializeMessageEmbedding(key, storedChunks);
     await message.update((msg) => {
-      msg._setRaw("chunks", JSON.stringify(storedChunks));
+      msg._setRaw("chunks", serializedChunks);
       if (msg.origin === "chunks_discarded" && storedChunks.length > 0 && storedChunks.every((chunk) => chunk.vector.length > 0)) {
         msg._setRaw("origin", "message");
       }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
-  return messageToStored(message, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+  return messageToStored(message, ctx.walletAddress);
+}
+async function resolveMessageEmbeddingKey(address, content) {
+  if (!address || !isEncrypted(content)) return null;
+  const key = await getEncryptionKey(address, "v3");
+  const contentKey = content.startsWith("enc:v2:") ? await getEncryptionKey(address, "v2") : key;
+  try {
+    await decryptDataWithKey(content.slice("enc:vN:".length), contentKey);
+  } catch {
+    throw new Error("Cannot decrypt the message with the cached wallet key");
+  }
+  return key;
+}
+async function serializeMessageEmbedding(key, value) {
+  const json4 = JSON.stringify(value);
+  return key ? `enc:v3:${await encryptDataWithKey(json4, key)}` : json4;
+}
+async function decryptMessageEmbeddingFields(raw, walletAddress) {
+  const result = { ...raw };
+  for (const column of ["vector", "chunks"]) {
+    const value = raw[column];
+    if (typeof value !== "string" || !value.startsWith("enc:")) continue;
+    const decrypted = walletAddress ? await decryptField(value, walletAddress) : value;
+    result[column] = decrypted.startsWith("enc:") ? void 0 : decrypted;
+  }
+  return result;
+}
+async function sealLegacyMessageEmbeddingsOp(ctx) {
+  if (!ctx.walletAddress) return 0;
+  const candidateIds = await ctx.messagesCollection.query(
+    import_watermelondb2.Q.where("content", import_watermelondb2.Q.like("enc:%")),
+    import_watermelondb2.Q.or(
+      ...["vector", "chunks"].map(
+        (column) => import_watermelondb2.Q.and(
+          import_watermelondb2.Q.where(column, import_watermelondb2.Q.notEq(null)),
+          import_watermelondb2.Q.where(column, import_watermelondb2.Q.notEq("")),
+          import_watermelondb2.Q.where(column, import_watermelondb2.Q.notLike("enc:%"))
+        )
+      )
+    )
+  ).fetchIds();
+  let sealed = 0;
+  let undecryptable = 0;
+  for (let start = 0; start < candidateIds.length; start += 50) {
+    const pageIds = candidateIds.slice(start, start + 50);
+    await ctx.database.write(async () => {
+      const pageMessages = await ctx.messagesCollection.query(import_watermelondb2.Q.where("id", import_watermelondb2.Q.oneOf(pageIds))).fetch();
+      for (const message of pageMessages) {
+        if (!isEncrypted(message.content)) continue;
+        const content = await decryptFieldDetailed(message.content, ctx.walletAddress);
+        if (content.status === "key_missing") {
+          throw new Error("Cannot decrypt the message with the cached wallet key");
+        }
+        if (content.status !== "ok") {
+          undecryptable++;
+          continue;
+        }
+        const plaintext = content.value;
+        const key = await resolveMessageEmbeddingKey(ctx.walletAddress, message.content);
+        const updates = {};
+        for (const column of ["vector", "chunks"]) {
+          const value = message._getRaw(column);
+          if (typeof value !== "string" || value.startsWith("enc:") || !value) continue;
+          let parsed;
+          try {
+            parsed = JSON.parse(value);
+          } catch {
+            updates[column] = null;
+            continue;
+          }
+          if (!Array.isArray(parsed)) {
+            updates[column] = null;
+            continue;
+          }
+          if (column === "chunks") {
+            const chunks3 = parsed;
+            if (chunks3.some(
+              (chunk) => !chunk || !Number.isInteger(chunk.startOffset) || !Number.isInteger(chunk.endOffset) || chunk.startOffset < 0 || chunk.endOffset <= chunk.startOffset || chunk.endOffset > plaintext.length
+            )) {
+              updates.chunks = null;
+              updates.vector = null;
+              updates.origin = message.origin === "tool_result" ? "tool_result" : "chunks_discarded";
+              break;
+            }
+            parsed = chunks3.map(({ text: _text, ...rest }) => rest);
+          }
+          updates[column] = await serializeMessageEmbedding(key, parsed);
+        }
+        if (Object.keys(updates).length === 0) continue;
+        const updatedAt = Number(message._getRaw("updated_at"));
+        const discarded = updates.origin !== void 0;
+        await message.update((msg) => {
+          for (const [column, value] of Object.entries(updates)) msg._setRaw(column, value);
+          msg._setRaw("updated_at", discarded ? Math.max(Date.now(), updatedAt + 1) : updatedAt);
+        });
+        sealed++;
+      }
+    });
+  }
+  if (undecryptable > 0) {
+    getLogger().warn(
+      `Skipped ${undecryptable} message embeddings that the cached wallet key cannot decrypt`
+    );
+  }
+  return sealed;
 }
 var RAW_REPAIR_PAGE_SIZE = 100;
 function chunksWithoutText(raw) {
@@ -33357,6 +33469,8 @@ function useChatStorage(options) {
       embeddedWalletSigner
     ]
   );
+  const storageCtxRef = (0, import_react6.useRef)(storageCtx);
+  storageCtxRef.current = storageCtx;
   const mediaCtx = (0, import_react6.useMemo)(
     () => ({
       database,
@@ -33493,14 +33607,6 @@ function useChatStorage(options) {
     refreshQueueStatus();
     return queueManager.onQueueChange(walletAddress, refreshQueueStatus);
   }, [walletAddress, refreshQueueStatus]);
-  (0, import_react6.useEffect)(() => {
-    if (!walletAddress || !enableQueue || !autoFlushOnKeyAvailable || !signMessage) return;
-    return onKeyAvailable(walletAddress, () => {
-      flushQueue().catch((err) => {
-        getLogger().warn("[useChatStorage] Auto-flush failed:", err);
-      });
-    });
-  }, [walletAddress, enableQueue, autoFlushOnKeyAvailable, signMessage, flushQueue]);
   (0, import_react6.useEffect)(() => {
     if (!getWalletAddress || walletAddress) return;
     const poller = new WalletPoller();
@@ -33524,34 +33630,49 @@ function useChatStorage(options) {
     pendingOpsRef.current = [];
     refreshQueueStatus();
   }, [walletAddress, enableQueue, refreshQueueStatus]);
+  (0, import_react6.useEffect)(() => {
+    if (!walletAddress || !enableQueue || !autoFlushOnKeyAvailable || !signMessage) return;
+    return onKeyAvailable(walletAddress, () => {
+      flushQueue().catch((err) => {
+        getLogger().warn("[useChatStorage] Auto-flush failed:", err);
+      });
+    });
+  }, [walletAddress, enableQueue, autoFlushOnKeyAvailable, signMessage, flushQueue]);
   const writeOrQueue = (0, import_react6.useCallback)(
     async (opType, payload, directWrite, makeSynthetic, dependencies = []) => {
-      if (isEncryptionReady()) {
-        const result2 = await directWrite();
+      const currentStorageCtx = storageCtx.walletAddress ? storageCtx : storageCtxRef.current;
+      const currentWalletAddress = currentStorageCtx.walletAddress;
+      if (enableQueue && signMessage && !currentWalletAddress && getWalletAddress) {
+        pendingOpsRef.current.push({ type: opType, payload, dependencies });
+        return { result: makeSynthetic(), queued: true };
+      }
+      if (!currentWalletAddress || !currentStorageCtx.signMessage || hasEncryptionKey(currentWalletAddress)) {
+        const result2 = await directWrite(currentStorageCtx);
         return { result: result2, queued: false };
       }
       if (!enableQueue) {
-        const result2 = await directWrite();
+        const result2 = await directWrite(currentStorageCtx);
         return { result: result2, queued: false };
       }
-      if (!walletAddress && getWalletAddress) {
-        pendingOpsRef.current.push({ type: opType, payload, dependencies });
-        return { result: makeSynthetic(), queued: true };
-      }
-      if (walletAddress) {
-        const queueId = queueManager.queueOperation(walletAddress, opType, payload, dependencies);
+      if (currentWalletAddress) {
+        const queueId = queueManager.queueOperation(
+          currentWalletAddress,
+          opType,
+          payload,
+          dependencies
+        );
         if (queueId === null) {
           getLogger().warn("[useChatStorage] Queue full, falling back to direct write");
-          const result2 = await directWrite();
+          const result2 = await directWrite(currentStorageCtx);
           return { result: result2, queued: false };
         }
-        refreshQueueStatus();
+        setQueueStatus(queueManager.getStatus(currentWalletAddress));
         return { result: makeSynthetic(), queued: true, queueId };
       }
-      const result = await directWrite();
+      const result = await directWrite(currentStorageCtx);
       return { result, queued: false };
     },
-    [isEncryptionReady, enableQueue, walletAddress, getWalletAddress, refreshQueueStatus]
+    [storageCtx, enableQueue, signMessage, getWalletAddress]
   );
   const embedMessageAsync = (0, import_react6.useCallback)(
     async (message, mask = maskForEmbedding) => {
@@ -33891,7 +34012,7 @@ function useChatStorage(options) {
       const { result, queued, queueId } = await writeOrQueue(
         "createConversation",
         conversationOpts,
-        () => createConversationOp(storageCtx, conversationOpts, defaultConversationTitle),
+        (ctx) => createConversationOp(ctx, conversationOpts, defaultConversationTitle),
         () => makeSyntheticStoredConversation(conversationOpts, defaultConversationTitle)
       );
       if (queued) {
@@ -33903,7 +34024,7 @@ function useChatStorage(options) {
       setCurrentConversationId(result.conversationId);
       return result;
     },
-    [storageCtx, defaultConversationTitle, writeOrQueue]
+    [defaultConversationTitle, writeOrQueue]
   );
   const getConversation = (0, import_react6.useCallback)(
     async (id) => {
@@ -33919,24 +34040,24 @@ function useChatStorage(options) {
       const { result } = await writeOrQueue(
         "updateConversationTitle",
         { conversationId: id, title },
-        () => updateConversationTitleOp(storageCtx, id, title),
+        (ctx) => updateConversationTitleOp(ctx, id, title),
         () => true
       );
       return result;
     },
-    [storageCtx, writeOrQueue]
+    [writeOrQueue]
   );
   const updateConversationPinned = (0, import_react6.useCallback)(
     async (id, pinned) => {
       const { result } = await writeOrQueue(
         "updateConversationPinned",
         { conversationId: id, pinned },
-        () => updateConversationPinnedOp(storageCtx, id, pinned),
+        (ctx) => updateConversationPinnedOp(ctx, id, pinned),
         () => true
       );
       return result;
     },
-    [storageCtx, writeOrQueue]
+    [writeOrQueue]
   );
   const deleteConversation = (0, import_react6.useCallback)(
     async (id) => {
@@ -34595,7 +34716,7 @@ function useChatStorage(options) {
         const userMsgResult = await writeOrQueue(
           "createMessage",
           userMsgOpts,
-          () => createMessageOp(storageCtx, userMsgOpts),
+          (ctx) => createMessageOp(ctx, userMsgOpts),
           () => makeSyntheticStoredMessage(userMsgOpts),
           convId && syntheticConvQueueIdsRef.current.has(convId) ? [syntheticConvQueueIdsRef.current.get(convId)] : []
         );
@@ -34916,7 +35037,7 @@ function useChatStorage(options) {
         const assistantMsgResult = await writeOrQueue(
           "createMessage",
           assistantMsgOpts,
-          () => createMessageOp(storageCtx, assistantMsgOpts),
+          (ctx) => createMessageOp(ctx, assistantMsgOpts),
           () => makeSyntheticStoredMessage(assistantMsgOpts),
           userMsgQueueId ? [userMsgQueueId] : []
         );
@@ -34967,7 +35088,7 @@ function useChatStorage(options) {
           const toolResultsWrite = await writeOrQueue(
             "createMessage",
             toolResultsOpts,
-            () => createMessageOp(storageCtx, toolResultsOpts),
+            (ctx) => createMessageOp(ctx, toolResultsOpts),
             () => makeSyntheticStoredMessage(toolResultsOpts),
             assistantMsgQueueId ? [assistantMsgQueueId] : []
           );
@@ -38196,6 +38317,7 @@ function useWalletBinding(options = {}) {
   decryptDataBytes,
   decryptDataBytesFromBytes,
   decryptDataWithKey,
+  decryptMessageEmbeddingFields,
   defaultServerToolsFilter,
   deleteAllAppFilesOp,
   deleteAllVaultMemoriesForUserOp,
@@ -38376,6 +38498,7 @@ function useWalletBinding(options = {}) {
   sdkMigrations,
   sdkModelClasses,
   sdkSchema,
+  sealLegacyMessageEmbeddingsOp,
   searchChunksOp,
   searchMediaOp,
   searchMessagesOp,