@blade-hq/agent-kit 0.5.5 → 0.5.7

package/dist/{chunk-S6EPGDAL.js → chunk-K5EE7X2D.js}

@@ -8,13 +8,13 @@ import {
   getCodeLanguageFromFilename,
   parseAskUserQuestion,
   useHighlightedCodeHtml
-} from "./chunk-557R7QZV.js";
+} from "./chunk-GIE2Q2MB.js";
 import {
   Collapsible,
   CollapsibleContent,
   CollapsibleTrigger,
   resolveSessionFilePreviewTarget
-} from "./chunk-3ZEWA2YM.js";
+} from "./chunk-X3S36RR2.js";
 import {
   buildMessageContent,
   buildToolPreviewKey,
@@ -57,7 +57,7 @@ import {
   useUiBridgeStore,
   useUiStore,
   writeFile
-} from "./chunk-HWOVPFWG.js";
+} from "./chunk-UM7G65GH.js";
 import {
   registerBridgeIframe,
   tapBridgeEvent
@@ -66,7 +66,7 @@ import {
   ModelOption,
   ModelsConfig,
   ModelsResource
-} from "./chunk-Q6CSM5DE.js";
+} from "./chunk-ETHPRRT2.js";
 import {
   cn,
   copyToClipboard
@@ -77,7 +77,7 @@ import {
 
 // src/react/components/chat/ChatView.tsx
 import { Eye } from "lucide-react";
-import { useCallback as useCallback13 } from "react";
+import { useCallback as useCallback13, useMemo as useMemo18 } from "react";
 
 // src/react/hooks/use-chat.ts
 import { useCallback, useEffect, useRef, useState } from "react";
@@ -2121,11 +2121,22 @@ function isImageAttachment(attachment) {
 function buildSubmitPayload(text, attachments, pendingContexts, onBeforeSend) {
   const validAttachments = attachments.filter((attachment) => attachment.status !== "failed");
   const payload = buildMessageContent(text, validAttachments);
-  const contextParts = pendingContexts.map((context) => ({
-    type: "text",
-    text: `[\u4E0A\u4E0B\u6587: ${context.label}]
+  const contextParts = pendingContexts.flatMap((context) => {
+    const parts = [
+      {
+        type: "text",
+        text: `[\u4E0A\u4E0B\u6587: ${context.label}]
 ${context.content}`
-  }));
+      }
+    ];
+    if (context.imageUrl) {
+      parts.push({
+        type: "image_url",
+        image_url: { url: context.imageUrl }
+      });
+    }
+    return parts;
+  });
   const mergedPayload = contextParts.length === 0 ? payload : typeof payload === "string" ? payload.trim() ? [...contextParts, { type: "text", text: payload }] : contextParts : [...contextParts, ...payload];
   return onBeforeSend ? onBeforeSend(mergedPayload) : mergedPayload;
 }
@@ -2430,12 +2441,23 @@ function ComposerContextPill({
   id,
   label,
   content,
+  imageUrl,
   onRemove
 }) {
   const [showDetail, setShowDetail] = useState8(false);
   const tokenK = formatTokenK(content);
   return /* @__PURE__ */ jsxs8(Fragment2, { children: [
     /* @__PURE__ */ jsxs8("div", { className: "flex shrink-0 items-center gap-1.5 rounded-full border border-[hsl(var(--border))] bg-[hsl(var(--accent))] px-2.5 py-1 text-[11px] text-[hsl(var(--foreground))]", children: [
+      imageUrl ? /* @__PURE__ */ jsx9(
+        "button",
+        {
+          type: "button",
+          onClick: () => setShowDetail(true),
+          className: "h-6 w-6 overflow-hidden rounded-md border border-[hsl(var(--border))] bg-white",
+          title: "\u67E5\u770B\u9009\u4E2D\u533A\u57DF\u622A\u56FE",
+          children: /* @__PURE__ */ jsx9("img", { src: imageUrl, alt: "\u9009\u4E2D\u533A\u57DF\u622A\u56FE", className: "h-full w-full object-cover" })
+        }
+      ) : null,
       /* @__PURE__ */ jsx9(
         "button",
         {
@@ -2481,7 +2503,10 @@ function ComposerContextPill({
             onKeyDown: (e) => e.stopPropagation(),
             children: [
               /* @__PURE__ */ jsxs8("div", { className: "flex items-center justify-between border-b border-[hsl(var(--border))] px-6 py-3", children: [
-                /* @__PURE__ */ jsx9("h4", { className: "text-xs font-semibold text-[hsl(var(--foreground))]", children: label }),
+                /* @__PURE__ */ jsxs8("div", { className: "min-w-0", children: [
+                  /* @__PURE__ */ jsx9("h4", { className: "truncate text-xs font-semibold text-[hsl(var(--foreground))]", children: label }),
+                  imageUrl ? /* @__PURE__ */ jsx9("p", { className: "mt-0.5 text-[11px] text-[hsl(var(--muted-foreground))]", children: "\u5DF2\u9644\u52A0\u9009\u4E2D\u533A\u57DF\u622A\u56FE\uFF0C\u975E\u591A\u6A21\u6001\u6A21\u578B\u4F1A\u81EA\u52A8\u5FFD\u7565\u56FE\u7247\u3002" }) : null
+                ] }),
                 /* @__PURE__ */ jsx9(
                   "button",
                   {
@@ -2492,7 +2517,10 @@ function ComposerContextPill({
                   }
                 )
               ] }),
-              /* @__PURE__ */ jsx9("div", { className: "min-h-0 flex-1 overflow-y-auto px-6 py-4", children: /* @__PURE__ */ jsx9("pre", { className: "whitespace-pre-wrap break-words font-mono text-[11px] leading-[1.6] text-[hsl(var(--foreground)/0.85)]", children: content }) })
+              /* @__PURE__ */ jsxs8("div", { className: "min-h-0 flex-1 overflow-y-auto px-6 py-4", children: [
+                imageUrl ? /* @__PURE__ */ jsx9("div", { className: "mb-4 rounded-xl border border-[hsl(var(--border))] bg-white p-2", children: /* @__PURE__ */ jsx9("img", { src: imageUrl, alt: "\u9009\u4E2D\u533A\u57DF\u622A\u56FE", className: "max-h-72 max-w-full rounded-lg object-contain" }) }) : null,
+                /* @__PURE__ */ jsx9("pre", { className: "whitespace-pre-wrap break-words font-mono text-[11px] leading-[1.6] text-[hsl(var(--foreground)/0.85)]", children: content })
+              ] })
             ]
           }
         )
@@ -3387,6 +3415,7 @@ function ChatInput({
                       id: context.id,
                       label: context.label,
                       content: context.content,
+                      imageUrl: context.imageUrl,
                       onRemove: (contextId) => {
                         if (!activeSessionId) {
                           return;
@@ -4184,7 +4213,7 @@ function useStickToBottomContext() {
 
 // src/react/components/chat/AssistantTurnBlock.tsx
 import { AlertCircle, BookOpen, Check as Check4, ChevronRight as ChevronRight5 } from "lucide-react";
-import { useCallback as useCallback11, useEffect as useEffect14, useMemo as useMemo15, useRef as useRef11, useState as useState18 } from "react";
+import { memo as memo3, useCallback as useCallback11, useEffect as useEffect14, useMemo as useMemo15, useRef as useRef11, useState as useState18 } from "react";
 
 // src/react/routes.ts
 var MEMORIES_ROUTE = "/memories";
@@ -5670,6 +5699,8 @@ function ToolCallBlock({
       ) : null
     ] }),
     expanded && /* @__PURE__ */ jsx20("div", { className: "ml-4 mt-1 rounded-xl bg-[hsl(var(--card))] px-3 py-3", children: Renderer ? /* @__PURE__ */ jsx20(Renderer, { toolCall, sessionId: resolvedSessionId }) : /* @__PURE__ */ jsxs16(Fragment5, { children: [
+      /* @__PURE__ */ jsx20("div", { className: "mb-1 text-[10px] uppercase tracking-wider text-[hsl(var(--muted-foreground))]", children: "\u5DE5\u5177" }),
+      /* @__PURE__ */ jsx20("div", { className: "mb-3 font-mono text-[11px] text-[hsl(var(--foreground))]", children: normalizedName }),
       /* @__PURE__ */ jsx20("div", { className: "mb-1 text-[10px] uppercase tracking-wider text-[hsl(var(--muted-foreground))]", children: "\u53C2\u6570" }),
       /* @__PURE__ */ jsx20("pre", { className: "overflow-x-auto whitespace-pre-wrap rounded-md bg-[hsl(var(--muted))] p-2 font-mono text-[11px] text-[hsl(var(--foreground))]", children: formatArgs(toolCall.arguments) }),
       toolCall.result != null && /* @__PURE__ */ jsxs16(Fragment5, { children: [
@@ -6414,6 +6445,19 @@ function safeParseDescription(argsStr) {
     return "\u5B50\u667A\u80FD\u4F53";
   }
 }
+function safeParsePrompt(argsStr) {
+  try {
+    const prompt = JSON.parse(argsStr)?.prompt ?? "";
+    const marker = "# \u4F60\u7684\u4EFB\u52A1";
+    const idx = prompt.indexOf(marker);
+    if (idx >= 0) {
+      return prompt.slice(idx + marker.length).trim();
+    }
+    return prompt;
+  } catch {
+    return "";
+  }
+}
 function stripFileExtension(fileName) {
   return fileName.replace(/\.[^.]+$/, "");
 }
@@ -6654,33 +6698,42 @@ function AgentLoopBlock({ toolCall, sessionId, reasoning }) {
             inlineToolUiBlocks.length > 0 ? /* @__PURE__ */ jsx29("div", { className: "flex flex-col gap-2 px-4 pb-3", children: inlineToolUiBlocks.map((block) => /* @__PURE__ */ jsx29(ResourceIframe, { ui: block.ui, sessionId }, block.key)) }) : null
           ] }) : null
         ] }),
-        expanded && /* @__PURE__ */ jsx29("div", { className: "border-t border-[hsl(var(--border))] py-2", children: childMessages.length > 0 ? /* @__PURE__ */ jsx29("div", { className: "flex flex-col gap-3 px-3", children: childMessages.map(
-          (message, index) => message.role === "assistant" ? /* @__PURE__ */ jsx29(
-            ExpandedChildAssistantMessage,
-            {
-              message,
-              sessionId,
-              isStreaming: message.status === "streaming"
-            },
-            message.entry_id ?? `${message.timestamp ?? "child"}-${index}`
-          ) : message.role === "user" ? /* @__PURE__ */ jsx29(
-            ExpandedChildUserMessage,
-            {
-              message
-            },
-            message.entry_id ?? `${message.timestamp ?? "user"}-${index}`
-          ) : message.role === "error" ? /* @__PURE__ */ jsx29(
-            "div",
-            {
-              className: "border-l-[2px] border-l-red-500 px-3 py-2 text-[11px] text-red-200",
-              children: typeof message.content === "string" ? message.content : "\u5B50\u667A\u80FD\u4F53\u6267\u884C\u51FA\u9519"
-            },
-            message.entry_id ?? `${message.timestamp ?? "error"}-${index}`
-          ) : null
-        ) }) : status === "running" ? /* @__PURE__ */ jsxs24("div", { className: "px-3 py-2 text-[11px] text-[hsl(var(--muted-foreground))]", children: [
-          /* @__PURE__ */ jsx29(Loader24, { size: 12, className: "mr-1.5 inline animate-spin" }),
-          "\u6B63\u5728\u542F\u52A8..."
-        ] }) : toolCall.result ? /* @__PURE__ */ jsx29("pre", { className: "max-h-[400px] overflow-auto px-3 py-2 whitespace-pre-wrap text-[11px] text-[hsl(var(--muted-foreground))]", children: typeof toolCall.result === "string" ? toolCall.result : JSON.stringify(toolCall.result, null, 2) }) : null })
+        expanded && /* @__PURE__ */ jsxs24("div", { className: "border-t border-[hsl(var(--border))] py-2", children: [
+          (() => {
+            const taskPrompt = safeParsePrompt(toolCall.arguments);
+            return taskPrompt ? /* @__PURE__ */ jsxs24("details", { className: "mx-3 mb-2 rounded-md bg-[hsl(var(--muted))]/40", children: [
+              /* @__PURE__ */ jsx29("summary", { className: "cursor-pointer px-3 py-1.5 text-[11px] font-medium text-[hsl(var(--muted-foreground))]", children: "\u4EFB\u52A1" }),
+              /* @__PURE__ */ jsx29("div", { className: "px-3 pb-2", children: /* @__PURE__ */ jsx29("pre", { className: "whitespace-pre-wrap text-[11px] leading-relaxed text-[hsl(var(--foreground))]", children: taskPrompt }) })
+            ] }) : null;
+          })(),
+          childMessages.length > 0 ? /* @__PURE__ */ jsx29("div", { className: "flex flex-col gap-3 px-3", children: childMessages.map(
+            (message, index) => message.role === "assistant" ? /* @__PURE__ */ jsx29(
+              ExpandedChildAssistantMessage,
+              {
+                message,
+                sessionId,
+                isStreaming: message.status === "streaming"
+              },
+              message.entry_id ?? `${message.timestamp ?? "child"}-${index}`
+            ) : message.role === "user" ? /* @__PURE__ */ jsx29(
+              ExpandedChildUserMessage,
+              {
+                message
+              },
+              message.entry_id ?? `${message.timestamp ?? "user"}-${index}`
+            ) : message.role === "error" ? /* @__PURE__ */ jsx29(
+              "div",
+              {
+                className: "border-l-[2px] border-l-red-500 px-3 py-2 text-[11px] text-red-200",
+                children: typeof message.content === "string" ? message.content : "\u5B50\u667A\u80FD\u4F53\u6267\u884C\u51FA\u9519"
+              },
+              message.entry_id ?? `${message.timestamp ?? "error"}-${index}`
+            ) : null
+          ) }) : status === "running" ? /* @__PURE__ */ jsxs24("div", { className: "px-3 py-2 text-[11px] text-[hsl(var(--muted-foreground))]", children: [
+            /* @__PURE__ */ jsx29(Loader24, { size: 12, className: "mr-1.5 inline animate-spin" }),
+            "\u6B63\u5728\u542F\u52A8..."
+          ] }) : toolCall.result ? /* @__PURE__ */ jsx29("pre", { className: "max-h-[400px] overflow-auto px-3 py-2 whitespace-pre-wrap text-[11px] text-[hsl(var(--muted-foreground))]", children: typeof toolCall.result === "string" ? toolCall.result : JSON.stringify(toolCall.result, null, 2) }) : null
+        ] })
       ]
     }
   );
@@ -6796,6 +6849,8 @@ function getLoopCardStyles(status) {
 
 // src/react/components/chat/AssistantTurnBlock.tsx
 import { jsx as jsx30, jsxs as jsxs25 } from "react/jsx-runtime";
+var EMPTY_MESSAGES3 = [];
+var EMPTY_AGENT_LOOPS2 = {};
 function defaultTurnDisplayMode({
   forceExpanded,
   hasActionableToolCall
@@ -6930,7 +6985,7 @@ function collectTurnMessages({
   });
   return merged;
 }
-function AssistantTurnBlock({
+function AssistantTurnBlockBase({
   sessionId,
   messages: rawMessages,
   isStreaming = false,
@@ -6943,6 +6998,10 @@ function AssistantTurnBlock({
 }) {
   const messages = Array.isArray(rawMessages) ? rawMessages : [];
   const allToolCalls = useMemo15(() => messages.flatMap((m) => m.tool_calls ?? []), [messages]);
+  const hasAgentToolCall = useMemo15(
+    () => allToolCalls.some((toolCall) => formatToolName(toolCall.name) === "Agent"),
+    [allToolCalls]
+  );
   const hasActionableToolCall = useMemo15(
     () => allToolCalls.some(
       (tc) => tc.status === "error" || tc.status === "cancelled" || formatToolName(tc.name) === "AskUserQuestion" && tc.status === "awaiting_answer"
@@ -6959,11 +7018,15 @@ function AssistantTurnBlock({
     }
     wasStreamingRef.current = isStreaming;
   }, [forceExpanded, hasActionableToolCall, isStreaming]);
-  const sessionMessages = useChatStore((state) => state.messages[sessionId] ?? []);
-  const agentLoops = useChatStore((state) => state.agentLoops[sessionId] ?? {});
+  const sessionMessages = useChatStore(
+    (state) => hasAgentToolCall ? state.messages[sessionId] ?? EMPTY_MESSAGES3 : EMPTY_MESSAGES3
+  );
+  const agentLoops = useChatStore(
+    (state) => hasAgentToolCall ? state.agentLoops[sessionId] ?? EMPTY_AGENT_LOOPS2 : EMPTY_AGENT_LOOPS2
+  );
   const turnMessages = useMemo15(
-    () => collectTurnMessages({ rootMessages: messages, sessionMessages, agentLoops }),
-    [agentLoops, messages, sessionMessages]
+    () => hasAgentToolCall ? collectTurnMessages({ rootMessages: messages, sessionMessages, agentLoops }) : messages,
+    [agentLoops, hasAgentToolCall, messages, sessionMessages]
   );
   const resourceBlocks = useMemo15(() => getInlineResourceBlocks(turnMessages), [turnMessages]);
   const finalMessage = useMemo15(() => getLastTextMessage(turnMessages), [turnMessages]);
@@ -7037,6 +7100,10 @@ function AssistantTurnBlock({
     }
   );
 }
+function areAssistantTurnBlockPropsEqual(prev, next) {
+  return prev.turnKey === next.turnKey && prev.sessionId === next.sessionId && prev.messages === next.messages && prev.isStreaming === next.isStreaming && prev.isLastTurn === next.isLastTurn && prev.askAnswers === next.askAnswers && prev.onAnswer === next.onAnswer && prev.sessionStatus === next.sessionStatus && prev.level === next.level && prev.forceExpanded === next.forceExpanded && prev.customization === next.customization;
+}
+var AssistantTurnBlock = memo3(AssistantTurnBlockBase, areAssistantTurnBlockPropsEqual);
 function AssistantMessages({
   messages,
   isStreaming,
@@ -7822,45 +7889,34 @@ function getAssistantTurnPreview(messages) {
   return "\u672C\u8F6E\u56DE\u590D";
 }
 function getMessageMeasureSignature(message) {
-  const contentSize = typeof message.content === "string" ? message.content.length : message.content.reduce((size, part) => size + JSON.stringify(part).length, 0);
+  const contentSize = typeof message.content === "string" ? message.content.length : message.content.length;
   return [
     message.entry_id ?? message.timestamp ?? message.role,
     message.status ?? "",
     message.reasoning?.length ?? 0,
     message.tool_calls?.length ?? 0,
+    message.blocks?.length ?? 0,
     contentSize
   ].join(":");
 }
 function getMessagesMeasureSignature(messages) {
   return messages.map((message) => getMessageMeasureSignature(message)).join("|");
 }
-function safeJson(value) {
-  try {
-    return JSON.stringify(value);
-  } catch {
-    return String(value);
-  }
-}
-function hashString(value) {
-  let hash = 0;
-  for (let index = 0; index < value.length; index += 1) {
-    hash = hash * 31 + value.charCodeAt(index) | 0;
-  }
-  return hash.toString(36);
-}
-function getMessageRenderSignature(message) {
+function getMessageResetSignature(message) {
+  const contentSize = typeof message.content === "string" ? message.content.length : message.content.length;
+  const toolSignature = message.tool_calls?.map((toolCall) => `${toolCall.id}:${toolCall.status}:${toolCall.result == null ? 0 : 1}`).join(",") ?? "";
   return [
     message.entry_id ?? message.timestamp ?? message.role,
     message.status ?? "",
-    hashString(safeJson(message.content)),
-    hashString(message.reasoning ?? ""),
-    hashString(safeJson(message.tool_calls ?? [])),
-    hashString(safeJson(message.blocks ?? [])),
-    hashString(safeJson(message.memory_refs ?? []))
+    contentSize,
+    message.reasoning?.length ?? 0,
+    message.blocks?.length ?? 0,
+    message.memory_refs?.length ?? 0,
+    toolSignature
   ].join(":");
 }
-function getMessagesRenderSignature(messages) {
-  return messages.map((message) => getMessageRenderSignature(message)).join("|");
+function getMessagesResetSignature(messages) {
+  return messages.map((message) => getMessageResetSignature(message)).join("|");
 }
 function findScrollContainer(start) {
   let current = start;
@@ -7885,7 +7941,17 @@ function MessageList({
   const agentLoops = useChatStore((state) => state.agentLoops[sessionId]);
   const askAnswers = useChatStore((state) => state.askAnswers[sessionId]);
   const agentLoopNameSet = useMemo17(() => new Set(Object.keys(agentLoops ?? {})), [agentLoops]);
+  const assistantMessagesCacheRef = useRef12(/* @__PURE__ */ new Map());
   const renderBlocks = useMemo17(() => {
+    const stableAssistantMessages = (key, buffer) => {
+      const previous = assistantMessagesCacheRef.current.get(key);
+      if (previous && previous.length === buffer.length && previous.every((message, index) => message === buffer[index])) {
+        return previous;
+      }
+      const next = [...buffer];
+      assistantMessagesCacheRef.current.set(key, next);
+      return next;
+    };
     const visible = messages.filter((message) => {
       const loopName = message.loop_name ?? "root";
       if (loopName !== "root") return false;
@@ -7903,13 +7969,15 @@ function MessageList({
       assistantTurnCount += 1;
       const last = assistantBuffer[assistantBuffer.length - 1];
       const isStreaming = assistantBuffer.some((message) => message.status === "streaming");
+      const key = last.entry_id ?? `assistant-${blocks.length}`;
+      const stableMessages = stableAssistantMessages(key, assistantBuffer);
       blocks.push({
         type: "assistant_turn",
-        messages: assistantBuffer,
-        key: last.entry_id ?? `assistant-${blocks.length}`,
+        messages: stableMessages,
+        key,
         anchorId: `chat-turn-${assistantTurnCount}`,
         isStreaming,
-        railTitle: lastUserPreview || getAssistantTurnPreview(assistantBuffer)
+        railTitle: lastUserPreview || getAssistantTurnPreview(stableMessages)
       });
       assistantBuffer = [];
     };
@@ -7948,6 +8016,14 @@ function MessageList({
       });
     }
     flushAssistant();
+    const activeAssistantKeys = new Set(
+      blocks.filter((block) => block.type === "assistant_turn").map((block) => block.key)
+    );
+    for (const key of assistantMessagesCacheRef.current.keys()) {
+      if (!activeAssistantKeys.has(key)) {
+        assistantMessagesCacheRef.current.delete(key);
+      }
+    }
     return blocks;
   }, [agentLoopNameSet, messages]);
   const hasInterruptedTurn = useMemo17(
@@ -8167,7 +8243,7 @@ function MessageListContent({
                 {
                   label: "\u52A9\u624B\u6D88\u606F",
                   details: block.key,
-                  resetKey: getMessagesRenderSignature(block.messages),
+                  resetKey: getMessagesResetSignature(block.messages),
                   children: customization?.components?.AssistantTurn ? /* @__PURE__ */ jsx35(
                     customization.components.AssistantTurn,
                     {
@@ -8300,7 +8376,10 @@ function ChatView({
     },
     [send]
   );
-  const customization = { classNames, components, renderers };
+  const customization = useMemo18(
+    () => ({ classNames, components, renderers }),
+    [classNames, components, renderers]
+  );
   const SkillStatus = components?.SkillStatusBar;
   return /* @__PURE__ */ jsxs31("div", { className: `flex min-h-0 flex-1 flex-col overflow-hidden ${classNames?.root ?? ""}`, children: [
     isViewer && /* @__PURE__ */ jsxs31("div", { className: `flex items-center justify-center gap-2 border-b border-[hsl(var(--border))] bg-[hsl(var(--card))] py-2 text-xs text-[hsl(var(--muted-foreground))] ${classNames?.viewerBanner ?? ""}`, children: [
@@ -8396,4 +8475,4 @@ use-stick-to-bottom/dist/StickToBottom.js:
    *  Licensed under the MIT License. See License.txt in the project root for license information.
    *--------------------------------------------------------------------------------------------*)
 */
-//# sourceMappingURL=chunk-S6EPGDAL.js.map
+//# sourceMappingURL=chunk-K5EE7X2D.js.map