@anuma/sdk 6.21.1 → 6.23.0

package/dist/expo/{chunk-CIMZZ7GQ.mjs → chunk-6BULYLRY.mjs}

@@ -942,6 +942,12 @@ async function decryptJsonField(value, address) {
   }
 }
 
+// src/lib/db/extractionJobs/models.ts
+import { Model } from "@nozbe/watermelondb";
+var ExtractionJob = class extends Model {
+};
+ExtractionJob.table = "memory_extraction_jobs";
+
 // src/lib/db/chat/conversationEncryption.ts
 async function encryptConversationFields(conversation, address, signMessage, embeddedWalletSigner) {
   if (!address || !signMessage) {
@@ -1584,6 +1590,8 @@ async function deleteConversationOp(ctx, id) {
       await results[0].update((conv) => {
         conv._setRaw("is_deleted", true);
       });
+      const jobs = ctx.database.schema?.tables[ExtractionJob.table] ? await ctx.database.get(ExtractionJob.table).query(Q.where("conversation_id", id)).fetch() : [];
+      for (const job of jobs) await job.destroyPermanently();
     });
     return true;
   }
@@ -1704,8 +1712,16 @@ async function getToolCallEventIdsOp(ctx, convId) {
   return ids;
 }
 async function clearMessagesOp(ctx, convId) {
-  const messages = await ctx.messagesCollection.query(Q.where("conversation_id", convId)).fetch();
   await ctx.database.write(async () => {
+    const messages = await ctx.messagesCollection.query(Q.where("conversation_id", convId)).fetch();
+    const jobs = ctx.database.schema?.tables[ExtractionJob.table] ? await ctx.database.get(ExtractionJob.table).query(Q.where("conversation_id", convId)).fetch() : [];
+    for (const job of jobs) {
+      await job.update((row) => {
+        row._setRaw("message_ids", "[]");
+        row._setRaw("watermark", null);
+        row._setRaw("watermark_seq", null);
+      });
+    }
     for (const message of messages) {
       await message.update((msg) => {
         msg._setRaw("file_ids", null);
@@ -1715,6 +1731,30 @@ async function clearMessagesOp(ctx, convId) {
     }
   });
 }
+async function deleteMessageOp(ctx, uniqueId) {
+  let message;
+  try {
+    message = await ctx.messagesCollection.find(uniqueId);
+  } catch {
+    return null;
+  }
+  await ctx.database.write(async () => {
+    const jobs = ctx.database.schema?.tables[ExtractionJob.table] ? await ctx.database.get(ExtractionJob.table).query(Q.where("conversation_id", message.conversationId)).fetch() : [];
+    for (const job of jobs) {
+      const ids = JSON.parse(String(job._getRaw("message_ids")));
+      await job.update((row) => {
+        row._setRaw("message_ids", JSON.stringify(ids.filter((id) => id !== uniqueId)));
+        if (row._getRaw("watermark") === uniqueId) row._setRaw("watermark", null);
+      });
+    }
+    await message.update((msg) => {
+      msg._setRaw("file_ids", null);
+      msg._setRaw("files", null);
+    });
+    await message.destroyPermanently();
+  });
+  return uniqueId;
+}
 async function encryptMessageOptsIfNeeded(ctx, opts) {
   return ctx.walletAddress && ctx.signMessage ? await encryptMessageFields(opts, ctx.walletAddress, ctx.signMessage, ctx.embeddedWalletSigner) : opts;
 }
@@ -2557,6 +2597,7 @@ function isOwnedByCtxUser(ctx, record) {
 }
 function baseVaultConditions(ctx, options) {
   return [
+    ...options?.memoryIds !== void 0 ? [Q3.where("id", Q3.oneOf(options.memoryIds))] : [],
     ...options?.includeDeleted ? [] : [Q3.where("is_deleted", false)],
     ...options?.includeArchived ? [] : [Q3.where("archived_at", Q3.eq(null))],
     ...options?.includeQuarantined ? [] : [Q3.where("trust_tier", Q3.notEq("quarantined"))],
@@ -2641,6 +2682,8 @@ async function createVaultMemoryOp(ctx, opts) {
     ctx.embeddedWalletSigner
   ) : opts.content;
   const created = await ctx.database.write(async () => {
+    if (ctx.canWrite && !await ctx.canWrite())
+      throw new Error("Memory source is no longer eligible");
     return ctx.vaultMemoryCollection.create((record) => {
       record._setRaw("content", encryptedContent);
       record._setRaw("scope", scope);
@@ -2690,6 +2733,8 @@ async function createSupersedingMemoryOp(ctx, opts, targetId) {
   ) : opts.content;
   let createdRecord = null;
   await ctx.database.write(async () => {
+    if (ctx.canWrite && !await ctx.canWrite())
+      throw new Error("Memory source is no longer eligible");
     let target;
     try {
       target = await ctx.vaultMemoryCollection.find(targetId);
@@ -2794,6 +2839,8 @@ async function createVaultMemoriesBatchOp(ctx, optionsArray) {
     })
   );
   const created = await ctx.database.write(async () => {
+    if (ctx.canWrite && !await ctx.canWrite())
+      throw new Error("Memory source is no longer eligible");
     const prepared = optionsArray.map(
       (opts, i) => ctx.vaultMemoryCollection.prepareCreate((record) => {
         record._setRaw("content", encryptedContents[i]);
@@ -2980,6 +3027,11 @@ async function getVaultCandidateKeysOp(ctx, options) {
     });
     const clauses = [base.sql];
     const args = [...base.args];
+    if (options?.memoryIds !== void 0) {
+      if (options.memoryIds.length === 0) return [];
+      clauses.push(`"id" in (${options.memoryIds.map(() => "?").join(",")})`);
+      args.push(...options.memoryIds);
+    }
     if (options?.scopes?.length) {
       clauses.push(`"scope" in (${options.scopes.map(() => "?").join(",")})`);
       args.push(...options.scopes);
@@ -3001,6 +3053,7 @@ async function getVaultCandidateKeysOp(ctx, options) {
     );
     const conditions = [
       ...baseVaultConditions(ctx, {
+        memoryIds: options?.memoryIds,
         ...options?.includeArchived !== void 0 && { includeArchived: options.includeArchived }
       }),
       ...options?.scopes?.length ? [Q3.where("scope", Q3.oneOf(options.scopes))] : [],
@@ -3088,10 +3141,23 @@ async function updateVaultMemoryOp(ctx, id, opts) {
     const record = probe;
     const originalUpdatedAt = record.updatedAt.getTime();
     await ctx.database.write(async () => {
+      if (ctx.canWrite && !await ctx.canWrite())
+        throw new Error("Memory source is no longer eligible");
       if (record.isDeleted || record.supersededBy || !isOwnedByCtxUser(ctx, record)) {
         stale = true;
         return;
       }
+      let observedSources = [];
+      let replayedObservation = false;
+      if (opts.observationSourceIds?.length) {
+        try {
+          const parsed = JSON.parse(record.sourceChunkIds ?? "[]");
+          if (Array.isArray(parsed))
+            observedSources = parsed.filter((id2) => typeof id2 === "string");
+        } catch {
+        }
+        replayedObservation = opts.observationSourceIds.every((id2) => observedSources.includes(id2));
+      }
       await record.update((r) => {
         r._setRaw("content", encryptedContent);
         if (opts.scope !== void 0) {
@@ -3105,9 +3171,15 @@ async function updateVaultMemoryOp(ctx, id, opts) {
           r._setRaw("embedding_model", opts.embeddingModel ?? null);
         }
         if (opts.sourceChunkIds !== void 0) {
-          r._setRaw("source_chunk_ids", JSON.stringify(opts.sourceChunkIds));
+          r._setRaw(
+            "source_chunk_ids",
+            JSON.stringify(
+              opts.observationSourceIds?.length ? [.../* @__PURE__ */ new Set([...observedSources, ...opts.sourceChunkIds])] : opts.sourceChunkIds
+            )
+          );
         }
-        if (opts.proofCountIncrement !== void 0) {
+        if (replayedObservation) {
+        } else if (opts.proofCountIncrement !== void 0) {
           const current = r.proofCount ?? 1;
           r._setRaw("proof_count", current + opts.proofCountIncrement);
         } else if (opts.proofCount !== void 0) {
@@ -3124,7 +3196,7 @@ async function updateVaultMemoryOp(ctx, id, opts) {
         if (opts.topicsUserManaged !== void 0) {
           r._setRaw("topics_user_managed", opts.topicsUserManaged);
         }
-        if (opts.lastObservedAt !== void 0) {
+        if (opts.lastObservedAt !== void 0 && !replayedObservation) {
           r._setRaw("last_observed_at", opts.lastObservedAt);
         }
         if (opts.factType !== void 0) {
@@ -3250,6 +3322,8 @@ async function supersedeVaultMemoryOp(ctx, id, supersededById) {
     }
     let stale = false;
     await ctx.database.write(async () => {
+      if (ctx.canWrite && !await ctx.canWrite())
+        throw new Error("Memory source is no longer eligible");
       if (record.isDeleted || record.supersededBy || !isOwnedByCtxUser(ctx, record)) {
         stale = true;
         return;
@@ -3542,6 +3616,7 @@ async function getDecayCandidatesRawOp(ctx) {
     eventTimeEnd: raw.event_time_end ?? null,
     eventTimeKind: raw.event_time_kind ?? null,
     updatedAt: raw.updated_at,
+    lastObservedAt: raw.last_observed_at ?? null,
     archivedAt: raw.archived_at ?? null,
     source: raw.source ?? null,
     trustTier: raw.trust_tier ?? null
@@ -3558,7 +3633,7 @@ async function archiveVaultMemoryOp(ctx, id, opts) {
         stale = true;
         return;
       }
-      if (opts?.expectedUpdatedAt !== void 0 && record.updatedAt.getTime() !== opts.expectedUpdatedAt) {
+      if (opts?.expectedUpdatedAt !== void 0 && record.updatedAt.getTime() !== opts.expectedUpdatedAt || opts?.expectedLastObservedAt !== void 0 && (record.lastObservedAt ?? null) !== opts.expectedLastObservedAt) {
         stale = true;
         return;
       }
@@ -6388,6 +6463,7 @@ async function prepareVaultCandidates(query, vaultCtx, embeddingOptions, cache,
   const scopes = searchOptions?.scopes;
   const folderId = searchOptions?.folderId;
   const queryOpts = {};
+  if (searchOptions?.memoryIds !== void 0) queryOpts.memoryIds = searchOptions.memoryIds;
   if (scopes?.length) queryOpts.scopes = scopes;
   if (folderId !== void 0) queryOpts.folderId = folderId;
   if (searchOptions?.factTypes?.length) queryOpts.factTypes = searchOptions.factTypes;
@@ -6912,7 +6988,7 @@ function createMemoryVaultSearchTool(vaultCtx, embeddingOptions, cache, searchOp
             subQueries = decomp.subQueries;
           }
         }
-        const { recall: recall2 } = await import("./recall-FQQGE6A2.mjs");
+        const { recall: recall2 } = await import("./recall-CUTJWQDQ.mjs");
         let recallDegraded = [];
         const result = await recall2(
           query,
@@ -7479,7 +7555,9 @@ function dedupeBy(items, ...keys) {
   return out;
 }
 async function recall(query, ctx, options = {}) {
-  const types = options.types ?? ["fact"];
+  const requestedTypes = options.types ?? ["fact"];
+  const chunksScopeRestricted = options.memoryIds !== void 0 && requestedTypes.includes("chunk");
+  const types = options.memoryIds !== void 0 ? requestedTypes.filter((kind) => kind === "fact") : requestedTypes;
   const limit = options.limit ?? DEFAULT_LIMIT;
   const usedBudget = options.budget ?? DEFAULT_BUDGET;
   const flags = flagsForBudget(usedBudget);
@@ -7531,7 +7609,10 @@ async function recall(query, ctx, options = {}) {
     if (embeddingsUnavailable) degraded.push("embeddings-unavailable");
     if (graphLaneFailed) degraded.push("graph-lane-failed");
     if (temporalLaneFailed) degraded.push("temporal-lane-failed");
-    const laneRan = types.includes("fact") && !!ctx.vaultCtx && !!ctx.vaultCache || types.includes("chunk") && !!ctx.storageCtx;
+    if (chunksScopeRestricted) degraded.push("chunks-scope-restricted");
+    const laneRan = types.includes("fact") && !!ctx.vaultCtx && !!ctx.vaultCache || types.includes("chunk") && !!ctx.storageCtx || // The chunk lane was wired and deliberately dropped for the scope, which
+    // is not the "requested kinds have no store" wiring bug `no-lanes` means.
+    chunksScopeRestricted && !!ctx.storageCtx;
     const scores = admitted.map((m) => m.score);
     const minScoreApplied = factResults.length > 0 ? factFloor : chunkResults.length > 0 ? chunkFloor : factFloor >= 0 ? factFloor : chunkFloor;
     const diagnostics = {
@@ -7676,6 +7757,7 @@ async function recall(query, ctx, options = {}) {
         ...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 }
@@ -7940,6 +8022,7 @@ export {
   decryptField,
   encryptJsonField,
   decryptJsonField,
+  ExtractionJob,
   getToolCallEvents,
   getToolsChecksum,
   getImageModel,
@@ -7965,6 +8048,7 @@ export {
   getMessageSkeletonsOp,
   getToolCallEventIdsOp,
   clearMessagesOp,
+  deleteMessageOp,
   createMessageOp,
   upsertMessageOp,
   getMessageOp,