stashes 0.1.52 → 0.1.54

package/dist/cli.js

@@ -548,7 +548,6 @@ import { writeFileSync as writeFileSync2, existsSync as existsSync4, mkdirSync a
 import { join as join4 } from "path";
 import { tmpdir } from "os";
 var CLAUDE_BIN = "/opt/homebrew/bin/claude";
-var DEFAULT_TOOLS = ["Read", "Write", "Edit", "Glob", "Grep", "Bash"];
 function getPlaywrightMcpConfigPath() {
   const configDir = join4(tmpdir(), "stashes-mcp");
   const configPath = join4(configDir, "playwright.json");
@@ -562,11 +561,29 @@ function getPlaywrightMcpConfigPath() {
   }
   return configPath;
 }
+var OVERHEAD_TOOLS = [
+  "Agent",
+  "TodoWrite",
+  "TaskCreate",
+  "TaskUpdate",
+  "TaskList",
+  "TaskGet",
+  "Skill",
+  "ToolSearch",
+  "EnterPlanMode",
+  "ExitPlanMode",
+  "WebSearch",
+  "WebFetch",
+  "NotebookEdit",
+  "mcp__UseAI__*",
+  "mcp__stashes__*",
+  "mcp__plugin_drills*",
+  "mcp__plugin_coverit*"
+];
 var processes = new Map;
 function startAiProcess(idOrOpts, prompt, cwd, resumeSessionId, model) {
   const opts = typeof idOrOpts === "string" ? { id: idOrOpts, prompt, cwd, resumeSessionId, model } : idOrOpts;
-  const bare = opts.bare ?? true;
-  const tools = opts.tools ?? DEFAULT_TOOLS;
+  const restricted = opts.tools !== undefined;
   killAiProcess(opts.id);
   logger.info("claude", `spawning process: ${opts.id}`, {
     cwd: opts.cwd,
@@ -574,15 +591,14 @@ function startAiProcess(idOrOpts, prompt, cwd, resumeSessionId, model) {
     promptPreview: opts.prompt.substring(0, 100),
     resumeSessionId: opts.resumeSessionId,
     model: opts.model,
-    bare,
-    tools: bare ? tools.join(",") : "all"
+    restricted,
+    tools: restricted ? opts.tools.join(",") || "none" : "all"
   });
   const cmd = [CLAUDE_BIN, "-p", opts.prompt, "--output-format=stream-json", "--verbose", "--dangerously-skip-permissions"];
-  if (bare) {
-    cmd.push("--bare");
-    if (tools.length > 0) {
-      cmd.push("--tools", tools.join(","));
-    }
+  if (restricted) {
+    cmd.push("--tools", opts.tools.length > 0 ? opts.tools.join(",") : '""');
+    cmd.push("--disallowedTools", OVERHEAD_TOOLS.join(","));
+    cmd.push("--strict-mcp-config");
     if (opts.mcpConfigPath) {
       cmd.push("--mcp-config", opts.mcpConfigPath);
     }
@@ -788,33 +804,49 @@ ${truncatedDiff}`;
   }
 }
 function buildScreenshotPrompt(port, diff, screenshotDir, stashId) {
+  const outputPath = join6(screenshotDir, `${stashId}.png`);
   return [
-    "You are a screenshot assistant. Take screenshots of a running web app using Playwright MCP tools.",
+    "You are a screenshot assistant. Take a screenshot of a running web app.",
     "Be fast \u2014 you have a strict time limit.",
     "",
-    `## The app is running at: http://localhost:${port}`,
+    `The app is running at: http://localhost:${port}`,
     "",
-    "## Git diff of changes:",
+    "## Git diff of changes (tells you what was modified):",
     "```",
     diff,
     "```",
     "",
-    "## Steps:",
-    `1. Call mcp__playwright__browser_navigate with url "http://localhost:${port}" to open the app`,
-    "2. From the diff, determine which page/route shows the changed components",
-    "3. If the change is on a different route, navigate to it",
-    "4. If the change is behind an interaction (tab click, modal open), perform it with mcp__playwright__browser_click",
-    `5. Call mcp__playwright__browser_take_screenshot to capture the page`,
-    "6. Respond with the JSON below",
+    "## Instructions",
+    "Use mcp__playwright__browser_run_code to navigate AND screenshot in a single call.",
+    "From the diff, determine the route/section that shows the changes.",
+    "",
+    "**Call this tool:**",
+    "```",
+    "mcp__playwright__browser_run_code",
+    "```",
     "",
-    `## Screenshot save directory: ${screenshotDir}`,
-    `Use filename "${stashId}.png" for the primary screenshot.`,
+    "**With this code parameter** (adjust the URL path and scroll if needed):",
+    "```javascript",
+    `async (page) => {`,
+    `  await page.setViewportSize({ width: 1280, height: 720 });`,
+    `  await page.goto('http://localhost:${port}', { waitUntil: 'networkidle', timeout: 15000 });`,
+    `  // If the changed component is further down, scroll to it:`,
+    `  // await page.evaluate(() => window.scrollTo(0, 500));`,
+    `  // await page.waitForTimeout(500);`,
+    `  await page.waitForTimeout(2000);`,
+    `  await page.screenshot({ path: '${outputPath.replace(/'/g, "\\'")}', type: 'png' });`,
+    `  return 'saved';`,
+    `}`,
+    "```",
     "",
-    "## After taking screenshots, respond with ONLY this JSON:",
+    "If the diff shows changes to a component that requires scrolling or navigation,",
+    "adjust the code: change the URL path, add scrollTo, or click a tab first.",
+    "",
+    `After the tool call succeeds, respond with ONLY this JSON:`,
     "{",
     '  "screenshots": [',
     "    {",
-    `      "path": "${join6(screenshotDir, `${stashId}.png`)}",`,
+    `      "path": "${outputPath}",`,
     '      "label": "Short description of what is shown",',
     '      "route": "/the-url-path",',
     '      "isPrimary": true',
@@ -872,7 +904,6 @@ async function captureSmartScreenshots(opts) {
     prompt,
     cwd: worktreePath,
     model: modelFlag,
-    bare: true,
     tools: [],
     mcpConfigPath: getPlaywrightMcpConfigPath()
   });
@@ -1041,8 +1072,7 @@ async function generate(opts) {
     const aiProcess = startAiProcess({
       id: stashId,
       prompt: stashPrompt,
-      cwd: worktree.path,
-      bare: false
+      cwd: worktree.path
     });
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
@@ -1110,8 +1140,53 @@ async function generate(opts) {
         return;
       }
       const generatedStash = { ...stash, status: "screenshotting" };
-      completedStashes.push(generatedStash);
       persistence.saveStash(generatedStash);
+      emit(onProgress, { type: "screenshotting", stashId });
+      try {
+        const port = await allocatePort();
+        const screenshotWorktree = await worktreeManager.createForGeneration(`screenshot-${stashId}`);
+        const screenshotGit = simpleGit3(screenshotWorktree.path);
+        await screenshotGit.checkout(["--detach", stash.branch]);
+        const devServer = spawn3({
+          cmd: ["npm", "run", "dev"],
+          cwd: screenshotWorktree.path,
+          stdin: "ignore",
+          stdout: "pipe",
+          stderr: "pipe",
+          env: { ...process.env, PORT: String(port), BROWSER: "none" }
+        });
+        try {
+          await waitForPort(port, 60000);
+          await new Promise((r) => setTimeout(r, 2000));
+          const mainGit = simpleGit3(projectPath);
+          const parentBranch = (await mainGit.revparse(["HEAD"])).trim();
+          const { primary, screenshots } = await captureSmartScreenshots({
+            projectPath,
+            stashId,
+            stashBranch: stash.branch,
+            parentBranch,
+            worktreePath: screenshotWorktree.path,
+            port,
+            model: opts.screenshotModel,
+            timeout: opts.screenshotTimeout
+          });
+          const readyStash = { ...generatedStash, status: "ready", screenshotUrl: primary, screenshots };
+          completedStashes.push(readyStash);
+          persistence.saveStash(readyStash);
+          emit(onProgress, { type: "ready", stashId, screenshotPath: primary });
+        } finally {
+          devServer.kill();
+        }
+        await worktreeManager.removeGeneration(`screenshot-${stashId}`);
+      } catch (err) {
+        logger.error("generation", `screenshot failed for ${stashId}`, {
+          error: err instanceof Error ? err.message : String(err)
+        });
+        const readyStash = { ...generatedStash, status: "ready" };
+        completedStashes.push(readyStash);
+        persistence.saveStash(readyStash);
+        emit(onProgress, { type: "ready", stashId, screenshotPath: "" });
+      }
     } catch (error) {
       const errorMsg = error instanceof Error ? error.message : "Unknown error";
       persistence.saveStash({ ...stash, status: "error", error: errorMsg });
@@ -1121,53 +1196,6 @@ async function generate(opts) {
     }
   });
   await Promise.all(stashPromises);
-  for (const stash of completedStashes) {
-    emit(onProgress, { type: "screenshotting", stashId: stash.id });
-    try {
-      const port = await allocatePort();
-      const worktree = await worktreeManager.createForGeneration(`screenshot-${stash.id}`);
-      const screenshotGit = simpleGit3(worktree.path);
-      await screenshotGit.checkout(["--detach", stash.branch]);
-      const devServer = spawn3({
-        cmd: ["npm", "run", "dev"],
-        cwd: worktree.path,
-        stdin: "ignore",
-        stdout: "pipe",
-        stderr: "pipe",
-        env: { ...process.env, PORT: String(port), BROWSER: "none" }
-      });
-      try {
-        await waitForPort(port, 60000);
-        await new Promise((r) => setTimeout(r, 2000));
-        const mainGit = simpleGit3(projectPath);
-        const parentBranch = (await mainGit.revparse(["HEAD"])).trim();
-        const { primary, screenshots } = await captureSmartScreenshots({
-          projectPath,
-          stashId: stash.id,
-          stashBranch: stash.branch,
-          parentBranch,
-          worktreePath: worktree.path,
-          port,
-          model: opts.screenshotModel,
-          timeout: opts.screenshotTimeout
-        });
-        const updatedStash = { ...stash, status: "ready", screenshotUrl: primary, screenshots };
-        persistence.saveStash(updatedStash);
-        const idx = completedStashes.indexOf(stash);
-        completedStashes[idx] = updatedStash;
-        emit(onProgress, { type: "ready", stashId: stash.id, screenshotPath: primary });
-      } finally {
-        devServer.kill();
-      }
-      await worktreeManager.removeGeneration(`screenshot-${stash.id}`);
-    } catch (err) {
-      logger.error("generation", `screenshot failed for ${stash.id}`, {
-        error: err instanceof Error ? err.message : String(err)
-      });
-      persistence.saveStash({ ...stash, status: "ready" });
-      emit(onProgress, { type: "ready", stashId: stash.id, screenshotPath: "" });
-    }
-  }
   return completedStashes;
 }
 // ../core/dist/vary.js
@@ -1242,8 +1270,7 @@ async function vary(opts) {
   const aiProcess = startAiProcess({
     id: stashId,
     prompt: varyPrompt,
-    cwd: worktree.path,
-    bare: false
+    cwd: worktree.path
   });
   try {
     for await (const chunk of parseClaudeStream(aiProcess.process)) {
@@ -1914,7 +1941,6 @@ class StashService {
       prompt,
       cwd: this.projectPath,
       model: "claude-haiku-4-5-20251001",
-      bare: true,
       tools: ["Read", "Grep", "Glob", "Bash"]
     });
     let resolvedPath = "";
@@ -2030,8 +2056,7 @@ ${sourceCode.substring(0, 3000)}
       id: "chat",
       prompt: chatPrompt,
       cwd: this.projectPath,
-      resumeSessionId: existingSessionId,
-      bare: false
+      resumeSessionId: existingSessionId
     });
     let thinkingBuf = "";
     let textBuf = "";
@@ -2130,11 +2155,13 @@ ${sourceCode.substring(0, 3000)}
             toolResult = parsed.result ?? chunk.content;
             isError = !!parsed.is_error;
           } catch {}
+          const endToolName = chunk.toolName ?? "unknown";
           save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
             type: "tool_end",
+            toolName: endToolName,
             toolStatus: isError ? "error" : "completed",
             toolResult: toolResult.substring(0, 300),
             stashActivity,
@@ -2145,6 +2172,7 @@ ${sourceCode.substring(0, 3000)}
             content: chunk.content,
             streamType: "tool_end",
             source: "chat",
+            toolName: endToolName,
             toolStatus: isError ? "error" : "completed",
             toolResult: toolResult.substring(0, 300)
           });

