@blade-hq/agent-kit 0.5.7 → 0.5.11

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

@@ -16,6 +16,7 @@ import {
   resolveSessionFilePreviewTarget
 } from "./chunk-X3S36RR2.js";
 import {
+  apiFetchResponse,
   buildMessageContent,
   buildToolPreviewKey,
   contentPreview,
@@ -3780,10 +3781,10 @@ function ConnectionBanner() {
 import { ChevronDown as ChevronDown3, Lightbulb as Lightbulb2, MessageSquare } from "lucide-react";
 import {
   useCallback as useCallback12,
-  useEffect as useEffect16,
+  useEffect as useEffect17,
   useEffectEvent as useEffectEvent4,
   useMemo as useMemo17,
-  useRef as useRef12,
+  useRef as useRef13,
   useState as useState21
 } from "react";
 
@@ -4213,7 +4214,7 @@ function useStickToBottomContext() {
 
 // src/react/components/chat/AssistantTurnBlock.tsx
 import { AlertCircle, BookOpen, Check as Check4, ChevronRight as ChevronRight5 } from "lucide-react";
-import { memo as memo3, 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 useEffect15, useMemo as useMemo15, useRef as useRef12, useState as useState18 } from "react";
 
 // src/react/routes.ts
 var MEMORIES_ROUTE = "/memories";
@@ -4495,7 +4496,7 @@ ReasoningContent.displayName = "ReasoningContent";
 
 // src/react/components/chat/AgentLoopBlock.tsx
 import { Bot, Check as Check3, ChevronRight as ChevronRight4, FileText as FileText6, Loader2 as Loader24, MessageSquareMore as MessageSquareMore2 } from "lucide-react";
-import { useEffect as useEffect13, useMemo as useMemo14, useState as useState17 } from "react";
+import { useEffect as useEffect14, useMemo as useMemo14, useState as useState17 } from "react";
 
 // src/react/components/chat/ResourceIframe.tsx
 import { useEffect as useEffect10, useEffectEvent as useEffectEvent3, useRef as useRef9, useState as useState11 } from "react";
@@ -5781,7 +5782,8 @@ function buildAskUserPayload(argumentsJson) {
 }
 
 // src/react/components/chat/UserMessageBubble.tsx
-import { useState as useState16 } from "react";
+import { useQueries } from "@tanstack/react-query";
+import { useEffect as useEffect13, useRef as useRef11, useState as useState16 } from "react";
 
 // src/react/lib/preview-dispatch.ts
 var IMAGE_EXTS = [
@@ -6282,29 +6284,47 @@ function UserMessageBubble({ message, className }) {
       )
     ] }) });
   }
