@sentry/junior-memory 0.89.0 → 0.90.0

package/dist/index.js

@@ -667,9 +667,12 @@ function createMemoryCliCommand() {
 import { Type } from "@sinclair/typebox";
 import { Value } from "@sinclair/typebox/value";
 import {
+  definePluginTool,
   getSourceKey,
-  PluginToolInputError
+  PluginToolInputError,
+  pluginToolResultSchema
 } from "@sentry/junior-plugin-api";
+import { z as z4 } from "zod";
 
 // src/store.ts
 import { createHash, randomUUID } from "crypto";
@@ -1892,59 +1895,24 @@ function requireMemoryContent(value) {
   }
   return value;
 }
-var createMemoryInputSchema2 = Type.Object(
-  {
-    content: Type.String({
-      minLength: 1,
-      maxLength: MAX_TOOL_CONTENT_CHARS,
-      description: "Self-contained public/shareable memory candidate. Include the subject in natural language when it matters; do not rely on surrounding chat context."
-    }),
-    expires_at: Type.Optional(
-      Type.String({
-        minLength: 1,
-        description: 'Expiration selector. Omit or use "never" when the memory should not expire, or use an exact ISO timestamp such as "2027-06-21T00:00:00Z".'
-      })
-    )
-  },
-  { additionalProperties: false }
-);
-var removeMemoryInputSchema = Type.Object(
-  {
-    id: Type.String({
-      minLength: 1,
-      description: "Memory id or unambiguous short id prefix to remove."
-    })
-  },
-  { additionalProperties: false }
-);
-var listMemoriesInputSchema2 = Type.Object(
-  {
-    limit: Type.Optional(
-      Type.Number({
-        minimum: 1,
-        maximum: 50,
-        description: "Maximum number of visible memories to return."
-      })
-    )
-  },
-  { additionalProperties: false }
-);
-var searchMemoriesInputSchema2 = Type.Object(
-  {
-    query: Type.String({
-      minLength: 1,
-      description: "Search query for visible memory content."
-    }),
-    limit: Type.Optional(
-      Type.Number({
-        minimum: 1,
-        maximum: 50,
-        description: "Maximum number of matching memories to return."
-      })
-    )
-  },
-  { additionalProperties: false }
-);
+var createMemoryInputSchema2 = z4.object({
+  content: z4.string().min(1).max(MAX_TOOL_CONTENT_CHARS).describe(
+    "Self-contained public/shareable memory candidate. Include the subject in natural language when it matters; do not rely on surrounding chat context."
+  ),
+  expires_at: z4.string().min(1).describe(
+    'Expiration selector. Omit or use "never" when the memory should not expire, or use an exact ISO timestamp such as "2027-06-21T00:00:00Z".'
+  ).optional()
+}).strict();
+var removeMemoryInputSchema = z4.object({
+  id: z4.string().min(1).describe("Memory id or unambiguous short id prefix to remove.")
+}).strict();
+var listMemoriesInputSchema2 = z4.object({
+  limit: z4.number().min(1).max(50).describe("Maximum number of visible memories to return.").optional()
+}).strict();
+var searchMemoriesInputSchema2 = z4.object({
+  query: z4.string().min(1).describe("Search query for visible memory content."),
+  limit: z4.number().min(1).max(50).describe("Maximum number of matching memories to return.").optional()
+}).strict();
 var memoryToolProjectionSchema = Type.Object(
   {
     id: Type.String({ minLength: 1 }),
@@ -1955,17 +1923,53 @@ var memoryToolProjectionSchema = Type.Object(
   },
   { additionalProperties: false }
 );
-function parseToolInput(schema, input) {
-  try {
-    if (!Value.Check(schema, input)) {
-      throw new Error("Input does not match memory tool schema.");
-    }
-    return Value.Parse(schema, input);
-  } catch (error) {
+var memoryProjectionOutputSchema = z4.object({
+  id: z4.string(),
+  content: z4.string(),
+  createdAtMs: z4.number(),
+  observedAtMs: z4.number(),
+  expiresAtMs: z4.number().optional()
+});
+var memoryCreateDataOutputSchema = z4.object({
+  created: z4.boolean(),
+  memory: memoryProjectionOutputSchema
+});
+var memorySingleDataOutputSchema = z4.object({
+  memory: memoryProjectionOutputSchema
+});
+var memoryManyDataOutputSchema = z4.object({
+  memories: z4.array(memoryProjectionOutputSchema)
+});
+var memoryCreateOutputSchema = pluginToolResultSchema.extend({
+  ok: z4.literal(true),
+  status: z4.literal("success"),
+  target: z4.string(),
+  data: memoryCreateDataOutputSchema,
+  created: z4.boolean(),
+  memory: memoryProjectionOutputSchema
+});
+var memorySingleOutputSchema = pluginToolResultSchema.extend({
+  ok: z4.literal(true),
+  status: z4.literal("success"),
+  target: z4.string(),
+  data: memorySingleDataOutputSchema,
+  memory: memoryProjectionOutputSchema
+});
+var memoryManyOutputSchema = pluginToolResultSchema.extend({
+  ok: z4.literal(true),
+  status: z4.literal("success"),
+  target: z4.string(),
+  data: memoryManyDataOutputSchema,
+  memories: z4.array(memoryProjectionOutputSchema)
+});
+function parseMemoryToolInput(schema, input) {
+  const result = schema.safeParse(input);
+  if (!result.success) {
     throw new PluginToolInputError("Invalid memory tool input.", {
-      cause: error
+      cause: result.error
     });
   }
+  return result.data;
 }
 function sourceIdempotencyKey(context) {
   const sourceKey2 = getSourceKey(context.source);
@@ -1997,16 +2001,23 @@ function compactMemory(memory) {
     ...memory.expiresAtMs !== void 0 ? { expiresAtMs: memory.expiresAtMs } : {}
   });
 }
-function createMemoryCreateTool(context) {
+function memoryToolResult(target, data) {
   return {
+    ok: true,
+    status: "success",
+    target,
+    data,
+    ...data
+  };
+}
+function createMemoryCreateTool(context) {
+  return definePluginTool({
     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.",
     executionMode: "sequential",
     inputSchema: createMemoryInputSchema2,
+    outputSchema: memoryCreateOutputSchema,
     execute: async (input, options) => {
-      const parsedInput = parseToolInput(
-        createMemoryInputSchema2,
-        input
-      );
+      const parsedInput = parseMemoryToolInput(createMemoryInputSchema2, input);
       const toolCallId = requireToolCallId(options.toolCallId);
       const requestedExpiresAtMs = parseExpiresAt(parsedInput.expires_at);
       const runtimeContext = memoryRuntimeContext(context);
@@ -2064,24 +2075,21 @@ function createMemoryCreateTool(context) {
           asToolInputError(error);
         }
       })();
-      return {
-        ok: true,
+      return memoryToolResult("createMemory", {
         created: result.created,
         memory: compactMemory(result.memory)
-      };
+      });
     }
-  };
+  });
 }
 function createMemoryRemoveTool(context) {
-  return {
+  return definePluginTool({
     description: "Forget one memory visible in the active context. Use only ids or short id prefixes returned by listMemories or searchMemories. Never remove memories by hidden actor, Slack, scope, or subject identifiers.",
     executionMode: "sequential",
     inputSchema: removeMemoryInputSchema,
+    outputSchema: memorySingleOutputSchema,
     execute: async (input) => {
-      const parsedInput = parseToolInput(
-        removeMemoryInputSchema,
-        input
-      );
+      const parsedInput = parseMemoryToolInput(removeMemoryInputSchema, input);
       const memory = await (async () => {
         try {
           return await memoryStore(context).archiveMemory({
@@ -2092,40 +2100,37 @@ function createMemoryRemoveTool(context) {
           asToolInputError(error);
         }
       })();
-      return {
-        ok: true,
+      return memoryToolResult("removeMemory", {
         memory: compactMemory(memory)
-      };
+      });
     }
-  };
+  });
 }
 function createMemoryListTool(context) {
-  return {
+  return definePluginTool({
     description: "List active memories visible in the current context. Use when the user asks what Junior remembers or when memory ids are needed before removing a memory.",
     annotations: { readOnlyHint: true, destructiveHint: false },
     inputSchema: listMemoriesInputSchema2,
+    outputSchema: memoryManyOutputSchema,
     execute: async (input) => {
-      const parsedInput = parseToolInput(
-        listMemoriesInputSchema2,
-        input
-      );
+      const parsedInput = parseMemoryToolInput(listMemoriesInputSchema2, input);
       const memories = await memoryStore(context).listMemories({
         limit: boundedLimit2(parsedInput.limit, DEFAULT_RESULT_LIMIT)
       });
-      return {
-        ok: true,
+      return memoryToolResult("listMemories", {
         memories: memories.map(compactMemory)
-      };
+      });
     }
-  };
+  });
 }
 function createMemorySearchTool(context) {
-  return {
+  return definePluginTool({
     description: "Search active memories visible in the current context. Use when the model needs targeted memory recall. The tool searches only the current requester and active conversation scopes.",
     annotations: { readOnlyHint: true, destructiveHint: false },
     inputSchema: searchMemoriesInputSchema2,
+    outputSchema: memoryManyOutputSchema,
     execute: async (input) => {
-      const parsedInput = parseToolInput(
+      const parsedInput = parseMemoryToolInput(
         searchMemoriesInputSchema2,
         input
       );
@@ -2133,12 +2138,11 @@ function createMemorySearchTool(context) {
         query: parsedInput.query,
         limit: boundedLimit2(parsedInput.limit, DEFAULT_SEARCH_LIMIT2)
       });
-      return {
-        ok: true,
+      return memoryToolResult("searchMemories", {
         memories: memories.map(compactMemory)
-      };
+      });
     }
-  };
+  });
 }
 
 // src/process-session.ts
@@ -2147,7 +2151,7 @@ import {
   getSourceKey as getSourceKey2,
   isPrivateSource as isPrivateSource2
 } from "@sentry/junior-plugin-api";
-import { z as z4 } from "zod";
+import { z as z5 } from "zod";
 var MEMORY_TOOL_NAMES = /* @__PURE__ */ new Set([
   "createMemory",
   "listMemories",
@@ -2155,11 +2159,11 @@ var MEMORY_TOOL_NAMES = /* @__PURE__ */ new Set([
   "searchMemories"
 ]);
 var MEMORY_TASK_STATE_TTL_MS = 7 * 24 * 60 * 60 * 1e3;
-var extractedMemoryCacheSchema = z4.array(
-  z4.object({
-    content: z4.string().min(1),
-    expiresAtMs: z4.number().finite().nullable(),
-    kind: z4.enum(MEMORY_KINDS)
+var extractedMemoryCacheSchema = z5.array(
+  z5.object({
+    content: z5.string().min(1),
+    expiresAtMs: z5.number().finite().nullable(),
+    kind: z5.enum(MEMORY_KINDS)
   }).strict().transform(parseExtractedMemory)
 );
 function targetForKind2(kind) {