@ash-ai/ui 0.0.4 → 0.0.6

package/dist/index.cjs

@@ -402,6 +402,22 @@ ${fileRefs}` : fileRefs;
           setMessages((prev) => prev.map(
             (m) => m.id === assistantId ? { ...m, content: fullContent || "(no response)", toolCalls: [...toolCalls], isStreaming: false, timestamp: (/* @__PURE__ */ new Date()).toISOString() } : m
           ));
+        } else if (event.type === "message") {
+          const data = event.data;
+          if (data.type === "assistant" && Array.isArray(data.message?.content)) {
+            let text = "";
+            for (const block of data.message.content) {
+              if (block.type === "text" && block.text) {
+                text += (text ? "\n" : "") + block.text;
+              }
+            }
+            if (text) {
+              fullContent = text;
+              setMessages((prev) => prev.map(
+                (m) => m.id === assistantId ? { ...m, content: fullContent, toolCalls: [...toolCalls], isStreaming: true } : m
+              ));
+            }
+          }
         } else if (event.type === "error") {
           const data = event.data;
           const errText = data.error || "Unknown error";
@@ -410,12 +426,14 @@ ${fileRefs}` : fileRefs;
           ));
         }
       }
-      for (const tc of toolCalls) {
-        if (tc.state === "running" || tc.state === "pending") tc.state = "completed";
+      if (fullContent || toolCalls.length > 0) {
+        for (const tc of toolCalls) {
+          if (tc.state === "running" || tc.state === "pending") tc.state = "completed";
+        }
+        setMessages((prev) => prev.map(
+          (m) => m.id === assistantId && m.isStreaming ? { ...m, content: fullContent || "(no response)", toolCalls: toolCalls.length > 0 ? [...toolCalls] : void 0, isStreaming: false, timestamp: (/* @__PURE__ */ new Date()).toISOString() } : m
+        ));
       }
-      setMessages((prev) => prev.map(
-        (m) => m.id === assistantId && m.isStreaming ? { ...m, content: fullContent || "(no response)", toolCalls: toolCalls.length > 0 ? [...toolCalls] : void 0, isStreaming: false, timestamp: (/* @__PURE__ */ new Date()).toISOString() } : m
-      ));
     } catch (err) {
       const errMsg = err instanceof Error ? err.message : "Failed to send";
       setMessages((prev) => prev.map(
@@ -1269,13 +1287,13 @@ function Terminal2({ logs, connected, onClear, className }) {
         ref: containerRef,
         onScroll: handleScroll,
         className: "flex-1 overflow-auto p-3 font-mono text-xs leading-5 scrollbar-thin min-h-[120px]",
-        children: filteredLogs.length === 0 ? /* @__PURE__ */ (0, import_jsx_runtime11.jsx)("div", { className: "flex h-full items-center justify-center text-white/20 text-xs", children: logs.length === 0 ? "Waiting for sandbox output..." : "No matching lines" }) : filteredLogs.map((entry) => /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)("div", { className: "flex hover:bg-white/5 rounded px-1 -mx-1", children: [
+        children: filteredLogs.length === 0 ? /* @__PURE__ */ (0, import_jsx_runtime11.jsx)("div", { className: "flex h-full items-center justify-center text-white/20 text-xs", children: logs.length === 0 ? "Waiting for sandbox output..." : "No matching lines" }) : filteredLogs.map((entry, i) => /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)("div", { className: "flex hover:bg-white/5 rounded px-1 -mx-1", children: [
           showTimestamps && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)("span", { className: "mr-3 shrink-0 select-none text-white/20", children: formatTimestamp(entry.ts) }),
           /* @__PURE__ */ (0, import_jsx_runtime11.jsx)("span", { className: cn(
             "whitespace-pre-wrap break-all",
             levelColors[entry.level] || "text-white/60"
           ), children: entry.text })
-        ] }, entry.index))
+        ] }, `${entry.index}-${i}`))
       }
     )
   ] });