stashes 0.1.20 → 0.1.21

package/dist/cli.js

@@ -1417,11 +1417,27 @@ ${sourceCode.substring(0, 3000)}
     ].filter(Boolean).join(`
 `);
     const aiProcess = startAiProcess("chat", chatPrompt, this.projectPath);
-    let fullResponse = "";
+    let thinkingBuf = "";
+    let textBuf = "";
+    const pendingMessages = [];
+    const now = new Date().toISOString();
+    function flushThinking() {
+      if (!thinkingBuf)
+        return;
+      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
+      thinkingBuf = "";
+    }
+    function flushText() {
+      if (!textBuf)
+        return;
+      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
+      textBuf = "";
+    }
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
         if (chunk.type === "text") {
-          fullResponse += chunk.content;
+          flushThinking();
+          textBuf += chunk.content;
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
@@ -1429,6 +1445,7 @@ ${sourceCode.substring(0, 3000)}
             source: "chat"
           });
         } else if (chunk.type === "thinking") {
+          thinkingBuf += chunk.content;
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
@@ -1436,36 +1453,66 @@ ${sourceCode.substring(0, 3000)}
             source: "chat"
           });
         } else if (chunk.type === "tool_use") {
+          flushThinking();
+          flushText();
+          let toolName = "unknown";
+          let toolParams = {};
+          try {
+            const parsed = JSON.parse(chunk.content);
+            toolName = parsed.tool ?? "unknown";
+            toolParams = parsed.input ?? {};
+          } catch {}
+          pendingMessages.push({
+            id: crypto.randomUUID(),
+            role: "assistant",
+            content: chunk.content,
+            type: "tool_start",
+            toolName,
+            toolParams,
+            toolStatus: "running",
+            createdAt: now
+          });
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
             streamType: "tool_start",
             source: "chat",
-            toolName: chunk.toolName ?? "unknown",
-            toolParams: chunk.toolInput ?? {},
+            toolName,
+            toolParams,
             toolStatus: "running"
           });
         } else if (chunk.type === "tool_result") {
+          let toolResult = chunk.content;
+          let isError = false;
+          try {
+            const parsed = JSON.parse(chunk.content);
+            toolResult = parsed.result ?? chunk.content;
+            isError = !!parsed.is_error;
+          } catch {}
+          pendingMessages.push({
+            id: crypto.randomUUID(),
+            role: "assistant",
+            content: chunk.content,
+            type: "tool_end",
+            toolStatus: isError ? "error" : "completed",
+            toolResult: toolResult.substring(0, 300),
+            createdAt: now
+          });
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
             streamType: "tool_end",
             source: "chat",
-            toolName: chunk.toolName ?? "unknown",
-            toolStatus: chunk.isError ? "error" : "completed",
-            toolResult: chunk.content.substring(0, 300)
+            toolStatus: isError ? "error" : "completed",
+            toolResult: toolResult.substring(0, 300)
           });
         }
       }
       await aiProcess.process.exited;
-      if (fullResponse) {
-        this.persistence.saveChatMessage(projectId, chatId, {
-          id: crypto.randomUUID(),
-          role: "assistant",
-          content: fullResponse,
-          type: "text",
-          createdAt: new Date().toISOString()
-        });
+      flushThinking();
+      flushText();
+      for (const msg of pendingMessages) {
+        this.persistence.saveChatMessage(projectId, chatId, msg);
       }
     } catch (err) {
       this.broadcast({

package/dist/mcp.js

@@ -1613,11 +1613,27 @@ ${sourceCode.substring(0, 3000)}
     ].filter(Boolean).join(`
 `);
     const aiProcess = startAiProcess("chat", chatPrompt, this.projectPath);
-    let fullResponse = "";
+    let thinkingBuf = "";
+    let textBuf = "";
+    const pendingMessages = [];
+    const now = new Date().toISOString();
+    function flushThinking() {
+      if (!thinkingBuf)
+        return;
+      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: thinkingBuf, type: "thinking", createdAt: now });
+      thinkingBuf = "";
+    }
+    function flushText() {
+      if (!textBuf)
+        return;
+      pendingMessages.push({ id: crypto.randomUUID(), role: "assistant", content: textBuf, type: "text", createdAt: now });
+      textBuf = "";
+    }
     try {
       for await (const chunk of parseClaudeStream(aiProcess.process)) {
         if (chunk.type === "text") {
-          fullResponse += chunk.content;
+          flushThinking();
+          textBuf += chunk.content;
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
@@ -1625,6 +1641,7 @@ ${sourceCode.substring(0, 3000)}
             source: "chat"
           });
         } else if (chunk.type === "thinking") {
+          thinkingBuf += chunk.content;
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
@@ -1632,36 +1649,66 @@ ${sourceCode.substring(0, 3000)}
             source: "chat"
           });
         } else if (chunk.type === "tool_use") {
+          flushThinking();
+          flushText();
+          let toolName = "unknown";
+          let toolParams = {};
+          try {
+            const parsed = JSON.parse(chunk.content);
+            toolName = parsed.tool ?? "unknown";
+            toolParams = parsed.input ?? {};
+          } catch {}
+          pendingMessages.push({
+            id: crypto.randomUUID(),
+            role: "assistant",
+            content: chunk.content,
+            type: "tool_start",
+            toolName,
+            toolParams,
+            toolStatus: "running",
+            createdAt: now
+          });
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
             streamType: "tool_start",
             source: "chat",
-            toolName: chunk.toolName ?? "unknown",
-            toolParams: chunk.toolInput ?? {},
+            toolName,
+            toolParams,
             toolStatus: "running"
           });
         } else if (chunk.type === "tool_result") {
+          let toolResult = chunk.content;
+          let isError = false;
+          try {
+            const parsed = JSON.parse(chunk.content);
+            toolResult = parsed.result ?? chunk.content;
+            isError = !!parsed.is_error;
+          } catch {}
+          pendingMessages.push({
+            id: crypto.randomUUID(),
+            role: "assistant",
+            content: chunk.content,
+            type: "tool_end",
+            toolStatus: isError ? "error" : "completed",
+            toolResult: toolResult.substring(0, 300),
+            createdAt: now
+          });
           this.broadcast({
             type: "ai_stream",
             content: chunk.content,
             streamType: "tool_end",
             source: "chat",
-            toolName: chunk.toolName ?? "unknown",
-            toolStatus: chunk.isError ? "error" : "completed",
-            toolResult: chunk.content.substring(0, 300)
+            toolStatus: isError ? "error" : "completed",
+            toolResult: toolResult.substring(0, 300)
           });
         }
       }
       await aiProcess.process.exited;
-      if (fullResponse) {
-        this.persistence.saveChatMessage(projectId, chatId, {
-          id: crypto.randomUUID(),
-          role: "assistant",
-          content: fullResponse,
-          type: "text",
-          createdAt: new Date().toISOString()
-        });
+      flushThinking();
+      flushText();
+      for (const msg of pendingMessages) {
+        this.persistence.saveChatMessage(projectId, chatId, msg);
       }
     } catch (err) {
       this.broadcast({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stashes",
-  "version": "0.1.20",
+  "version": "0.1.21",
   "type": "module",
   "description": "Generate AI-powered UI design explorations in your project",
   "keywords": [