stitchkit 0.94.0 → 0.94.1

package/dist/entrypoints/agent-runtime.js

@@ -24,7 +24,7 @@ import {
   renderAgentStateSlots,
   repairedSearchCall,
   selectAgentHistory
-} from "../index-97ac4gph.js";
+} from "../index-7f68ghdn.js";
 import"../index-83bdg97s.js";
 import {
   AgentAdmissionEventSchema,
@@ -78,7 +78,7 @@ import {
 } from "../index-qmy6zq75.js";
 import {
   createMemoryAgentRuntimeStore
-} from "../index-xsqjkha9.js";
+} from "../index-sp1tzwnb.js";
 import {
   ACTIVE_AGENT_RUN_STATES,
   AgentAdmissionReceiptSchema,
@@ -91,7 +91,7 @@ import {
   AgentStoredRunSchema,
   createAgentRuntimeStore,
   purgeAgentConversation
-} from "../index-58k4fp34.js";
+} from "../index-wsm5w4ee.js";
 import {
   AcceptInputAndAssignRunSchema,
   AcquireAgentRunSchema,
@@ -123,7 +123,7 @@ import {
   encodeAgentConversationArchive,
   isAssistantHistoryEvidence,
   isCompleteAgentHistoryTurn
-} from "../index-1nd48tm9.js";
+} from "../index-gxrt5h6v.js";
 import {
   AgentAssistantPlaceholderSchema,
   AgentControlPartSchema,
@@ -164,7 +164,7 @@ import {
 import {
   buildToolManifest,
   mountAgent
-} from "../index-dcpdvjsc.js";
+} from "../index-863ea9jb.js";
 import {
   DURABILITY_STEP_EVENT_KIND,
   ParkAbortedError,
@@ -172,10 +172,10 @@ import {
   StepAbortedError,
   StepResultDecodeError,
   StepResultNotSerializableError
-} from "../index-cq4qfycf.js";
-import"../index-vcvkq30w.js";
-import"../index-2cr1cqhn.js";
-import"../index-j05z5gn4.js";
+} from "../index-79z7m7dt.js";
+import"../index-hmp4800j.js";
+import"../index-khtgb21r.js";
+import"../index-55hkhm67.js";
 import"../index-zfrhfn82.js";
 import"../index-35mh5bhj.js";
 import {
@@ -186,8 +186,8 @@ import"../index-1zbbzgfh.js";
 import {
   defineRuntimeTool,
   executableAgentRuntimeTools
-} from "../index-v38qyerb.js";
-import"../index-dydjcp8z.js";
+} from "../index-xw3s75nj.js";
+import"../index-np4m8hd3.js";
 import"../index-eqqvgdb3.js";
 import"../index-znfs20ss.js";
 import"../index-p5r7qpdd.js";
@@ -1973,44 +1973,192 @@ function createAgentScheduleTools(service) {
   ];
 }
 // src/agent-runtime/schedules.ts
+import { randomUUID as randomUUID3 } from "node:crypto";
+import { z as z10 } from "zod";
+
+// src/agent-runtime/schedule-records.ts
 import { randomUUID as randomUUID2 } from "node:crypto";
 import { z as z9 } from "zod";
