@blank-utils/llm 0.4.15 → 0.4.17

package/dist/{chunk-AUJSSEB5.js → chunk-MV5JXFWL.js}

@@ -915,9 +915,10 @@ function ChatInput({
   const textareaRef = useRef2(null);
   const fileInputRef = useRef2(null);
   const [isDragging, setIsDragging] = useState2(false);
-  const captionerRef = useRef2(null);
+  const captionerPromiseRef = useRef2(null);
   useEffect2(() => {
     let mounted = true;
+    if (captionerPromiseRef.current) return;
     const initCaptioner = async () => {
       try {
         console.log("[ImagePipeline] Initializing Transformers.js background captioner...");
@@ -925,18 +926,16 @@ function ChatInput({
         env.allowLocalModels = false;
         env.useBrowserCache = true;
         const captioner = await pipeline2("image-to-text", "Xenova/vit-gpt2-image-captioning", {
-          device: "wasm",
-          dtype: "q8"
+          device: "wasm"
         });
-        if (mounted) {
-          console.log("[ImagePipeline] Captioner loaded successfully!");
-          captionerRef.current = captioner;
-        }
+        if (mounted) console.log("[ImagePipeline] Captioner loaded successfully!");
+        return captioner;
       } catch (err) {
         console.warn("Failed to initialize background captioner:", err);
+        return null;
       }
     };
-    initCaptioner();
+    captionerPromiseRef.current = initCaptioner();
     return () => {
       mounted = false;
     };
@@ -1059,22 +1058,27 @@ ${newText}` : newText);
         const id = Math.random().toString(36).substring(7);
         const dataUrl = e.target.result;
         let extractedText;
-        if (captionerRef.current) {
-          console.log(`[ImagePipeline] Generating caption for ${file.name}...`);
+        if (captionerPromiseRef.current) {
+          console.log(`[ImagePipeline] Generating caption for ${file.name} (Waiting for captioner)...`);
           try {
-            const out = await captionerRef.current(dataUrl);
-            console.log("[ImagePipeline] Raw captioner output:", out);
-            if (Array.isArray(out) && out[0] && out[0].generated_text) {
-              extractedText = out[0].generated_text;
-            } else if (!Array.isArray(out) && out.generated_text) {
-              extractedText = out.generated_text;
+            const captioner = await captionerPromiseRef.current;
+            if (captioner) {
+              const out = await captioner(dataUrl);
+              console.log("[ImagePipeline] Raw captioner output:", out);
+              if (Array.isArray(out) && out[0] && out[0].generated_text) {
+                extractedText = out[0].generated_text;
+              } else if (!Array.isArray(out) && out.generated_text) {
+                extractedText = out.generated_text;
+              }
+              console.log("[ImagePipeline] Extracted caption text:", extractedText);
+            } else {
+              console.log("[ImagePipeline] Captioner initialized to null, skipping caption generation.");
             }
-            console.log("[ImagePipeline] Extracted caption text:", extractedText);
           } catch (err) {
             console.warn("[ImagePipeline] Background captioning failed for image:", err);
           }
         } else {
-          console.log("[ImagePipeline] Captioner ref is null, skipping caption generation.");
+          console.log("[ImagePipeline] Captioner promise ref is null, skipping caption generation.");
         }
         onImageAdd?.({ id, dataUrl, file, name: file.name, extractedText });
       }
@@ -1615,7 +1619,15 @@ ${systemPrompt}` : systemPrompt;
             mermaid: mermaidOptions,
             controls: streamdownControls,
             parseMarkdownIntoBlocksFn: sanitizeMarkdownLanguageBlocks,
-            children: msg.content.includes("\u{1F4C4} PDF:") ? (msg.content.split("\u{1F4C4} PDF:")[0] || "").trim() : msg.content
+            children: [
+              { match: "\u{1F4C4} PDF:", index: msg.content.indexOf("\u{1F4C4} PDF:") },
+              { match: "\u{1F4C4} SVG Source Code", index: msg.content.indexOf("\u{1F4C4} SVG Source Code") },
+              { match: "\u{1F5BC}\uFE0F Image Auto-Caption", index: msg.content.indexOf("\u{1F5BC}\uFE0F Image Auto-Caption") }
+            ].filter((m) => m.index !== -1).reduce((min, m) => m.index < min ? m.index : min, msg.content.length) !== msg.content.length ? msg.content.substring(0, [
+              { match: "\u{1F4C4} PDF:", index: msg.content.indexOf("\u{1F4C4} PDF:") },
+              { match: "\u{1F4C4} SVG Source Code", index: msg.content.indexOf("\u{1F4C4} SVG Source Code") },
+              { match: "\u{1F5BC}\uFE0F Image Auto-Caption", index: msg.content.indexOf("\u{1F5BC}\uFE0F Image Auto-Caption") }
+            ].filter((m) => m.index !== -1).reduce((min, m) => m.index < min ? m.index : min, msg.content.length)).trim() : msg.content
           }
         ) })
       ] }) : /* @__PURE__ */ jsxs3("div", { className: "prose prose-base dark:prose-invert max-w-none py-2 w-full min-w-0 font-mono bg-transparent mt-2 !text-current", children: [

package/dist/index.js

@@ -26,7 +26,7 @@ import {
   useCompletion,
   useLLM,
   useStream
-} from "./chunk-AUJSSEB5.js";
+} from "./chunk-MV5JXFWL.js";
 export {
   Chat,
   ChatInput,

package/dist/react/index.js

@@ -9,7 +9,7 @@ import {
   useCompletion,
   useLLM,
   useStream
-} from "../chunk-AUJSSEB5.js";
+} from "../chunk-MV5JXFWL.js";
 export {
   Chat,
   ChatApp,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blank-utils/llm",
-  "version": "0.4.15",
+  "version": "0.4.17",
   "description": "Run LLMs directly in your browser with WebGPU acceleration. Supports React hooks and eager background loading.",
   "type": "module",
   "main": "./dist/index.js",