@lasso-ai/cli 1.1.3 → 1.1.5

package/dist/cli/agent.js

@@ -36,6 +36,7 @@ async function detectLocalAgents() {
 }
 const ignored = new Set(["node_modules", ".git", ".next", "dist", "build", ".turbo"]);
 const sourceExtensions = /\.(tsx?|jsx?|vue|svelte|css|scss|html)$/i;
+const layoutFiles = /^(layout|_layout|_app|app|_document|document|template|_template)\.(tsx?|jsx?)$/i;
 const sourceFileCache = new Map();
 async function sourceFiles(directory, budget = { remaining: 25 }) {
     if (budget.remaining <= 0)
@@ -48,6 +49,8 @@ async function sourceFiles(directory, budget = { remaining: 25 }) {
     }
     const entries = await promises_1.default.readdir(directory, { withFileTypes: true });
     const files = [];
+    const layoutFilesList = [];
+    const otherFiles = [];
     for (const entry of entries) {
         if (ignored.has(entry.name) || entry.name.startsWith("."))
             continue;
@@ -59,10 +62,17 @@ async function sourceFiles(directory, budget = { remaining: 25 }) {
             files.push(...nested);
         }
         else if (sourceExtensions.test(entry.name)) {
-            files.push(fullPath);
-            budget.remaining -= 1;
+            if (layoutFiles.test(entry.name)) {
+                layoutFilesList.push(fullPath);
+            }
+            else {
+                otherFiles.push(fullPath);
+            }
         }
     }
+    // Prioritize layout files first, then other files
+    files.push(...layoutFilesList, ...otherFiles);
+    budget.remaining -= files.length;
     sourceFileCache.set(directory, { expiresAt: Date.now() + 30000, files });
     return files;
 }
@@ -70,6 +80,11 @@ async function contextFor(cwd, element) {
     const needle = element.label.replace(/^[^.#]+[.#]?/, "");
     const hintedPath = element.sourceHint?.split(":")[0];
     const sourceFile = hintedPath ? node_path_1.default.basename(hintedPath) : "";
+    // Extract unique class names and attributes from the element for better matching
+    const classMatches = element.html?.match(/class="([^"]+)"/);
+    const classNames = classMatches ? classMatches[1].split(/\s+/).filter((c) => c.length > 3) : [];
+    const hrefMatch = element.html?.match(/href="([^"]+)"/);
+    const href = hrefMatch ? hrefMatch[1] : "";
     if (hintedPath) {
         const candidates = [
             node_path_1.default.isAbsolute(hintedPath) ? hintedPath : node_path_1.default.resolve(cwd, hintedPath),
@@ -91,11 +106,31 @@ async function contextFor(cwd, element) {
     }
     const files = await sourceFiles(cwd);
     const snippets = [];
+    const scoredFiles = [];
     const results = await Promise.all(files.map(async (file) => {
         try {
             const content = await promises_1.default.readFile(file, "utf8");
-            if (!needle || (sourceFile && file.endsWith(sourceFile)) || content.includes(needle) || content.includes(element.label)) {
-                return `FILE: ${node_path_1.default.relative(cwd, file)}\n${content.slice(0, 6000)}`;
+            let score = 0;
+            // Score files based on relevance
+            if (sourceFile && file.endsWith(sourceFile))
+                score += 10;
+            if (content.includes(needle))
+                score += 5;
+            if (content.includes(element.label))
+                score += 5;
+            if (href && content.includes(href))
+                score += 8;
+            // Check for class name matches
+            for (const className of classNames) {
+                if (content.includes(className))
+                    score += 3;
+            }
+            // Boost score for component files
+            if (file.includes("component") || file.includes("nav") || file.includes("header") || file.includes("layout")) {
+                score += 2;
+            }
+            if (score > 0) {
+                return { file, score, content };
             }
         }
         catch {
@@ -103,9 +138,12 @@ async function contextFor(cwd, element) {
         }
         return null;
     }));
-    for (const result of results) {
-        if (result && snippets.length < 4)
-            snippets.push(result);
+    // Sort by score and take top files
+    const sortedResults = results.filter((r) => r !== null).sort((a, b) => b.score - a.score);
+    for (const result of sortedResults) {
+        if (snippets.length < 8) {
+            snippets.push(`FILE: ${node_path_1.default.relative(cwd, result.file)}\n${result.content.slice(0, 6000)}`);
+        }
     }
     return snippets.join("\n\n---\n\n");
 }

package/dist/overlay.js

@@ -18928,7 +18928,7 @@
   function findReusableAgentTask(instruction, element) {
     const normalized = normalizedInstruction(instruction);
     return state.agentTasks.find(
-      (task) => task.status !== "complete" && task.status !== "error" && task.status !== "stopped" && normalizedInstruction(task.instruction) === normalized && task.element === (element || void 0)
+      (task) => normalizedInstruction(task.instruction) === normalized && task.element === (element || void 0)
     );
   }
   function getComponentConversation(selectionId) {
@@ -20033,18 +20033,31 @@
     const model = state.selectedModel;
     const screenshotPromise = state.screenshotPromise;
     const reusableTask = findReusableAgentTask(instruction, selectedElement);
+    let task;
     if (reusableTask) {
       selectAgentTask(reusableTask.id);
-      return;
+      if (reusableTask.status === "complete" || reusableTask.status === "error" || reusableTask.status === "stopped") {
+        updateAgentTask(reusableTask.id, { status: "thinking", message: "Thinking\u2026", activity: [...reusableTask.activity, "Restarting\u2026"] });
+        appendChat("user", instruction);
+        resetPromptSession(selectedElement);
+        state.promptTaskId = reusableTask.id;
+        state.lastInstruction = instruction;
+        task = reusableTask;
+      } else {
+        return;
+      }
+    } else {
+      appendChat("user", instruction);
+      const messages2 = state.chatHistory.map((message) => ({ ...message }));
+      const changesHistory2 = state.changesHistory.map((entry) => ({ ...entry, changes: entry.changes.map((change) => ({ ...change })) }));
+      task = createAgentTask(instruction, messages2, changesHistory2, selectedElement, selectionId);
+      updateAgentTask(task.id, { status: "thinking", message: "Thinking\u2026", activity: [...task.activity, "Thinking\u2026"] });
+      resetPromptSession(selectedElement);
+      state.promptTaskId = task.id;
+      state.lastInstruction = instruction;
     }
-    appendChat("user", instruction);
     const messages = state.chatHistory.map((message) => ({ ...message }));
     const changesHistory = state.changesHistory.map((entry) => ({ ...entry, changes: entry.changes.map((change) => ({ ...change })) }));
-    const task = createAgentTask(instruction, messages, changesHistory, selectedElement, selectionId);
-    updateAgentTask(task.id, { status: "thinking", message: "Thinking\u2026", activity: [...task.activity, "Thinking\u2026"] });
-    resetPromptSession(selectedElement);
-    state.promptTaskId = task.id;
-    state.lastInstruction = instruction;
     const rect = selectedElement.getBoundingClientRect();
     const computed = getComputedStyle(selectedElement);
     const attributes = Object.fromEntries(

package/package.json

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