@lasso-ai/cli 1.3.7 → 1.3.9

package/dist/cli/bridge.js

@@ -1183,47 +1183,14 @@ async function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort
             }
             else if (msg.type === "apply") {
                 try {
-                    socket.send(JSON.stringify({
-                        type: "agent_status",
-                        taskId: msg.taskId,
-                        status: "working",
-                        message: "Re-generating changes with current file state..."
-                    }));
-                    // Re-run the edit request to get fresh changes based on current file state
-                    const editRequest = editRequests.get(msg.taskId);
-                    const editConfig = editConfigs.get(msg.taskId);
-                    if (!editRequest || !editConfig) {
-                        socket.send(JSON.stringify({
-                            type: "agent_status",
-                            taskId: msg.taskId,
-                            status: "error",
-                            message: "Could not re-generate changes: original request not found."
-                        }));
-                        return;
-                    }
-                    const controller = new AbortController();
-                    taskControllers.set(msg.taskId, controller);
-                    const freshProposal = await (0, agent_1.proposeChanges)(cwd, editRequest, editConfig, controller.signal, (message, detail, level) => {
-                        if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
-                            socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: level === "error" ? "error" : "working", message, detail }));
-                        }
-                    }, (prompt) => {
-                        if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
-                            socket.send(JSON.stringify({ type: "agent_prompt", taskId: msg.taskId, prompt }));
-                        }
-                    });
-                    if (controller.signal.aborted) {
-                        socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: "stopped", message: "Agent stopped." }));
-                        return;
-                    }
-                    // Apply the fresh changes
-                    const filePaths = [...new Set(freshProposal.changes.map(c => c.filePath))];
+                    // Create snapshot before applying changes
+                    const filePaths = [...new Set(msg.changes.map(c => c.filePath))];
                     const parentSnapshot = (0, snapshot_1.getSnapshotByTaskId)(cwd, msg.taskId);
                     // Create pre-apply snapshot
-                    const preSnapshot = (0, snapshot_1.createSnapshot)(cwd, msg.taskId, `Pre-apply: ${freshProposal.changes.length} file${freshProposal.changes.length === 1 ? "" : "s"}`, filePaths, parentSnapshot?.id);
+                    const preSnapshot = (0, snapshot_1.createSnapshot)(cwd, msg.taskId, `Pre-apply: ${msg.changes.length} file${msg.changes.length === 1 ? "" : "s"}`, filePaths, parentSnapshot?.id);
                     const snapshots = [];
                     const planned = new Map();
-                    for (const change of freshProposal.changes) {
+                    for (const change of msg.changes) {
                         const isNew = !change.oldString;
                         const filePath = resolveProposedFile(cwd, change.filePath, true);
                         const exists = node_fs_1.default.existsSync(filePath);
@@ -1253,7 +1220,7 @@ async function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort
                         node_fs_1.default.writeFileSync(filePath, nextContent);
                     }
                     // Create post-apply snapshot
-                    (0, snapshot_1.createSnapshot)(cwd, msg.taskId, `Applied: ${freshProposal.changes.length} file${freshProposal.changes.length === 1 ? "" : "s"}`, filePaths, preSnapshot.id);
+                    (0, snapshot_1.createSnapshot)(cwd, msg.taskId, `Applied: ${msg.changes.length} file${msg.changes.length === 1 ? "" : "s"}`, filePaths, preSnapshot.id);
                     // Run build validation after applying changes
                     const validation = await validateBuild(cwd);
                     if (!validation.passed) {
@@ -1270,17 +1237,17 @@ async function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort
                         buildErrorContext.set(msg.taskId, {
                             errors: validation.errors,
                             command: validation.command,
-                            originalChanges: freshProposal.changes,
+                            originalChanges: msg.changes,
                             retryCount: 0
                         });
                         // Trigger automatic build fix
-                        await triggerBuildFix(cwd, msg.taskId, validation.errors, validation.command, freshProposal.changes, resolvedLassoKey, agentConfig, collabConfig, socket, taskControllers, taskSnapshots, buildErrorContext);
+                        await triggerBuildFix(cwd, msg.taskId, validation.errors, validation.command, msg.changes, resolvedLassoKey, agentConfig, collabConfig, socket, taskControllers, taskSnapshots, buildErrorContext);
                         return;
                     }
                     socket.send(JSON.stringify({
                         type: "applied",
                         taskId: msg.taskId,
-                        message: `${freshProposal.changes.length} file${freshProposal.changes.length === 1 ? "" : "s"} updated. Your dev server will reload.`,
+                        message: `${msg.changes.length} file${msg.changes.length === 1 ? "" : "s"} updated. Your dev server will reload.`,
                         validation
                     }));
                 }

