@vellumai/assistant 0.8.1 → 0.8.2

package/src/workspace/migrations/runner.ts

@@ -22,6 +22,13 @@ export type CheckpointFile = {
     string,
     { appliedAt: string; status?: WorkspaceMigrationStatus }
   >;
+  /** Persisted "fresh workspace" flag. Set to `true` by the runner when the
+   *  checkpoint file is being created for the first time; cleared after the
+   *  initial migration sweep finishes. Survives crashes mid-first-boot so
+   *  seeding migrations that fall later in the sequence still observe the
+   *  brand-new state. Absent on workspaces that pre-date this field — those
+   *  are treated as not-new. */
+  isNewWorkspace?: boolean;
 };
 
 function getCheckpointPath(workspaceDir: string): string {
@@ -72,29 +79,43 @@ export async function runWorkspaceMigrations(
   workspaceDir: string,
   migrations: WorkspaceMigration[],
 ): Promise<void> {
-  const seen = new Set<string>();
+  const migrationsById = new Map<string, WorkspaceMigration>();
   for (const m of migrations) {
-    if (seen.has(m.id)) {
+    if (migrationsById.has(m.id)) {
       throw new Error(`Duplicate workspace migration id: "${m.id}"`);
     }
-    seen.add(m.id);
+    migrationsById.set(m.id, m);
   }
 
-  // Detect a brand-new workspace: no checkpoint file existed before this run.
-  // Captured before loadCheckpoints so seeding migrations can distinguish a
-  // first-ever boot from an upgrade where the user may have cleared seeded files.
+  // The checkpoint file is written *before* each migration's run() (to record
+  // the "started" status), so file-absence alone cannot be used to detect a
+  // brand-new workspace — a crash mid-first-boot would flip the next boot's
+  // verdict to "upgrade" before later seeding migrations have run. Persist
+  // the flag inside the file instead so it survives across reboots.
+  const checkpointExisted =
+    readTextFileSync(getCheckpointPath(workspaceDir)) != null;
+  const checkpoints = loadCheckpoints(workspaceDir);
+  if (!checkpointExisted) {
+    checkpoints.isNewWorkspace = true;
+  }
   const ctx: MigrationRunContext = {
-    isNewWorkspace: readTextFileSync(getCheckpointPath(workspaceDir)) == null,
+    isNewWorkspace: checkpoints.isNewWorkspace === true,
   };
 
-  const checkpoints = loadCheckpoints(workspaceDir);
-
   for (const [id, entry] of Object.entries(checkpoints.applied)) {
     if (entry.status === "started" || entry.status === "rolling_back") {
       log.warn(
         `Workspace migration "${id}" was interrupted during a previous run; will re-run`,
       );
       delete checkpoints.applied[id];
+    } else if (
+      entry.status === "failed" &&
+      migrationsById.get(id)?.retryFailedCheckpoint === true
+    ) {
+      log.warn(
+        `Workspace migration "${id}" failed during a previous run; will retry`,
+      );
+      delete checkpoints.applied[id];
     }
   }
 
@@ -136,6 +157,15 @@ export async function runWorkspaceMigrations(
     };
     saveCheckpoints(workspaceDir, checkpoints);
   }
+
+  // First-boot sweep finished cleanly — clear the flag so future runs (and
+  // future seeding migrations added later) treat the workspace as an upgrade.
+  // A crash above this point leaves the flag set, so the retry on the next
+  // boot still observes a fresh workspace.
+  if (checkpoints.isNewWorkspace === true) {
+    checkpoints.isNewWorkspace = false;
+    saveCheckpoints(workspaceDir, checkpoints);
+  }
 }
 
 /**

package/src/workspace/migrations/types.ts

@@ -18,6 +18,10 @@ export interface WorkspaceMigration {
    *  it). Must be idempotent — safe to re-run if it was interrupted.
    *  Both synchronous and asynchronous migrations are supported. */
   run(workspaceDir: string, ctx?: MigrationRunContext): void | Promise<void>;
+  /** Retry a prior failed checkpoint on the next run.
+   *  Only set this for migrations whose run() is safe to retry after a
+   *  partially successful failed attempt. */
+  retryFailedCheckpoint?: boolean;
   /** Reverse the migration. Receives the workspace directory path.
    *  Must be idempotent — safe to re-run if it was interrupted.
    *  Both synchronous and asynchronous rollbacks are supported. */

package/examples/plugins/echo/bun.lock

@@ -1,25 +0,0 @@
-{
-  "lockfileVersion": 1,
-  "configVersion": 1,
-  "workspaces": {
-    "": {
-      "name": "@vellumai/plugin-echo-example",
-      "devDependencies": {
-        "@types/bun": "1.3.10",
-        "@types/node": "25.5.0",
-        "typescript": "5.9.3",
-      },
-    },
-  },
-  "packages": {
-    "@types/bun": ["@types/bun@1.3.10", "", { "dependencies": { "bun-types": "1.3.10" } }, "sha512-0+rlrUrOrTSskibryHbvQkDOWRJwJZqZlxrUs1u4oOoTln8+WIXBPmAuCF35SWB2z4Zl3E84Nl/D0P7803nigQ=="],
-
-    "@types/node": ["@types/node@25.5.0", "", { "dependencies": { "undici-types": "~7.18.0" } }, "sha512-jp2P3tQMSxWugkCUKLRPVUpGaL5MVFwF8RDuSRztfwgN1wmqJeMSbKlnEtQqU8UrhTmzEmZdu2I6v2dpp7XIxw=="],
-
-    "bun-types": ["bun-types@1.3.10", "", { "dependencies": { "@types/node": "*" } }, "sha512-tcpfCCl6XWo6nCVnpcVrxQ+9AYN1iqMIzgrSKYMB/fjLtV2eyAVEg7AxQJuCq/26R6HpKWykQXuSOq/21RYcbg=="],
-
-    "typescript": ["typescript@5.9.3", "", { "bin": { "tsc": "bin/tsc", "tsserver": "bin/tsserver" } }, "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw=="],
-
-    "undici-types": ["undici-types@7.18.2", "", {}, "sha512-AsuCzffGHJybSaRrmr5eHr81mwJU3kjw6M+uprWvCXiNeN9SOGwQ3Jn8jb8m3Z6izVgknn1R0FTCEAP2QrLY/w=="],
-  }
-}