package/dist/mcp.js

@@ -533,7 +533,6 @@ import { writeFileSync as writeFileSync2, existsSync as existsSync4, mkdirSync a
 import { join as join4 } from "path";
 import { tmpdir } from "os";
 var CLAUDE_BIN = "/opt/homebrew/bin/claude";
-var DEFAULT_TOOLS = ["Read", "Write", "Edit", "Glob", "Grep", "Bash"];
 function getPlaywrightMcpConfigPath() {
   const configDir = join4(tmpdir(), "stashes-mcp");
   const configPath = join4(configDir, "playwright.json");
@@ -547,11 +546,29 @@ function getPlaywrightMcpConfigPath() {
   }
   return configPath;
 }
+var OVERHEAD_TOOLS = [
+  "Agent",
+  "TodoWrite",
+  "TaskCreate",
+  "TaskUpdate",
+  "TaskList",
+  "TaskGet",
+  "Skill",
+  "ToolSearch",
+  "EnterPlanMode",
+  "ExitPlanMode",
+  "WebSearch",
+  "WebFetch",
+  "NotebookEdit",
+  "mcp__UseAI__*",
+  "mcp__stashes__*",
+  "mcp__plugin_drills*",
+  "mcp__plugin_coverit*"
+];
 var processes = new Map;
 function startAiProcess(idOrOpts, prompt, cwd, resumeSessionId, model) {
   const opts = typeof idOrOpts === "string" ? { id: idOrOpts, prompt, cwd, resumeSessionId, model } : idOrOpts;
-  const bare = opts.bare ?? true;
-  const tools = opts.tools ?? DEFAULT_TOOLS;
+  const restricted = opts.tools !== undefined;
   killAiProcess(opts.id);
   logger.info("claude", `spawning process: ${opts.id}`, {
     cwd: opts.cwd,
@@ -559,15 +576,14 @@ function startAiProcess(idOrOpts, prompt, cwd, resumeSessionId, model) {
     promptPreview: opts.prompt.substring(0, 100),
     resumeSessionId: opts.resumeSessionId,
     model: opts.model,
-    bare,
-    tools: bare ? tools.join(",") : "all"
+    restricted,
+    tools: restricted ? opts.tools.join(",") || "none" : "all"
   });
   const cmd = [CLAUDE_BIN, "-p", opts.prompt, "--output-format=stream-json", "--verbose", "--dangerously-skip-permissions"];
-  if (bare) {
-    cmd.push("--bare");
-    if (tools.length > 0) {
-      cmd.push("--tools", tools.join(","));
-    }
+  if (restricted) {
+    cmd.push("--tools", opts.tools.length > 0 ? opts.tools.join(",") : '""');
+    cmd.push("--disallowedTools", OVERHEAD_TOOLS.join(","));
+    cmd.push("--strict-mcp-config");
     if (opts.mcpConfigPath) {
       cmd.push("--mcp-config", opts.mcpConfigPath);
     }
@@ -773,33 +789,49 @@ ${truncatedDiff}`;
   }
 }
 function buildScreenshotPrompt(port, diff, screenshotDir, stashId) {
+  const outputPath = join6(screenshotDir, `${stashId}.png`);
   return [
-    "You are a screenshot assistant. Take screenshots of a running web app using Playwright MCP tools.",
+    "You are a screenshot assistant. Take a screenshot of a running web app.",
     "Be fast \u2014 you have a strict time limit.",
     "",
-    `## The app is running at: http://localhost:${port}`,
+    `The app is running at: http://localhost:${port}`,
     "",
-    "## Git diff of changes:",
+    "## Git diff of changes (tells you what was modified):",
     "```",
     diff,
     "```",
     "",
-    "## Steps:",
-    `1. Call mcp__playwright__browser_navigate with url "http://localhost:${port}" to open the app`,
-    "2. From the diff, determine which page/route shows the changed components",
-    "3. If the change is on a different route, navigate to it",
-    "4. If the change is behind an interaction (tab click, modal open), perform it with mcp__playwright__browser_click",
-    `5. Call mcp__playwright__browser_take_screenshot to capture the page`,
-    "6. Respond with the JSON below",
+    "## Instructions",
+    "Use mcp__playwright__browser_run_code to navigate AND screenshot in a single call.",
+    "From the diff, determine the route/section that shows the changes.",
+    "",
+    "**Call this tool:**",
+    "```",
+    "mcp__playwright__browser_run_code",
+    "```",
     "",
-    `## Screenshot save directory: ${screenshotDir}`,
-    `Use filename "${stashId}.png" for the primary screenshot.`,
+    "**With this code parameter** (adjust the URL path and scroll if needed):",
+    "```javascript",
+    `async (page) => {`,
+    `  await page.setViewportSize({ width: 1280, height: 720 });`,
+    `  await page.goto('http://localhost:${port}', { waitUntil: 'networkidle', timeout: 15000 });`,
+    `  // If the changed component is further down, scroll to it:`,
+    `  // await page.evaluate(() => window.scrollTo(0, 500));`,
+    `  // await page.waitForTimeout(500);`,
+    `  await page.waitForTimeout(2000);`,
+    `  await page.screenshot({ path: '${outputPath.replace(/'/g, "\\'")}', type: 'png' });`,
+    `  return 'saved';`,
+    `}`,
+    "```",
     "",
-    "## After taking screenshots, respond with ONLY this JSON:",
+    "If the diff shows changes to a component that requires scrolling or navigation,",
+    "adjust the code: change the URL path, add scrollTo, or click a tab first.",
+    "",
+    `After the tool call succeeds, respond with ONLY this JSON:`,
     "{",
     '  "screenshots": [',
     "    {",
-    `      "path": "${join6(screenshotDir, `${stashId}.png`)}",`,
+    `      "path": "${outputPath}",`,
     '      "label": "Short description of what is shown",',
     '      "route": "/the-url-path",',
     '      "isPrimary": true',
@@ -857,7 +889,6 @@ async function captureSmartScreenshots(opts) {
     prompt,
     cwd: worktreePath,
     model: modelFlag,
-    bare: true,
     tools: [],
     mcpConfigPath: getPlaywrightMcpConfigPath()
   });
@@ -1026,8 +1057,7 @@ async function generate(opts) {
     const aiProcess = startAiProcess({
       id: stashId,
       prompt: stashPrompt,
-      cwd: worktree.path,
-      bare: false
+      cwd: worktree.path
     });
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
@@ -1095,8 +1125,53 @@ async function generate(opts) {
         return;
       }
       const generatedStash = { ...stash, status: "screenshotting" };
-      completedStashes.push(generatedStash);
       persistence.saveStash(generatedStash);
+      emit(onProgress, { type: "screenshotting", stashId });
+      try {
+        const port = await allocatePort();
+        const screenshotWorktree = await worktreeManager.createForGeneration(`screenshot-${stashId}`);
+        const screenshotGit = simpleGit3(screenshotWorktree.path);
+        await screenshotGit.checkout(["--detach", stash.branch]);
+        const devServer = spawn3({
+          cmd: ["npm", "run", "dev"],
+          cwd: screenshotWorktree.path,
+          stdin: "ignore",
+          stdout: "pipe",
+          stderr: "pipe",
+          env: { ...process.env, PORT: String(port), BROWSER: "none" }
+        });
+        try {
+          await waitForPort(port, 60000);
+          await new Promise((r) => setTimeout(r, 2000));
+          const mainGit = simpleGit3(projectPath);
+          const parentBranch = (await mainGit.revparse(["HEAD"])).trim();
+          const { primary, screenshots } = await captureSmartScreenshots({
+            projectPath,
+            stashId,
+            stashBranch: stash.branch,
+            parentBranch,
+            worktreePath: screenshotWorktree.path,
+            port,
+            model: opts.screenshotModel,
+            timeout: opts.screenshotTimeout
+          });
+          const readyStash = { ...generatedStash, status: "ready", screenshotUrl: primary, screenshots };
+          completedStashes.push(readyStash);
+          persistence.saveStash(readyStash);
+          emit(onProgress, { type: "ready", stashId, screenshotPath: primary });
+        } finally {
+          devServer.kill();
+        }
+        await worktreeManager.removeGeneration(`screenshot-${stashId}`);
+      } catch (err) {
+        logger.error("generation", `screenshot failed for ${stashId}`, {
+          error: err instanceof Error ? err.message : String(err)
+        });
+        const readyStash = { ...generatedStash, status: "ready" };
+        completedStashes.push(readyStash);
+        persistence.saveStash(readyStash);
+        emit(onProgress, { type: "ready", stashId, screenshotPath: "" });
+      }
     } catch (error) {
       const errorMsg = error instanceof Error ? error.message : "Unknown error";
       persistence.saveStash({ ...stash, status: "error", error: errorMsg });
@@ -1106,53 +1181,6 @@ async function generate(opts) {
     }
   });
   await Promise.all(stashPromises);
-  for (const stash of completedStashes) {
-    emit(onProgress, { type: "screenshotting", stashId: stash.id });
-    try {
-      const port = await allocatePort();
-      const worktree = await worktreeManager.createForGeneration(`screenshot-${stash.id}`);
-      const screenshotGit = simpleGit3(worktree.path);
-      await screenshotGit.checkout(["--detach", stash.branch]);
-      const devServer = spawn3({
-        cmd: ["npm", "run", "dev"],
-        cwd: worktree.path,
-        stdin: "ignore",
-        stdout: "pipe",
-        stderr: "pipe",
-        env: { ...process.env, PORT: String(port), BROWSER: "none" }
-      });
-      try {
-        await waitForPort(port, 60000);
-        await new Promise((r) => setTimeout(r, 2000));
-        const mainGit = simpleGit3(projectPath);
-        const parentBranch = (await mainGit.revparse(["HEAD"])).trim();
-        const { primary, screenshots } = await captureSmartScreenshots({
-          projectPath,
-          stashId: stash.id,
-          stashBranch: stash.branch,
-          parentBranch,
-          worktreePath: worktree.path,
-          port,
-          model: opts.screenshotModel,
-          timeout: opts.screenshotTimeout
-        });
-        const updatedStash = { ...stash, status: "ready", screenshotUrl: primary, screenshots };
-        persistence.saveStash(updatedStash);
-        const idx = completedStashes.indexOf(stash);
-        completedStashes[idx] = updatedStash;
-        emit(onProgress, { type: "ready", stashId: stash.id, screenshotPath: primary });
-      } finally {
-        devServer.kill();
-      }
-      await worktreeManager.removeGeneration(`screenshot-${stash.id}`);
-    } catch (err) {
-      logger.error("generation", `screenshot failed for ${stash.id}`, {
-        error: err instanceof Error ? err.message : String(err)
-      });
-      persistence.saveStash({ ...stash, status: "ready" });
-      emit(onProgress, { type: "ready", stashId: stash.id, screenshotPath: "" });
-    }
-  }
   return completedStashes;
 }
 // ../core/dist/vary.js
@@ -1227,8 +1255,7 @@ async function vary(opts) {
   const aiProcess = startAiProcess({
     id: stashId,
     prompt: varyPrompt,
-    cwd: worktree.path,
-    bare: false
+    cwd: worktree.path
   });
   try {
     for await (const chunk of parseClaudeStream(aiProcess.process)) {
@@ -2138,7 +2165,6 @@ class StashService {
       prompt,
       cwd: this.projectPath,
       model: "claude-haiku-4-5-20251001",
-      bare: true,
       tools: ["Read", "Grep", "Glob", "Bash"]
     });
     let resolvedPath = "";
@@ -2254,8 +2280,7 @@ ${sourceCode.substring(0, 3000)}
       id: "chat",
       prompt: chatPrompt,
       cwd: this.projectPath,
-      resumeSessionId: existingSessionId,
-      bare: false
+      resumeSessionId: existingSessionId
     });
     let thinkingBuf = "";
     let textBuf = "";
@@ -2354,11 +2379,13 @@ ${sourceCode.substring(0, 3000)}
             toolResult = parsed.result ?? chunk.content;
             isError = !!parsed.is_error;
           } catch {}
+          const endToolName = chunk.toolName ?? "unknown";
           save({
             id: crypto.randomUUID(),
             role: "assistant",
             content: chunk.content,
             type: "tool_end",
+            toolName: endToolName,
             toolStatus: isError ? "error" : "completed",
             toolResult: toolResult.substring(0, 300),
             stashActivity,
@@ -2369,6 +2396,7 @@ ${sourceCode.substring(0, 3000)}
             content: chunk.content,
             streamType: "tool_end",
             source: "chat",
+            toolName: endToolName,
             toolStatus: isError ? "error" : "completed",
             toolResult: toolResult.substring(0, 300)
           });