-var AgentScheduleSchema = z9.object({
-  id: z9.string().min(1),
-  conversationId: z9.string().min(1),
-  kind: z9.enum(["at", "after", "every"]),
-  nextAt: z9.iso.datetime({ offset: true }),
-  intervalMs: z9.int().positive().optional(),
-  timeZone: z9.string().min(1).optional(),
-  input: z9.json(),
-  state: z9.enum(["scheduled", "cancelled", "completed"]),
-  occurrence: z9.int().nonnegative(),
-  createdAt: z9.iso.datetime({ offset: true }),
-  updatedAt: z9.iso.datetime({ offset: true })
+function newSchedule(request, observedAt) {
+  const modes = [
+    request.at !== undefined,
+    request.afterMs !== undefined,
+    request.everyMs !== undefined
+  ];
+  if (modes.filter(Boolean).length !== 1)
+    throw new TypeError("Declare exactly one schedule mode");
+  if (request.everyMs !== undefined && !request.timeZone) {
+    throw new TypeError("Repeating schedules require an explicit timeZone");
+  }
+  if (request.timeZone) {
+    try {
+      new Intl.DateTimeFormat("en", { timeZone: request.timeZone }).format(new Date(0));
+    } catch {
+      throw new TypeError(`Unknown schedule timeZone: ${request.timeZone}`);
+    }
+  }
+  const kind = request.at !== undefined ? "at" : request.afterMs !== undefined ? "after" : "every";
+  const intervalMs = request.everyMs;
+  const delay = request.afterMs ?? request.everyMs;
+  if (delay !== undefined && (!Number.isSafeInteger(delay) || delay < 1)) {
+    throw new TypeError("Schedule delay must be a positive safe integer");
+  }
+  const nextAt = request.at ? new Date(z9.iso.datetime({ offset: true }).parse(request.at)).toISOString() : new Date(observedAt.getTime() + (delay ?? 0)).toISOString();
+  const schedule = AgentScheduleSchema.parse({
+    id: randomUUID2(),
+    conversationId: request.conversationId,
+    kind,
+    nextAt,
+    ...intervalMs !== undefined && { intervalMs },
+    ...request.timeZone && { timeZone: request.timeZone },
+    input: z9.json().parse(request.input),
+    state: "scheduled",
+    occurrence: 0,
+    createdAt: observedAt.toISOString(),
+    updatedAt: observedAt.toISOString()
+  });
+  return schedule;
+}
+async function insertSchedule(sqlite, schedule) {
+  await sqlite.transaction(async (scope) => {
+    scope.database.prepare(`
+        INSERT INTO stitchkit_agent_runtime_schedules (
+          id, conversation_id, kind, next_at, interval_ms, time_zone,
+          input_payload, state, occurrence, created_at, updated_at
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, 'scheduled', 0, ?, ?)
+      `).run(schedule.id, schedule.conversationId, schedule.kind, schedule.nextAt, schedule.intervalMs ?? null, schedule.timeZone ?? null, JSON.stringify(schedule.input), schedule.createdAt, schedule.updatedAt);
+    await scope.appendEvent({
+      conversationId: schedule.conversationId,
+      kind: "schedule/set",
+      occurredAt: schedule.createdAt,
+      payload: { id: schedule.id, kind: schedule.kind, nextAt: schedule.nextAt }
+    });
+  });
+}
+async function recordDispatchFailure(sqlite, { schedule, occurrence, at, owner }, error) {
+  await sqlite.transaction(async (scope) => {
+    scope.database.prepare(`
+        UPDATE stitchkit_agent_runtime_schedules
+        SET claim_owner = NULL, claim_until = NULL, updated_at = ?
+        WHERE id = ? AND claim_owner = ?
+      `).run(at, schedule.id, owner);
+    await scope.appendEvent({
+      conversationId: schedule.conversationId,
+      kind: "schedule/failed",
+      occurredAt: at,
+      payload: {
+        id: schedule.id,
+        occurrence,
+        message: error instanceof Error ? error.message : String(error)
+      }
+    });
+  });
+}
+async function settleFiring(sqlite, { schedule, occurrence, at, owner }, observedAt, lateByMs) {
+  let nextAt = schedule.nextAt;
+  if (schedule.kind === "every" && schedule.intervalMs) {
+    let next = new Date(schedule.nextAt).getTime() + schedule.intervalMs;
+    while (next <= observedAt.getTime())
+      next += schedule.intervalMs;
+    nextAt = new Date(next).toISOString();
+  }
+  const state = schedule.kind === "every" ? "scheduled" : "completed";
+  await sqlite.transaction(async (scope) => {
+    const held = scope.database.prepare(`
+        SELECT state, claim_owner FROM stitchkit_agent_runtime_schedules WHERE id = ?
+      `).get(schedule.id);
+    const settled = held !== null && held !== undefined && (() => {
+      const row = FinalizeRowSchema.parse(held);
+      return row.state === "scheduled" && row.claim_owner === owner;
+    })();
+    if (settled) {
+      scope.database.prepare(`
+          UPDATE stitchkit_agent_runtime_schedules
+          SET state = ?, occurrence = ?, next_at = ?, updated_at = ?,
+            claim_owner = NULL, claim_until = NULL
+          WHERE id = ?
+        `).run(state, occurrence, nextAt, at, schedule.id);
+    }
+    if (lateByMs > 0) {
+      await scope.appendEvent({
+        conversationId: schedule.conversationId,
+        kind: "schedule/late",
+        occurredAt: at,
+        payload: { id: schedule.id, occurrence, lateByMs }
+      });
+    }
+    await scope.appendEvent({
+      conversationId: schedule.conversationId,
+      kind: "schedule/fired",
+      occurredAt: at,
+      payload: {
+        id: schedule.id,
+        occurrence,
+        lateByMs,
+        nextAt,
+        state: settled ? state : "cancelled"
+      }
+    });
+  });
+}
+function claimSchedule(sqlite, owner, id, until, at) {
+  return sqlite.transaction(async (scope) => {
+    const raw = scope.database.prepare(`
+        SELECT state, claim_until FROM stitchkit_agent_runtime_schedules WHERE id = ?
+      `).get(id);
+    if (raw === null || raw === undefined)
+      return false;
+    const row = ClaimRowSchema.parse(raw);
+    if (row.state !== "scheduled")
+      return false;
+    if (row.claim_until !== null && row.claim_until >= at)
+      return false;
+    scope.database.prepare(`
+        UPDATE stitchkit_agent_runtime_schedules
+        SET claim_owner = ?, claim_until = ?, updated_at = ?
+        WHERE id = ?
+      `).run(owner, until, at, id);
+    return true;
+  });
+}
+
+// src/agent-runtime/schedules.ts
+var AgentScheduleSchema = z10.object({
+  id: z10.string().min(1),
+  conversationId: z10.string().min(1),
+  kind: z10.enum(["at", "after", "every"]),
+  nextAt: z10.iso.datetime({ offset: true }),
+  intervalMs: z10.int().positive().optional(),
+  timeZone: z10.string().min(1).optional(),
+  input: z10.json(),
+  state: z10.enum(["scheduled", "cancelled", "completed"]),
+  occurrence: z10.int().nonnegative(),
+  createdAt: z10.iso.datetime({ offset: true }),
+  updatedAt: z10.iso.datetime({ offset: true })
 }).strict();
-var ScheduleRowSchema = z9.object({
-  id: z9.string(),
-  conversation_id: z9.string(),
-  kind: z9.enum(["at", "after", "every"]),
-  next_at: z9.string(),
-  interval_ms: z9.number().int().positive().nullable(),
-  time_zone: z9.string().nullable(),
-  input_payload: z9.string(),
-  state: z9.enum(["scheduled", "cancelled", "completed"]),
-  occurrence: z9.int().nonnegative(),
-  created_at: z9.string(),
-  updated_at: z9.string()
+var ScheduleRowSchema = z10.object({
+  id: z10.string(),
+  conversation_id: z10.string(),
+  kind: z10.enum(["at", "after", "every"]),
+  next_at: z10.string(),
+  interval_ms: z10.number().int().positive().nullable(),
+  time_zone: z10.string().nullable(),
+  input_payload: z10.string(),
+  state: z10.enum(["scheduled", "cancelled", "completed"]),
+  occurrence: z10.int().nonnegative(),
+  created_at: z10.string(),
+  updated_at: z10.string()
 });
-var ClaimRowSchema = z9.object({
-  state: z9.enum(["scheduled", "cancelled", "completed"]),
-  claim_until: z9.string().nullable()
+var ClaimRowSchema = z10.object({
+  state: z10.enum(["scheduled", "cancelled", "completed"]),
+  claim_until: z10.string().nullable()
 });
-var FinalizeRowSchema = z9.object({
-  state: z9.enum(["scheduled", "cancelled", "completed"]),
-  claim_owner: z9.string().nullable()
+var FinalizeRowSchema = z10.object({
+  state: z10.enum(["scheduled", "cancelled", "completed"]),
+  claim_owner: z10.string().nullable()
 });
-var StateRowSchema = z9.object({
-  state: z9.enum(["scheduled", "cancelled", "completed"])
+var StateRowSchema = z10.object({
+  state: z10.enum(["scheduled", "cancelled", "completed"])
 });
 function parseSchedule(raw) {
   const row = ScheduleRowSchema.parse(raw);
@@ -2066,27 +2214,10 @@ function createAgentScheduleService(input) {
     const delay = Math.max(0, new Date(raw.due_at).getTime() - now().getTime());
     timer = setTimer(() => void tick(), Math.min(delay, 2147483647));
   };
-  const owner = randomUUID2();
+  const owner = randomUUID3();
   const CLAIM_MS = 60000;
   let ticking;
-  const claim = (id, until, at) => input.sqlite.transaction(async (scope) => {
-    const raw = scope.database.prepare(`
-          SELECT state, claim_until FROM stitchkit_agent_runtime_schedules WHERE id = ?
-        `).get(id);
-    if (raw === null || raw === undefined)
-      return false;
-    const row = ClaimRowSchema.parse(raw);
-    if (row.state !== "scheduled")
-      return false;
-    if (row.claim_until !== null && row.claim_until >= at)
-      return false;
-    scope.database.prepare(`
-          UPDATE stitchkit_agent_runtime_schedules
-          SET claim_owner = ?, claim_until = ?, updated_at = ?
-          WHERE id = ?
-        `).run(owner, until, at, id);
-    return true;
-  });
+  const claim = (id, until, at) => claimSchedule(input.sqlite, owner, id, until, at);
   const runTick = async () => {
     if (closed)
       return;
@@ -2112,70 +2243,10 @@ function createAgentScheduleService(input) {
           schedule: { id: schedule.id, occurrence, lateByMs }
         });
       } catch (error) {
-        await input.sqlite.transaction(async (scope) => {
-          scope.database.prepare(`
-              UPDATE stitchkit_agent_runtime_schedules
-              SET claim_owner = NULL, claim_until = NULL, updated_at = ?
-              WHERE id = ? AND claim_owner = ?
-            `).run(at, schedule.id, owner);
-          await scope.appendEvent({
-            conversationId: schedule.conversationId,
-            kind: "schedule/failed",
-            occurredAt: at,
-            payload: {
-              id: schedule.id,
-              occurrence,
-              message: error instanceof Error ? error.message : String(error)
-            }
-          });
-        });
+        await recordDispatchFailure(input.sqlite, { schedule, occurrence, at, owner }, error);
         continue;
       }
-      let nextAt = schedule.nextAt;
-      if (schedule.kind === "every" && schedule.intervalMs) {
-        let next = new Date(schedule.nextAt).getTime() + schedule.intervalMs;
-        while (next <= observedAt.getTime())
-          next += schedule.intervalMs;
-        nextAt = new Date(next).toISOString();
-      }
-      const state = schedule.kind === "every" ? "scheduled" : "completed";
-      await input.sqlite.transaction(async (scope) => {
-        const held = scope.database.prepare(`
-            SELECT state, claim_owner FROM stitchkit_agent_runtime_schedules WHERE id = ?
-          `).get(schedule.id);
-        const settled = held !== null && held !== undefined && (() => {
-          const row = FinalizeRowSchema.parse(held);
-          return row.state === "scheduled" && row.claim_owner === owner;
-        })();
-        if (settled) {
-          scope.database.prepare(`
-              UPDATE stitchkit_agent_runtime_schedules
-              SET state = ?, occurrence = ?, next_at = ?, updated_at = ?,
-                claim_owner = NULL, claim_until = NULL
-              WHERE id = ?
-            `).run(state, occurrence, nextAt, at, schedule.id);
-        }
-        if (lateByMs > 0) {
-          await scope.appendEvent({
-            conversationId: schedule.conversationId,
-            kind: "schedule/late",
-            occurredAt: at,
-            payload: { id: schedule.id, occurrence, lateByMs }
-          });
-        }
-        await scope.appendEvent({
-          conversationId: schedule.conversationId,
-          kind: "schedule/fired",
-          occurredAt: at,
-          payload: {
-            id: schedule.id,
-            occurrence,
-            lateByMs,
-            nextAt,
-            state: settled ? state : "cancelled"
-          }
-        });
-      });
+      await settleFiring(input.sqlite, { schedule, occurrence, at, owner }, observedAt, lateByMs);
     }
   };
   const tick = () => {
@@ -2188,58 +2259,8 @@ function createAgentScheduleService(input) {
     return ticking;
   };
   const scheduleInput = async (request) => {
-    const modes = [
-      request.at !== undefined,
-      request.afterMs !== undefined,
-      request.everyMs !== undefined
-    ];
-    if (modes.filter(Boolean).length !== 1)
-      throw new TypeError("Declare exactly one schedule mode");
-    if (request.everyMs !== undefined && !request.timeZone) {
-      throw new TypeError("Repeating schedules require an explicit timeZone");
-    }
-    if (request.timeZone) {
-      try {
-        new Intl.DateTimeFormat("en", { timeZone: request.timeZone }).format(new Date(0));
-      } catch {
-        throw new TypeError(`Unknown schedule timeZone: ${request.timeZone}`);
-      }
-    }
-    const observedAt = now();
-    const kind = request.at !== undefined ? "at" : request.afterMs !== undefined ? "after" : "every";
-    const intervalMs = request.everyMs;
-    const delay = request.afterMs ?? request.everyMs;
-    if (delay !== undefined && (!Number.isSafeInteger(delay) || delay < 1)) {
-      throw new TypeError("Schedule delay must be a positive safe integer");
-    }
-    const nextAt = request.at ? new Date(z9.iso.datetime({ offset: true }).parse(request.at)).toISOString() : new Date(observedAt.getTime() + (delay ?? 0)).toISOString();
-    const schedule = AgentScheduleSchema.parse({
-      id: randomUUID2(),
-      conversationId: request.conversationId,
-      kind,
-      nextAt,
-      ...intervalMs !== undefined && { intervalMs },
-      ...request.timeZone && { timeZone: request.timeZone },
-      input: z9.json().parse(request.input),
-      state: "scheduled",
-      occurrence: 0,
-      createdAt: observedAt.toISOString(),
-      updatedAt: observedAt.toISOString()
-    });
-    await input.sqlite.transaction(async (scope) => {
-      scope.database.prepare(`
-          INSERT INTO stitchkit_agent_runtime_schedules (
-            id, conversation_id, kind, next_at, interval_ms, time_zone,
-            input_payload, state, occurrence, created_at, updated_at
-          ) VALUES (?, ?, ?, ?, ?, ?, ?, 'scheduled', 0, ?, ?)
-        `).run(schedule.id, schedule.conversationId, schedule.kind, schedule.nextAt, schedule.intervalMs ?? null, schedule.timeZone ?? null, JSON.stringify(schedule.input), schedule.createdAt, schedule.updatedAt);
-      await scope.appendEvent({
-        conversationId: schedule.conversationId,
-        kind: "schedule/set",
-        occurredAt: schedule.createdAt,
-        payload: { id: schedule.id, kind: schedule.kind, nextAt: schedule.nextAt }
-      });
-    });
+    const schedule = newSchedule(request, now());
+    await insertSchedule(input.sqlite, schedule);
     arm();
     return schedule;
   };
@@ -2284,17 +2305,17 @@ function createAgentScheduleService(input) {
   };
 }
 // src/agent-runtime/spill.ts
-import { createHash, randomUUID as randomUUID3 } from "node:crypto";
-import { z as z10 } from "zod";
-var SpillRowSchema = z10.object({
-  payload: z10.instanceof(Uint8Array),
-  bytes: z10.int().nonnegative(),
-  authorization_payload: z10.string().nullable()
+import { createHash, randomUUID as randomUUID4 } from "node:crypto";
+import { z as z11 } from "zod";
+var SpillRowSchema = z11.object({
+  payload: z11.instanceof(Uint8Array),
+  bytes: z11.int().nonnegative(),
+  authorization_payload: z11.string().nullable()
 });
-var SpillMetadataRowSchema = z10.object({
-  reference_id: z10.string(),
-  bytes: z10.int().nonnegative(),
-  created_at: z10.string()
+var SpillMetadataRowSchema = z11.object({
+  reference_id: z11.string(),
+  bytes: z11.int().nonnegative(),
+  created_at: z11.string()
 });
 function decodeAuthorization(value) {
   if (value === null)
@@ -2316,7 +2337,7 @@ function createSqliteAgentSpillStore(input) {
   };
   const authorization = (reference) => decodeAuthorization(rowFor(reference).authorization_payload);
   const write = async (request) => {
-    const reference = randomUUID3();
+    const reference = randomUUID4();
     const createdAt = now();
     const expiresAt = input.retentionMs ? new Date(createdAt.getTime() + input.retentionMs).toISOString() : null;
     const sha256 = createHash("sha256").update(request.data).digest("hex");
@@ -2381,14 +2402,14 @@ function createSqliteAgentSpillStore(input) {
   return { authorization, write, read, search, cleanup };
 }
 // src/agent-runtime/state-tools.ts
-import { z as z11 } from "zod";
-var GoalSchema = z11.object({
-  objective: z11.string().min(1),
-  status: z11.enum(["active", "complete", "blocked"])
+import { z as z12 } from "zod";
+var GoalSchema = z12.object({
+  objective: z12.string().min(1),
+  status: z12.enum(["active", "complete", "blocked"])
 }).strict();
-var TodoItemSchema = z11.object({
-  step: z11.string().min(1),
-  status: z11.enum(["pending", "in_progress", "completed"])
+var TodoItemSchema = z12.object({
+  step: z12.string().min(1),
+  status: z12.enum(["pending", "in_progress", "completed"])
 }).strict();
 function createAgentStateTools(input) {
   const writeGoal = async (value) => {
@@ -2405,7 +2426,7 @@ function createAgentStateTools(input) {
       name: "create_goal",
       description: "Create the durable objective for this conversation.",
       identity: { serviceName: "agent-state", action: "create-goal", method: "POST" },
-      input: z11.object({ objective: z11.string().min(1) }).strict(),
+      input: z12.object({ objective: z12.string().min(1) }).strict(),
       output: GoalSchema,
       transports: ["AGENT"],
       handler: async ({ input: request }) => {
@@ -2419,9 +2440,9 @@ function createAgentStateTools(input) {
       name: "update_goal",
       description: "Update the durable objective or lifecycle status.",
       identity: { serviceName: "agent-state", action: "update-goal", method: "POST" },
-      input: z11.object({
-        objective: z11.string().min(1).optional(),
-        status: z11.enum(["active", "complete", "blocked"]).optional()
+      input: z12.object({
+        objective: z12.string().min(1).optional(),
+        status: z12.enum(["active", "complete", "blocked"]).optional()
       }).strict().refine((request) => request.objective !== undefined || request.status !== undefined, {
         message: "At least one goal field must be supplied"
       }),
@@ -2439,8 +2460,8 @@ function createAgentStateTools(input) {
       name: "get_goal",
       description: "Read the durable objective for this conversation.",
       identity: { serviceName: "agent-state", action: "get-goal", method: "GET" },
-      input: z11.object({}).strict(),
-      output: z11.object({ goal: GoalSchema.optional() }).strict(),
+      input: z12.object({}).strict(),
+      output: z12.object({ goal: GoalSchema.optional() }).strict(),
       transports: ["AGENT"],
       handler: async () => {
         const current = await input.state.get(input.conversationId, "goal");
@@ -2451,8 +2472,8 @@ function createAgentStateTools(input) {
       name: "todo_write",
       description: "Replace the durable ordered work plan for this conversation.",
       identity: { serviceName: "agent-state", action: "write-todo", method: "POST" },
-      input: z11.object({ items: z11.array(TodoItemSchema) }).strict(),
-      output: z11.object({ items: z11.array(TodoItemSchema) }).strict(),
+      input: z12.object({ items: z12.array(TodoItemSchema) }).strict(),
+      output: z12.object({ items: z12.array(TodoItemSchema) }).strict(),
       transports: ["AGENT"],
       handler: async ({ input: request }) => {
         const stored = await input.state.set({
@@ -2461,7 +2482,7 @@ function createAgentStateTools(input) {
           value: request,
           actor: "agent"
         });
-        return z11.object({ items: z11.array(TodoItemSchema) }).strict().parse(stored.value);
+        return z12.object({ items: z12.array(TodoItemSchema) }).strict().parse(stored.value);
       }
     })
   ];