puckeditor-plugin-ai 0.6.5 → 0.6.6

package/dist/index.js

@@ -323,6 +323,23 @@ function filesToAttachedImages(files) {
     return results;
   });
 }
+function mediaTypeFromDataUrl(dataUrl) {
+  if (!dataUrl.startsWith("data:")) return "application/octet-stream";
+  const rest = dataUrl.slice(5);
+  const semi = rest.indexOf(";");
+  const comma = rest.indexOf(",");
+  const end = semi === -1 ? comma : semi;
+  if (end <= 0) return "application/octet-stream";
+  return rest.slice(0, end) || "application/octet-stream";
+}
+function attachedImageToFileUIPart(img) {
+  return {
+    type: "file",
+    mediaType: mediaTypeFromDataUrl(img.dataUrl),
+    url: img.dataUrl,
+    filename: img.name
+  };
+}
 var DEFAULT_AI_SETTINGS = {
   thinkingLevel: "none",
   urlContext: false,
@@ -1574,8 +1591,12 @@ function Chat({
     setError("");
     setPromptValue("");
     pendingSendImagesRef.current = attachedImages.map((img) => img.dataUrl);
+    const fileParts = attachedImages.map(attachedImageToFileUIPart);
     setAttachedImages([]);
-    sendMessage({ text }).catch((e) => {
+    const sendPayload = {};
+    if (text) sendPayload.text = text;
+    if (fileParts.length > 0) sendPayload.files = fileParts;
+    sendMessage(sendPayload).catch((e) => {
       console.error(e);
     });
   };