@anuma/sdk 10.1.0 → 10.2.0

package/dist/server/index.cjs

@@ -25480,17 +25480,19 @@ init_clientConfig();
 
 // src/lib/db/chat/summaryOperations.ts
 var import_watermelondb4 = __toESM(require_watermelondb());
-function createSummaryContext(database) {
+init_logger();
+function createSummaryContext(database, encryption = {}) {
   return {
+    ...encryption,
     database,
     summariesCollection: database.get("conversation_summaries")
   };
 }
-function modelToStored(record) {
+function modelToStored(record, summary) {
   return {
     uniqueId: record.id,
     conversationId: record.conversationId,
-    summary: record.summary,
+    summary,
     summarizedUpTo: record.summarizedUpTo,
     tokenCount: record.tokenCount,
     createdAt: record.createdAt,
@@ -25500,28 +25502,62 @@ function modelToStored(record) {
 async function getConversationSummaryOp(ctx, conversationId) {
   const results = await ctx.summariesCollection.query(import_watermelondb4.Q.where("conversation_id", conversationId)).fetch();
   if (results.length === 0) return null;
-  return modelToStored(results[0]);
+  const record = results[0];
+  const stored = modelToStored(record, record.summary);
+  if (ctx.walletAddress && isEncrypted(stored.summary)) {
+    if (ctx.signMessage) {
+      try {
+        await requestEncryptionKey(ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+      } catch (error) {
+        getLogger().warn("Failed to request the encryption key for the summary:", error);
+      }
+    }
+    stored.summary = await decryptField(stored.summary, ctx.walletAddress);
+  } else if (ctx.walletAddress && ctx.signMessage && stored.summary) {
+    try {
+      const encryptedSummary = await encryptField(
+        stored.summary,
+        ctx.walletAddress,
+        ctx.signMessage,
+        ctx.embeddedWalletSigner
+      );
+      await ctx.database.write(async () => {
+        const current = await ctx.summariesCollection.query(import_watermelondb4.Q.where("id", record.id)).fetch();
+        if (current[0]?.summary === stored.summary) {
+          await current[0].update((rec) => {
+            rec.summary = encryptedSummary;
+          });
+        }
+      });
+    } catch (error) {
+      getLogger().warn("Failed to encrypt the legacy summary:", error);
+    }
+  }
+  return stored;
 }
 async function upsertConversationSummaryOp(ctx, conversationId, summary, summarizedUpTo, tokenCount) {
+  const revision = getConversationRevision(ctx.database, conversationId);
   let record;
+  const encryptedSummary = ctx.walletAddress ? await encryptField(summary, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner) : summary;
   await ctx.database.write(async () => {
+    assertConversationRevision(ctx.database, conversationId, revision);
     const existing = await ctx.summariesCollection.query(import_watermelondb4.Q.where("conversation_id", conversationId)).fetch();
     if (existing.length > 0) {
       record = await existing[0].update((rec) => {
-        rec.summary = summary;
+        rec.summary = encryptedSummary;
         rec.summarizedUpTo = summarizedUpTo;
         rec.tokenCount = tokenCount;
       });
     } else {
       record = await ctx.summariesCollection.create((rec) => {
         rec.conversationId = conversationId;
-        rec.summary = summary;
+        rec.summary = encryptedSummary;
         rec.summarizedUpTo = summarizedUpTo;
         rec.tokenCount = tokenCount;
       });
     }
   });
-  return modelToStored(record);
+  return modelToStored(record, summary);
 }
 async function deleteConversationSummaryOp(ctx, conversationId) {
   await ctx.database.write(async () => {
@@ -25789,7 +25825,7 @@ async function summarizeHistoryAtRevision(options, revision) {
       const cachedSummary = await summaryStore(options).getSummary(conversationId);
       return {
         messagesToConvert: messages,
-        summarySystemMessage: cachedSummary?.summary ? summaryToSystemMessage(cachedSummary.summary) : null
+        summarySystemMessage: cachedSummary?.summary && !isEncrypted(cachedSummary.summary) ? summaryToSystemMessage(cachedSummary.summary) : null
       };
     } catch {
       return { messagesToConvert: messages, summarySystemMessage: null };
@@ -25844,14 +25880,15 @@ async function doSummarizeHistory(options, revision) {
   } = options;
   try {
     const summaries = summaryStore(options);
-    const persistSummary = (summary, summarizedUpTo, tokenCount) => database.write(async (writer) => {
+    const persistSummary = (summary, summarizedUpTo, tokenCount) => {
       assertConversationRevision(database, conversationId, revision);
-      await writer.callWriter(
-        () => summaries.upsertSummary(conversationId, summary, summarizedUpTo, tokenCount)
-      );
-    });
+      return summaries.upsertSummary(conversationId, summary, summarizedUpTo, tokenCount);
+    };
     let cachedSummary = await summaries.getSummary(conversationId);
     assertConversationRevision(database, conversationId, revision);
+    if (cachedSummary && isEncrypted(cachedSummary.summary)) {
+      throw new Error("The cached summary could not be decrypted");
+    }
     const lastCompacted = lastCompactionTime.get(conversationId) ?? 0;
     const compactionCooledDown = Date.now() - lastCompacted > COMPACTION_COOLDOWN_MS;
     if (cachedSummary && cachedSummary.tokenCount > summaryTokenThreshold * MAX_SUMMARY_TOKEN_RATIO && compactionCooledDown) {

package/dist/server/index.d.mts

@@ -8028,12 +8028,12 @@ declare function clearLazyTitleCache(): void;
 declare function decryptConversationTitle(encryptedTitle: string, address: string): Promise<string>;
 
 /** Context needed for summary operations */
-interface SummaryOperationsContext {
+interface SummaryOperationsContext extends Pick<StorageOperationsContext, "walletAddress" | "signMessage" | "embeddedWalletSigner"> {
     database: Database;
     summariesCollection: Collection<ConversationSummary>;
 }
 /** Create a summary operations context from a database instance */
-declare function createSummaryContext(database: Database): SummaryOperationsContext;
+declare function createSummaryContext(database: Database, encryption?: Pick<SummaryOperationsContext, "walletAddress" | "signMessage" | "embeddedWalletSigner">): SummaryOperationsContext;
 /**
  * Get the cached summary for a conversation.
  * Returns null if no summary exists yet.

package/dist/server/index.d.ts

@@ -8028,12 +8028,12 @@ declare function clearLazyTitleCache(): void;
 declare function decryptConversationTitle(encryptedTitle: string, address: string): Promise<string>;
 
 /** Context needed for summary operations */
-interface SummaryOperationsContext {
+interface SummaryOperationsContext extends Pick<StorageOperationsContext, "walletAddress" | "signMessage" | "embeddedWalletSigner"> {
     database: Database;
     summariesCollection: Collection<ConversationSummary>;
 }
 /** Create a summary operations context from a database instance */
-declare function createSummaryContext(database: Database): SummaryOperationsContext;
+declare function createSummaryContext(database: Database, encryption?: Pick<SummaryOperationsContext, "walletAddress" | "signMessage" | "embeddedWalletSigner">): SummaryOperationsContext;
 /**
  * Get the cached summary for a conversation.
  * Returns null if no summary exists yet.

package/dist/server/index.mjs

@@ -23753,17 +23753,18 @@ function createStockPricePreProcessor(options = {}) {
 
 // src/lib/db/chat/summaryOperations.ts
 var import_watermelondb4 = __toESM(require_watermelondb());
-function createSummaryContext(database) {
+function createSummaryContext(database, encryption = {}) {
   return {
+    ...encryption,
     database,
     summariesCollection: database.get("conversation_summaries")
   };
 }
-function modelToStored(record) {
+function modelToStored(record, summary) {
   return {
     uniqueId: record.id,
     conversationId: record.conversationId,
-    summary: record.summary,
+    summary,
     summarizedUpTo: record.summarizedUpTo,
     tokenCount: record.tokenCount,
     createdAt: record.createdAt,
@@ -23773,28 +23774,62 @@ function modelToStored(record) {
 async function getConversationSummaryOp(ctx, conversationId) {
   const results = await ctx.summariesCollection.query(import_watermelondb4.Q.where("conversation_id", conversationId)).fetch();
   if (results.length === 0) return null;
-  return modelToStored(results[0]);
+  const record = results[0];
+  const stored = modelToStored(record, record.summary);
+  if (ctx.walletAddress && isEncrypted(stored.summary)) {
+    if (ctx.signMessage) {
+      try {
+        await requestEncryptionKey(ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner);
+      } catch (error) {
+        getLogger().warn("Failed to request the encryption key for the summary:", error);
+      }
+    }
+    stored.summary = await decryptField(stored.summary, ctx.walletAddress);
+  } else if (ctx.walletAddress && ctx.signMessage && stored.summary) {
+    try {
+      const encryptedSummary = await encryptField(
+        stored.summary,
+        ctx.walletAddress,
+        ctx.signMessage,
+        ctx.embeddedWalletSigner
+      );
+      await ctx.database.write(async () => {
+        const current = await ctx.summariesCollection.query(import_watermelondb4.Q.where("id", record.id)).fetch();
+        if (current[0]?.summary === stored.summary) {
+          await current[0].update((rec) => {
+            rec.summary = encryptedSummary;
+          });
+        }
+      });
+    } catch (error) {
+      getLogger().warn("Failed to encrypt the legacy summary:", error);
+    }
+  }
+  return stored;
 }
 async function upsertConversationSummaryOp(ctx, conversationId, summary, summarizedUpTo, tokenCount) {
+  const revision = getConversationRevision(ctx.database, conversationId);
   let record;
+  const encryptedSummary = ctx.walletAddress ? await encryptField(summary, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner) : summary;
   await ctx.database.write(async () => {
+    assertConversationRevision(ctx.database, conversationId, revision);
     const existing = await ctx.summariesCollection.query(import_watermelondb4.Q.where("conversation_id", conversationId)).fetch();
     if (existing.length > 0) {
       record = await existing[0].update((rec) => {
-        rec.summary = summary;
+        rec.summary = encryptedSummary;
         rec.summarizedUpTo = summarizedUpTo;
         rec.tokenCount = tokenCount;
       });
     } else {
       record = await ctx.summariesCollection.create((rec) => {
         rec.conversationId = conversationId;
-        rec.summary = summary;
+        rec.summary = encryptedSummary;
         rec.summarizedUpTo = summarizedUpTo;
         rec.tokenCount = tokenCount;
       });
     }
   });
-  return modelToStored(record);
+  return modelToStored(record, summary);
 }
 async function deleteConversationSummaryOp(ctx, conversationId) {
   await ctx.database.write(async () => {
@@ -24059,7 +24094,7 @@ async function summarizeHistoryAtRevision(options, revision) {
       const cachedSummary = await summaryStore(options).getSummary(conversationId);
       return {
         messagesToConvert: messages,
-        summarySystemMessage: cachedSummary?.summary ? summaryToSystemMessage(cachedSummary.summary) : null
+        summarySystemMessage: cachedSummary?.summary && !isEncrypted(cachedSummary.summary) ? summaryToSystemMessage(cachedSummary.summary) : null
       };
     } catch {
       return { messagesToConvert: messages, summarySystemMessage: null };
@@ -24114,14 +24149,15 @@ async function doSummarizeHistory(options, revision) {
   } = options;
   try {
     const summaries = summaryStore(options);
-    const persistSummary = (summary, summarizedUpTo, tokenCount) => database.write(async (writer) => {
+    const persistSummary = (summary, summarizedUpTo, tokenCount) => {
       assertConversationRevision(database, conversationId, revision);
-      await writer.callWriter(
-        () => summaries.upsertSummary(conversationId, summary, summarizedUpTo, tokenCount)
-      );
-    });
+      return summaries.upsertSummary(conversationId, summary, summarizedUpTo, tokenCount);
+    };
     let cachedSummary = await summaries.getSummary(conversationId);
     assertConversationRevision(database, conversationId, revision);
+    if (cachedSummary && isEncrypted(cachedSummary.summary)) {
+      throw new Error("The cached summary could not be decrypted");
+    }
     const lastCompacted = lastCompactionTime.get(conversationId) ?? 0;
     const compactionCooledDown = Date.now() - lastCompacted > COMPACTION_COOLDOWN_MS;
     if (cachedSummary && cachedSummary.tokenCount > summaryTokenThreshold * MAX_SUMMARY_TOKEN_RATIO && compactionCooledDown) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anuma/sdk",
-  "version": "10.1.0",
+  "version": "10.2.0",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",