@sentry/junior-memory 0.80.1 → 0.81.0

package/src/tools.ts

@@ -1,4 +1,4 @@
-import { Type, type TSchema } from "@sinclair/typebox";
+import { Type, type Static, type TSchema } from "@sinclair/typebox";
 import { Value } from "@sinclair/typebox/value";
 import {
   getSourceKey,
@@ -13,6 +13,7 @@ import {
   type MemoryEmbeddingProvider,
   type MemoryDb,
   type MemoryRecord,
+  type MemorySupersessionDecider,
 } from "./store";
 import {
   parseCreateMemoryRequest,
@@ -54,6 +55,10 @@ export interface MemoryToolContext {
   userText?: string;
 }
 
+export interface MemoryCreateToolContext extends MemoryToolContext {
+  supersessionDecider?: MemorySupersessionDecider;
+}
+
 function throwToolInputError(message: string): never {
   throw new PluginToolInputError(message);
 }
@@ -83,9 +88,15 @@ function memoryRuntimeContext(
   });
 }
 
-function memoryStore(context: MemoryToolContext) {
+function memoryStore(
+  context: MemoryToolContext,
+  options: { supersessionDecider?: MemorySupersessionDecider } = {},
+) {
   return createMemoryStore(context.db, memoryRuntimeContext(context), {
     embedder: context.embedder,
+    ...(options.supersessionDecider
+      ? { supersessionDecider: options.supersessionDecider }
+      : {}),
   });
 }
 
@@ -280,6 +291,18 @@ const searchMemoriesInputSchema = Type.Object(
   { additionalProperties: false },
 );
 
+const memoryToolProjectionSchema = Type.Object(
+  {
+    id: Type.String({ minLength: 1 }),
+    content: Type.String({ minLength: 1 }),
+    createdAtMs: Type.Number(),
+    observedAtMs: Type.Number(),
+    expiresAtMs: Type.Optional(Type.Number()),
+  },
+  { additionalProperties: false },
+);
+type MemoryToolProjection = Static<typeof memoryToolProjectionSchema>;
+
 function parseToolInput<T>(schema: TSchema, input: unknown): T {
   try {
     if (!Value.Check(schema, input)) {
@@ -324,19 +347,20 @@ function targetForKind(kind: MemoryKind): "requester" | "conversation" {
 }
 
 /** Return the model-visible projection without hidden ownership/source fields. */
-function compactMemory(memory: MemoryRecord) {
-  return {
+function compactMemory(memory: MemoryRecord): MemoryToolProjection {
+  return Value.Parse(memoryToolProjectionSchema, {
     id: memory.id,
     content: memory.content,
     createdAtMs: memory.createdAtMs,
+    observedAtMs: memory.observedAtMs,
     ...(memory.expiresAtMs !== undefined
       ? { expiresAtMs: memory.expiresAtMs }
       : {}),
-  };
+  });
 }
 
 /** Create a tool that submits an explicit memory candidate for storage. */
-export function createMemoryCreateTool(context: MemoryToolContext) {
+export function createMemoryCreateTool(context: MemoryCreateToolContext) {
   return {
     description:
       "Explicit memory-write tool. Use only when the latest user message directly asks Junior to remember, store, save, or forget-and-replace a public/shareable fact. Do not use for ordinary statements like 'I prefer X', 'I use Y', or 'X goes before Y' unless the user also asks you to remember/store/save it; passive memory learning handles those after the visible reply. Pass one self-contained natural-language candidate preserving the user's explicit memory intent. Do not ask the user to rephrase ordinary first-person facts, and do not rewrite them into display-name or third-person wording. Do not include secrets, private personal details, medical/legal/financial/sensitive facts, or another person's personal preference, opinion, habit, identity, relationship, workflow, or private life. Runtime context derives actor, scope, source, and subject ids; the memory agent decides canonical stored content and memory kind, then the plugin derives storage target from kind.",
@@ -350,7 +374,9 @@ export function createMemoryCreateTool(context: MemoryToolContext) {
       const toolCallId = requireToolCallId(options.toolCallId);
       const requestedExpiresAtMs = parseExpiresAt(parsedInput.expires_at);
       const runtimeContext = memoryRuntimeContext(context);
-      const store = memoryStore(context);
+      const store = memoryStore(context, {
+        supersessionDecider: context.supersessionDecider,
+      });
       const review = await (async () => {
         try {
           return parseMemoryReview(