-  const imageTextAttachments = activeSessionId ? textAttachments.flatMap((attachment) => {
-    if (!isImageTextAttachment(attachment.name, attachment.uploadedPath)) {
-      return [];
-    }
-    const pathForUrl = attachment.uploadedPath ?? attachment.name;
-    return [
-      {
-        ...attachment,
-        url: buildSessionFileUrl(activeSessionId, pathForUrl)
-      }
-    ];
+  const imageTextAttachmentMeta = activeSessionId ? textAttachments.flatMap((attachment) => {
+    if (!isImageTextAttachment(attachment.name, attachment.uploadedPath)) return [];
+    return [{ ...attachment, filePath: attachment.uploadedPath ?? attachment.name }];
   }) : [];
   const nonImageTextAttachments = textAttachments.filter(
     (attachment) => !isImageTextAttachment(attachment.name, attachment.uploadedPath)
   );
+  const imageAttachmentQueries = useQueries({
+    queries: imageTextAttachmentMeta.map((att) => ({
+      queryKey: ["session-image", activeSessionId, att.filePath],
+      queryFn: async ({ signal }) => {
+        const apiPath = `/api/sessions/${encodeURIComponent(activeSessionId)}/files/${encodeURIComponent(att.filePath)}`;
+        const res = await apiFetchResponse(apiPath, { signal });
+        const blob = await res.blob();
+        return URL.createObjectURL(blob);
+      },
+      enabled: !!activeSessionId
+    }))
+  });
+  const blobUrlMapRef = useRef11(/* @__PURE__ */ new Map());
+  for (let i = 0; i < imageAttachmentQueries.length; i++) {
+    const url = imageAttachmentQueries[i].data;
+    const key = imageTextAttachmentMeta[i]?.filePath ?? String(i);
+    const prev = blobUrlMapRef.current.get(key);
+    if (prev && url && prev !== url) URL.revokeObjectURL(prev);
+    if (url) blobUrlMapRef.current.set(key, url);
+  }
+  useEffect13(() => {
+    const urls = blobUrlMapRef.current;
+    return () => {
+      for (const u of urls.values()) URL.revokeObjectURL(u);
+    };
+  }, []);
   const lightboxImages = [
     ...imageParts.map((part) => ({
       src: part.image_url.url,
       alt: "\u7528\u6237\u4E0A\u4F20\u7684\u56FE\u7247"
     })),
-    ...imageTextAttachments.map((attachment) => ({
-      src: attachment.url,
-      alt: attachment.name || "\u7528\u6237\u4E0A\u4F20\u7684\u56FE\u7247"
+    ...imageAttachmentQueries.map((q, i) => ({
+      src: q.data ?? "",
+      alt: imageTextAttachmentMeta[i]?.name || "\u7528\u6237\u4E0A\u4F20\u7684\u56FE\u7247"
     }))
   ];
   const [lightboxIndex, setLightboxIndex] = useState16(null);
@@ -6328,7 +6348,7 @@ function UserMessageBubble({ message, className }) {
     setPreview({ filename: attachment.name, url, mode });
   };
   return /* @__PURE__ */ jsx28("div", { className: cn("flex justify-end", className), children: /* @__PURE__ */ jsxs23("div", { className: "group flex max-w-[72%] flex-col items-end gap-3", children: [
-    (imageParts.length > 0 || imageTextAttachments.length > 0) && /* @__PURE__ */ jsxs23("div", { className: "grid gap-2", children: [
+    (imageParts.length > 0 || imageTextAttachmentMeta.length > 0) && /* @__PURE__ */ jsxs23("div", { className: "grid gap-2", children: [
       imageParts.map((part, idx) => /* @__PURE__ */ jsx28(
         "button",
         {
@@ -6347,24 +6367,34 @@ function UserMessageBubble({ message, className }) {
         },
         part.image_url.url
       )),
-      imageTextAttachments.map((attachment, idx) => /* @__PURE__ */ jsx28(
-        "button",
-        {
-          type: "button",
-          onClick: () => setLightboxIndex(imageParts.length + idx),
-          className: "cursor-zoom-in",
-          title: "\u67E5\u770B\u5927\u56FE",
-          children: /* @__PURE__ */ jsx28(
-            "img",
-            {
-              src: attachment.url,
-              alt: attachment.name || "\u7528\u6237\u4E0A\u4F20\u7684\u56FE\u7247",
-              className: "max-h-64 rounded-xl border border-[hsl(var(--user-msg-border))] object-cover"
-            }
-          )
-        },
-        `${attachment.uploadedPath}:${attachment.name}`
-      ))
+      imageTextAttachmentMeta.map((att, idx) => {
+        const blobUrl = imageAttachmentQueries[idx]?.data;
+        return blobUrl ? /* @__PURE__ */ jsx28(
+          "button",
+          {
+            type: "button",
+            onClick: () => setLightboxIndex(imageParts.length + idx),
+            className: "cursor-zoom-in",
+            title: "\u67E5\u770B\u5927\u56FE",
+            children: /* @__PURE__ */ jsx28(
+              "img",
+              {
+                src: blobUrl,
+                alt: att.name || "\u7528\u6237\u4E0A\u4F20\u7684\u56FE\u7247",
+                className: "max-h-64 rounded-xl border border-[hsl(var(--user-msg-border))] object-cover"
+              }
+            )
+          },
+          `${att.uploadedPath}:${att.name}`
+        ) : imageAttachmentQueries[idx]?.isError ? null : /* @__PURE__ */ jsx28(
+          "div",
+          {
+            className: "flex h-20 items-center justify-center rounded-xl border border-[hsl(var(--user-msg-border))] bg-[hsl(var(--muted)/0.3)] text-xs text-[hsl(var(--muted-foreground))]",
+            children: "\u52A0\u8F7D\u4E2D..."
+          },
+          `${att.uploadedPath}:${att.name}`
+        );
+      })
     ] }),
     lightboxImages.length > 0 && /* @__PURE__ */ jsx28(
       ImageLightbox,
@@ -6525,7 +6555,7 @@ function AgentLoopBlock({ toolCall, sessionId, reasoning }) {
     ),
     [childMessages]
   );
-  useEffect13(() => {
+  useEffect14(() => {
     if (hasAwaitingAnswer) {
       setExpanded(true);
     }
@@ -7011,8 +7041,8 @@ function AssistantTurnBlockBase({
   const [displayMode, setDisplayMode] = useState18(
     defaultTurnDisplayMode({ forceExpanded, hasActionableToolCall })
   );
-  const wasStreamingRef = useRef11(isStreaming);
-  useEffect14(() => {
+  const wasStreamingRef = useRef12(isStreaming);
+  useEffect15(() => {
     if (wasStreamingRef.current && !isStreaming && !forceExpanded) {
       setDisplayMode(defaultTurnDisplayMode({ forceExpanded, hasActionableToolCall }));
     }
@@ -7673,7 +7703,7 @@ import {
   TerminalSquare,
   WandSparkles
 } from "lucide-react";
-import { useEffect as useEffect15, useMemo as useMemo16, useState as useState20 } from "react";
+import { useEffect as useEffect16, useMemo as useMemo16, useState as useState20 } from "react";
 import { jsx as jsx33, jsxs as jsxs28 } from "react/jsx-runtime";
 var EMPTY_EVENTS = [];
 function formatElapsedDuration(durationMs) {
@@ -7753,7 +7783,7 @@ function getTurnStartedAt(events) {
 }
 function useElapsedDuration(startedAt, active) {
   const [now, setNow] = useState20(() => Date.now());
-  useEffect15(() => {
+  useEffect16(() => {
     if (!active || startedAt == null) {
       return;
     }
@@ -7941,7 +7971,7 @@ 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 assistantMessagesCacheRef = useRef13(/* @__PURE__ */ new Map());
   const renderBlocks = useMemo17(() => {
     const stableAssistantMessages = (key, buffer) => {
       const previous = assistantMessagesCacheRef.current.get(key);
@@ -8053,9 +8083,9 @@ function MessageList({
     ) ?? null,
     [lastTurnId, renderBlocks]
   );
-  const containerRef = useRef12(null);
-  const scrollContainerRef = useRef12(null);
-  const frameRef = useRef12(null);
+  const containerRef = useRef13(null);
+  const scrollContainerRef = useRef13(null);
+  const frameRef = useRef13(null);
   const [activeTurnId, setActiveTurnId] = useState21(lastTurnId);
   const layoutSignature = useMemo17(
     () => getMessagesMeasureSignature(messages),
@@ -8097,7 +8127,7 @@ function MessageList({
       measureActiveTurn();
     });
   });
-  useEffect16(() => {
+  useEffect17(() => {
     scrollContainerRef.current = findScrollContainer(containerRef.current);
     const sc = scrollContainerRef.current;
     if (!sc) return;
@@ -8113,7 +8143,7 @@ function MessageList({
       }
     };
   }, [scheduleMeasure]);
-  useEffect16(() => {
+  useEffect17(() => {
     scheduleMeasure();
   }, [lastTurnId, layoutSignature, scheduleMeasure]);
   const handleSelectTurn = useCallback12((turnId) => {
@@ -8475,4 +8505,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-K5EE7X2D.js.map
+//# sourceMappingURL=chunk-G2LJZTPX.js.map