Update: Search checkpoint across all Runs in Job for --resume-from

packages/perstack/src/lib/context.ts

@@ -2,7 +2,7 @@ import type { Checkpoint, PerstackConfig, ProviderConfig, ProviderName } from "@
 import { getEnv } from "./get-env.js"
 import { getPerstackConfig } from "./perstack-toml.js"
 import { getProviderConfig } from "./provider-config.js"
-import { getCheckpointById, getMostRecentCheckpoint, getMostRecentRunInJob } from "./run-manager.js"
+import { findCheckpointInJob, getMostRecentCheckpoint, getMostRecentRunInJob } from "./run-manager.js"
 
 const defaultProvider: ProviderName = "anthropic"
 const defaultModel = "claude-sonnet-4-5"
@@ -37,8 +37,7 @@ export async function resolveRunContext(input: ResolveRunContextInput): Promise<
       throw new Error("--resume-from requires --continue or --continue-job")
     }
     const jobId = input.continueJob ?? (await getMostRecentCheckpoint()).jobId
-    const run = await getMostRecentRunInJob(jobId)
-    checkpoint = await getCheckpointById(jobId, run.runId, input.resumeFrom)
+    checkpoint = await findCheckpointInJob(jobId, input.resumeFrom)
   } else if (input.continueJob) {
     const run = await getMostRecentRunInJob(input.continueJob)
     checkpoint = await getMostRecentCheckpoint(run.jobId, run.runId)

packages/perstack/src/lib/run-manager.ts

@@ -185,6 +185,26 @@ export async function getCheckpointById(
   const checkpoint = await readFile(checkpointPath, "utf-8")
   return checkpointSchema.parse(JSON.parse(checkpoint))
 }
+
+export async function findCheckpointInJob(jobId: string, checkpointId: string): Promise<Checkpoint> {
+  const runs = await getRunsByJobId(jobId)
+  for (const run of runs) {
+    const runDir = getRunDir(jobId, run.runId)
+    if (!existsSync(runDir)) {
+      continue
+    }
+    const files = await readdir(runDir)
+    const checkpointFile = files.find(
+      (file) => file.startsWith("checkpoint-") && file.includes(`-${checkpointId}.`),
+    )
+    if (checkpointFile) {
+      const checkpointPath = path.resolve(runDir, checkpointFile)
+      const checkpoint = await readFile(checkpointPath, "utf-8")
+      return checkpointSchema.parse(JSON.parse(checkpoint))
+    }
+  }
+  throw new Error(`Checkpoint ${checkpointId} not found in job ${jobId}`)
+}
 export async function getCheckpointsWithDetails(
   jobId: string,
   runId: string,