@gendive/chatllm 0.21.8 → 0.22.0

package/dist/react/index.mjs

@@ -2184,6 +2184,16 @@ var convertAttachmentsToBase64 = async (attachments) => Promise.all(
     size: att.size
   }))
 );
+var convertAttachmentsWithUploader = async (attachments, uploader) => Promise.all(
+  attachments.map(async (att) => ({
+    name: att.name,
+    mimeType: att.mimeType,
+    base64: "",
+    url: await uploader(att.file),
+    size: att.size,
+    source: "uploader"
+  }))
+);
 var findPreviousResultImage = (messages) => {
   for (let i = messages.length - 1; i >= 0; i--) {
     const msg = messages[i];
@@ -2245,6 +2255,8 @@ var useChatUI = (options) => {
     // Image upload
     onUploadImage,
     onImageError,
+    // File upload
+    fileUploader,
     // External storage options
     useExternalStorage = false,
     startWithNewSession = false,
@@ -2352,6 +2364,7 @@ var useChatUI = (options) => {
   const onLoadModelsRef = useRef5(onLoadModels);
   const onUploadImageRef = useRef5(onUploadImage);
   const onImageErrorRef = useRef5(onImageError);
+  const fileUploaderRef = useRef5(fileUploader);
   const globalMemoryRef = useRef5(null);
   useEffect4(() => {
     onSendMessageRef.current = onSendMessage;
@@ -2373,6 +2386,7 @@ var useChatUI = (options) => {
     onSessionContextChangeRef.current = onSessionContextChange;
     onUploadImageRef.current = onUploadImage;
     onImageErrorRef.current = onImageError;
+    fileUploaderRef.current = fileUploader;
     onLoadModelsRef.current = onLoadModels;
   });
   const abortControllersRef = useRef5(/* @__PURE__ */ new Map());
@@ -3560,7 +3574,7 @@ ${finalContent}`;
           })
         );
         try {
-          const filesToPass = skillConfig.autoConvertBase64 ? await convertAttachmentsToBase64(matchedFiles) : matchedFiles;
+          const filesToPass = fileUploaderRef.current ? await convertAttachmentsWithUploader(matchedFiles, fileUploaderRef.current) : skillConfig.autoConvertBase64 ? await convertAttachmentsToBase64(matchedFiles) : matchedFiles;
           const result = await skillConfig.execute({ files: filesToPass, userMessage: finalContent });
           const attachResultType = result.metadata?.type || result.metadata?.resultType || "text";
           const toolResultPart = {
@@ -3618,7 +3632,12 @@ ${finalContent}`;
       const hasUserText = finalContent.trim().length > 0;
       if (hasImageAttachments && hasUserText) {
         const imageAttachments = currentAttachments.filter((a) => a.type === "image");
-        pendingAttachmentDataRef.current = await convertAttachmentsToBase64(imageAttachments);
+        try {
+          pendingAttachmentDataRef.current = fileUploaderRef.current ? await convertAttachmentsWithUploader(imageAttachments, fileUploaderRef.current) : await convertAttachmentsToBase64(imageAttachments);
+        } catch (err) {
+          console.error("[chatllm] pendingAttachment conversion failed:", err);
+          pendingAttachmentDataRef.current = null;
+        }
       } else {
         pendingAttachmentDataRef.current = null;
       }
@@ -3766,7 +3785,9 @@ ${attachmentContext}
           }
         }
       } else if (attachmentResults.length === 0 && pendingAttachmentDataRef.current !== null) {
-        const isReferenceImage = pendingAttachmentDataRef.current.some((d) => d.url && !d.base64);
+        const isReferenceImage = pendingAttachmentDataRef.current.some(
+          (d) => d.url && !d.base64 && d.source !== "uploader"
+        );
         if (isReferenceImage) {
           chatMessages.push({
             role: "user",
@@ -6495,9 +6516,9 @@ var ImageWithCopyButton = ({ src, alt, imageKey }) => {
         console.log("Canvas method failed (CORS), trying fetch...");
       }
     }
-    if (!src.startsWith("data:")) {
+    if (!imgSrc.startsWith("data:")) {
       try {
-        const response = await fetch(src, { mode: "cors" });
+        const response = await fetch(imgSrc, { mode: "cors" });
         if (response.ok) {
           const blob = await response.blob();
           if (blob.type === "image/png") return blob;
@@ -6537,7 +6558,7 @@ var ImageWithCopyButton = ({ src, alt, imageKey }) => {
     setCopyState("copying");
     try {
       if (!navigator.clipboard?.write) {
-        await navigator.clipboard.writeText(src);
+        await navigator.clipboard.writeText(imgSrc);
         setCopyState("copied");
         setTimeout(() => setCopyState("idle"), 2e3);
         return;
@@ -6547,7 +6568,7 @@ var ImageWithCopyButton = ({ src, alt, imageKey }) => {
         await navigator.clipboard.write([new ClipboardItem({ "image/png": blob })]);
         setCopyState("copied");
       } else {
-        await navigator.clipboard.writeText(src);
+        await navigator.clipboard.writeText(imgSrc);
         setCopyState("copied");
       }
       setTimeout(() => setCopyState("idle"), 2e3);
@@ -6560,7 +6581,7 @@ var ImageWithCopyButton = ({ src, alt, imageKey }) => {
   const handleDownload = async () => {
     try {
       const blob = await getImageBlob();
-      const url = blob ? URL.createObjectURL(blob) : src;
+      const url = blob ? URL.createObjectURL(blob) : imgSrc;
       const link = document.createElement("a");
       link.href = url;
       link.download = alt || `image-${Date.now()}.png`;
@@ -6570,7 +6591,7 @@ var ImageWithCopyButton = ({ src, alt, imageKey }) => {
       if (blob) URL.revokeObjectURL(url);
     } catch (error) {
       console.error("Failed to download image:", error);
-      window.open(src, "_blank");
+      window.open(imgSrc, "_blank");
     }
   };
   return /* @__PURE__ */ jsxs2(
@@ -15077,7 +15098,9 @@ var ChatUIWithHook = ({
   onAnalyzePatterns,
   // Image upload
   onUploadImage,
-  onImageError
+  onImageError,
+  // File upload
+  fileUploader
 }) => {
   const hookOptions = {
     models,
@@ -15128,7 +15151,9 @@ var ChatUIWithHook = ({
     onAnalyzePatterns,
     // Image upload
     onUploadImage,
-    onImageError
+    onImageError,
+    // File upload
+    fileUploader
   };
   const state = useChatUI(hookOptions);
   return /* @__PURE__ */ jsx23(ImageErrorContext.Provider, { value: onImageError || null, children: /* @__PURE__ */ jsx23(