@sentry/junior-memory 0.81.0 → 0.82.0

package/dist/index.js

@@ -770,6 +770,7 @@ var MAX_MEMORY_CONTENT_CHARS = 4e3;
 var EMBEDDING_METRIC = "cosine";
 var ONE_DAY_MS = 24 * 60 * 60 * 1e3;
 var RELEVANCE_NEAR_TIE_DELTA = 0.01;
+var SOURCE_CHANNEL_BOOST = 0.05;
 var SUPERSESSION_STOP_TERMS = /* @__PURE__ */ new Set([
   "and",
   "for",
@@ -907,6 +908,12 @@ function sourceKey(ctx) {
   }
   return `slack:${ctx.source.teamId}:${ctx.source.channelId}:${threadKey}`;
 }
+function sourceChannelPrefix(ctx) {
+  if (ctx.source.platform !== "slack") {
+    return void 0;
+  }
+  return `slack:${ctx.source.teamId}:${ctx.source.channelId}:`;
+}
 function parseMemoryRow(row) {
   const parsed = memoryRowSchema.parse(row);
   return memoryRecordSchema.parse({
@@ -1339,7 +1346,11 @@ async function searchVisibleMemories(args) {
       )
     )
   );
-  return rows.map(parseMemoryRow);
+  return rows.map((row) => ({
+    memory: parseMemoryRow(row),
+    score: 0,
+    sourceKey: row.sourceKey
+  }));
 }
 async function searchVisibleVectorMemories(args) {
   if (!args.embedder) {
@@ -1384,14 +1395,24 @@ async function searchVisibleVectorMemories(args) {
     if (row.distance === null || !Number.isFinite(distanceValue) || hashEmbeddedContent(row.memory.content) !== row.contentHash) {
       return [];
     }
+    if (args.maxDistance !== void 0 && distanceValue > args.maxDistance) {
+      return [];
+    }
     return [
       {
         memory: parseMemoryRow(row.memory),
-        score: 1 / (1 + Math.max(0, distanceValue))
+        score: 1 / (1 + Math.max(0, distanceValue)),
+        sourceKey: row.memory.sourceKey
       }
     ];
   });
 }
+function sourceBoost(candidate, currentChannelPrefix) {
+  if (!currentChannelPrefix) {
+    return 0;
+  }
+  return candidate.sourceKey.startsWith(currentChannelPrefix) ? SOURCE_CHANNEL_BOOST : 0;
+}
 function observedAgeBoost(memory, nowMs) {
   const ageMs = Math.max(0, nowMs - memory.observedAtMs);
   if (ageMs <= 7 * ONE_DAY_MS) {
@@ -1405,7 +1426,7 @@ function observedAgeBoost(memory, nowMs) {
   }
   return 0;
 }
-function mergeSearchCandidates(candidates, nowMs) {
+function mergeSearchCandidates(candidates, nowMs, currentChannelKey) {
   const byId = /* @__PURE__ */ new Map();
   for (const candidate of candidates) {
     const existing = byId.get(candidate.memory.id);
@@ -1415,7 +1436,8 @@ function mergeSearchCandidates(candidates, nowMs) {
     }
     byId.set(candidate.memory.id, {
       memory: candidate.memory,
-      score: candidate.score
+      score: candidate.score,
+      sourceKey: candidate.sourceKey
     });
   }
   return [...byId.values()].sort((left, right) => {
@@ -1423,6 +1445,10 @@ function mergeSearchCandidates(candidates, nowMs) {
     if (Math.abs(scoreDelta) > RELEVANCE_NEAR_TIE_DELTA) {
       return scoreDelta;
     }
+    const sourceDelta = sourceBoost(right, currentChannelKey) - sourceBoost(left, currentChannelKey);
+    if (sourceDelta !== 0) {
+      return sourceDelta;
+    }
     return observedAgeBoost(right.memory, nowMs) - observedAgeBoost(left.memory, nowMs) || right.memory.observedAtMs - left.memory.observedAtMs || left.memory.id.localeCompare(right.memory.id);
   }).map((candidate) => candidate.memory);
 }
@@ -1430,6 +1456,7 @@ function createMemoryStore(db, context, options = {}) {
   const runtimeContext = memoryRuntimeContextSchema.parse(context);
   const parsedOptions = memoryStoreOptionsSchema.parse({ now: options.now });
   const embedder = options.embedder;
+  const maxVectorDistance = options.maxVectorDistance;
   const supersessionDecider = options.supersessionDecider;
   const getNowMs = parsedOptions.now ?? Date.now;
   async function archiveExpiredVisibleMemories(input, nowMs) {
@@ -1681,6 +1708,7 @@ function createMemoryStore(db, context, options = {}) {
         db,
         embedder,
         limit: limit * VECTOR_SEARCH_OVERFETCH,
+        ...maxVectorDistance !== void 0 ? { maxDistance: maxVectorDistance } : {},
         nowMs,
         query: input.query,
         scopes
@@ -1692,10 +1720,14 @@ function createMemoryStore(db, context, options = {}) {
         scopes
       });
       const terms = searchTerms(input.query);
-      const lexicalCandidates = candidates.map((memory) => ({ memory, score: searchScore(memory, terms) })).filter((item) => item.score > 0);
+      const lexicalCandidates = candidates.map((candidate) => ({
+        ...candidate,
+        score: searchScore(candidate.memory, terms)
+      })).filter((item) => item.score > 0);
       return mergeSearchCandidates(
         [...vectorCandidates, ...lexicalCandidates],
-        nowMs
+        nowMs,
+        sourceChannelPrefix(runtimeContext)
       ).slice(0, limit);
     },
     async archiveMemory(input) {
@@ -2220,8 +2252,8 @@ async function processMemorySession(context) {
 
 // src/recall.ts
 var DEFAULT_RECALL_LIMIT = 5;
-var MAX_PROMPT_CHARS = 1600;
-var MAX_MEMORY_LINE_CHARS = 320;
+var MAX_PROMPT_CHARS = 4e3;
+var MAX_MEMORY_LINE_CHARS = 600;
 function trimContent(content, maxLength) {
   const trimmed = content.trim();
   if (trimmed.length <= maxLength) {
@@ -2268,7 +2300,10 @@ async function createMemoryPromptMessages(context) {
   const memories = await createMemoryStore(
     context.db,
     runtimeContext,
-    context.embedder ? { embedder: context.embedder } : {}
+    {
+      ...context.embedder ? { embedder: context.embedder } : {},
+      ...context.maxVectorDistance !== void 0 ? { maxVectorDistance: context.maxVectorDistance } : {}
+    }
   ).searchMemories({
     query: context.text,
     limit: DEFAULT_RECALL_LIMIT
@@ -2279,6 +2314,8 @@ async function createMemoryPromptMessages(context) {
 
 // src/plugin.ts
 var MEMORY_MODEL_ENV = "AI_MEMORY_MODEL";
+var MEMORY_RECALL_MAX_VECTOR_DISTANCE_ENV = "MEMORY_RECALL_MAX_VECTOR_DISTANCE";
+var DEFAULT_RECALL_MAX_VECTOR_DISTANCE = 0.45;
 function memoryModelId(options) {
   const explicitModelId = options.modelId?.trim();
   if (explicitModelId) {
@@ -2287,6 +2324,19 @@ function memoryModelId(options) {
   const envModelId = process.env[MEMORY_MODEL_ENV]?.trim();
   return envModelId || void 0;
 }
+function recallMaxVectorDistance(options) {
+  if (options.recallMaxVectorDistance !== void 0) {
+    return options.recallMaxVectorDistance;
+  }
+  const raw = process.env[MEMORY_RECALL_MAX_VECTOR_DISTANCE_ENV]?.trim();
+  if (raw) {
+    const parsed = Number(raw);
+    if (Number.isFinite(parsed) && parsed > 0) {
+      return Math.min(parsed, 1);
+    }
+  }
+  return DEFAULT_RECALL_MAX_VECTOR_DISTANCE;
+}
 function memoryToolContext(ctx) {
   return {
     agent: ctx.agent,
@@ -2354,6 +2404,7 @@ function createMemoryPlugin(options = {}) {
           ...ctx.requester ? { requester: ctx.requester } : {},
           db: ctx.db,
           embedder: ctx.embedder,
+          maxVectorDistance: recallMaxVectorDistance(options),
           source: ctx.source,
           text: ctx.text
         });