stitchkit 0.64.0 → 0.65.1

package/dist/agent-runtime.js

@@ -28,8 +28,9 @@ import {
   AgentToolCallPartSchema,
   AgentToolResultPartSchema,
   AgentUsageSchema,
-  AgentUsageValueSchema
-} from "./index-b1k33127.js";
+  AgentUsageValueSchema,
+  runStateForTerminalReason
+} from "./index-sbkyvacf.js";
 import"./index-6djpbnda.js";
 import"./index-cby4ar3v.js";
 import {
@@ -395,7 +396,7 @@ function advanceAgentRuntimeEventCursor(rawCursor, event) {
       return { status: "duplicate", cursor };
     }
     return {
-      status: previous !== undefined && event.snapshotVersion > previous + 1 ? "gap" : "accepted",
+      status: "accepted",
       cursor: {
         ...cursor,
         snapshotVersion: event.snapshotVersion,
@@ -561,13 +562,7 @@ function interruptedSystemNote(message) {
     return { rendered };
   rendered.add("text");
   rendered.add("control");
-  return {
-    message: modelMessageSchema.parse({
-      role: "system",
-      content: `[interrupted] partial response: ${text}`
-    }),
-    rendered
-  };
+  return { text: `[interrupted] partial response: ${text}`, rendered };
 }
 function completeToolChronology(message) {
   const calls = new Set(message.parts.filter((part) => part.type === "tool-call").map((part) => part.callId));
@@ -576,12 +571,13 @@ function completeToolChronology(message) {
 }
 async function projectAgentHistoryDetailed(messages, options = {}) {
   const projected = [];
+  const system = [];
   const decisions = [];
   const interruptedRule = options.interruptedAssistant ?? "assistant-marked";
   let observedUser = false;
   for (const message of messages) {
-    if (message.status === "superseded") {
-      decisions.push(decide(message, "omitted", "superseded"));
+    if (message.role === "assistant" && !isSpeakableAssistantStatus(message.status)) {
+      decisions.push(decide(message, "omitted", message.status === "superseded" ? "superseded" : "draft-or-failed"));
       continue;
     }
     if (message.status === "streaming" || message.status === "failed") {
@@ -602,7 +598,7 @@ async function projectAgentHistoryDetailed(messages, options = {}) {
     if (message.role === "system" || message.role === "summary") {
       const content = textContent(message.parts);
       if (content) {
-        projected.push(modelMessageSchema.parse({ role: "system", content }));
+        system.push(content);
         decisions.push(decide(message, "projected", "projected", new Set(["text"])));
       } else {
         decisions.push(decide(message, "omitted", "empty"));
@@ -623,8 +619,8 @@ async function projectAgentHistoryDetailed(messages, options = {}) {
     }
     if (interrupted && interruptedRule === "system-note") {
       const note = interruptedSystemNote(message);
-      if (note.message) {
-        projected.push(note.message);
+      if (note.text) {
+        system.push(note.text);
         decisions.push(decide(message, "projected", "projected", note.rendered));
       } else {
         decisions.push(decide(message, "omitted", "empty"));
@@ -642,7 +638,7 @@ async function projectAgentHistoryDetailed(messages, options = {}) {
     projected.push(...assistant.messages);
     decisions.push(assistant.messages.length > 0 ? decide(message, "projected", "projected", assistant.rendered) : decide(message, "omitted", "empty"));
   }
-  return { messages: projected, decisions };
+  return { messages: projected, decisions, system };
 }
 async function projectAgentHistory(messages, options = {}) {
   return [...(await projectAgentHistoryDetailed(messages, options)).messages];
@@ -882,7 +878,7 @@ async function selectAgentHistory(options) {
   if (!Number.isSafeInteger(keepRecentTurns) || keepRecentTurns < 0) {
     throw new TypeError("keepRecentTurns must be a non-negative safe integer");
   }
-  const spoken = options.messages.filter((message) => message.status !== "superseded");
+  const spoken = options.messages.filter((message) => message.role !== "assistant" || isSpeakableAssistantStatus(message.status));
   const counts = new Map;
   let total = 0;
   let estimated = false;
@@ -895,8 +891,8 @@ async function selectAgentHistory(options) {
         messages: [...spoken],
         decisions: options.messages.map((candidate) => ({
           messageId: candidate.id,
-          action: candidate.status === "superseded" ? "removed" : "kept",
-          reason: candidate.status === "superseded" ? "superseded" : "token-count-unavailable",
+          action: candidate.role === "assistant" && !isSpeakableAssistantStatus(candidate.status) ? "removed" : "kept",
+          reason: candidate.role === "assistant" && !isSpeakableAssistantStatus(candidate.status) ? "unspeakable" : "token-count-unavailable",
           tokens: counts.get(candidate.id) ?? { provenance: "unavailable" }
         })),
         totalTokens: { provenance: "unavailable" },
@@ -922,11 +918,11 @@ async function selectAgentHistory(options) {
   }
   const messages = spoken.filter((message) => !removed.has(message.id));
   const decisions = options.messages.map((message) => {
-    if (message.status === "superseded") {
+    if (message.role === "assistant" && !isSpeakableAssistantStatus(message.status)) {
       return {
         messageId: message.id,
         action: "removed",
-        reason: "superseded",
+        reason: "unspeakable",
         tokens: { provenance: "unavailable" }
       };
     }
@@ -1126,6 +1122,7 @@ function interruptedCandidate(candidate, run, now) {
     AgentMessagePartSchema.parse({ type: "control", reason: "run-interrupted" })
   ];
   return {
+    ...candidate.usage && { usage: candidate.usage },
     run,
     assistant: AgentMessageSchema.parse({
       ...candidate.assistant,
@@ -1394,6 +1391,7 @@ function createRunExecutor(dependencies) {
     let terminalReason = "success";
     let usage = input.acceptedRun.usage;
     let sawProviderFinish = false;
+    let contextRefusal = false;
     let step = 0;
     let selectedModel;
     let internalCause;
@@ -1510,56 +1508,38 @@ function createRunExecutor(dependencies) {
         config.tools(runtimeContext)
       ]);
       if (prompt.contextDecision === "oversized") {
+        contextRefusal = true;
         throw new Error("Agent context exceeds the configured model budget");
       }
       if (prompt.contextDecision === "requires-compaction") {
+        contextRefusal = true;
         throw new Error("Agent context still exceeds the model budget after compaction");
       }
-      const projectHistory = (source) => config.history?.project ? config.history.project(source.messages) : projectAgentHistory(source.messages, {
-        ...config.history?.resolveFile && { resolveFile: config.history.resolveFile },
-        ...config.history?.unresolvedFile && {
-          unresolvedFile: config.history.unresolvedFile
-        },
-        ...config.history?.interruptedAssistant && {
-          interruptedAssistant: config.history.interruptedAssistant
-        }
-      });
-      const history = await projectHistory(snapshot);
-      const absorbPending = async () => {
-        if (!input.absorbable?.size || executionSignal.aborted)
-          return;
-        const latest = await config.store.loadSnapshot(run.conversationId);
-        const current = latest.runs.find((candidate) => candidate.id === run.id);
-        if (current?.state !== "running" || current.ownerId !== runtimeEpoch) {
-          return;
-        }
-        const pending = latest.runs.find((candidate) => candidate.state === "queued" && input.absorbable?.has(candidate.id));
-        if (!pending)
-          return;
-        const absorbed = await config.store.absorbQueuedRun({
-          conversationId: run.conversationId,
-          runningRunId: current.id,
-          runningExpectedRevision: current.revision,
-          ownerId: runtimeEpoch,
-          ...current.fencingToken !== undefined && { fencingToken: current.fencingToken },
-          queuedRunId: pending.id,
-          queuedExpectedRevision: pending.revision
-        });
-        if (absorbed.outcome !== "applied")
-          return;
-        input.onAbsorbed?.(pending.id);
-        snapshot = absorbed.snapshot;
-        run = findRun(snapshot.runs, run.id);
-        await publish({
-          type: "run-state",
-          eventId: agentDurableEventId("run-state", run.id, snapshot.version),
-          conversationId: run.conversationId,
-          runId: run.id,
-          snapshotVersion: snapshot.version,
-          state: run.state,
-          emittedAt: now().toISOString()
+      let carriedSystem = [];
+      const projectHistory = async (source) => {
+        if (config.history?.project)
+          return config.history.project(source.messages);
+        const detailed = await projectAgentHistoryDetailed(source.messages, {
+          ...config.history?.resolveFile && { resolveFile: config.history.resolveFile },
+          ...config.history?.unresolvedFile && {
+            unresolvedFile: config.history.unresolvedFile
+          },
+          ...config.history?.interruptedAssistant && {
+            interruptedAssistant: config.history.interruptedAssistant
+          }
         });
-        return [...await projectHistory(snapshot)];
+        carriedSystem = detailed.system;
+        return [...detailed.messages];
+      };
+      const history = await projectHistory(snapshot);
+      const withCarriedSystem = (instructions) => {
+        if (carriedSystem.length === 0)
+          return instructions;
+        const composed = typeof instructions === "string" ? [{ role: "system", content: instructions }] : Array.isArray(instructions) ? instructions : [instructions];
+        return [
+          ...composed,
+          ...carriedSystem.map((content) => ({ role: "system", content }))
+        ];
       };
       const maxStepCondition = stepCountIs(maxSteps);
       const stopConditions = [
@@ -1581,15 +1561,13 @@ function createRunExecutor(dependencies) {
       const result = streamText({
         model: selectedModel.model,
         tools,
-        instructions: prompt.instructions,
+        instructions: withCarriedSystem(prompt.instructions),
         messages: history,
         abortSignal: executionSignal,
         maxRetries: 0,
         stopWhen: stopConditions,
-        prepareStep: async (options) => {
-          const absorbed = await absorbPending();
-          const prepared = await config.loop?.prepareStep?.({ ...options, ...runtimeContext }) ?? {};
-          return absorbed && !prepared.messages ? { ...prepared, messages: absorbed } : prepared;
+        ...config.loop?.prepareStep && {
+          prepareStep: (options) => config.loop?.prepareStep?.({ ...options, ...runtimeContext })
         }
       });
       for await (const part of result.stream) {
@@ -1854,7 +1832,7 @@ function createRunExecutor(dependencies) {
           reason: isToolExecutionControlError(error) && error.reason === "stale_run" ? "stale-run" : "run-interrupted"
         }));
       } else {
-        terminalReason = "provider_failure";
+        terminalReason = contextRefusal ? "context_overflow" : "provider_failure";
       }
     } finally {
       idleDeadline.dispose();
@@ -1963,7 +1941,8 @@ function createAgentRuntime(config) {
   const waitForAdmissionAcceptances = admissionLanes.waitForAcceptances;
   const checkpointEveryEvents = config.loop?.checkpointEveryEvents ?? 20;
   const maxSteps = config.loop?.maxSteps ?? 50;
-  const idleTimeoutMs = config.loop?.idleTimeoutMs;
+  const declaredIdleTimeoutMs = config.loop?.idleTimeoutMs;
+  const idleTimeoutMs = declaredIdleTimeoutMs === null ? undefined : declaredIdleTimeoutMs ?? 60000;
   if (!Number.isSafeInteger(checkpointEveryEvents) || checkpointEveryEvents < 1) {
     throw new TypeError("checkpointEveryEvents must be a positive safe integer");
   }
@@ -1973,9 +1952,6 @@ function createAgentRuntime(config) {
   if (idleTimeoutMs !== undefined && (!Number.isSafeInteger(idleTimeoutMs) || idleTimeoutMs < 1)) {
     throw new TypeError("idleTimeoutMs must be a positive safe integer");
   }
-  const absorbable = new Set;
-  const absorbedInto = new Map;
-  const runResults = new Map;
   const policyNames = new Set(["max-steps"]);
   for (const policy of config.loop?.stopPolicies ?? []) {
     if (!policy.name || policyNames.has(policy.name)) {
@@ -2108,8 +2084,7 @@ function createAgentRuntime(config) {
       if (existingTicket)
         return existingTicket;
       const key = config.runs?.key?.(input) ?? input.conversationId;
-      const declaredPolicy = typeof config.runs?.inputPolicy === "function" ? config.runs.inputPolicy(input) : config.runs?.inputPolicy ?? "queue";
-      const policy = declaredPolicy === "inject" ? "queue" : declaredPolicy;
+      const policy = typeof config.runs?.inputPolicy === "function" ? config.runs.inputPolicy(input) : config.runs?.inputPolicy ?? "queue";
       const nowIso = now().toISOString();
       const inputMessageId = rawInput.recordIds?.inputMessageId ?? generateId();
       const runId = rawInput.recordIds?.runId ?? generateId();
@@ -2235,8 +2210,6 @@ function createAgentRuntime(config) {
             state: acceptedRun.state,
             emittedAt: now().toISOString()
           });
-          if (declaredPolicy === "inject")
-            absorbable.add(acceptedRun.id);
           outerAccepted.resolve();
           if (acceptance.outcome === "duplicate") {
             if (!acceptedRun.terminalReason) {
@@ -2285,25 +2258,7 @@ function createAgentRuntime(config) {
                 await waitForAdmissionAcceptances(reservation.lane);
               return {
                 runId: acceptedRun.id,
-                execute: () => {
-                  const answeredBy = absorbedInto.get(acceptedRun.id);
-                  const answer = answeredBy ? runResults.get(answeredBy) : undefined;
-                  if (answer)
-                    return answer;
-                  const running = executeRun({
-                    acceptedRun,
-                    context,
-                    signal,
-                    absorbable,
-                    onAbsorbed: (absorbedRunId) => {
-                      absorbedInto.set(absorbedRunId, acceptedRun.id);
-                    }
-                  });
-                  runResults.set(acceptedRun.id, running);
-                  return running.finally(() => {
-                    absorbable.delete(acceptedRun.id);
-                  });
-                }
+                execute: () => executeRun({ acceptedRun, context, signal })
               };
             }
           });
@@ -2550,15 +2505,6 @@ var CommitRunTerminalSchema = z6.object({
   policyName: z6.string().min(1).optional(),
   usage: AgentUsageSchema.optional()
 });
-var AbsorbQueuedRunSchema = z6.object({
-  conversationId: AgentRecordIdSchema,
-  runningRunId: AgentRecordIdSchema,
-  runningExpectedRevision: AgentRecordVersionSchema,
-  ownerId: z6.string().min(1),
-  fencingToken: AgentRecordVersionSchema.optional(),
-  queuedRunId: AgentRecordIdSchema,
-  queuedExpectedRevision: AgentRecordVersionSchema
-});
 var RequestRunInterruptSchema = z6.object({
   conversationId: AgentRecordIdSchema,
   runId: AgentRecordIdSchema,
@@ -2699,24 +2645,17 @@ function replaceRun(runs, next) {
 function replaceMessage(messages, next) {
   return messages.some((message) => message.id === next.id) ? messages.map((message) => message.id === next.id ? next : message) : [...messages, next];
 }
+var ACTIVE_AGENT_RUN_STATES = [
+  "queued",
+  "running",
+  "interrupt_requested"
+];
+function isActiveRunState(state) {
+  return ACTIVE_AGENT_RUN_STATES.includes(state);
+}
 function conflict(actualVersion) {
   return { outcome: "conflict", actualVersion };
 }
-function terminalState(reason) {
-  if (reason === "success" || reason === "policy_stop" || reason === "provider_stop") {
-    return "completed";
-  }
-  if (reason === "interrupted")
-    return "interrupted";
-  if (reason === "superseded")
-    return "superseded";
-  if (reason === "cancelled" || reason === "shutdown" || reason === "timeout") {
-    return "cancelled";
-  }
-  if (reason === "abandoned")
-    return "abandoned";
-  return "failed";
-}
 function applied(current, input, effects) {
   return {
     outcome: "applied",
@@ -2727,7 +2666,6 @@ function applied(current, input, effects) {
       messages: input.messages ?? current.messages
     }),
     ...effects?.runRecord && { runRecord: effects.runRecord },
-    ...effects?.secondaryRunRecord && { secondaryRunRecord: effects.secondaryRunRecord },
     ...effects?.admissionReceipt && { admissionReceipt: effects.admissionReceipt },
     ...effects?.historyMutation && { historyMutation: effects.historyMutation }
   };
@@ -2771,39 +2709,6 @@ function reduceStore(current, operation) {
       historyMutation: { type: "admit", input: input.input }
     });
   }
-  if (operation.type === "absorb") {
-    const input = operation.input;
-    if (current.conversationId !== input.conversationId)
-      return { outcome: "not_found" };
-    const running = current.runs.find((candidate) => candidate.id === input.runningRunId);
-    const queued = current.runs.find((candidate) => candidate.id === input.queuedRunId);
-    if (!running || !queued)
-      return { outcome: "not_found" };
-    if (running.revision !== input.runningExpectedRevision || running.state !== "running" || running.ownerId !== input.ownerId || input.fencingToken !== undefined && running.fencingToken !== input.fencingToken) {
-      return conflict(running.revision);
-    }
-    if (queued.revision !== input.queuedExpectedRevision || queued.state !== "queued" || queued.ownerId !== undefined || queued.terminalReason !== undefined || queued.id === running.id) {
-      return conflict(queued.revision);
-    }
-    const stamp = new Date().toISOString();
-    const grown = AgentRunSchema.parse({
-      ...running,
-      inputMessageIds: [...running.inputMessageIds, ...queued.inputMessageIds],
-      revision: running.revision + 1,
-      updatedAt: stamp
-    });
-    const emptied = AgentRunSchema.parse({
-      ...queued,
-      state: "absorbed",
-      absorbedIntoRunId: running.id,
-      revision: queued.revision + 1,
-      updatedAt: stamp
-    });
-    return applied(current, { runs: replaceRun(replaceRun(current.runs, grown), emptied) }, {
-      runRecord: AgentStoredRunSchema.parse({ schemaVersion: 1, run: grown }),
-      secondaryRunRecord: AgentStoredRunSchema.parse({ schemaVersion: 1, run: emptied })
-    });
-  }
   const conversationId = operation.input.conversationId;
   const run = operation.type === "compact" ? undefined : current.runs.find((candidate) => candidate.id === operation.input.runId);
   if (operation.type !== "compact" && !run)
@@ -2861,22 +2766,18 @@ function reduceStore(current, operation) {
   }
   if (operation.type === "recover" && run) {
     const input = operation.input;
-    if (run.revision !== input.expectedRevision || !["queued", "running", "interrupt_requested"].includes(run.state)) {
+    if (run.revision !== input.expectedRevision || !isActiveRunState(run.state)) {
       return conflict(run.revision);
     }
     if (input.action === "requeue" && run.state !== "queued" && input.replaySafe !== true) {
       throw new TypeError("Recovering an acquired run requires explicit replaySafe evidence");
     }
+    const { ownerId: _released, ...carried } = run;
     const next = AgentRunSchema.parse({
-      schemaVersion: 1,
-      id: run.id,
-      conversationId: run.conversationId,
-      inputMessageIds: run.inputMessageIds,
-      assistantMessageId: run.assistantMessageId,
+      ...carried,
       state: input.action === "requeue" ? "queued" : "abandoned",
       revision: run.revision + 1,
       ...input.action === "abandon" && { terminalReason: "abandoned" },
-      createdAt: run.createdAt,
       updatedAt: new Date().toISOString()
     });
     if (input.action === "abandon") {
@@ -2917,7 +2818,7 @@ function reduceStore(current, operation) {
     }
     const next = AgentRunSchema.parse({
       ...run,
-      state: terminalState(input.reason),
+      state: runStateForTerminalReason(input.reason),
       terminalReason: input.reason,
       ...input.policyName && { terminalPolicyName: input.policyName },
       ...input.usage && { usage: input.usage },
@@ -2946,6 +2847,10 @@ function reduceStore(current, operation) {
     if (!input.replacedMessageIds.every((id) => current.messages.some((m) => m.id === id))) {
       return { outcome: "not_found" };
     }
+    const liveAssistant = current.runs.find((candidate) => isActiveRunState(candidate.state) && replaced.has(candidate.assistantMessageId));
+    if (liveAssistant) {
+      throw new TypeError(`Compaction may not replace the assistant message of run ${liveAssistant.id}, which has not finished`);
+    }
     const positions = current.messages.map((message, index) => replaced.has(message.id) ? index : undefined).filter((index) => index !== undefined);
     const first = positions[0];
     if (first === undefined || positions.some((position, offset) => position !== first + offset) || input.summary.conversationId !== input.conversationId || input.summary.runId !== undefined || input.summary.role !== "summary" || input.summary.status !== "committed" || current.messages.some((message) => message.id === input.summary.id) || current.runs.some((candidate) => candidate.assistantMessageId === input.summary.id)) {
@@ -3009,7 +2914,7 @@ function createAgentRuntimeStore(driver) {
   });
   const mutate = (operation) => driver.transaction(async (transaction) => {
     const conversationId = operationConversationId(operation);
-    const operationRunId = operation.type === "accept" ? operation.input.coalesceIntoRunId : operation.type === "compact" ? undefined : operation.type === "absorb" ? operation.input.queuedRunId : operation.input.runId;
+    const operationRunId = operation.type === "accept" ? operation.input.coalesceIntoRunId : operation.type === "compact" ? undefined : operation.input.runId;
     const [stored, messages, activeRecords, operationRecord, duplicateReceipt] = await Promise.all([
       driver.head.load(transaction, conversationId),
       driver.history.load(transaction, conversationId),
@@ -3090,9 +2995,6 @@ function createAgentRuntimeStore(driver) {
       return conflict(outcome.actualVersion);
     if (reduced.runRecord)
       await driver.runs.save(transaction, reduced.runRecord);
-    if (reduced.secondaryRunRecord) {
-      await driver.runs.save(transaction, reduced.secondaryRunRecord);
-    }
     if (reduced.admissionReceipt) {
       await driver.admissions.create(transaction, reduced.admissionReceipt);
     }
@@ -3116,7 +3018,6 @@ function createAgentRuntimeStore(driver) {
       type: "interrupt",
       input: RequestRunInterruptSchema.parse(input)
     }),
-    absorbQueuedRun: (input) => mutate({ type: "absorb", input: AbsorbQueuedRunSchema.parse(input) }),
     recoverRun: (input) => mutate({ type: "recover", input: RecoverAgentRunSchema.parse(input) }),
     commitRunTerminal: (input) => mutate({ type: "terminal", input: CommitRunTerminalSchema.parse(input) }),
     replaceCompactedRange: (input) => mutate({
@@ -3212,7 +3113,7 @@ function createMemoryAgentRuntimeStore() {
         });
       },
       async listActive(transaction, conversationId) {
-        return [...transaction.runs.get(conversationId)?.values() ?? []].filter((record) => ["queued", "running", "interrupt_requested"].includes(record.run.state)).map((record) => AgentStoredRunSchema.parse(structuredClone(record)));
+        return [...transaction.runs.get(conversationId)?.values() ?? []].filter((record) => isActiveRunState(record.run.state)).map((record) => AgentStoredRunSchema.parse(structuredClone(record)));
       },
       async save(transaction, rawRecord) {
         const record = AgentStoredRunSchema.parse(structuredClone(rawRecord));
@@ -3274,9 +3175,12 @@ function createMemoryAgentRuntimeStore() {
       }
     },
     async scanRecoverable(input) {
-      const descriptors = [...runs].flatMap(([conversationId, conversationRuns]) => [...conversationRuns.values()].filter((record) => ["queued", "running", "interrupt_requested"].includes(record.run.state)).map((record) => ({ conversationId, run: record.run }))).sort((left, right) => left.conversationId.localeCompare(right.conversationId) || left.run.id.localeCompare(right.run.id));
+      const descriptors = [...runs].flatMap(([conversationId, conversationRuns]) => [...conversationRuns.values()].filter((record) => isActiveRunState(record.run.state)).map((record) => ({ conversationId, run: record.run }))).sort((left, right) => left.conversationId.localeCompare(right.conversationId) || left.run.id.localeCompare(right.run.id));
       const cursorTuple = input.cursor ? parseRecoverableCursor(input.cursor) : undefined;
-      const start = cursorTuple ? descriptors.findIndex((item) => item.conversationId === cursorTuple[0] && item.run.id === cursorTuple[1]) + 1 : 0;
+      const start = cursorTuple ? descriptors.findIndex((item) => item.conversationId.localeCompare(cursorTuple[0]) > 0 || item.conversationId === cursorTuple[0] && item.run.id.localeCompare(cursorTuple[1]) > 0) : 0;
+      if (start === -1) {
+        return AgentRecoverablePageSchema.parse({ items: [] });
+      }
       const items = descriptors.slice(start, start + input.limit);
       const last = items.at(-1);
       const hasMore = start + items.length < descriptors.length;
@@ -3289,7 +3193,7 @@ function createMemoryAgentRuntimeStore() {
   return createAgentRuntimeStore(driver);
 }
 export {
-  AbsorbQueuedRunSchema,
+  ACTIVE_AGENT_RUN_STATES,
   AcceptInputAndAssignRunSchema,
   AcquireAgentRunSchema,
   AgentAdmissionEventSchema,
@@ -3368,6 +3272,7 @@ export {
   defineModelRegistry,
   projectAgentHistory,
   projectAgentHistoryDetailed,
+  runStateForTerminalReason,
   selectAgentHistory,
   structuredCompaction,
   validateAgentModelSnapshot

package/dist/{index-b1k33127.js → index-sbkyvacf.js}

@@ -99,7 +99,6 @@ var AgentRunStateSchema = z.enum([
   "completed",
   "interrupted",
   "superseded",
-  "absorbed",
   "failed",
   "cancelled",
   "abandoned"
@@ -114,9 +113,24 @@ var AgentTerminalReasonSchema = z.enum([
   "timeout",
   "shutdown",
   "provider_failure",
-  "tool_failure",
+  "context_overflow",
   "abandoned"
 ]);
+function runStateForTerminalReason(reason) {
+  if (reason === "success" || reason === "policy_stop" || reason === "provider_stop") {
+    return "completed";
+  }
+  if (reason === "interrupted")
+    return "interrupted";
+  if (reason === "superseded")
+    return "superseded";
+  if (reason === "cancelled" || reason === "shutdown" || reason === "timeout") {
+    return "cancelled";
+  }
+  if (reason === "abandoned")
+    return "abandoned";
+  return "failed";
+}
 var AgentUsageValueSchema = z.object({
   value: z.number().nonnegative().optional(),
   provenance: z.enum(["provider-reported", "computed", "estimated", "unavailable"])
@@ -134,7 +148,7 @@ var AgentUsageSchema = z.object({
   cacheWriteTokens: AgentUsageValueSchema.optional(),
   cost: AgentCostValueSchema.optional()
 });
-var AgentRunSchema = z.object({
+var AgentRunFieldsSchema = z.object({
   schemaVersion: z.literal(1),
   id: AgentRecordIdSchema,
   conversationId: AgentRecordIdSchema,
@@ -146,11 +160,45 @@ var AgentRunSchema = z.object({
   fencingToken: AgentRecordVersionSchema.optional(),
   terminalReason: AgentTerminalReasonSchema.optional(),
   terminalPolicyName: z.string().min(1).optional(),
-  absorbedIntoRunId: AgentRecordIdSchema.optional(),
   usage: AgentUsageSchema.optional(),
   createdAt: AgentTimestampSchema,
   updatedAt: AgentTimestampSchema
 });
+var AgentRunSchema = AgentRunFieldsSchema.superRefine((run, ctx) => {
+  if (run.terminalReason === undefined) {
+    if (TERMINAL_RUN_STATES.has(run.state)) {
+      ctx.addIssue({
+        code: "custom",
+        path: ["terminalReason"],
+        message: `A run in terminal state "${run.state}" must say why it ended`
+      });
+    }
+    return;
+  }
+  const expected = runStateForTerminalReason(run.terminalReason);
+  if (run.state !== expected) {
+    ctx.addIssue({
+      code: "custom",
+      path: ["state"],
+      message: `Terminal reason "${run.terminalReason}" ends a run in state "${expected}", not "${run.state}"`
+    });
+  }
+  if (run.terminalReason === "policy_stop" && run.terminalPolicyName === undefined) {
+    ctx.addIssue({
+      code: "custom",
+      path: ["terminalPolicyName"],
+      message: "A policy stop names the policy that stopped the run"
+    });
+  }
+});
+var TERMINAL_RUN_STATES = new Set([
+  "completed",
+  "interrupted",
+  "superseded",
+  "cancelled",
+  "abandoned",
+  "failed"
+]);
 var AgentSnapshotSchema = z.object({
   schemaVersion: z.literal(1),
   conversationId: AgentRecordIdSchema,
@@ -160,9 +208,9 @@ var AgentSnapshotSchema = z.object({
 });
 var AgentRunMetricsSchema = z.object({
   partial: z.boolean(),
-  usage: AgentUsageSchema.optional(),
+  usage: AgentUsageSchema,
   durationMs: z.number().nonnegative().optional(),
   ttftMs: z.number().nonnegative().optional()
 });
 
-export { AgentRecordIdSchema, AgentRecordVersionSchema, AgentTimestampSchema, AgentJsonObjectSchema, AgentProviderEnvelopeSchema, AgentTextPartSchema, AgentReasoningPartSchema, AgentFilePartSchema, AgentSourcePartSchema, AgentToolCallPartSchema, AgentToolResultPartSchema, AgentOpaquePartSchema, AgentControlPartSchema, AgentMessagePartSchema, AgentMessageRoleSchema, AgentMessageStatusSchema, AgentMessageSchema, AgentAssistantPlaceholderSchema, AgentRunStateSchema, AgentTerminalReasonSchema, AgentUsageValueSchema, AgentCostValueSchema, AgentUsageSchema, AgentRunSchema, AgentSnapshotSchema, AgentRunMetricsSchema };
+export { AgentRecordIdSchema, AgentRecordVersionSchema, AgentTimestampSchema, AgentJsonObjectSchema, AgentProviderEnvelopeSchema, AgentTextPartSchema, AgentReasoningPartSchema, AgentFilePartSchema, AgentSourcePartSchema, AgentToolCallPartSchema, AgentToolResultPartSchema, AgentOpaquePartSchema, AgentControlPartSchema, AgentMessagePartSchema, AgentMessageRoleSchema, AgentMessageStatusSchema, AgentMessageSchema, AgentAssistantPlaceholderSchema, AgentRunStateSchema, AgentTerminalReasonSchema, runStateForTerminalReason, AgentUsageValueSchema, AgentCostValueSchema, AgentUsageSchema, AgentRunSchema, AgentSnapshotSchema, AgentRunMetricsSchema };

package/dist/testing/agent-store-conformance.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-store-conformance.d.ts","sourceRoot":"","sources":["../../src/testing/agent-store-conformance.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE,MAAM,WAAW,2BAA2B;IAC1C,WAAW,IAAI,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;CAC/D;AAwCD,gFAAgF;AAChF,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,2BAA2B,GAClC,OAAO,CAAC,IAAI,CAAC,CA4Sf"}
+{"version":3,"file":"agent-store-conformance.d.ts","sourceRoot":"","sources":["../../src/testing/agent-store-conformance.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE,MAAM,WAAW,2BAA2B;IAC1C,WAAW,IAAI,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;CAC/D;AAwCD,gFAAgF;AAChF,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,2BAA2B,GAClC,OAAO,CAAC,IAAI,CAAC,CAiXf"}