@lasso-ai/cli 1.2.2 → 1.2.4

package/dist/cli/auth.js

@@ -30,7 +30,9 @@ function serverUrlFrom(fileEnv) {
         fileEnv.LASSO_AUTH_URL ||
         process.env.LASSO_API_URL ||
         fileEnv.LASSO_API_URL ||
-        "https://api.lasso.byorello.space").replace(/\/$/, "");
+        process.env.LASSO_SERVER_URL ||
+        fileEnv.LASSO_SERVER_URL ||
+        (process.env.NODE_ENV === "development" ? "http://localhost:3005" : "https://api.lasso.byorello.space")).replace(/\/$/, "");
 }
 function fetchWithTimeout(url, init, timeoutMs = 8000) {
     const controller = new AbortController();

package/dist/cli/bridge.d.ts

@@ -17,7 +17,7 @@ export type BridgeMessage = {
     taskId: string;
     instruction: string;
     model: string;
-    provider?: "anthropic" | "openai" | "google" | "ollama" | "cli";
+    provider?: "anthropic" | "openai" | "google" | "ollama" | "nvidia" | "cli";
     messages?: Array<{
         role: string;
         content: string;
@@ -52,7 +52,7 @@ export type BridgeMessage = {
     taskId: string;
     question: string;
     model: string;
-    provider?: "anthropic" | "openai" | "google" | "ollama" | "cli";
+    provider?: "anthropic" | "openai" | "google" | "ollama" | "nvidia" | "cli";
     messages?: Array<{
         role: string;
         content: string;
@@ -112,7 +112,7 @@ export type BridgeMessage = {
 } | {
     type: "git_generate_message";
     model: string;
-    provider?: "anthropic" | "openai" | "google" | "ollama" | "cli";
+    provider?: "anthropic" | "openai" | "google" | "ollama" | "nvidia" | "cli";
 } | {
     type: "git_init";
 } | {
@@ -134,14 +134,20 @@ export type BridgeMessage = {
 } | {
     type: "oneshot";
     prompt: string;
-    scope: "project" | "component";
+    scope?: "project" | "component";
     model?: string;
-    provider?: "anthropic" | "openai" | "google" | "ollama" | "cli";
+    provider?: "anthropic" | "openai" | "google" | "ollama" | "nvidia" | "cli";
+    messages?: Array<{
+        role: "user" | "assistant";
+        content: string;
+    }>;
+    taskId?: string;
+    pageContext?: any;
 };
 type ModelOption = {
     id: string;
     label: string;
-    provider: "anthropic" | "openai" | "google" | "ollama" | "cli";
+    provider: "anthropic" | "openai" | "google" | "ollama" | "nvidia" | "cli";
 };
 type PageEntry = {
     name: string;
@@ -154,6 +160,10 @@ export type ServerBridgeMessage = {
     agentConfigured: boolean;
     models: ModelOption[];
     collab?: CollabConfig | null;
+    user?: {
+        name?: string;
+        email?: string;
+    } | null;
 } | {
     type: "git_state";
     git: GitState;
@@ -175,6 +185,12 @@ export type ServerBridgeMessage = {
     message: string;
     detail?: string;
     changes?: SourceChange[];
+    thinking?: Array<{
+        title: string;
+        detail?: string;
+        durationMs?: number;
+    }>;
+    totalThinkingTimeMs?: number;
 } | {
     type: "agent_prompt";
     taskId: string;

package/dist/cli/bridge.js

@@ -15,6 +15,7 @@ const node_child_process_1 = require("node:child_process");
 const node_util_1 = require("node:util");
 const chalk_1 = __importDefault(require("chalk"));
 const agent_1 = require("./agent");
+const project_1 = require("./project");
 const auth_1 = require("./auth");
 exports.DEFAULT_BRIDGE_PORT = 3056;
 const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
@@ -52,7 +53,7 @@ function fileLineEnding(content) {
 function withLineEnding(value, lineEnding) {
     return value.replace(/\r\n?|\n/g, lineEnding);
 }
-function resolveProposedFile(cwd, proposedPath) {
+function resolveProposedFile(cwd, proposedPath, allowNew = false) {
     const root = node_path_1.default.resolve(cwd);
     const candidate = node_path_1.default.resolve(root, proposedPath);
     if (candidate.startsWith(`${root}${node_path_1.default.sep}`) && node_fs_1.default.existsSync(candidate))
@@ -65,10 +66,15 @@ function resolveProposedFile(cwd, proposedPath) {
         const rebased = node_path_1.default.join(root, normalized.slice(srcIndex + 1));
         if (rebased.startsWith(`${root}${node_path_1.default.sep}`) && node_fs_1.default.existsSync(rebased))
             return rebased;
+        if (allowNew && (rebased.startsWith(`${root}${node_path_1.default.sep}`) || rebased === root))
+            return rebased;
+    }
+    if (candidate.startsWith(`${root}${node_path_1.default.sep}`) || candidate === root) {
+        if (allowNew)
+            return candidate;
+        throw new Error(`Could not find the proposed file in the current project: ${proposedPath}`);
     }
-    if (!candidate.startsWith(`${root}${node_path_1.default.sep}`))
-        throw new Error(`The proposed file is outside the current project: ${proposedPath}`);
-    throw new Error(`Could not find the proposed file in the current project: ${proposedPath}`);
+    throw new Error(`The proposed file is outside the current project: ${proposedPath}`);
 }
 function occurrenceCount(content, needle) {
     if (!needle)
@@ -117,6 +123,8 @@ function prepareChange(content, change) {
 function proposalMatchesCurrentSource(cwd, changes) {
     try {
         for (const change of changes) {
+            if (!change.oldString)
+                continue;
             const filePath = resolveProposedFile(cwd, change.filePath);
             const content = node_fs_1.default.readFileSync(filePath, "utf8");
             // Try exact match first
@@ -171,24 +179,33 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
         ...readEnvFile(root, ".env.local"),
         ...readEnvFile(root, ".env.production"),
     }), {});
-    const lassoKeyConfigured = Boolean(process.env.VITE_LASSO_API_KEY || process.env.NEXT_LASSO_API_KEY || fileEnv.VITE_LASSO_API_KEY || fileEnv.NEXT_LASSO_API_KEY);
+    const resolvedLassoKey = (0, project_1.resolveLassoApiKey)(fileEnv) ||
+        process.env.LASSO_API_KEY ||
+        fileEnv.LASSO_API_KEY ||
+        process.env.VITE_LASSO_API_KEY ||
+        process.env.NEXT_LASSO_API_KEY ||
+        fileEnv.VITE_LASSO_API_KEY ||
+        fileEnv.NEXT_LASSO_API_KEY ||
+        "";
+    const lassoKeyConfigured = Boolean(resolvedLassoKey);
+    const localProviderKeys = {
+        google: process.env.GOOGLE_GENERATIVE_AI_API_KEY || fileEnv.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GEMINI_API_KEY || fileEnv.GEMINI_API_KEY,
+        openai: process.env.OPENAI_API_KEY || fileEnv.OPENAI_API_KEY,
+        anthropic: process.env.ANTHROPIC_API_KEY || fileEnv.ANTHROPIC_API_KEY,
+        nvidia: process.env.NVIDIA_API_KEY || fileEnv.NVIDIA_API_KEY,
+        ollama: process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || process.env.OLLAMA_MODEL || fileEnv.OLLAMA_MODEL ? "ollama" : undefined,
+    };
     let agentConfig = (() => {
         const provider = (process.env.LASSO_AGENT_PROVIDER || fileEnv.LASSO_AGENT_PROVIDER || process.env.AI_PROVIDER || fileEnv.AI_PROVIDER || "").toLowerCase();
-        const keys = {
-            google: process.env.GOOGLE_GENERATIVE_AI_API_KEY || fileEnv.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GEMINI_API_KEY || fileEnv.GEMINI_API_KEY,
-            openai: process.env.OPENAI_API_KEY || fileEnv.OPENAI_API_KEY,
-            anthropic: process.env.ANTHROPIC_API_KEY || fileEnv.ANTHROPIC_API_KEY,
-            ollama: process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || process.env.OLLAMA_MODEL || fileEnv.OLLAMA_MODEL,
-        };
-        const selectedProvider = provider === "google" || provider === "gemini" ? "google" : provider === "openai" ? "openai" : provider === "ollama" ? "ollama" : provider === "anthropic" ? "anthropic" : keys.ollama ? "ollama" : keys.google ? "google" : keys.openai ? "openai" : "anthropic";
-        const apiKey = selectedProvider === "ollama" ? "ollama" : keys[selectedProvider];
-        if (!apiKey)
+        const selectedProvider = provider === "google" || provider === "gemini" ? "google" : provider === "openai" ? "openai" : provider === "ollama" ? "ollama" : provider === "anthropic" ? "anthropic" : provider === "nvidia" ? "nvidia" : localProviderKeys.ollama ? "ollama" : localProviderKeys.nvidia ? "nvidia" : localProviderKeys.google ? "google" : localProviderKeys.openai ? "openai" : localProviderKeys.anthropic ? "anthropic" : "google";
+        const apiKey = selectedProvider === "ollama" ? "ollama" : localProviderKeys[selectedProvider];
+        if (!apiKey && !lassoKeyConfigured)
             return null;
         return {
             provider: selectedProvider,
-            apiKey,
+            apiKey: apiKey || resolvedLassoKey,
             model: process.env.LASSO_AGENT_MODEL || fileEnv.LASSO_AGENT_MODEL || process.env.AI_MODEL || fileEnv.AI_MODEL,
-            baseUrl: process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL ? `${(process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || "http://localhost:11434/api").replace(/\/api\/?$/, "")}/v1` : undefined,
+            baseUrl: selectedProvider === "nvidia" ? "https://integrate.api.nvidia.com/v1" : process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL ? `${(process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || "http://localhost:11434/api").replace(/\/api\/?$/, "")}/v1` : undefined,
         };
     })();
     let activeAgentController = null;
@@ -293,9 +310,26 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             const base = process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || "http://localhost:11434/api";
             agentConfig = { provider: "ollama", apiKey: "ollama", model: process.env.OLLAMA_MODEL || fileEnv.OLLAMA_MODEL, baseUrl: `${base.replace(/\/api\/?$/, "")}/v1` };
         }
-        const isPaidPlan = collabConfig?.plan && collabConfig.plan !== "free";
+        const isPaidPlan = Boolean(collabConfig?.plan && collabConfig.plan.toLowerCase() !== "free");
         const configuredProviders = collabConfig?.configuredProviders || [];
         const allModels = [
+            // NVIDIA NIM models (confirmed working with server API key)
+            { id: "meta/llama-3.2-11b-vision-instruct", label: "Llama 3.2 11B Vision", provider: "nvidia" },
+            { id: "openai/gpt-oss-20b", label: "GPT-OSS 20B (NVIDIA)", provider: "nvidia" },
+            { id: "deepseek-ai/deepseek-coder-6.7b-instruct", label: "DeepSeek Coder 6.7B", provider: "nvidia" },
+            { id: "deepseek-ai/deepseek-v4.1-flash", label: "DeepSeek V4.1 Flash", provider: "nvidia" },
+            { id: "ibm/granite-34b-code-instruct", label: "Granite 34B Code", provider: "nvidia" },
+            { id: "ibm/granite-3.0-8b-instruct", label: "Granite 3.0 8B", provider: "nvidia" },
+            { id: "mistralai/mistral-large-2-instruct", label: "Mistral Large 2", provider: "nvidia" },
+            { id: "mistralai/codestral-22b-instruct-v0.1", label: "Codestral 22B", provider: "nvidia" },
+            { id: "mistralai/mistral-large", label: "Mistral Large", provider: "nvidia" },
+            { id: "nv-mistralai/mistral-nemo-12b-instruct", label: "Mistral Nemo 12B", provider: "nvidia" },
+            { id: "nvidia/nemotron-3.5-lightning-30b-a3b", label: "Nemotron Lightning 30B", provider: "nvidia" },
+            { id: "microsoft/phi-3.5-moe-instruct", label: "Phi-3.5 MoE", provider: "nvidia" },
+            { id: "google/gemma-4-31b-it", label: "Gemma 4 31B", provider: "nvidia" },
+            { id: "moonshotai/kimi-k2.6", label: "Kimi K2.6", provider: "nvidia" },
+            { id: "poolside/laguna-xs-2.1", label: "Laguna XS 2.1", provider: "nvidia" },
+            // Existing models
             { id: "claude-sonnet-4-5-20250929", label: "Claude Sonnet 4.5", provider: "anthropic" },
             { id: "claude-opus-4-1-20250805", label: "Claude Opus 4.1", provider: "anthropic" },
             { id: "gpt-4.1", label: "GPT-4.1", provider: "openai" },
@@ -340,6 +374,7 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             anthropic: "ANTHROPIC_API_KEY",
             openai: "OPENAI_API_KEY",
             google: "GOOGLE_GENERATIVE_AI_API_KEY",
+            nvidia: "NVIDIA_API_KEY",
             ollama: "OLLAMA_BASE_URL",
         };
         return allModels.map((model) => {
@@ -355,7 +390,8 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             const envKey = providerKeyMapping[model.provider];
             const isConfigured = configuredProviders.includes(model.provider) ||
                 Boolean(process.env[envKey]) ||
-                Boolean(fileEnv[envKey]);
+                Boolean(fileEnv[envKey]) ||
+                Boolean(localProviderKeys[model.provider]);
             if (!isConfigured) {
                 return {
                     ...model,
@@ -369,10 +405,14 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
     wss.on("connection", (socket) => {
         overlaySocket = socket;
         console.log(chalk_1.default.green("✓") + " Overlay connected");
-        socket.send(JSON.stringify({ type: "config", apiKeyConfigured: lassoKeyConfigured, agentConfigured: Boolean(agentConfig) || localAgents.size > 0, models: [], collab: collabConfig?.registered ? { ...collabConfig, plan: collabConfig.plan, configuredProviders: collabConfig.configuredProviders } : null }));
+        const creds = (0, auth_1.loadCredentials)();
+        const userInfo = creds?.userName
+            ? { name: creds.userName, email: creds.userEmail }
+            : (process.env.USER ? { name: process.env.USER } : null);
+        socket.send(JSON.stringify({ type: "config", apiKeyConfigured: lassoKeyConfigured, agentConfigured: lassoKeyConfigured || Boolean(agentConfig) || localAgents.size > 0, models: [], collab: collabConfig?.registered ? { ...collabConfig, plan: collabConfig.plan, configuredProviders: collabConfig.configuredProviders } : null, user: userInfo }));
         void availableModels(collabConfig).then((models) => {
             if (socket.readyState === socket.OPEN)
-                socket.send(JSON.stringify({ type: "config", apiKeyConfigured: lassoKeyConfigured, agentConfigured: Boolean(agentConfig) || localAgents.size > 0, models, collab: collabConfig?.registered ? { ...collabConfig, plan: collabConfig.plan, configuredProviders: collabConfig.configuredProviders } : null }));
+                socket.send(JSON.stringify({ type: "config", apiKeyConfigured: lassoKeyConfigured, agentConfigured: lassoKeyConfigured || Boolean(agentConfig) || localAgents.size > 0, models, collab: collabConfig?.registered ? { ...collabConfig, plan: collabConfig.plan, configuredProviders: collabConfig.configuredProviders } : null, user: userInfo }));
         });
         void getGitState(cwd).then((git) => {
             if (socket.readyState === socket.OPEN)
@@ -514,9 +554,29 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                 taskControllers.get(msg.taskId)?.abort();
                 const controller = new AbortController();
                 taskControllers.set(msg.taskId, controller);
-                const selectedConfig = localProvider
-                    ? { provider: cliProvider, model: msg.model }
-                    : { ...agentConfig, provider: (msg.provider || agentConfig.provider), model: msg.model };
+                let selectedConfig;
+                if (localProvider) {
+                    selectedConfig = { provider: cliProvider, model: msg.model };
+                }
+                else {
+                    const requestedProvider = (msg.provider || agentConfig.provider);
+                    const providerLocalKey = localProviderKeys[requestedProvider];
+                    const apiKey = providerLocalKey || resolvedLassoKey;
+                    const baseUrl = requestedProvider === "nvidia"
+                        ? "https://integrate.api.nvidia.com/v1"
+                        : requestedProvider === "ollama"
+                            ? (agentConfig.baseUrl || "http://localhost:11434/v1")
+                            : agentConfig.baseUrl;
+                    selectedConfig = {
+                        ...agentConfig,
+                        provider: requestedProvider,
+                        model: msg.model,
+                        apiKey,
+                        baseUrl,
+                        lassoKey: resolvedLassoKey,
+                        serverUrl: collabConfig?.serverUrl || (0, auth_1.serverUrlFrom)(fileEnv),
+                    };
+                }
                 socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: "working", message: "Working…" }));
                 void (0, agent_1.answerQuestion)(cwd, { taskId: msg.taskId, question: msg.question, context: msg.context, element: msg.element, messages: msg.messages }, selectedConfig, controller.signal, (message, detail, level) => {
                     if (!controller.signal.aborted && socket.readyState === socket.OPEN)
@@ -546,9 +606,29 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                     socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: "error", message: "Add a supported agent key: GOOGLE_GENERATIVE_AI_API_KEY, OPENAI_API_KEY, or ANTHROPIC_API_KEY." }));
                     return;
                 }
-                const selectedConfig = localProvider
-                    ? { provider: cliProvider, model: msg.model }
-                    : { ...agentConfig, provider: (msg.provider || agentConfig.provider), model: msg.model };
+                let selectedConfig;
+                if (localProvider) {
+                    selectedConfig = { provider: cliProvider, model: msg.model };
+                }
+                else {
+                    const requestedProvider = (msg.provider || agentConfig.provider);
+                    const providerLocalKey = localProviderKeys[requestedProvider];
+                    const apiKey = providerLocalKey || resolvedLassoKey;
+                    const baseUrl = requestedProvider === "nvidia"
+                        ? "https://integrate.api.nvidia.com/v1"
+                        : requestedProvider === "ollama"
+                            ? (agentConfig.baseUrl || "http://localhost:11434/v1")
+                            : agentConfig.baseUrl;
+                    selectedConfig = {
+                        ...agentConfig,
+                        provider: requestedProvider,
+                        model: msg.model,
+                        apiKey,
+                        baseUrl,
+                        lassoKey: resolvedLassoKey,
+                        serverUrl: collabConfig?.serverUrl || (0, auth_1.serverUrlFrom)(fileEnv),
+                    };
+                }
                 editRequests.set(msg.taskId, msg);
                 editConfigs.set(msg.taskId, selectedConfig);
                 reviewRefreshAttempts.set(msg.taskId, 0);
@@ -633,10 +713,15 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             }
             else if (msg.type === "transcribe") {
                 const serverUrl = (0, auth_1.serverUrlFrom)(fileEnv);
+                const creds = (0, auth_1.loadCredentials)();
+                const apiKey = resolvedLassoKey || agentConfig?.apiKey || collabConfig?.apiKey || creds?.apiKey;
                 try {
                     const resp = await fetch(`${serverUrl}/api/v1/transcribe`, {
                         method: "POST",
-                        headers: { "Content-Type": "application/json" },
+                        headers: {
+                            "Content-Type": "application/json",
+                            ...(apiKey ? { Authorization: `Bearer ${apiKey}` } : {}),
+                        },
                         body: JSON.stringify({
                             audio: msg.audio,
                             mimeType: msg.mimeType,
@@ -664,22 +749,38 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             }
             else if (msg.type === "oneshot") {
                 console.log("[lasso] one-shot request:", msg.prompt, msg.scope);
-                if (!lassoKeyConfigured && !agentConfig) {
-                    socket.send(JSON.stringify({ type: "agent_status", status: "error", message: "Set your Lasso API key or configure an AI provider before using one-shot mode." }));
+                const taskId = msg.taskId || `oneshot-${Date.now()}`;
+                const localProvider = msg.provider === "cli" || msg.model?.startsWith("claude-code:") || msg.model?.startsWith("opencode:") || msg.model?.startsWith("cursor:") || msg.model?.startsWith("codex:");
+                const cliProvider = msg.model?.startsWith("claude-code:") ? "claude-code" : msg.model?.startsWith("opencode:") ? "opencode" : msg.model?.startsWith("cursor:") ? "cursor" : "codex";
+                const apiKey = resolvedLassoKey || agentConfig?.apiKey || collabConfig?.apiKey;
+                if (!localProvider && !lassoKeyConfigured && !agentConfig && !apiKey) {
+                    socket.send(JSON.stringify({
+                        type: "agent_status",
+                        taskId,
+                        status: "error",
+                        message: "No AI provider or Lasso key configured. Sign in with 'lasso auth login' or configure a provider key in .env."
+                    }));
                     return;
                 }
-                const localProvider = msg.provider === "cli";
-                const cliProvider = msg.model?.startsWith("claude-code:") ? "claude-code" : msg.model?.startsWith("opencode:") ? "opencode" : msg.model?.startsWith("cursor:") ? "cursor" : "codex";
+                const provider = (msg.provider || agentConfig?.provider || "google");
+                const model = msg.model || agentConfig?.model || (provider === "nvidia" ? "nvidia/llama-3.1-nemotron-70b-instruct" : provider === "google" ? "gemini-2.5-flash" : "claude-3-7-sonnet");
+                const localKey = localProviderKeys[provider];
                 const selectedConfig = localProvider
                     ? { provider: cliProvider, model: msg.model || "claude-code:sonnet" }
-                    : { ...agentConfig, provider: (msg.provider || agentConfig.provider), model: msg.model || agentConfig.model };
-                const taskId = `oneshot-${Date.now()}`;
+                    : {
+                        ...(agentConfig || {}),
+                        provider,
+                        model,
+                        apiKey: localKey || apiKey,
+                        baseUrl: provider === "nvidia" ? "https://integrate.api.nvidia.com/v1" : provider === "ollama" ? `${(process.env.OLLAMA_BASE_URL || fileEnv.OLLAMA_BASE_URL || "http://localhost:11434/api").replace(/\/api\/?$/, "")}/v1` : undefined,
+                    };
                 const controller = new AbortController();
                 taskControllers.set(taskId, controller);
                 socket.send(JSON.stringify({ type: "agent_status", taskId, status: "thinking", message: "Planning your request..." }));
-                // Import the one-shot agent function
                 const { runOneShotAgent } = await import("./oneshot.js");
-                runOneShotAgent(cwd, msg.prompt, msg.scope, selectedConfig, controller.signal, (status, message, detail) => {
+                const serverUrl = collabConfig?.serverUrl || (0, auth_1.serverUrlFrom)(fileEnv);
+                const targetApiUrl = (serverUrl.includes("collab.lasso.byorello.space") ? "https://api.lasso.byorello.space" : serverUrl).replace(/\/api\/v1\/?$/, "");
+                runOneShotAgent(cwd, msg.prompt, msg.scope || "project", selectedConfig, controller.signal, (status, message, detail) => {
                     if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
                         socket.send(JSON.stringify({ type: "agent_status", taskId, status, message, detail }));
                     }
@@ -687,10 +788,20 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                     if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
                         socket.send(JSON.stringify({ type: "agent_prompt", taskId, prompt }));
                     }
-                }).then((result) => {
+                }, msg.messages, targetApiUrl, apiKey, taskId, msg.pageContext).then((result) => {
                     if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
                         if (result.ok) {
-                            socket.send(JSON.stringify({ type: "agent_status", taskId, status: "review", message: result.summary, changes: result.changes }));
+                            const hasChanges = Array.isArray(result.changes) && result.changes.length > 0;
+                            socket.send(JSON.stringify({
+                                type: "agent_status",
+                                taskId,
+                                status: hasChanges ? "review" : "complete",
+                                message: result.summary,
+                                todo: result.todo,
+                                changes: result.changes,
+                                thinking: result.thinking,
+                                totalThinkingTimeMs: result.totalThinkingTimeMs,
+                            }));
                         }
                         else {
                             socket.send(JSON.stringify({ type: "agent_status", taskId, status: "error", message: result.error || "One-shot agent failed" }));
@@ -710,25 +821,33 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                     const snapshots = [];
                     const planned = new Map();
                     for (const change of msg.changes) {
-                        const filePath = resolveProposedFile(cwd, change.filePath);
-                        const content = node_fs_1.default.readFileSync(filePath, "utf8");
-                        const prepared = prepareChange(content, change);
+                        const isNew = !change.oldString;
+                        const filePath = resolveProposedFile(cwd, change.filePath, true);
+                        const exists = node_fs_1.default.existsSync(filePath);
+                        const content = exists ? node_fs_1.default.readFileSync(filePath, "utf8") : "";
+                        const prepared = (!exists || isNew)
+                            ? { start: 0, end: content.length, oldString: content, newString: change.newString }
+                            : prepareChange(content, change);
                         const fileChanges = planned.get(filePath) || [];
                         if (fileChanges.some((item) => prepared.start < item.end && item.start < prepared.end)) {
                             throw new Error(`Could not safely apply ${change.filePath}. Proposed changes overlap.`);
                         }
-                        fileChanges.push({ filePath, content, ...prepared });
+                        fileChanges.push({ filePath, content, exists, ...prepared });
                         planned.set(filePath, fileChanges);
                     }
                     // Validate every change before writing any file, then apply each
                     // file's replacements from the end toward the beginning.
                     taskSnapshots.set(msg.taskId, snapshots);
                     for (const [filePath, changes] of planned) {
+                        const exists = changes[0].exists;
                         const content = changes[0].content;
-                        snapshots.push({ filePath, content });
-                        const nextContent = [...changes]
-                            .sort((a, b) => b.start - a.start)
-                            .reduce((value, change) => value.slice(0, change.start) + change.newString + value.slice(change.end), content);
+                        snapshots.push({ filePath, content: exists ? content : null });
+                        const nextContent = exists
+                            ? [...changes]
+                                .sort((a, b) => b.start - a.start)
+                                .reduce((value, change) => value.slice(0, change.start) + change.newString + value.slice(change.end), content)
+                            : changes[0].newString;
+                        node_fs_1.default.mkdirSync(node_path_1.default.dirname(filePath), { recursive: true });
                         node_fs_1.default.writeFileSync(filePath, nextContent);
                     }
                     socket.send(JSON.stringify({ type: "applied", taskId: msg.taskId, message: `${msg.changes.length} file${msg.changes.length === 1 ? "" : "s"} updated. Your dev server will reload.` }));
@@ -736,8 +855,15 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                 catch (error) {
                     // Validation happens before writes, but restore this task's snapshot
                     // if a filesystem error occurs during the write phase.
-                    for (const snapshot of taskSnapshots.get(msg.taskId) || [])
-                        node_fs_1.default.writeFileSync(snapshot.filePath, snapshot.content);
+                    for (const snapshot of taskSnapshots.get(msg.taskId) || []) {
+                        if (snapshot.content === null) {
+                            if (node_fs_1.default.existsSync(snapshot.filePath))
+                                node_fs_1.default.unlinkSync(snapshot.filePath);
+                        }
+                        else {
+                            node_fs_1.default.writeFileSync(snapshot.filePath, snapshot.content);
+                        }
+                    }
                     taskSnapshots.delete(msg.taskId);
                     const message = error instanceof Error ? error.message : "The change could not be applied.";
                     const sourceChanged = message.includes("The source changed after the suggestion was generated");
@@ -755,8 +881,15 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             }
             else if (msg.type === "undo") {
                 const snapshots = taskSnapshots.get(msg.taskId || "") || [];
-                for (const snapshot of snapshots)
-                    node_fs_1.default.writeFileSync(snapshot.filePath, snapshot.content);
+                for (const snapshot of snapshots) {
+                    if (snapshot.content === null) {
+                        if (node_fs_1.default.existsSync(snapshot.filePath))
+                            node_fs_1.default.unlinkSync(snapshot.filePath);
+                    }
+                    else {
+                        node_fs_1.default.writeFileSync(snapshot.filePath, snapshot.content);
+                    }
+                }
                 taskSnapshots.delete(msg.taskId || "");
                 socket.send(JSON.stringify({ type: "undone", taskId: msg.taskId, message: snapshots.length ? "The accepted change was reverted." : "There is no accepted change to undo." }));
             }

package/dist/cli/index.js

@@ -264,6 +264,16 @@ program.command("register [domain]")
         console.error(chalk_1.default.red("✗") + ` "${domain}" is not a valid .lasso domain (lowercase letters, numbers, and hyphens).`);
         process.exit(1);
     }
+    // Register with Lasso workspace & Host
+    console.log(chalk_1.default.dim("Registering project with Lasso workspace…"));
+    const initResult = await (0, project_1.initProject)(cwd, env, domain || undefined);
+    if (initResult.ok) {
+        console.log(chalk_1.default.green("✓") + ` Registered project: ${chalk_1.default.cyan(initResult.projectId)}`);
+        console.log(chalk_1.default.green("✓") + ` Local domain: ${chalk_1.default.cyan(initResult.domain || domain)}`);
+        console.log(chalk_1.default.green("✓") + ` Points to ${chalk_1.default.dim(cwd)}`);
+        return;
+    }
+    // If server registration was not possible (e.g. offline/no key), fall back to local host registration
     if (!domain) {
         if (config?.domain) {
             domain = config.domain;

package/dist/cli/oneshot.d.ts

@@ -1,8 +1,21 @@
 import type { AgentConfig, SourceChange } from "./agent";
+export interface OneShotThinkingStep {
+    title: string;
+    detail?: string;
+    durationMs?: number;
+}
+export interface OneShotTodoItem {
+    file: string;
+    action: "modify" | "create" | "delete";
+    reason: string;
+}
 export interface OneShotResult {
     ok: boolean;
     summary?: string;
+    todo?: OneShotTodoItem[];
     changes?: SourceChange[];
+    thinking?: OneShotThinkingStep[];
+    totalThinkingTimeMs?: number;
     error?: string;
 }
 export interface ProgressCallback {
@@ -14,4 +27,9 @@ export interface PromptCallback {
         options?: string[];
     }): void;
 }
-export declare function runOneShotAgent(cwd: string, prompt: string, scope: "project" | "component", config: AgentConfig, signal: AbortSignal, onProgress: ProgressCallback, onPrompt: PromptCallback): Promise<OneShotResult>;
+export interface ConversationMessage {
+    role: "user" | "assistant";
+    content: string;
+}
+export declare function isConversationalPrompt(prompt: string): boolean;
+export declare function runOneShotAgent(cwd: string, prompt: string, scope: "project" | "component", config: AgentConfig, signal: AbortSignal, onProgress: ProgressCallback, onPrompt: PromptCallback, messages?: ConversationMessage[], serverUrl?: string, apiKey?: string, taskId?: string, pageContext?: any): Promise<OneShotResult>;