feat(workflow): persist exact replay values

Author: Waishnav

src/db/migrations.ts

@@ -37,6 +37,11 @@ const migrations: Migration[] = [
     name: "workflow-replay-provenance",
     up: migrateWorkflowReplayProvenance,
   },
+  {
+    version: 7,
+    name: "workflow-exact-replay",
+    up: migrateWorkflowExactReplay,
+  },
 ];
 
 export function migrateDatabase(sqlite: Database.Database): void {
@@ -270,6 +275,7 @@ function migrateWorkflowJournal(sqlite: Database.Database): void {
       provider_session_id text,
       response_text text,
       structured_json text,
+      return_value_json text,
       error text,
       isolation text not null default 'shared',
       worktree_path text,
@@ -302,6 +308,10 @@ function migrateWorkflowReplayProvenance(sqlite: Database.Database): void {
   `);
 }
 
+function migrateWorkflowExactReplay(sqlite: Database.Database): void {
+  addColumnIfMissing(sqlite, "workflow_agent_calls", "return_value_json", "text");
+}
+
 function addColumnIfMissing(
   sqlite: Database.Database,
   table: "workspace_sessions" | "local_agent_sessions" | "workflow_agent_calls",

src/db/schema.ts

@@ -169,6 +169,7 @@ export const workflowAgentCalls = sqliteTable(
     providerSessionId: text("provider_session_id"),
     responseText: text("response_text"),
     structuredJson: text("structured_json"),
+    returnValueJson: text("return_value_json"),
     error: text("error"),
     errorKind: text("error_kind"),
     replayMatch: text("replay_match"),

src/oauth-store.test.ts

@@ -47,6 +47,7 @@ async function testDatabaseConfiguration(stateDir: string): Promise<void> {
       { version: 4, name: "local-agent-effort-rename" },
       { version: 5, name: "workflow-journal" },
       { version: 6, name: "workflow-replay-provenance" },
+      { version: 7, name: "workflow-exact-replay" },
     ]);
   } finally {
     database.close();

src/workflow-store.test.ts

@@ -85,6 +85,7 @@ try {
     callIndex: 0,
     responseText: "done",
     structuredJson: JSON.stringify({ ok: true }),
+    returnValueJson: JSON.stringify({ ok: true, exact: true }),
     providerSessionId: "sess_1",
     dirty: true,
   });
@@ -95,6 +96,7 @@ try {
   assert.equal(call?.providerSessionId, "sess_1");
   assert.equal(call?.effort, "high");
   assert.equal(call?.prompt, "review");
+  assert.equal(call?.returnValueJson, JSON.stringify({ ok: true, exact: true }));
   assert.equal(call?.replayReason, "identity_changed:prompt");
 
   store.beginAgentCall({

src/workflow-store.ts

@@ -70,6 +70,7 @@ export interface CompleteAgentCallInput {
   callIndex: number;
   responseText?: string;
   structuredJson?: string;
+  returnValueJson?: string;
   providerSessionId?: string;
   dirty?: boolean;
   worktreePath?: string;
@@ -153,6 +154,7 @@ interface WorkflowAgentCallRow {
   provider_session_id: string | null;
   response_text: string | null;
   structured_json: string | null;
+  return_value_json: string | null;
   error: string | null;
   error_kind: string | null;
   replay_match: string | null;
@@ -673,6 +675,13 @@ export class WorkflowStore {
     if (input.structuredJson !== undefined) {
       assertTextSize(input.structuredJson, WORKFLOW_LIMITS.structuredJsonBytes, "structuredJson");
     }
+    if (input.returnValueJson !== undefined) {
+      assertTextSize(
+        input.returnValueJson,
+        WORKFLOW_LIMITS.replayValueJsonBytes,
+        "returnValueJson",
+      );
+    }
     const now = isoNow();
     const status: WorkflowAgentCallStatus = input.fromCache ? "from_cache" : "completed";
     this.database.sqlite
@@ -682,6 +691,7 @@ export class WorkflowStore {
           from_cache = ?,
           response_text = ?,
           structured_json = ?,
+          return_value_json = ?,
           provider_session_id = coalesce(?, provider_session_id),
           worktree_path = coalesce(?, worktree_path),
           dirty = ?,
@@ -694,6 +704,7 @@ export class WorkflowStore {
         input.fromCache ? "true" : "false",
         input.responseText ?? null,
         input.structuredJson ?? null,
+        input.returnValueJson ?? null,
         input.providerSessionId ?? null,
         input.worktreePath ?? null,
         input.dirty === undefined ? null : input.dirty ? "true" : "false",
@@ -894,6 +905,7 @@ function rowToAgentCall(row: WorkflowAgentCallRow): WorkflowAgentCallRecord {
     providerSessionId: row.provider_session_id ?? undefined,
     responseText: row.response_text ?? undefined,
     structuredJson: row.structured_json ?? undefined,
+    returnValueJson: row.return_value_json ?? undefined,
     error: row.error ?? undefined,
     errorKind: (row.error_kind as WorkflowErrorKind | null) ?? undefined,
     replayMatch:

src/workflow-types.ts

@@ -60,6 +60,7 @@ export const WORKFLOW_LIMITS = {
   eventDataJsonBytes: 8 * 1024,
   responseTextBytes: 1 * 1024 * 1024,
   structuredJsonBytes: 256 * 1024,
+  replayValueJsonBytes: 1 * 1024 * 1024,
   resultJsonBytes: 256 * 1024,
   argsJsonBytes: 64 * 1024,
   scriptSourceBytes: 512 * 1024,
@@ -149,6 +150,7 @@ export interface WorkflowAgentCallRecord {
   providerSessionId?: string;
   responseText?: string;
   structuredJson?: string;
+  returnValueJson?: string;
   error?: string;
   errorKind?: WorkflowErrorKind;
   replayMatch?: "same_index" | "compatible_key";