@gendive/chatllm 0.21.9 → 0.22.1

package/dist/react/index.mjs

@@ -2184,6 +2184,23 @@ var convertAttachmentsToBase64 = async (attachments) => Promise.all(
     size: att.size
   }))
 );
+var convertAttachmentsWithUploaderCached = async (attachments, uploader, cache) => Promise.all(
+  attachments.map(async (att) => {
+    let url = cache.get(att.id);
+    if (!url) {
+      url = await uploader(att.file);
+      cache.set(att.id, url);
+    }
+    return {
+      name: att.name,
+      mimeType: att.mimeType,
+      base64: "",
+      url,
+      size: att.size,
+      source: "uploader"
+    };
+  })
+);
 var findPreviousResultImage = (messages) => {
   for (let i = messages.length - 1; i >= 0; i--) {
     const msg = messages[i];
@@ -2245,6 +2262,8 @@ var useChatUI = (options) => {
     // Image upload
     onUploadImage,
     onImageError,
+    // File upload
+    fileUploader,
     // External storage options
     useExternalStorage = false,
     startWithNewSession = false,
@@ -2352,6 +2371,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 +2393,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());
@@ -3452,6 +3473,7 @@ ${finalContent}`;
     const actionPrompt = selectedAction?.systemPrompt;
     const isHidden = options2?.hiddenUserMessage ?? false;
     const currentAttachments = attachments;
+    const uploadedUrlMap = /* @__PURE__ */ new Map();
     let userContentParts;
     if (currentAttachments.length > 0) {
       userContentParts = [];
@@ -3460,8 +3482,19 @@ ${finalContent}`;
       }
       for (const att of currentAttachments) {
         if (att.type === "image" && att.file) {
-          const dataUri = await fileToDataUri(att.file);
-          const imageUrl = onUploadImageRef.current ? await onUploadImageRef.current(dataUri, att.name) : dataUri;
+          let imageUrl;
+          try {
+            if (fileUploaderRef.current) {
+              imageUrl = await fileUploaderRef.current(att.file);
+            } else {
+              const dataUri = await fileToDataUri(att.file);
+              imageUrl = onUploadImageRef.current ? await onUploadImageRef.current(dataUri, att.name) : dataUri;
+            }
+          } catch (err) {
+            console.error("[chatllm] image upload failed, falling back to data URI:", err);
+            imageUrl = await fileToDataUri(att.file);
+          }
+          uploadedUrlMap.set(att.id, imageUrl);
           userContentParts.push({ type: "image", url: imageUrl, alt: att.name, fileName: att.name });
         } else {
           userContentParts.push({ type: "file", name: att.name, url: att.previewUrl || "", mimeType: att.mimeType, size: att.size });
@@ -3560,7 +3593,7 @@ ${finalContent}`;
           })
         );
         try {
-          const filesToPass = skillConfig.autoConvertBase64 ? await convertAttachmentsToBase64(matchedFiles) : matchedFiles;
+          const filesToPass = fileUploaderRef.current ? await convertAttachmentsWithUploaderCached(matchedFiles, fileUploaderRef.current, uploadedUrlMap) : 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 +3651,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 convertAttachmentsWithUploaderCached(imageAttachments, fileUploaderRef.current, uploadedUrlMap) : await convertAttachmentsToBase64(imageAttachments);
+        } catch (err) {
+          console.error("[chatllm] pendingAttachment conversion failed:", err);
+          pendingAttachmentDataRef.current = null;
+        }
       } else {
         pendingAttachmentDataRef.current = null;
       }
@@ -3766,7 +3804,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",
@@ -15077,7 +15117,9 @@ var ChatUIWithHook = ({
   onAnalyzePatterns,
   // Image upload
   onUploadImage,
-  onImageError
+  onImageError,
+  // File upload
+  fileUploader
 }) => {
   const hookOptions = {
     models,
@@ -15128,7 +15170,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(