package/dist/cli/host/daemon.js

@@ -17,6 +17,8 @@ const runtime_1 = require("./runtime");
 const dns_1 = require("./dns");
 const paths_1 = require("./paths");
 const certificates_1 = require("./certificates");
+const package_json_1 = __importDefault(require("../../../package.json"));
+const CURRENT_VERSION = package_json_1.default.version;
 exports.SERVICE_NAME = "lasso-host";
 function writeJson(res, status, body) {
     res.writeHead(status, { "content-type": "application/json" });
@@ -247,7 +249,9 @@ class LassoHost {
         writeJson(res, 200, {
             ok: true,
             service: exports.SERVICE_NAME,
-            version: this.opts.version || "unknown",
+            version: this.opts.version || CURRENT_VERSION,
+            installedVersion: CURRENT_VERSION,
+            versionMismatch: (this.opts.version || CURRENT_VERSION) !== CURRENT_VERSION,
             pid: process.pid,
             dnsPort: this.dnsHandle?.port ?? null,
             proxyPort: this.opts.proxyPort,

package/dist/cli/host/install.d.ts

@@ -5,6 +5,7 @@ export declare function spawnDaemon(env: Record<string, string>): Promise<{
     alreadyRunning?: boolean;
     pid?: number;
     error?: string;
+    restarted?: boolean;
 }>;
 export declare function isAutoStartInstalled(): boolean;
 export declare function enableAutoStart(): Promise<{

package/dist/cli/host/install.js

@@ -17,9 +17,12 @@ const node_os_1 = __importDefault(require("node:os"));
 const node_path_1 = __importDefault(require("node:path"));
 const node_readline_1 = __importDefault(require("node:readline"));
 const node_child_process_1 = require("node:child_process");
+const chalk_1 = __importDefault(require("chalk"));
 const client_1 = require("./client");
 const dns_1 = require("./dns");
 const paths_1 = require("./paths");
+const package_json_1 = __importDefault(require("../../../package.json"));
+const CURRENT_VERSION = package_json_1.default.version;
 const AGENT_LABEL = "com.lasso.host";
 const LAUNCH_AGENT_FILE = node_path_1.default.join(node_os_1.default.homedir(), "Library", "LaunchAgents", `${AGENT_LABEL}.plist`);
 function cliEntry() {
@@ -31,8 +34,18 @@ function spawnDaemon(env) {
     const dnsPort = (0, paths_1.hostDnsPort)(env);
     const httpsPort = (0, paths_1.hostHttpsPort)(env);
     return (0, client_1.hostAlive)(port).then(async (health) => {
-        if (health)
-            return { ok: true, alreadyRunning: true };
+        if (health) {
+            // Check if the running daemon has a different version
+            if (health.version && health.version !== CURRENT_VERSION) {
+                console.log(chalk_1.default.yellow(`!`) + ` Lasso Host is running version ${health.version} but installed version is ${CURRENT_VERSION}. Restarting...`);
+                await stopDaemon(env);
+                // Wait a moment for the daemon to stop
+                await new Promise(resolve => setTimeout(resolve, 1000));
+            }
+            else {
+                return { ok: true, alreadyRunning: true };
+            }
+        }
         const entry = cliEntry();
         if (!entry)
             return { ok: false, error: "Could not resolve the Lasso CLI entry point. Run this from the installed CLI." };
@@ -46,7 +59,7 @@ function spawnDaemon(env) {
         child.unref();
         await new Promise((resolve) => setTimeout(resolve, 1800));
         if (await (0, client_1.hostAlive)(port))
-            return { ok: true, pid };
+            return { ok: true, pid, restarted: Boolean(health && health.version !== CURRENT_VERSION) };
         return { ok: false, error: `Lasso Host did not come up on port ${port}. Check ${paths_1.LOG_FILE}.` };
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lasso-ai/cli",
-  "version": "1.3.7",
+  "version": "1.3.9",
   "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",