@springbrand/agent-runtime 0.3.0-alpha.16 → 0.3.0-alpha.17

package/src/session/cloudflare/migration.ts

@@ -1,116 +1,30 @@
-import { isEqual } from "lodash-es";
 import {
-  decodePiToolRecoveryMilestone,
-  replayPiToolRecovery,
-} from "./legacy-tool-recovery";
-/** Versioned SQL data migration, called before the Runtime opens its transcript. */
-import {
-  branchTip,
-  entryLabel,
-  insertEntry,
-  insertUsage,
-  setValue,
-  deleteValue,
-  sessionName,
-  laneState,
-  operationMeta,
-  operationState,
-  pendingEntry,
-  sessionEntryToContextMessages,
-  value,
-  type Entry,
-  type JsonValue,
-  type Write,
-  type Value,
-  type LaneState,
-  type ToolExecution,
-} from "@earendil-works/pi-agent-core/harness/session";
-import type {
-  AgentToolResult,
-  AgentMessage,
-} from "@earendil-works/pi-agent-core";
-import type { Usage, ToolResultMessage } from "@earendil-works/pi-ai";
-import {
-  CloudflareSessionStorage,
-  CURRENT_PI_STORAGE_VERSION,
-} from "./storage";
-
-const RETIRED_SCHEDULE_CALLBACKS = [
-  "_cleanupStreamBuffers",
-  "_piPlannedContinuation",
-  "_piContinuationWatchdog",
-  "_piApprovalContinuation",
-  "_chatRecoveryRetry",
-] as const;
+  AgentDoc, GenerationTask, LiveDoc,
+  defineDoc, type ConversationId, type EntryRecord,
+  type GenerationCheckpoint, type Session,
+} from "@earendil-works/pi-durable";
+import { BACKGROUND_CONTEXT } from "@earendil-works/chord/context";
+import type { Message } from "@earendil-works/pi-ai";
+import { replayPiToolRecovery } from "./legacy-tool-recovery";
+import { RuntimeConversation } from "./storage";
 
-const LEGACY_TERMINAL_SUBMISSION_STATUSES = new Set([
-  "completed",
-  "failed",
-  "aborted",
-  "skipped",
-]);
+const Migration = defineDoc({
+  kind: "springbrand.pi-upgrade", version: 1, scope: "session",
+  initial: (): { completed: boolean } => ({ completed: false }),
+});
 
