@lasso-ai/cli 1.1.3 → 1.1.4

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;
 }

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.4",
   "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",