@sentry/junior-memory 0.81.0 → 0.83.0

package/dist/plugin.d.ts

@@ -1,5 +1,7 @@
 export interface MemoryPluginOptions {
     modelId?: string;
+    /** Maximum cosine distance for vector recall candidates. Defaults to MEMORY_RECALL_MAX_VECTOR_DISTANCE env or 0.45. */
+    recallMaxVectorDistance?: number;
 }
 /** Create Junior's long-term memory plugin registration. */
 export declare function createMemoryPlugin(options?: MemoryPluginOptions): import("@sentry/junior-plugin-api").PluginRegistration;

package/dist/recall.d.ts

@@ -4,6 +4,8 @@ export interface MemoryRecallContext {
     conversationId?: string;
     db: MemoryDb;
     embedder?: MemoryEmbeddingProvider;
+    /** Maximum cosine distance for vector recall. Passed through to the memory store. */
+    maxVectorDistance?: number;
     requester?: Requester;
     source: Source;
     text: string;

package/dist/store.d.ts

@@ -107,6 +107,8 @@ export interface MemorySupersessionDecider {
 }
 export interface MemoryStoreOptions {
     embedder?: MemoryEmbeddingProvider;
+    /** Maximum cosine distance for vector recall candidates. Model-dependent; tune when changing AI_EMBEDDING_MODEL. */
+    maxVectorDistance?: number;
     now?: () => number;
     supersessionDecider?: MemorySupersessionDecider;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sentry/junior-memory",
-  "version": "0.81.0",
+  "version": "0.83.0",
   "private": false,
   "publishConfig": {
     "access": "public"
@@ -27,7 +27,7 @@
     "commander": "^14.0.3",
     "drizzle-orm": "^0.45.2",
     "zod": "^4.4.3",
-    "@sentry/junior-plugin-api": "0.81.0"
+    "@sentry/junior-plugin-api": "0.83.0"
   },
   "devDependencies": {
     "@types/node": "^25.9.1",

package/src/plugin.ts

@@ -15,9 +15,13 @@ import { createMemoryPromptMessages } from "./recall";
 import type { MemoryDb } from "./store";
 
 const MEMORY_MODEL_ENV = "AI_MEMORY_MODEL";
+const MEMORY_RECALL_MAX_VECTOR_DISTANCE_ENV = "MEMORY_RECALL_MAX_VECTOR_DISTANCE";
+const DEFAULT_RECALL_MAX_VECTOR_DISTANCE = 0.45;
 
 export interface MemoryPluginOptions {
   modelId?: string;
+  /** Maximum cosine distance for vector recall candidates. Defaults to MEMORY_RECALL_MAX_VECTOR_DISTANCE env or 0.45. */
+  recallMaxVectorDistance?: number;
 }
 
 function memoryModelId(options: MemoryPluginOptions): string | undefined {
@@ -29,6 +33,20 @@ function memoryModelId(options: MemoryPluginOptions): string | undefined {
   return envModelId || undefined;
 }
 
+function recallMaxVectorDistance(options: MemoryPluginOptions): number {
+  if (options.recallMaxVectorDistance !== undefined) {
+    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: {
   agent: MemoryReviewer;
   conversationId?: string;
@@ -118,6 +136,7 @@ export function createMemoryPlugin(options: MemoryPluginOptions = {}) {
           ...(ctx.requester ? { requester: ctx.requester } : {}),
           db: ctx.db as MemoryDb,
           embedder: ctx.embedder,
+          maxVectorDistance: recallMaxVectorDistance(options),
           source: ctx.source,
           text: ctx.text,
         });

package/src/recall.ts

@@ -12,13 +12,15 @@ import {
 import { memoryRuntimeContextSchema } from "./types";
 
 const DEFAULT_RECALL_LIMIT = 5;
-const MAX_PROMPT_CHARS = 1_600;
-const MAX_MEMORY_LINE_CHARS = 320;
+const MAX_PROMPT_CHARS = 4_000;
+const MAX_MEMORY_LINE_CHARS = 600;
 
 export interface MemoryRecallContext {
   conversationId?: string;
   db: MemoryDb;
   embedder?: MemoryEmbeddingProvider;
+  /** Maximum cosine distance for vector recall. Passed through to the memory store. */
+  maxVectorDistance?: number;
   requester?: Requester;
   source: Source;
   text: string;
@@ -78,7 +80,12 @@ export async function createMemoryPromptMessages(
   const memories = await createMemoryStore(
     context.db,
     runtimeContext,
-    context.embedder ? { embedder: context.embedder } : {},
+    {
+      ...(context.embedder ? { embedder: context.embedder } : {}),
+      ...(context.maxVectorDistance !== undefined
+        ? { maxVectorDistance: context.maxVectorDistance }
+        : {}),
+    },
   ).searchMemories({
     query: context.text,
     limit: DEFAULT_RECALL_LIMIT,

package/src/store.ts

@@ -56,6 +56,7 @@ const MAX_MEMORY_CONTENT_CHARS = 4_000;
 const EMBEDDING_METRIC = "cosine";
 const ONE_DAY_MS = 24 * 60 * 60 * 1000;
 const RELEVANCE_NEAR_TIE_DELTA = 0.01;
+const SOURCE_CHANNEL_BOOST = 0.05;
 const SUPERSESSION_STOP_TERMS = new Set([
   "and",
   "for",
@@ -73,6 +74,7 @@ export type MemoryDb = PgDatabase<PgQueryResultHKT, typeof memorySqlSchema>;
 interface SearchCandidate {
   memory: MemoryRecord;
   score: number;
+  sourceKey: string;
 }
 
 interface MemoryEmbedding {
@@ -274,6 +276,8 @@ export interface MemorySupersessionDecider {
 
 export interface MemoryStoreOptions {
   embedder?: MemoryEmbeddingProvider;
+  /** Maximum cosine distance for vector recall candidates. Model-dependent; tune when changing AI_EMBEDDING_MODEL. */
+  maxVectorDistance?: number;
   now?: () => number;
   supersessionDecider?: MemorySupersessionDecider;
 }
@@ -343,6 +347,14 @@ function sourceKey(ctx: MemoryRuntimeContext): string {
   return `slack:${ctx.source.teamId}:${ctx.source.channelId}:${threadKey}`;
 }
 
+function sourceChannelPrefix(ctx: MemoryRuntimeContext): string | undefined {
+  if (ctx.source.platform !== "slack") {
+    return undefined;
+  }
+  // TODO(v0.82.0): Replace Slack source-key prefix matching with typed source proximity metadata.
+  return `slack:${ctx.source.teamId}:${ctx.source.channelId}:`;
+}
+
 /** Parse one SQL row into the public memory record projection. */
 function parseMemoryRow(row: unknown): MemoryRecord {
   const parsed = memoryRowSchema.parse(row);
@@ -959,7 +971,7 @@ async function searchVisibleMemories(args: {
   nowMs: number;
   query: string;
   scopes: ResolvedMemoryScope[];
-}): Promise<MemoryRecord[]> {
+}): Promise<SearchCandidate[]> {
   const terms = searchTerms(args.query);
   if (terms.length === 0) {
     return [];
@@ -981,7 +993,11 @@ async function searchVisibleMemories(args: {
         ),
       ),
     );
-  return rows.map(parseMemoryRow);
+  return rows.map((row) => ({
+    memory: parseMemoryRow(row),
+    score: 0,
+    sourceKey: row.sourceKey,
+  }));
 }
 
 /** Search active visible records with exact pgvector cosine distance. */
@@ -989,6 +1005,7 @@ async function searchVisibleVectorMemories(args: {
   db: MemoryDb;
   embedder: MemoryEmbeddingProvider | undefined;
   limit: number;
+  maxDistance?: number;
   nowMs: number;
   query: string;
   scopes: ResolvedMemoryScope[];
@@ -1045,15 +1062,31 @@ async function searchVisibleVectorMemories(args: {
     ) {
       return [];
     }
+    if (args.maxDistance !== undefined && distanceValue > args.maxDistance) {
+      return [];
+    }
     return [
       {
         memory: parseMemoryRow(row.memory),
         score: 1 / (1 + Math.max(0, distanceValue)),
+        sourceKey: row.memory.sourceKey,
       },
     ];
   });
 }
 
+function sourceBoost(
+  candidate: Pick<SearchCandidate, "sourceKey">,
+  currentChannelPrefix: string | undefined,
+): number {
+  if (!currentChannelPrefix) {
+    return 0;
+  }
+  return candidate.sourceKey.startsWith(currentChannelPrefix)
+    ? SOURCE_CHANNEL_BOOST
+    : 0;
+}
+
 /** Score only observation age for near-tie reranking; this is not lifecycle decay. */
 function observedAgeBoost(memory: MemoryRecord, nowMs: number): number {
   const ageMs = Math.max(0, nowMs - memory.observedAtMs);
@@ -1073,12 +1106,14 @@ function observedAgeBoost(memory: MemoryRecord, nowMs: number): number {
 function mergeSearchCandidates(
   candidates: SearchCandidate[],
   nowMs: number,
+  currentChannelKey?: string,
 ): MemoryRecord[] {
   const byId = new Map<
     string,
     {
       memory: MemoryRecord;
       score: number;
+      sourceKey: string;
     }
   >();
   for (const candidate of candidates) {
@@ -1090,6 +1125,7 @@ function mergeSearchCandidates(
     byId.set(candidate.memory.id, {
       memory: candidate.memory,
       score: candidate.score,
+      sourceKey: candidate.sourceKey,
     });
   }
   return [...byId.values()]
@@ -1098,6 +1134,12 @@ function mergeSearchCandidates(
       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) ||
@@ -1117,6 +1159,7 @@ export function createMemoryStore(
   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;
 
@@ -1404,6 +1447,9 @@ export function createMemoryStore(
         db,
         embedder,
         limit: limit * VECTOR_SEARCH_OVERFETCH,
+        ...(maxVectorDistance !== undefined
+          ? { maxDistance: maxVectorDistance }
+          : {}),
         nowMs,
         query: input.query,
         scopes,
@@ -1416,11 +1462,15 @@ export function createMemoryStore(
       });
       const terms = searchTerms(input.query);
       const lexicalCandidates = candidates
-        .map((memory) => ({ memory, score: searchScore(memory, terms) }))
+        .map((candidate) => ({
+          ...candidate,
+          score: searchScore(candidate.memory, terms),
+        }))
         .filter((item) => item.score > 0);
       return mergeSearchCandidates(
         [...vectorCandidates, ...lexicalCandidates],
         nowMs,
+        sourceChannelPrefix(runtimeContext),
       ).slice(0, limit);
     },