-type LegacyRecord = Record<string, unknown> & {
-  id: string;
-  parentId: string | null;
-  type: string;
-  timestamp: string;
+type LegacyEntry = {
+  id: string; parentId: string | null; type: string; timestamp: number | string;
+  message?: Message; summary?: string; firstKeptEntryId?: string;
+  retainedTail?: Message[]; [key: string]: unknown;
 };
-export interface PiSessionMigrationReport {
-  sourceEntries: number;
-  entries: number;
-  tip: string | null;
-  alreadyMigrated: boolean;
-}
-
-function object(input: unknown): Record<string, unknown> {
-  if (!input || typeof input !== "object" || Array.isArray(input))
-    throw new Error("Invalid legacy object");
-  return input as Record<string, unknown>;
-}
-function text(input: unknown): string {
-  if (typeof input !== "string") throw new Error("Invalid legacy string");
-  return input;
-}
-function time(input: unknown): number {
-  const timestamp = Date.parse(text(input));
-  if (!Number.isSafeInteger(timestamp) || timestamp < 0)
-    throw new Error("Invalid legacy timestamp");
-  return timestamp;
-}
-function finite(input: unknown): number {
-  if (typeof input !== "number" || !Number.isFinite(input) || input < 0)
-    throw new Error("Invalid legacy number");
-  return input;
-}
-function message(input: unknown): AgentMessage {
-  const record = object(input);
-  if (
-    ![
-      "user",
-      "assistant",
-      "toolResult",
-      "custom",
-      "bashExecution",
-      "branchSummary",
-      "compactionSummary",
-    ].includes(text(record.role))
-  )
-    throw new Error("Unknown legacy message role");
-  finite(record.timestamp);
-  if (record.role === "assistant" && record.stopReason === "pending")
-    throw new Error("Unsettled legacy assistant message");
-  return normalizeLegacyAskUserMessage(record) as AgentMessage;
-}
 function hasTable(sql: SqlStorage, name: string): boolean {
-  return (
-    sql
-      .exec(
-        "SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = ?",
-        name,
-      )
-      .toArray().length > 0
-  );
+  return sql.exec("SELECT 1 FROM sqlite_master WHERE type='table' AND name=?", name).toArray().length > 0;
+}
+function timestamp(entry: LegacyEntry): number {
+  const value = typeof entry.timestamp === "number" ? entry.timestamp : Date.parse(entry.timestamp);
+  if (!Number.isFinite(value)) throw new Error(`Invalid historical entry time: ${entry.id}`);
+  return value;
 }
 
 const ASK_USER_RESPONSE_TYPES = new Set([
@@ -178,855 +92,223 @@ function normalizeLegacyAskUserEntry(value: unknown): unknown {
   return value;
 }
 
-function migrateLegacyAskUserData(sql: SqlStorage): void {
-  if (hasTable(sql, "pi_entries")) {
-    for (const row of sql.exec<{ id: string; body: string }>(
-      "SELECT id, body FROM pi_entries",
-    ).toArray()) {
-      sql.exec(
-        "UPDATE pi_entries SET body = ? WHERE id = ?",
-        JSON.stringify(normalizeLegacyAskUserEntry(JSON.parse(row.body))),
-        row.id,
-      );
-    }
-  }
-  if (hasTable(sql, "pi_tool_interactions")) {
-    for (const row of sql.exec<{ interaction_id: string; input_json: string }>(
-      "SELECT interaction_id, input_json FROM pi_tool_interactions WHERE tool_name = 'ask_user'",
-    ).toArray()) {
-      sql.exec(
-        "UPDATE pi_tool_interactions SET input_json = ? WHERE interaction_id = ?",
-        JSON.stringify(normalizeLegacyAskUserInput(JSON.parse(row.input_json))),
-        row.interaction_id,
-      );
-    }
+/** Converts records once; historical tables are retained until the upgrade is accepted. */
+export async function migratePiSession(storage: DurableObjectStorage, session: Session): Promise<void> {
+  if ((await session.snapshot(Migration, BACKGROUND_CONTEXT))?.completed) return;
+  const sql = storage.sql;
+  if (hasTable(sql, "pi_storage_state")) {
+    const version = sql.exec<{ storage_version: number }>("SELECT storage_version FROM pi_storage_state WHERE singleton=1").one().storage_version;
+    if (![1, 2, 3].includes(version)) throw new Error(`Unsupported historical Pi storage version: ${version}`);
   }
-  if (hasTable(sql, "pi_values")) {
-    for (const row of sql.exec<{ key: string; body: string }>(
-      "SELECT key, body FROM pi_values WHERE namespace = 'pi.op.state'",
-    ).toArray()) {
-      const state = JSON.parse(row.body) as {
-        batch?: {
-          assistantEntryId?: unknown;
-          turnId?: unknown;
-        };
-      };
-      const assistantEntryId = state.batch?.assistantEntryId;
-      const turnId = state.batch?.turnId;
-      if (typeof assistantEntryId !== "string" || typeof turnId !== "string") {
-        continue;
+  const current = hasTable(sql, "pi_entries");
+  const older = hasTable(sql, "pi_session_entries");
+  const rows = current || older
+    ? sql.exec<{ body: string }>(`SELECT body FROM ${current ? "pi_entries" : "pi_session_entries"} ORDER BY seq`).toArray()
+    : [];
+  const entries = rows.map(({ body }) => normalizeLegacyAskUserEntry(JSON.parse(body)) as LegacyEntry);
+  const byId = new Map(entries.map((entry) => [entry.id, entry]));
+  if (byId.size !== entries.length) throw new Error("Historical Pi entry IDs are not unique");
+  const values = current && hasTable(sql, "pi_values")
+    ? sql.exec<{ namespace: string; key: string; body: string }>("SELECT namespace,key,body FROM pi_values").toArray()
+    : [];
+  const read = <T>(namespace: string, key: string): T | undefined => {
+    const row = values.find((row) => row.namespace === namespace && row.key === key);
+    return row ? JSON.parse(row.body) as T : undefined;
+  };
+  const tipValue = values.find((row) => row.namespace === "pi.branch.tip" && row.key === "main");
+  const last = entries.at(-1);
+  let oldTip: string | null = tipValue ? JSON.parse(tipValue.body) : last?.type === "leaf" ? last.targetId as string | null : last?.id ?? null;
+  const lane = read<{ currentOperationId?: string | null }>("pi.lane.state", "main");
+  const operationId = lane?.currentOperationId;
+  const result = operationId ? read<{ status?: string }>("pi.result", operationId) : undefined;
+  const legacyRunning = older && hasTable(sql, "pi_submissions") ? sql.exec<{ submission_id: string }>("SELECT submission_id FROM pi_submissions WHERE status='running'").toArray() : [];
+  if (legacyRunning.length > 1) throw new Error("Multiple legacy operations are active in one conversation");
+  const hostStatus = operationId && hasTable(sql, "pi_submissions")
+    ? sql.exec<{ status: string }>("SELECT status FROM pi_submissions WHERE submission_id=?", operationId).toArray()[0]?.status
+    : undefined;
+  const terminal = hostStatus && ["completed", "failed", "aborted", "skipped"].includes(hostStatus);
+  const active = operationId && result === undefined && !terminal ? operationId : legacyRunning[0]?.submission_id;
+  const links = hasTable(sql, "pi_session_entry_submission")
+    ? sql.exec<{ entry_id: string; submission_id: string }>("SELECT entry_id,submission_id FROM pi_session_entry_submission").toArray()
+    : hasTable(sql, "pi_session_message_submission") ? sql.exec<{ entry_id: string; submission_id: string }>("SELECT message_id AS entry_id,submission_id FROM pi_session_message_submission").toArray() : [];
+  const owners = new Map(links.map((row) => [row.entry_id, row.submission_id]));
+  const sidecars = hasTable(sql, "pi_message_ui")
+    ? sql.exec<{ entry_id: string; body: string }>("SELECT message_id AS entry_id,body FROM pi_message_ui").toArray()
+    : [];
+  const ui = new Map(sidecars.map((row) => [row.entry_id, row.body]));
+  // Older versions copied the initial user's view onto later Steers. Keep their canonical input instead.
+  if (current && hasTable(sql, "pi_submissions")) {
+    const initialUsers = new Map(sql.exec<{ submission_id: string; user_message_id: string | null }>(
+      "SELECT submission_id,user_message_id FROM pi_submissions",
+    ).toArray().map((row) => [row.submission_id, row.user_message_id]));
+    for (const [id, body] of ui) {
+      const owner = owners.get(id);
+      const entry = byId.get(id);
+      if (!owner || entry?.message?.role !== "user" || JSON.parse(body).id !== initialUsers.get(owner)) continue;
+      const visited = new Set<string>();
+      for (let parent = entry.parentId; parent; parent = byId.get(parent)?.parentId ?? null) {
+        if (visited.has(parent)) throw new Error(`Cyclic historical Pi ancestry: ${parent}`);
+        visited.add(parent);
+        if (owners.get(parent) === owner && byId.get(parent)?.message?.role === "user") {
+          ui.delete(id);
+          break;
+        }
       }
-      const entryRow = sql.exec<{ body: string }>(
-        "SELECT body FROM pi_entries WHERE id = ?",
-        assistantEntryId,
-      ).toArray()[0];
-      if (!entryRow) continue;
-      const entry = JSON.parse(entryRow.body) as {
-        type?: unknown;
-        message?: { role?: unknown; content?: unknown };
-      };
-      if (
-        entry.type !== "message" || entry.message?.role !== "assistant" ||
-        !Array.isArray(entry.message.content)
-      ) continue;
-      entry.message.content.forEach((part, sourceIndex) => {
-        if (!part || typeof part !== "object" || Array.isArray(part)) return;
-        const block = part as Record<string, unknown>;
-        if (block.type !== "toolCall" || block.name !== "ask_user") return;
-        const key = `${row.key}:${turnId}:${sourceIndex}`;
-        const args = sql.exec<{ body: string }>(
-          "SELECT body FROM pi_values WHERE namespace = 'pi.op.tool_args' AND key = ?",
-          key,
-        ).toArray()[0];
-        if (!args) return;
-        sql.exec(
-          "UPDATE pi_values SET body = ? WHERE namespace = 'pi.op.tool_args' AND key = ?",
-          JSON.stringify(normalizeLegacyAskUserInput(JSON.parse(args.body))),
-          key,
-        );
-      });
     }
   }
-}
-
-function removeRetiredRuntimeArtifacts(sql: SqlStorage): void {
-  sql.exec(
-    "CREATE TABLE IF NOT EXISTS pi_session_entry_submission (entry_id TEXT PRIMARY KEY, submission_id TEXT NOT NULL)",
-  );
-  if (hasTable(sql, "pi_session_message_submission")) {
-    sql.exec(`
-      INSERT OR IGNORE INTO pi_session_entry_submission (entry_id, submission_id)
-        SELECT message_id, submission_id FROM pi_session_message_submission;
-      DROP TABLE pi_session_message_submission;
-    `);
-  }
-  if (hasTable(sql, "cf_agents_schedules")) {
-    const placeholders = RETIRED_SCHEDULE_CALLBACKS.map(() => "?").join(", ");
-    sql.exec(
-      `DELETE FROM cf_agents_schedules WHERE callback IN (${placeholders})`,
-      ...RETIRED_SCHEDULE_CALLBACKS,
-    );
-  }
-  migrateLegacyAskUserData(sql);
-}
-
-/** Older Harness adapters copied the initial user sidecar onto consumed Steers. */
-function repairDuplicatedUserSidecars(sql: SqlStorage): void {
-  if (!hasTable(sql, "pi_message_ui") || !hasTable(sql, "pi_submissions")) return;
-  const entries = new Map(sql.exec<{
-    id: string;
-    parent_id: string | null;
-    body: string;
-    submission_id: string | null;
-  }>(`
-    SELECT entry.id, entry.parent_id, entry.body, link.submission_id
-    FROM pi_entries AS entry
-    LEFT JOIN pi_session_entry_submission AS link ON link.entry_id = entry.id
-  `).toArray().map((row) => [row.id, { ...row, entry: JSON.parse(row.body) as Entry }]));
-  const candidates = sql.exec<{ message_id: string }>(`
-    SELECT ui.message_id
-    FROM pi_message_ui AS ui
-    JOIN pi_session_entry_submission AS link ON link.entry_id = ui.message_id
-    JOIN pi_submissions AS submission ON submission.submission_id = link.submission_id
-    WHERE json_extract(ui.body, '$.id') = submission.user_message_id
-  `).toArray();
-  for (const candidate of candidates) {
-    const current = entries.get(candidate.message_id);
-    if (current?.entry.type !== "message" || current.entry.message.role !== "user") continue;
-    let parentId = current.parent_id;
-    const visited = new Set<string>();
-    while (parentId !== null) {
-      if (visited.has(parentId)) throw new Error(`Cyclic Pi ancestry: ${parentId}`);
-      visited.add(parentId);
-      const parent = entries.get(parentId);
-      if (!parent) throw new Error(`Missing Pi ancestor: ${parentId}`);
-      if (parent.submission_id === current.submission_id &&
-        parent.entry.type === "message" && parent.entry.message.role === "user") {
-        // The copied parts are also wrong: canonical Steer content is authoritative.
-        sql.exec("DELETE FROM pi_message_ui WHERE message_id = ?", candidate.message_id);
-        break;
-      }
-      parentId = parent.parent_id;
+  if (older && active && hasTable(sql, "pi_recovery_milestones")) {
+    const milestones = sql.exec<{ body: string }>("SELECT body FROM pi_recovery_milestones WHERE submission_id=? ORDER BY seq", active).toArray();
+    const recovered = replayPiToolRecovery(milestones.map((row) => row.body));
+    const existing = new Set(entries.flatMap((entry) => entry.message?.role === "toolResult" ? [entry.message.toolCallId] : []));
+    for (const call of Object.values(recovered.toolCalls)) {
+      if (existing.has(call.toolCallId)) continue;
+      const pending = Object.values(recovered.approvals).some((row) => row.toolCallId === call.toolCallId && row.status === "pending") || Object.values(recovered.interactions).some((row) => row.toolCallId === call.toolCallId && row.status === "pending");
+      const outcome = call.result ?? (!pending && call.retry === "non-idempotent" ? { role: "toolResult" as const, toolCallId: call.toolCallId, toolName: call.toolName, content: [{ type: "text" as const, text: "Interrupted legacy effect; its outcome is unknown. Reconcile it before repeating it." }], isError: true, timestamp: Date.now() } : undefined);
+      if (!outcome) continue;
+      const entry: LegacyEntry = { id: `legacy-result:${active}:${call.toolCallId}`, parentId: oldTip, type: "message", message: outcome, timestamp: outcome.timestamp };
+      entries.push(entry); byId.set(entry.id, entry); owners.set(entry.id, active); oldTip = entry.id;
     }
   }
-}
-
-function hasToolExecution(value: ToolExecution | undefined): boolean {
-  return value?.status === "blocked" ||
-    (value?.status === "executed" && value.args !== null &&
-      typeof value.args === "object" && !Array.isArray(value.args));
-}
-
-/** A terminal Submission cannot resume, so its stale lane pointer is not active work. */
-function clearTerminalOperationPointers(sql: SqlStorage): void {
-  if (!hasTable(sql, "pi_submissions")) return;
-  const lanes = sql.exec<{ key: string; body: string }>(
-    "SELECT key, body FROM pi_values WHERE namespace = ?",
-    laneState("").namespace,
-  ).toArray();
-  for (const lane of lanes) {
-    const state = JSON.parse(lane.body) as LaneState;
-    if (state.currentOperationId === null) continue;
-    const terminal = sql.exec(
-      `SELECT 1 FROM pi_submissions
-       WHERE submission_id = ?
-         AND status IN ('completed', 'failed', 'aborted', 'skipped')
-       LIMIT 1`,
-      state.currentOperationId,
-    ).toArray().length === 1;
-    if (!terminal) continue;
-    sql.exec(
-      `UPDATE pi_values
-       SET body = json_set(body, '$.currentOperationId', json('null'))
-       WHERE namespace = ? AND key = ?`,
-      laneState("").namespace,
-      lane.key,
-    );
-  }
-}
-
-/** Old active tool outcomes cannot be reconstructed from error text or approval facts. */
-function assertActiveToolProvenance(sql: SqlStorage): void {
-  const read = <T>(address: Value<T>): T | undefined => {
-    const row = sql.exec<{ body: string }>(
-      "SELECT body FROM pi_values WHERE namespace = ? AND key = ?",
-      address.namespace, address.key,
-    ).toArray()[0];
-    return row ? JSON.parse(row.body) as T : undefined;
-  };
-  const lanes = sql.exec<{ key: string; body: string }>(
-    "SELECT key, body FROM pi_values WHERE namespace = ?",
-    laneState("").namespace,
-  ).toArray();
-  for (const lane of lanes) {
-    const operationId = (JSON.parse(lane.body) as LaneState).currentOperationId;
-    if (operationId === null) continue;
-    const reject: (reason: string) => never = (reason) => {
-      throw new Error(
-        `Cannot upgrade Pi storage while operation ${operationId} is active: ${reason}. ` +
-        "Complete this operation with the previous runtime version, then retry the upgrade.",
-      );
-    };
-    const meta = read(operationMeta(operationId));
-    const state = read(operationState(operationId));
-    if (!meta || !state) reject("its durable operation metadata is incomplete");
-    let tip = read(branchTip(lane.key));
-    if (tip === undefined) reject("its lane branch tip is missing");
-    const visited = new Set<string>();
-    while (tip !== meta.sourceTipId) {
-      if (tip === null || visited.has(tip)) reject("its operation branch cannot be verified");
-      visited.add(tip);
-      const row = sql.exec<{ body: string }>("SELECT body FROM pi_entries WHERE id = ?", tip).toArray()[0];
-      if (!row) reject(`its canonical entry ${tip} is missing`);
-      const entry = JSON.parse(row.body) as Entry;
-      if (entry.type === "message" && entry.message.role === "toolResult" &&
-        !hasToolExecution(entry.toolExecution)) {
-        reject(`ToolResult ${entry.id} has no reliable execution provenance`);
-      }
-      tip = entry.parentId;
-    }
-    if (state.at !== "tools") continue;
-    for (const call of state.batch.calls) {
-      if (call.status === "planned" || call.status === "completed") continue;
-      if (call.status === "effect_pending") {
-        reject(`Tool invocation ${call.resultEntryId} has an unresolved execution outcome`);
+  const operation = active ? read<{ at: string; batch?: { assistantEntryId: string; turnId: string; calls: Array<{ status: string; sourceIndex: number; resultEntryId: string; replay?: string }> } }>("pi.op.state", active) : undefined;
+  if (active && operation?.at === "tools") {
+    const assistant = byId.get(operation.batch!.assistantEntryId)?.message;
+    if (assistant?.role !== "assistant") throw new Error("Historical Tool batch has no canonical assistant");
+    for (const call of operation.batch!.calls) {
+      const tool = assistant.content[call.sourceIndex];
+      if (tool?.type !== "toolCall") throw new Error("Historical Tool batch index is invalid");
+      if (call.status === "completed") continue;
+      const staged = read<{ type?: string; payload?: Message }>("pi.pending.entry", call.resultEntryId);
+      let outcome = staged?.payload;
+      if (call.status === "outcome_ready" && outcome?.role !== "toolResult") throw new Error("Historical Tool outcome is missing");
+      if (call.status === "effect_pending" && call.replay !== "safe") {
+        outcome = { role: "toolResult", toolCallId: tool.id, toolName: tool.name, content: [{ type: "text", text: "Tool execution was interrupted and its external outcome is unknown. Do not repeat the effect without reconciling it." }], isError: true, timestamp: Date.now() };
       }
-      const pending = read(pendingEntry(call.resultEntryId));
-      if (pending?.type !== "message" || pending.payload.role !== "toolResult" ||
-        !hasToolExecution(pending.toolExecution)) {
-        reject(`pending ToolResult ${call.resultEntryId} has no reliable execution provenance`);
+      if (outcome) {
+        if (outcome.role !== "toolResult" || outcome.toolCallId !== tool.id || outcome.toolName !== tool.name) throw new Error("Historical Tool outcome conflicts with its call");
+        const entry: LegacyEntry = { id: call.resultEntryId, parentId: oldTip, type: "message", message: outcome, timestamp: outcome.timestamp };
+        entries.push(entry); byId.set(entry.id, entry); owners.set(entry.id, active); oldTip = entry.id;
       }
     }
+  } else if (active && operation && !["starting", "checkpoint", "assistant.ready", "assistant.effect_pending", "assistant.retry_wait"].includes(operation.at)) {
+    throw new Error(`Historical operation ${operation.at} needs an explicit migration before it can run`);
   }
-}
-
-/** Upgrade an already-Harness-backed Session before normal reads begin. */
-export function migrateCurrentPiSession(
-  storage: DurableObjectStorage,
-): boolean {
-  return storage.transactionSync(() => {
-    const sql = storage.sql;
-    if (!hasTable(sql, "pi_storage_state")) return false;
-    const version = sql.exec<{ storage_version: number }>(
-      "SELECT storage_version FROM pi_storage_state WHERE singleton = 1",
-    ).one().storage_version;
-    if (version === CURRENT_PI_STORAGE_VERSION) return false;
-    if (version !== 1 && version !== 2) {
-      throw new Error(`Unsupported Cloudflare Pi storage version: ${version}`);
-    }
-    clearTerminalOperationPointers(sql);
-    assertActiveToolProvenance(sql);
-    if (version === 1) removeRetiredRuntimeArtifacts(sql);
-    repairDuplicatedUserSidecars(sql);
-    sql.exec(
-      "UPDATE pi_storage_state SET storage_version = ? WHERE singleton = 1",
-      CURRENT_PI_STORAGE_VERSION,
-    );
-    return true;
-  });
-}
-
-function assertSameToolResult(
-  left: ToolResultMessage,
-  right: ToolResultMessage,
-): void {
-  for (const key of [
-    "toolCallId",
-    "toolName",
-    "content",
-    "details",
-    "usage",
-    "isError",
-  ] as const) {
-    if (!isEqual(left[key], right[key]))
-      throw new Error(`Conflicting legacy Tool result: ${left.toolCallId}`);
-  }
-}
+  const converted = new Map<string, EntryRecord>();
+  const conversationByEntry = new Map<string, ConversationId>();
+  const tips = new Map<ConversationId, string>();
+  const copiedUi: Array<{ id: string; body: string }> = [];
+  const copiedLinks: Array<{ id: string; submissionId: string }> = [];
 
-function legacyToolResultMessage(input: {
-  toolCallId: string;
-  toolName: string;
-  result: AgentToolResult<unknown>;
-  isError: boolean;
-  timestamp: number;
-}): ToolResultMessage {
-  return {
-    role: "toolResult",
-    toolCallId: input.toolCallId,
-    toolName: input.toolName,
-    content: input.result.content ?? [],
-    details: input.result.details as JsonValue | undefined,
-    ...(input.result.usage ? { usage: input.result.usage } : {}),
-    isError: input.isError,
-    timestamp: input.timestamp,
-  };
-}
-
-/**
- * Called synchronously during DO startup, before admission or recovery can execute.
- * Durable pending work keeps its IDs and state; no old executor runs across this boundary.
- * A dry run exercises the real import then rolls back DDL, records, and the format switch together.
- */
-export function migrateLegacyPiSession(
-  storage: DurableObjectStorage,
-  options: { dryRun: boolean },
-): PiSessionMigrationReport {
-  const sql = storage.sql;
-  let report!: PiSessionMigrationReport;
-  const rollbackDryRun = new Error("rollback migration dry run");
-  try {
-    storage.transactionSync(() => {
-      if (hasTable(sql, "pi_storage_state")) {
-        migrateCurrentPiSession(storage);
-        const store = new CloudflareSessionStorage(storage);
-        report = {
-          sourceEntries: 0,
-          entries: store.entries().length,
-          tip: store.branch().at(-1)?.id ?? null,
-          alreadyMigrated: true,
-        };
-        return;
-      }
-      if (
-        !hasTable(sql, "pi_session_entries") ||
-        !hasTable(sql, "pi_session_metadata")
-      )
-        throw new Error("Missing legacy session snapshot");
-      const metadata = sql
-        .exec<{
-          id: string;
-          created_at: string;
-        }>("SELECT id, created_at FROM pi_session_metadata WHERE singleton = 1")
-        .one();
-      const createdAt = time(metadata.created_at);
-      const rows = sql
-        .exec<{
-          seq: number;
-          id: string;
-          body: string;
-        }>("SELECT seq, id, body FROM pi_session_entries ORDER BY seq")
-        .toArray();
-      const records = new Map<string, LegacyRecord>();
-      for (const row of rows) {
-        const raw = object(JSON.parse(row.body));
-        if (raw.id !== row.id || !row.id || records.has(row.id))
-          throw new Error("Invalid or duplicate legacy entry ID");
-        if (
-          raw.parentId !== null &&
-          (typeof raw.parentId !== "string" || !records.has(raw.parentId))
-        )
-          throw new Error(`Missing or forward legacy parent: ${row.id}`);
-        time(raw.timestamp);
-        text(raw.type);
-        records.set(row.id, raw as LegacyRecord);
-      }
-      const converted = new Map<string, Entry>();
-      const resolved = new Map<string, string | null>();
-      const values: Write[] = [];
-      const resolve = (id: unknown): string | null => {
-        if (id === null) return null;
-        const key = text(id);
-        if (converted.has(key)) return key;
-        if (!resolved.has(key))
-          throw new Error(`Unknown legacy reference: ${key}`);
-        return resolved.get(key)!;
-      };
-      for (const [index, row] of rows.entries()) {
-        const raw = records.get(row.id)!;
-        const parentId = resolve(raw.parentId);
-        const timestamp = time(raw.timestamp);
-        const base = { id: raw.id, parentId, timestamp, seq: index + 1 };
-        let entry: Entry | undefined;
-        switch (raw.type) {
-          case "message":
-            entry = { ...base, type: "message", message: message(raw.message) };
-            break;
-          case "custom_message":
-            entry = {
-              ...base,
-              type: "message",
-              message: {
-                role: "custom",
-                customType: text(raw.customType),
-                content: raw.content,
-                details: raw.details,
-                display: raw.display,
-                timestamp,
-              } as AgentMessage,
-            };
-            break;
-          case "custom":
-            entry = {
-              ...base,
-              type: "custom",
-              customType: text(raw.customType),
-              ...(raw.data === undefined
-                ? {}
-                : { data: raw.data as JsonValue }),
-            };
-            break;
-          case "branch_summary":
-            entry = {
-              ...base,
-              type: "branch_summary",
-              fromId: raw.fromId === "root" ? null : resolve(raw.fromId),
-              summary: text(raw.summary),
-              fromHook: raw.fromHook === true,
-              ...(raw.details === undefined
-                ? {}
-                : { details: raw.details as JsonValue }),
-              ...(raw.usage === undefined ? {} : { usage: raw.usage as Usage }),
-            };
-            break;
-          case "compaction": {
-            let retainedTail: AgentMessage[];
-            if (Array.isArray(raw.retainedTail))
-              retainedTail = raw.retainedTail.map(message);
-            else {
-              const kept = text(raw.firstKeptEntryId);
-              const tail: Entry[] = [];
-              let cursor = raw.parentId;
-              let found = false;
-              while (cursor !== null) {
-                const ancestor = records.get(cursor);
-                if (!ancestor)
-                  throw new Error(`Missing compaction ancestor: ${cursor}`);
-                const convertedEntry = converted.get(cursor);
-                if (convertedEntry) tail.unshift(convertedEntry);
-                if (cursor === kept) {
-                  found = true;
-                  break;
-                }
-                cursor = ancestor.parentId;
-              }
-              if (!found)
-                throw new Error(
-                  `Compaction retained tail is not on its historical branch: ${raw.id}`,
-                );
-              retainedTail = tail.flatMap(sessionEntryToContextMessages);
-            }
-            entry = {
-              ...base,
-              type: "compaction",
-              summary: text(raw.summary),
-              retainedTail,
-              tokensBefore: finite(raw.tokensBefore),
-              fromHook: raw.fromHook === true,
-              ...(raw.details === undefined
-                ? {}
-                : { details: raw.details as JsonValue }),
-              ...(raw.usage === undefined ? {} : { usage: raw.usage as Usage }),
-            };
-            break;
-          }
-          case "leaf":
-            resolved.set(raw.id, resolve(raw.targetId));
-            break;
-          case "label": {
-            const target = resolve(raw.targetId);
-            if (!target)
-              throw new Error(`Label has no content target: ${raw.id}`);
-            values.push(
-              raw.label === undefined
-                ? deleteValue(entryLabel(target))
-                : setValue(entryLabel(target), text(raw.label)),
-            );
-            resolved.set(raw.id, parentId);
-            break;
-          }
-          case "session_info":
-            if (raw.name !== undefined)
-              values.push(setValue(sessionName, text(raw.name)));
-            resolved.set(raw.id, parentId);
-            break;
-          case "model_change":
-          case "thinking_level_change":
-          case "active_tools_change": {
-            // Historical settings are values, not conversation entries or a second execution lane.
-            const setting =
-              raw.type === "model_change"
-                ? { provider: text(raw.provider), modelId: text(raw.modelId) }
-                : raw.type === "thinking_level_change"
-                  ? text(raw.thinkingLevel)
-                  : raw.activeToolNames;
-            if (setting === undefined)
-              throw new Error(`Missing legacy configuration: ${raw.id}`);
-            values.push(
-              setValue(value(`springbrand.history.${raw.type}`, raw.id), {
-                at: parentId,
-                timestamp,
-                setting,
-              }),
-            );
-            resolved.set(raw.id, parentId);
-            break;
-          }
-          default:
-            throw new Error(`Unsupported legacy entry type: ${raw.type}`);
-        }
-        if (entry) converted.set(raw.id, entry);
-      }
-      const tip = rows.length ? resolve(rows.at(-1)!.id) : null;
-      if (hasTable(sql, "pi_session_message_submission")) {
-        for (const link of sql
-          .exec<{
-            message_id: string;
-            submission_id: string;
-          }>(
-            "SELECT message_id, submission_id FROM pi_session_message_submission",
-          )
-          .toArray()) {
-          if (converted.get(link.message_id)?.type !== "message")
-            throw new Error(`Dangling message association: ${link.message_id}`);
-          const persistedSubmission =
-            hasTable(sql, "pi_submissions") &&
-            sql
-              .exec(
-                "SELECT 1 FROM pi_submissions WHERE submission_id = ?",
-                link.submission_id,
-              )
-              .toArray().length > 0;
-          if (!persistedSubmission) {
-            let ancestor = converted.get(link.message_id);
-            while (
-              ancestor &&
-              !(ancestor.type === "message" && ancestor.message.role === "user")
-            )
-              ancestor = ancestor.parentId
-                ? converted.get(ancestor.parentId)
-                : undefined;
-            if (!ancestor || link.submission_id !== `import:${ancestor.id}`)
-              throw new Error(
-                `Dangling submission association: ${link.submission_id}`,
-              );
-          }
-        }
-      }
-      const messageOwners = new Map(
-        hasTable(sql, "pi_session_message_submission")
-          ? sql
-              .exec<{ message_id: string; submission_id: string }>(
-                "SELECT message_id, submission_id FROM pi_session_message_submission",
-              )
-              .toArray()
-              .map((row) => [row.message_id, row.submission_id] as const)
-          : [],
-      );
-      const submissionStatuses = new Map(
-        hasTable(sql, "pi_submissions")
-          ? sql
-              .exec<{ submission_id: string; status: string }>(
-                "SELECT submission_id, status FROM pi_submissions",
-              )
-              .toArray()
-              .map((row) => [row.submission_id, row.status] as const)
-          : [],
-      );
-      const terminalSubmission = (submissionId: string): boolean =>
-        LEGACY_TERMINAL_SUBMISSION_STATUSES.has(
-          submissionStatuses.get(submissionId) ?? "",
-        );
-      const results = new Map<string, ToolResultMessage>();
-      for (const entry of converted.values()) {
-        if (entry.type !== "message" || entry.message.role !== "toolResult")
-          continue;
-        const result = entry.message;
-        let ancestor = entry.parentId
-          ? converted.get(entry.parentId)
-          : undefined;
-        let found = false;
-        while (ancestor) {
-          if (
-            ancestor.type === "message" &&
-            ancestor.message.role === "assistant"
-          ) {
-            const call = ancestor.message.content.find(
-              (block) =>
-                block.type === "toolCall" && block.id === result.toolCallId,
-            );
-            if (call?.type === "toolCall") {
-              if (call.name !== result.toolName)
-                throw new Error("Mismatched legacy Tool name");
-              found = true;
-              break;
-            }
-          }
-          ancestor = ancestor.parentId
-            ? converted.get(ancestor.parentId)
-            : undefined;
-        }
-        if (!found)
-          throw new Error(`Orphan legacy Tool result: ${result.toolCallId}`);
-        const owner = messageOwners.get(entry.id);
-        if (owner) {
-          const key = `${owner}:${result.toolCallId}`;
-          const previous = results.get(key);
-          if (previous) assertSameToolResult(previous, result);
-          results.set(key, result);
+  await storage.transaction(async () => {
+    const initial = await session.commit((tx) => tx.createConversation({ ownership: { kind: "ownerless" } }), BACKGROUND_CONTEXT);
+    let selected = initial.id;
+    for (const entry of entries) {
+      await session.commit(async (tx) => {
+        if (!entry.id || (entry.parentId !== null && !byId.has(entry.parentId))) {
+          throw new Error(`Historical Pi entry has a missing parent: ${entry.id}`);
         }
-      }
-      if (hasTable(sql, "pi_pending_steers")) {
-        for (const steer of sql
-          .exec<{
-            submission_id: string;
-            message_id: string;
-            canonical_json: string;
-          }>(
-            "SELECT submission_id, message_id, canonical_json FROM pi_pending_steers",
-          )
-          .toArray()) {
-          if (
-            !hasTable(sql, "pi_submissions") ||
-            !sql
-              .exec(
-                "SELECT 1 FROM pi_submissions WHERE submission_id = ?",
-                steer.submission_id,
-              )
-              .toArray().length
-          ) {
-            throw new Error("Dangling pending steer submission");
-          }
-          const pending = message(JSON.parse(steer.canonical_json));
-          if (pending.role !== "user" || converted.has(steer.message_id)) {
-            throw new Error("Invalid or already committed pending steer");
-          }
+        const parent = entry.parentId ? converted.get(entry.parentId) : undefined;
+        if (entry.parentId && !parent) throw new Error(`Historical Pi entry is out of order: ${entry.id}`);
+        let conversationId = parent ? conversationByEntry.get(entry.parentId!)! : initial.id;
+        if (parent && tips.get(conversationId) !== entry.parentId) {
+          conversationId = (await tx.forkConversation(conversationId, parent.id, { ownership: { kind: "ownerless" } })).id;
         }
-      }
-      // Business records keep their stable IDs and payloads. Validate ownership before switching.
-      if (hasTable(sql, "pi_submissions")) {
-        const submissions = sql
-          .exec<{
-            submission_id: string;
-            status: string;
-            queued_input_json: string | null;
-            queued_ui_message_json: string | null;
-            user_message_id: string | null;
-            regenerate_message_id: string | null;
-          }>("SELECT * FROM pi_submissions")
-          .toArray();
-        const knownSubmissions = new Set(
-          submissions.map((row) => row.submission_id),
-        );
-        for (const submission of submissions) {
-          if (submission.status !== "pending" || !submission.queued_input_json)
-            continue;
-          const queued = message(JSON.parse(submission.queued_input_json));
-          if (queued.role !== "user" || !submission.user_message_id) {
-            throw new Error("Invalid queued user message");
-          }
-          if (
-            submission.queued_ui_message_json &&
-            object(JSON.parse(submission.queued_ui_message_json)).role !==
-              "user"
-          ) {
-            throw new Error("Invalid queued UI user message");
-          }
-          if (submission.regenerate_message_id) {
-            const target = converted.get(submission.regenerate_message_id);
-            if (
-              target?.type !== "message" ||
-              target.message.role !== "user" ||
-              !isEqual(
-                typeof target.message.content === "string"
-                  ? [{ type: "text", text: target.message.content }]
-                  : target.message.content,
-                typeof queued.content === "string"
-                  ? [{ type: "text", text: queued.content }]
-                  : queued.content,
-              )
-            ) {
-              throw new Error("Invalid queued regeneration target");
-            }
-          } else if (converted.has(submission.user_message_id)) {
-            throw new Error("Queued user message is already committed");
-          }
-        }
-        const toolInputs = new Set<string>();
-        if (hasTable(sql, "pi_recovery_milestones")) {
-          const bySubmission = new Map<string, string[]>();
-          for (const row of sql
-            .exec<{
-              submission_id: string;
-              body: string;
-            }>(
-              "SELECT submission_id, body FROM pi_recovery_milestones ORDER BY submission_id, seq",
-            )
-            .toArray()) {
-            if (!knownSubmissions.has(row.submission_id))
-              throw new Error("Dangling recovery submission");
-            const record = decodePiToolRecoveryMilestone(row.body);
-            if (!record || record.turnId !== row.submission_id) {
-              if (LEGACY_TERMINAL_SUBMISSION_STATUSES.has(
-                submissionStatuses.get(row.submission_id) ?? "",
-              )) continue;
-              throw new Error("Invalid legacy recovery milestone");
-            }
-            const bodies = bySubmission.get(row.submission_id) ?? [];
-            bodies.push(row.body);
-            bySubmission.set(row.submission_id, bodies);
-            if (record.type === "tool-result" || record.type === "tool-error") {
-              for (const result of replayPiToolRecovery([row.body])
-                .canonicalToolResults) {
-                const key = `${row.submission_id}:${result.toolCallId}`;
-                const previous = results.get(key);
-                if (previous && !terminalSubmission(row.submission_id)) {
-                  assertSameToolResult(previous, result);
-                }
-                if (!previous || !terminalSubmission(row.submission_id)) {
-                  results.set(key, result);
-                }
-              }
-            }
-          }
-          for (const [submissionId, bodies] of bySubmission) {
-            const recovered = replayPiToolRecovery(bodies);
-            for (const call of Object.values(recovered.toolCalls)) {
-              if (call.input !== undefined || call.retry !== undefined)
-                toolInputs.add(`${submissionId}:${call.toolCallId}`);
-              if (call.result) {
-                const key = `${submissionId}:${call.toolCallId}`;
-                const previous = results.get(key);
-                if (previous && !terminalSubmission(submissionId)) {
-                  assertSameToolResult(previous, call.result);
-                }
-                if (!previous || !terminalSubmission(submissionId)) {
-                  results.set(key, call.result);
-                }
-              }
-            }
-          }
-        }
-        if (hasTable(sql, "pi_session_message_submission")) {
-          for (const row of sql
-            .exec<{
-              message_id: string;
-              submission_id: string;
-            }>(
-              "SELECT message_id, submission_id FROM pi_session_message_submission",
-            )
-            .toArray()) {
-            const entry = converted.get(row.message_id);
-            if (
-              entry?.type === "message" &&
-              entry.message.role === "assistant"
-            ) {
-              for (const block of entry.message.content)
-                if (block.type === "toolCall")
-                  toolInputs.add(`${row.submission_id}:${block.id}`);
+        const message = entry.message;
+        let record: EntryRecord;
+        if (entry.type === "message" && message) {
+          const role = (message as { role: string }).role;
+          record = await tx.appendEntry(conversationId, {
+            kind: role === "toolResult" ? "pi.tool-result" : role === "custom" ? "springbrand.custom" : `pi.${role}`,
+            ...(role === "custom" ? {} : { model: [message] }),
+            data: { displayId: entry.id, createdAt: timestamp(entry), ...(role === "custom" ? { message: message as unknown as Record<string, never> } : {}), ...(role === "toolResult" ? { diagnostics: [] } : {}) },
+          });
+        } else if (entry.type === "compaction" || entry.type === "branch_summary") {
+          record = await tx.appendEntry(conversationId, {
+            kind: "pi.compaction", model: [{ role: "user", content: `<summary>\n${entry.summary ?? ""}\n</summary>`, timestamp: timestamp(entry) }],
+            head: "self", data: { reason: "manual", displayId: entry.id, createdAt: timestamp(entry) },
+          });
+          let retainedTail = entry.retainedTail;
+          if (!retainedTail && entry.firstKeptEntryId) {
+            const tail: Message[] = [];
+            let cursor = entry.parentId;
+            while (cursor) {
+              const ancestor = byId.get(cursor);
+              if (!ancestor) throw new Error("Historical compaction has a missing ancestor");
+              if (ancestor.message) tail.unshift(ancestor.message);
+              if (cursor === entry.firstKeptEntryId) break;
+              cursor = ancestor.parentId;
             }
+            if (!cursor) throw new Error("Historical compaction tail is outside its branch");
+            retainedTail = tail;
           }
-        }
-        for (const table of [
-          "pi_tool_settlements",
-          "pi_approvals",
-          "pi_tool_interactions",
-        ]) {
-          if (!hasTable(sql, table)) continue;
-          for (const row of sql
-            .exec<{
-              submission_id: string;
-              tool_call_id: string;
-            }>(`SELECT submission_id, tool_call_id FROM ${table}`)
-            .toArray()) {
-            if (
-              !knownSubmissions.has(row.submission_id) ||
-              !toolInputs.has(`${row.submission_id}:${row.tool_call_id}`)
-            )
-              throw new Error(`Dangling Tool association in ${table}`);
-          }
-        }
-      }
-      if (hasTable(sql, "pi_tool_settlements")) {
-        for (const row of sql
-          .exec<{
-            submission_id: string;
-            tool_call_id: string;
-            tool_name: string;
-            result: string;
-            is_error: number;
-            created_at: number;
-          }>(
-            "SELECT submission_id, tool_call_id, tool_name, result, is_error, created_at FROM pi_tool_settlements",
-          )
-          .toArray()) {
-          const expected = legacyToolResultMessage({
-            toolCallId: row.tool_call_id,
-            toolName: row.tool_name,
-            result: JSON.parse(row.result) as AgentToolResult<unknown>,
-            isError: row.is_error === 1,
-            timestamp: row.created_at,
-          });
-          const recorded = results.get(
-            `${row.submission_id}:${row.tool_call_id}`,
-          );
-          if (recorded && !terminalSubmission(row.submission_id)) {
-            assertSameToolResult(recorded, expected);
+          for (const retained of retainedTail ?? []) {
+            await tx.appendEntry(conversationId, { kind: "springbrand.retained-context", model: [retained] });
           }
+        } else {
+          record = await tx.appendEntry(conversationId, { kind: "springbrand.history", data: { displayId: entry.id, createdAt: timestamp(entry), record: entry as unknown as Record<string, never> } });
         }
+        converted.set(entry.id, record);
+        conversationByEntry.set(entry.id, conversationId);
+        tips.set(conversationId, entry.id);
+        if (entry.id === oldTip) selected = conversationId;
+        if (owners.has(entry.id)) copiedLinks.push({ id: String(record.id), submissionId: owners.get(entry.id)! });
+        if (ui.has(entry.id)) copiedUi.push({ id: String(record.id), body: ui.get(entry.id)! });
+      }, BACKGROUND_CONTEXT);
+    }
+    if (oldTip === null && entries.length) {
+      selected = (await session.commit((tx) => tx.createConversation({ ownership: { kind: "ownerless" } }), BACKGROUND_CONTEXT)).id;
+    } else if (oldTip && tips.get(selected) !== oldTip) {
+      const tip = converted.get(oldTip);
+      if (!tip) throw new Error("Historical branch tip is missing");
+      selected = (await session.commit((tx) => tx.forkConversation(tip.conversationId, tip.id, { ownership: { kind: "ownerless" } }), BACKGROUND_CONTEXT)).id;
+    }
+    await session.commit(async (tx) => {
+      const pointer = await tx.doc(RuntimeConversation);
+      pointer.id = selected;
+      const config = read<Record<string, unknown>>("pi.lane.config", "main");
+      if (config) {
+        const agent = await tx.doc(AgentDoc, selected);
+        if (config.model) agent.model = config.model as { provider: string; modelId: string };
+        if (typeof config.thinkingLevel === "string") agent.thinkingLevel = config.thinkingLevel as typeof agent.thinkingLevel;
+        if (Array.isArray(config.activeToolNames)) agent.tools = config.activeToolNames as string[];
       }
-      // The old table remains an inert backup. Failure rolls back DDL, data and the version together.
-      sql.exec(
-        "ALTER TABLE pi_session_entries RENAME TO pi_legacy_entries_backup",
-      );
-      const store = new CloudflareSessionStorage(storage);
-      removeRetiredRuntimeArtifacts(sql);
-      for (const entry of converted.values()) {
-        const { seq: _seq, timestamp, ...next } = entry;
-        const writes: Write[] = [insertEntry(next)];
-        const usage =
-          entry.type === "message" &&
-          (entry.message.role === "assistant" ||
-            entry.message.role === "toolResult")
-            ? entry.message.usage
-            : entry.type === "compaction" || entry.type === "branch_summary"
-              ? entry.usage
-              : undefined;
-        if (usage) {
-          let usageId = `migration:usage:${entry.id}`;
-          while (records.has(usageId)) usageId = `migration:${usageId}`;
-          writes.push(
-            insertUsage({
-              id: usageId,
-              entryId: entry.id,
-              usage,
-              adjustment: false,
-            }),
-          );
+      if (active) {
+        const chain: LegacyEntry[] = [];
+        for (let id = oldTip; id; id = byId.get(id)?.parentId ?? null) {
+          const entry = byId.get(id);
+          if (!entry) throw new Error("Active historical operation has an invalid branch");
+          chain.unshift(entry);
         }
-        const result = store.commitSync(writes);
-        // Import timestamps belong to the historical snapshot, not the time of the migration.
-        sql.exec(
-          "UPDATE pi_entries SET timestamp = ?, body = json_set(body, '$.timestamp', ?) WHERE id = ?",
-          timestamp,
-          timestamp,
-          entry.id,
-        );
-        entry.seq = result.firstSeq;
+        const owned = chain.filter((entry) => owners.get(entry.id) === active);
+        const prompt = owned.find((entry) => entry.message?.role === "user");
+        if (!prompt) throw new Error("Active historical operation has no canonical user input");
+        const promptEntry = converted.get(prompt.id)!;
+        const submission = await tx.createSubmission({ conversationId: selected, requestId: active, type: "input", status: "placed", entry: promptEntry.id });
+        const assistant = [...owned].reverse().find((entry) => entry.message?.role === "assistant");
+        const settledCalls = new Set(owned.flatMap((entry) => entry.message?.role === "toolResult" ? [entry.message.toolCallId] : []));
+        const calls = assistant?.message?.role === "assistant"
+          ? assistant.message.content.filter((part): part is import("@earendil-works/pi-ai").ToolCall => part.type === "toolCall").filter((part) => !settledCalls.has(part.id))
+          : [];
+        // Initialize the official generation checkpoint; only unfinished calls enter its pending list.
+        const checkpoint: GenerationCheckpoint = assistant && calls.length
+          ? { phase: "tools", assistant: converted.get(assistant.id)!.id, tools: [], pending: calls.map((call) => call.id) }
+          : { phase: "prepare", attempt: 0 };
+        const generation = await tx.createTask({ definition: { ...GenerationTask.definition, initial: () => checkpoint } }, {}, { conversationId: selected, ownership: { kind: "conversation" } });
+        const live = await tx.doc(LiveDoc, selected);
+        live.run = { taskId: generation, inputs: [submission.id] };
+        if (calls.length) live.tools = calls.map((call) => ({ callId: call.id, name: call.name, status: "pending" }));
       }
-      const parents = new Set(
-        [...converted.values()].map((entry) => entry.parentId),
-      );
-      for (const entry of converted.values()) {
-        if (entry.id !== tip && !parents.has(entry.id))
-          values.push(setValue(branchTip(`history:${entry.id}`), entry.id));
-      }
-      store.commitSync([...values, setValue(branchTip("main"), tip)]);
-      repairDuplicatedUserSidecars(sql);
-      sql.exec(
-        "UPDATE pi_storage_state SET session_id = ?, created_at = ? WHERE singleton = 1",
-        metadata.id,
-        createdAt,
-      );
-      report = {
-        sourceEntries: rows.length,
-        entries: converted.size,
-        tip,
-        alreadyMigrated: false,
-      };
-      if (options.dryRun) throw rollbackDryRun;
-    });
-  } catch (error) {
-    if (error !== rollbackDryRun) throw error;
-  }
-  return report;
+      (await tx.doc(Migration)).completed = true;
+    }, BACKGROUND_CONTEXT);
+    sql.exec("CREATE TABLE IF NOT EXISTS pi_session_entry_submission (entry_id TEXT PRIMARY KEY, submission_id TEXT NOT NULL)");
+    for (const link of copiedLinks) sql.exec("INSERT OR IGNORE INTO pi_session_entry_submission VALUES (?,?)", link.id, link.submissionId);
+    for (const sidecar of copiedUi) sql.exec("INSERT OR IGNORE INTO pi_message_ui VALUES (?,?)", sidecar.id, sidecar.body);
+    for (const table of ["pi_approvals", "pi_tool_interactions"]) {
+      if (!hasTable(sql, table)) continue;
+      const rows = sql.exec<{ rowid: number; input_json: string }>(`SELECT rowid,input_json FROM ${table} WHERE tool_name='ask_user'`).toArray();
+      for (const row of rows) sql.exec(`UPDATE ${table} SET input_json=? WHERE rowid=?`, JSON.stringify(normalizeLegacyAskUserInput(JSON.parse(row.input_json))), row.rowid);
+    }
+    if (hasTable(sql, "cf_agents_schedules")) {
+      for (const callback of ["_cleanupStreamBuffers", "_piPlannedContinuation", "_piContinuationWatchdog", "_piApprovalContinuation", "_chatRecoveryRetry"]) sql.exec("DELETE FROM cf_agents_schedules WHERE callback=?", callback);
+    }
+  });
 }