stitchkit 0.88.0 → 0.89.0

package/dist/{index-qp3vshvc.js → index-fadnxwna.js}

@@ -7,9 +7,10 @@ import {
   AgentConversationPurgedError,
   AgentHistoryMutationSchema,
   AgentRuntimeHeadSchema,
+  AgentSeedReceiptSchema,
   AgentStoredRunSchema,
   createAgentRuntimeStore
-} from "./index-v1qk37y9.js";
+} from "./index-hrbajay7.js";
 import {
   AgentRecoverableDescriptorSchema,
   AgentRecoverablePageSchema,
@@ -18,7 +19,7 @@ import {
   AppendAgentStoreEventSchema,
   agentStoreEventDraft,
   canonicalAgentJson
-} from "./index-6krwcdcf.js";
+} from "./index-zwkgfq1s.js";
 import {
   AgentMessageSchema,
   AgentRunSchema
@@ -36,6 +37,7 @@ var OWNED_TABLES = [
   "stitchkit_agent_runtime_projections",
   "stitchkit_agent_runtime_events",
   "stitchkit_agent_runtime_messages",
+  "stitchkit_agent_runtime_seeds",
   "stitchkit_agent_runtime_admissions",
   "stitchkit_agent_runtime_runs",
   "stitchkit_agent_runtime_heads"
@@ -263,6 +265,23 @@ function createAgentRuntimeSqliteV2Tables(database) {
   createDurableTables(database);
 }
 
+// src/agent-runtime/store-migrations/v2-to-v3.ts
+function createAgentRuntimeSqliteV3Tables(database) {
+  database.exec(`
+    CREATE TABLE IF NOT EXISTS stitchkit_agent_runtime_seeds (
+      conversation_id TEXT NOT NULL,
+      seed_key TEXT NOT NULL,
+      message_ids TEXT NOT NULL,
+      created_at TEXT NOT NULL,
+      PRIMARY KEY (conversation_id, seed_key)
+    );
+  `);
+}
+function migrateAgentRuntimeSqliteV2ToV3(database) {
+  createAgentRuntimeSqliteV3Tables(database);
+  database.prepare("UPDATE stitchkit_agent_runtime_meta SET value = '3' WHERE key = 'schema_version'").run();
+}
+
 // src/agent-runtime/sqlite.ts
 var HeadRowSchema = z2.object({ version: z2.number().int().nonnegative() });
 var RunRowSchema = z2.object({
@@ -276,6 +295,11 @@ var AdmissionRowSchema2 = z2.object({
   run_id: z2.string(),
   assistant_message_id: z2.string()
 });
+var SeedRowSchema = z2.object({
+  conversation_id: z2.string(),
+  seed_key: z2.string(),
+  message_ids: z2.string()
+});
 var MessageRowSchema = z2.object({ payload: z2.string() });
 var PositionedMessageRowSchema = z2.object({ position: z2.number().int().nonnegative() });
 var RecoverableRowSchema = z2.object({
@@ -325,12 +349,13 @@ var ArchiveSpillRowSchema = z2.object({
   authorization_payload: z2.string().nullable(),
   payload: z2.instanceof(Uint8Array)
 });
-var SCHEMA_VERSION = 2;
+var SCHEMA_VERSION = 3;
 var TABLES = [
   "stitchkit_agent_runtime_heads",
   "stitchkit_agent_runtime_runs",
   "stitchkit_agent_runtime_admissions",
   "stitchkit_agent_runtime_messages",
+  "stitchkit_agent_runtime_seeds",
   "stitchkit_agent_runtime_events",
   "stitchkit_agent_runtime_projections",
   "stitchkit_agent_runtime_spills",
@@ -418,6 +443,13 @@ function initializeAgentRuntimeSqlite(database) {
       const version = Number(MetaRowSchema.parse(versionRow).value);
       if (version === 1) {
         migrateAgentRuntimeSqliteV1ToV2(database);
+        migrateAgentRuntimeSqliteV2ToV3(database);
+        initializeSqliteConversationPurge(database);
+        database.exec("COMMIT");
+        return;
+      }
+      if (version === 2) {
+        migrateAgentRuntimeSqliteV2ToV3(database);
         initializeSqliteConversationPurge(database);
         database.exec("COMMIT");
         return;
@@ -474,7 +506,8 @@ function initializeAgentRuntimeSqlite(database) {
       );
     `);
     createAgentRuntimeSqliteV2Tables(database);
-    database.prepare("INSERT INTO stitchkit_agent_runtime_meta (key, value) VALUES ('schema_version', '2')").run();
+    createAgentRuntimeSqliteV3Tables(database);
+    database.prepare("INSERT INTO stitchkit_agent_runtime_meta (key, value) VALUES ('schema_version', '3')").run();
     initializeSqliteConversationPurge(database);
     database.exec("COMMIT");
   } catch (error) {
@@ -646,6 +679,32 @@ function createSqliteAgentRuntimeStore(config) {
           `).run(receipt.conversationId, receipt.idempotencyKey, receipt.input.id, receipt.runId, receipt.assistantMessageId, encodeJson(receipt.input));
       }
     },
+    seeds: {
+      async load(transaction, input) {
+        const value = transaction.prepare(`
+            SELECT conversation_id, seed_key, message_ids
+            FROM stitchkit_agent_runtime_seeds
+            WHERE conversation_id = ? AND seed_key = ?
+          `).get(input.conversationId, input.seedKey);
+        if (missing(value))
+          return;
+        const row = SeedRowSchema.parse(value);
+        return AgentSeedReceiptSchema.parse({
+          schemaVersion: 1,
+          conversationId: row.conversation_id,
+          seedKey: row.seed_key,
+          messageIds: z2.array(z2.string().min(1)).parse(parseJson2(row.message_ids))
+        });
+      },
+      async create(transaction, rawReceipt) {
+        const receipt = AgentSeedReceiptSchema.parse(rawReceipt);
+        transaction.prepare(`
+            INSERT INTO stitchkit_agent_runtime_seeds (
+              conversation_id, seed_key, message_ids, created_at
+            ) VALUES (?, ?, ?, ?)
+          `).run(receipt.conversationId, receipt.seedKey, encodeJson(receipt.messageIds), new Date().toISOString());
+      }
+    },
     history: {
       async load(transaction, conversationId) {
         return transaction.prepare(`
@@ -653,9 +712,32 @@ function createSqliteAgentRuntimeStore(config) {
             WHERE conversation_id = ? AND active = 1 ORDER BY position ASC
           `).all(conversationId).map((value) => AgentMessageSchema.parse(parseJson2(MessageRowSchema.parse(value).payload)));
       },
+      async hasMessage(transaction, input) {
+        const row = transaction.prepare(`
+            SELECT count(*) AS count FROM stitchkit_agent_runtime_messages
+            WHERE conversation_id = ? AND id = ?
+          `).get(input.conversationId, input.messageId);
+        return CountRowSchema.parse(row).count > 0;
+      },
       async apply(transaction, rawMutation) {
         const mutation = AgentHistoryMutationSchema.parse(rawMutation);
-        const message = mutation.type === "admit" ? mutation.input : mutation.type === "upsert-assistant" ? mutation.message : mutation.summary;
+        const message = mutation.type === "admit" ? mutation.input : mutation.type === "upsert-assistant" ? mutation.message : mutation.type === "seed" ? mutation.message : mutation.summary;
+        if (mutation.type === "seed") {
+          transaction.prepare(`
+              UPDATE stitchkit_agent_runtime_messages SET position = position + 1
+              WHERE conversation_id = ?
+            `).run(mutation.message.conversationId);
+          transaction.prepare(`
+              INSERT INTO stitchkit_agent_runtime_messages
+                (conversation_id, id, position, active, payload)
+              VALUES (?, ?, 0, 1, ?)
+              ON CONFLICT (conversation_id, id) DO UPDATE SET
+                position = excluded.position,
+                active = excluded.active,
+                payload = excluded.payload
+            `).run(mutation.message.conversationId, mutation.message.id, encodeJson(mutation.message));
+          return;
+        }
         if (mutation.type === "replace-compacted-range") {
           const parameters = mutation.replacedMessageIds;
           const rows = transaction.prepare(`

package/dist/{index-v1qk37y9.js → index-hrbajay7.js}

@@ -14,12 +14,13 @@ import {
   RecoverAgentRunSchema,
   ReplaceCompactedRangeSchema,
   RequestRunInterruptSchema,
+  SeedConversationInputSchema,
   agentStoreEventDraft,
   assistantStatus,
   canonicalAgentJson,
   decodeAgentConversationArchive,
   encodeAgentConversationArchive
-} from "./index-6krwcdcf.js";
+} from "./index-zwkgfq1s.js";
 import {
   AgentMessageSchema,
   AgentRecordIdSchema,
@@ -105,6 +106,12 @@ var AgentAdmissionReceiptSchema = z2.object({
   runId: AgentRecordIdSchema,
   assistantMessageId: AgentRecordIdSchema
 });
+var AgentSeedReceiptSchema = z2.object({
+  schemaVersion: z2.literal(1),
+  conversationId: AgentRecordIdSchema,
+  seedKey: z2.string().min(1),
+  messageIds: z2.array(AgentRecordIdSchema).min(1)
+});
 var AgentHistoryMutationSchema = z2.discriminatedUnion("type", [
   z2.object({ type: z2.literal("admit"), input: AgentMessageSchema }),
   z2.object({
@@ -115,6 +122,10 @@ var AgentHistoryMutationSchema = z2.discriminatedUnion("type", [
     type: z2.literal("replace-compacted-range"),
     replacedMessageIds: z2.array(AgentRecordIdSchema).min(1),
     summary: AgentMessageSchema
+  }),
+  z2.object({
+    type: z2.literal("seed"),
+    message: AgentMessageSchema
   })
 ]);
 var AgentRecoverableScanInputSchema = z2.object({
@@ -279,6 +290,7 @@ function applied(current, input, effects) {
     }),
     ...effects?.runRecords?.length && { runRecords: effects.runRecords },
     ...effects?.admissionReceipt && { admissionReceipt: effects.admissionReceipt },
+    ...effects?.seedReceipt && { seedReceipt: effects.seedReceipt },
     ...effects?.historyMutations?.length && { historyMutations: effects.historyMutations }
   };
 }
@@ -318,6 +330,28 @@ function reduceStore(current, operation) {
       historyMutations: [{ type: "admit", input: input.input }]
     });
   }
+  if (operation.type === "seed") {
+    const conversationId2 = operation.input.conversationId;
+    if (current.conversationId !== conversationId2)
+      return { outcome: "not_found" };
+    for (const message of operation.input.inputs) {
+      if (message.conversationId !== conversationId2 || message.role !== "user" || message.status !== "committed" || message.runId !== undefined) {
+        throw new TypeError("Seeded conversation input must be committed unowned user messages");
+      }
+    }
+    return applied(current, { messages: [...operation.input.inputs, ...current.messages] }, {
+      seedReceipt: AgentSeedReceiptSchema.parse({
+        schemaVersion: 1,
+        conversationId: conversationId2,
+        seedKey: operation.input.seedKey,
+        messageIds: operation.input.inputs.map((message) => message.id)
+      }),
+      historyMutations: [...operation.input.inputs].reverse().map((message) => ({
+        type: "seed",
+        message
+      }))
+    });
+  }
   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)
@@ -619,7 +653,7 @@ function createAgentRuntimeStore(driver) {
     if (await driver.conversations?.isPurged(transaction, conversationId)) {
       throw new AgentConversationPurgedError;
     }
-    const operationRunId = operation.type === "accept" ? operation.input.coalesceIntoRunId : operation.type === "compact" ? undefined : operation.input.runId;
+    const operationRunId = operation.type === "accept" ? operation.input.coalesceIntoRunId : operation.type === "compact" || operation.type === "seed" ? undefined : operation.input.runId;
     const [stored, messages, activeRecords, operationRecord, duplicateReceipt] = await Promise.all([
       driver.head.load(transaction, conversationId),
       driver.history.load(transaction, conversationId),
@@ -637,6 +671,34 @@ function createAgentRuntimeStore(driver) {
     });
     const records = mergeRunRecords(activeRecords, referencedRecords, operationRecord ? [operationRecord] : []);
     const current = snapshotOf(head, messages, records);
+    if (operation.type === "seed") {
+      const existing = await driver.seeds.load(transaction, {
+        conversationId,
+        seedKey: operation.input.seedKey
+      });
+      if (existing)
+        return { outcome: "applied", snapshot: current };
+      let collisions = 0;
+      for (const message of operation.input.inputs) {
+        if (current.messages.some((candidate) => candidate.id === message.id) || driver.history.hasMessage && await driver.history.hasMessage(transaction, {
+          conversationId,
+          messageId: message.id
+        }))
+          collisions++;
+      }
+      if (collisions > 0 && collisions < operation.input.inputs.length) {
+        throw new TypeError("Partial instruction seed collision: restore the complete seed before retrying");
+      }
+      if (collisions > 0) {
+        await driver.seeds.create(transaction, AgentSeedReceiptSchema.parse({
+          schemaVersion: 1,
+          conversationId,
+          seedKey: operation.input.seedKey,
+          messageIds: operation.input.inputs.map((message) => message.id)
+        }));
+        return { outcome: "applied", snapshot: current };
+      }
+    }
     if (duplicateReceipt) {
       const duplicateRecord = await driver.runs.load(transaction, {
         conversationId,
@@ -712,6 +774,9 @@ function createAgentRuntimeStore(driver) {
     if (reduced.admissionReceipt) {
       await driver.admissions.create(transaction, reduced.admissionReceipt);
     }
+    if (reduced.seedReceipt) {
+      await driver.seeds.create(transaction, reduced.seedReceipt);
+    }
     for (const mutation of reduced.historyMutations ?? []) {
       await driver.history.apply(transaction, mutation);
     }
@@ -877,6 +942,10 @@ function createAgentRuntimeStore(driver) {
       type: "compact",
       input: ReplaceCompactedRangeSchema.parse(input)
     }),
+    seedConversationInput: (input) => mutate({
+      type: "seed",
+      input: SeedConversationInputSchema.parse(input)
+    }),
     async scanRecoverable(input) {
       const parsed = AgentRecoverableScanInputSchema.parse(input);
       return AgentRecoverablePageSchema.parse(await driver.scanRecoverable(parsed));
@@ -906,6 +975,7 @@ function createMemoryAgentRuntimeStore() {
   let heads = new Map;
   let runs = new Map;
   let admissions = new Map;
+  let seeds = new Map;
   let histories = new Map;
   let events = new Map;
   let transactionTail = Promise.resolve();
@@ -924,6 +994,7 @@ function createMemoryAgentRuntimeStore() {
         heads: cloneHeadMap(heads),
         runs: cloneNestedMap(runs, (record) => AgentStoredRunSchema.parse(structuredClone(record))),
         admissions: cloneNestedMap(admissions, (receipt) => AgentAdmissionReceiptSchema.parse(structuredClone(receipt))),
+        seeds: cloneNestedMap(seeds, (receipt) => AgentSeedReceiptSchema.parse(structuredClone(receipt))),
         histories: cloneHistoryMap(histories),
         events: new Map([...events].map(([conversationId, stored]) => [
           conversationId,
@@ -936,6 +1007,7 @@ function createMemoryAgentRuntimeStore() {
         heads = transaction.heads;
         runs = transaction.runs;
         admissions = transaction.admissions;
+        seeds = transaction.seeds;
         histories = transaction.histories;
         events = transaction.events;
         return result;
@@ -952,6 +1024,7 @@ function createMemoryAgentRuntimeStore() {
         transaction.heads.delete(conversationId);
         transaction.runs.delete(conversationId);
         transaction.admissions.delete(conversationId);
+        transaction.seeds.delete(conversationId);
         transaction.histories.delete(conversationId);
         transaction.events.delete(conversationId);
       }
@@ -1021,13 +1094,31 @@ function createMemoryAgentRuntimeStore() {
         transaction.admissions.set(receipt.conversationId, conversationAdmissions);
       }
     },
+    seeds: {
+      async load(transaction, input) {
+        const receipt = transaction.seeds.get(input.conversationId)?.get(input.seedKey);
+        return receipt ? AgentSeedReceiptSchema.parse(structuredClone(receipt)) : undefined;
+      },
+      async create(transaction, rawReceipt) {
+        const receipt = AgentSeedReceiptSchema.parse(structuredClone(rawReceipt));
+        const conversationSeeds = transaction.seeds.get(receipt.conversationId) ?? new Map;
+        if (conversationSeeds.has(receipt.seedKey)) {
+          throw new TypeError("Conversation seed identity is already reserved");
+        }
+        conversationSeeds.set(receipt.seedKey, receipt);
+        transaction.seeds.set(receipt.conversationId, conversationSeeds);
+      }
+    },
     history: {
       async load(transaction, conversationId) {
         return (transaction.histories.get(conversationId) ?? []).map((message) => AgentMessageSchema.parse(structuredClone(message)));
       },
+      async hasMessage(transaction, input) {
+        return (transaction.histories.get(input.conversationId) ?? []).some((message) => message.id === input.messageId);
+      },
       async apply(transaction, rawMutation) {
         const mutation = AgentHistoryMutationSchema.parse(rawMutation);
-        const conversationId = mutation.type === "admit" ? mutation.input.conversationId : mutation.type === "upsert-assistant" ? mutation.message.conversationId : mutation.summary.conversationId;
+        const conversationId = mutation.type === "admit" ? mutation.input.conversationId : mutation.type === "upsert-assistant" ? mutation.message.conversationId : mutation.type === "seed" ? mutation.message.conversationId : mutation.summary.conversationId;
         const current = transaction.histories.get(conversationId) ?? [];
         if (mutation.type === "admit") {
           transaction.histories.set(conversationId, [...current, mutation.input]);
@@ -1037,6 +1128,17 @@ function createMemoryAgentRuntimeStore() {
           transaction.histories.set(conversationId, replaceMessage(current, mutation.message));
           return;
         }
+        if (mutation.type === "seed") {
+          const existingIndex = current.findIndex((message) => message.id === mutation.message.id);
+          if (existingIndex === -1) {
+            transaction.histories.set(conversationId, [mutation.message, ...current]);
+          } else {
+            const next = [...current];
+            next[existingIndex] = mutation.message;
+            transaction.histories.set(conversationId, next);
+          }
+          return;
+        }
         const replaced = new Set(mutation.replacedMessageIds);
         const positions = current.map((message, index) => replaced.has(message.id) ? index : undefined).filter((index) => index !== undefined);
         const first = positions[0];
@@ -1091,4 +1193,4 @@ function createMemoryAgentRuntimeStore() {
   return createAgentRuntimeStore(driver);
 }
 
-export { AgentConversationPurgeInputSchema, AgentConversationPurgeResultSchema, AgentConversationPurgedError, purgeAgentConversation, AgentRuntimeHeadSchema, AgentStoredRunSchema, AgentAdmissionReceiptSchema, AgentHistoryMutationSchema, ACTIVE_AGENT_RUN_STATES, createAgentRuntimeStore, createMemoryAgentRuntimeStore };
+export { AgentConversationPurgeInputSchema, AgentConversationPurgeResultSchema, AgentConversationPurgedError, purgeAgentConversation, AgentRuntimeHeadSchema, AgentStoredRunSchema, AgentAdmissionReceiptSchema, AgentSeedReceiptSchema, AgentHistoryMutationSchema, ACTIVE_AGENT_RUN_STATES, createAgentRuntimeStore, createMemoryAgentRuntimeStore };