@lasso-ai/cli 2.1.13 → 2.1.14

package/dist/cli/runtime-client.js

@@ -88,9 +88,13 @@ class RuntimeClient {
         }
         const executor = new index_js_1.ActionExecutor(projectPath, opts?.browserContextProvider);
         const verifier = new index_js_1.VerificationEngine(projectPath);
+        // Idempotent: called on agent_completed AND on stream failure, so the
+        // diff is always cached before the safety-net restore wipes the edits.
+        let finalized = false;
         const finalizeGuard = async () => {
-            if (!guard)
+            if (!guard || finalized)
                 return;
+            finalized = true;
             try {
                 const changes = guard.diff();
                 const applyable = changes.filter((c) => !c.binary);
@@ -174,10 +178,11 @@ class RuntimeClient {
         }
         finally {
             reader.releaseLock();
-            // Safety net: if the stream died mid-run, still restore the workspace.
-            if (guard) {
-                guard.restore().catch(() => { });
-            }
+            // Stream died before agent_completed (network cut, server crash, model
+            // error surfaced as an error event): cache the diff first so the user's
+            // changes are recoverable, then restore the workspace. On the normal
+            // path finalizeGuard already ran and this is a no-op.
+            await finalizeGuard();
         }
     }
     /** Execute a tool_call request locally against the real project. */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lasso-ai/cli",
-  "version": "2.1.13",
+  "version": "2.1.14",
   "description": "Select any part of your running app, describe a change, and let AI edit the real source code.",
   "type": "commonjs",
   "main": "dist/cli/index.js",