@meetsmore-oss/use-ai-client 1.15.0 → 1.17.0

package/dist/bundled.js

@@ -1333,7 +1333,9 @@ var defaultStrings = {
     /** File size error (use {filename} and {maxSize} placeholders) */
     fileSizeError: 'File "{filename}" exceeds {maxSize}MB limit',
     /** File type error (use {type} placeholder) */
-    fileTypeError: 'File type "{type}" is not accepted'
+    fileTypeError: 'File type "{type}" is not accepted',
+    /** Error when too many files are attached (use {max} placeholder) */
+    maxAttachmentsError: "You can attach at most {max} files per message"
   },
   // Floating button
   floatingButton: {
@@ -18289,6 +18291,9 @@ async function getTransformedContent(files, transformer, context, onProgress) {
   transformationCache.set(cacheKey, results);
   return results;
 }
+function normalizeUploadResult(result) {
+  return typeof result === "string" ? { url: result } : result;
+}
 async function toContentPart(attachment, backend) {
   if (attachment.transformedContent !== void 0) {
     return {
@@ -18301,16 +18306,12 @@ async function toContentPart(attachment, backend) {
       }
     };
   }
-  const url3 = await backend.prepareForSend(attachment.file);
-  if (attachment.file.type.startsWith("image/")) {
-    return { type: "image", url: url3 };
+  const result = normalizeUploadResult(await backend.prepareForSend(attachment.file));
+  const isImage = attachment.file.type.startsWith("image/");
+  if ("ref" in result) {
+    return isImage ? { type: "image_ref", ref: result.ref } : { type: "file_ref", ref: result.ref, mimeType: attachment.file.type, name: attachment.file.name };
   }
-  return {
-    type: "file",
-    url: url3,
-    mimeType: attachment.file.type,
-    name: attachment.file.name
-  };
+  return isImage ? { type: "image_url", url: result.url } : { type: "file_url", url: result.url, mimeType: attachment.file.type, name: attachment.file.name };
 }
 async function processAttachments(attachments, config2) {
   const { getCurrentChat, backend = new EmbedFileUploadBackend(), transformers = {}, onFileProgress } = config2;
@@ -18464,7 +18465,12 @@ function useFileUpload({
   const enabled = config2 !== void 0;
   const maxFileSize = config2?.maxFileSize ?? DEFAULT_MAX_FILE_SIZE;
   const acceptedTypes = config2?.acceptedTypes;
+  const maxAttachments = config2?.maxAttachments;
   const transformers = config2?.transformers;
+  const attachmentsRef = useRef3(attachments);
+  useEffect3(() => {
+    attachmentsRef.current = attachments;
+  }, [attachments]);
   useEffect3(() => {
     if (fileError) {
       const timer = setTimeout(() => setFileError(null), 3e3);
@@ -18500,7 +18506,13 @@ function useFileUpload({
   }, [getCurrentChat, transformers]);
   const handleFiles = useCallback2(async (files) => {
     const fileArray = Array.from(files);
+    let currentCount = attachmentsRef.current.length;
     for (const file2 of fileArray) {
+      if (maxAttachments !== void 0 && currentCount >= maxAttachments) {
+        const errorMsg = strings.fileUpload.maxAttachmentsError.replace("{max}", String(maxAttachments));
+        setFileError(errorMsg);
+        break;
+      }
       if (file2.size > maxFileSize) {
         const errorMsg = strings.fileUpload.fileSizeError.replace("{filename}", file2.name).replace("{maxSize}", String(Math.round(maxFileSize / (1024 * 1024))));
         setFileError(errorMsg);
@@ -18519,14 +18531,17 @@ function useFileUpload({
         preview
       };
       setAttachments((prev) => [...prev, attachment]);
+      attachmentsRef.current = [...attachmentsRef.current, attachment];
+      currentCount++;
       const transformerKey = findTransformerPattern(file2.type, transformers);
       if (transformerKey) {
         runTransformer(attachmentId, file2, transformerKey);
       }
     }
-  }, [maxFileSize, acceptedTypes, strings, transformers, runTransformer]);
+  }, [maxFileSize, acceptedTypes, maxAttachments, strings, transformers, runTransformer]);
   const removeAttachment = useCallback2((id) => {
     setAttachments((prev) => prev.filter((a) => a.id !== id));
+    attachmentsRef.current = attachmentsRef.current.filter((a) => a.id !== id);
     setProcessingState((prev) => {
       const next = new Map(prev);
       next.delete(id);
@@ -18535,6 +18550,7 @@ function useFileUpload({
   }, []);
   const clearAttachments = useCallback2(() => {
     setAttachments([]);
+    attachmentsRef.current = [];
     setProcessingState(/* @__PURE__ */ new Map());
   }, []);
   const openFilePicker = useCallback2(() => {
@@ -19105,7 +19121,16 @@ function FeedbackButton({ type, isSelected, onClick, selectedColor, unselectedCo
   );
 }
 function hasFileContent(content3) {
-  return Array.isArray(content3) && content3.some((part) => part.type === "file");
+  return Array.isArray(content3) && content3.some((part) => part.type === "file" || part.type === "attachment_ref");
+}
+function fileChipInfo(part) {
+  if (part.type === "file") {
+    return { name: part.file.name, size: part.file.size };
+  }
+  if (part.type === "attachment_ref") {
+    return { name: part.name, size: part.size };
+  }
+  return null;
 }
 function UseAIChatPanel({
   onSendMessage,
@@ -19778,14 +19803,10 @@ function UseAIChatPanel({
                                   wordWrap: "break-word"
                                 },
                                 children: [
-                                  message.role === "user" && hasFileContent(message.content) && /* @__PURE__ */ jsx12("div", { style: { display: "flex", flexWrap: "wrap", gap: "6px", marginBottom: "8px" }, children: message.content.filter((part) => part.type === "file").map((part, idx) => /* @__PURE__ */ jsx12(
-                                    FilePlaceholder,
-                                    {
-                                      name: part.file.name,
-                                      size: part.file.size
-                                    },
-                                    idx
-                                  )) }),
+                                  message.role === "user" && hasFileContent(message.content) && /* @__PURE__ */ jsx12("div", { style: { display: "flex", flexWrap: "wrap", gap: "6px", marginBottom: "8px" }, children: message.content.flatMap((part, idx) => {
+                                    const info = fileChipInfo(part);
+                                    return info ? [/* @__PURE__ */ jsx12(FilePlaceholder, { name: info.name, size: info.size }, idx)] : [];
+                                  }) }),
                                   message.role === "assistant" ? /* @__PURE__ */ jsxs9(Fragment, { children: [
                                     message.reasoningParts && message.reasoningParts.length > 0 && /* @__PURE__ */ jsx12(
                                       Reasoning,
@@ -23995,31 +24016,13 @@ var UseAIClient = class {
   async sendPrompt(prompt, multimodalContent, forwardedProps) {
     let messageContent = prompt;
     if (multimodalContent && multimodalContent.length > 0) {
-      messageContent = multimodalContent.map((part) => {
-        if (part.type === "text") {
-          return { type: "text", text: part.text };
-        } else if (part.type === "image") {
-          return { type: "image", url: part.url };
-        } else if (part.type === "file") {
-          return {
-            type: "file",
-            url: part.url,
-            mimeType: part.mimeType
-          };
-        } else {
-          return {
-            type: "transformed_file",
-            text: part.text,
-            originalFile: part.originalFile
-          };
-        }
-      });
+      messageContent = multimodalContent;
     }
     const userMessage = {
       id: v4_default(),
       role: "user",
+      // @ag-ui/core declares Message.content as `string`; the wire also carries MultimodalContent[].
       content: messageContent
-      // Type cast needed for Message type compatibility
     };
     this._messages.push(userMessage);
     const runId = v4_default();
@@ -38484,6 +38487,7 @@ function buildPersistedParts(message, attachments, fileContent) {
     parts2.push({ type: "text", text: message });
   }
   const transformedByKey = /* @__PURE__ */ new Map();
+  const nonTransformedParts = [];
   for (const part of fileContent) {
     if (part.type === "transformed_file") {
       const key = `${part.originalFile.name}:${part.originalFile.size}:${part.originalFile.mimeType}`;
@@ -38493,6 +38497,8 @@ function buildPersistedParts(message, attachments, fileContent) {
       } else {
         transformedByKey.set(key, [part]);
       }
+    } else {
+      nonTransformedParts.push(part);
     }
   }
   for (const attachment of attachments) {
@@ -38504,6 +38510,18 @@ function buildPersistedParts(message, attachments, fileContent) {
         text: transformed.text,
         originalFile: transformed.originalFile
       });
+      continue;
+    }
+    const part = nonTransformedParts.shift();
+    const ref = part && (part.type === "image_ref" || part.type === "file_ref") ? part.ref : void 0;
+    if (ref) {
+      parts2.push({
+        type: "attachment_ref",
+        ref,
+        name: attachment.file.name,
+        mimeType: attachment.file.type,
+        size: attachment.file.size
+      });
     } else {
       parts2.push({
         type: "file",
@@ -38556,9 +38574,18 @@ function transformMessagesToClientFormat(persistedMessages) {
 ${p.text}`
             }];
           }
+          if (p.type === "attachment_ref") {
+            return [
+              p.mimeType.startsWith("image/") ? { type: "image_ref", ref: p.ref } : { type: "file_ref", ref: p.ref, mimeType: p.mimeType, name: p.name }
+            ];
+          }
           return [];
         });
-        return { id: msg.id, role: "user", content: parts2 };
+        return {
+          id: msg.id,
+          role: "user",
+          content: parts2
+        };
       }
     }
   });