puckeditor-plugin-ai 0.6.5 → 0.6.7

package/dist/index.d.mts

@@ -189,6 +189,11 @@ type AiSettings = {
     urlContext: boolean;
     googleSearch: boolean;
     enterpriseWebSearch: boolean;
+    /**
+     * When migrating from a page URL, run the full browser pipeline (DOM bundle + structured IR).
+     * Slower and uses more Worker time; leave off for screenshot-only + Vertex url_context.
+     */
+    pageMigrationIr: boolean;
     figmaToken: string;
 };
 type AiPluginProps = {

package/dist/index.d.ts

@@ -189,6 +189,11 @@ type AiSettings = {
     urlContext: boolean;
     googleSearch: boolean;
     enterpriseWebSearch: boolean;
+    /**
+     * When migrating from a page URL, run the full browser pipeline (DOM bundle + structured IR).
+     * Slower and uses more Worker time; leave off for screenshot-only + Vertex url_context.
+     */
+    pageMigrationIr: boolean;
     figmaToken: string;
 };
 type AiPluginProps = {

package/dist/index.js

@@ -323,11 +323,29 @@ 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,
   googleSearch: false,
   enterpriseWebSearch: false,
+  pageMigrationIr: false,
   figmaToken: ""
 };
 var AI_SETTINGS_STORAGE_KEY = "puck-ai-settings";
@@ -589,14 +607,32 @@ function FileMessagePart({
       }
     );
   }
-  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { className: "puck-ai-chat-message-file puck-ai-chat-message-file--document", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("a", { href: part.url, target: "_blank", rel: "noopener noreferrer", download: part.filename, children: (_b = part.filename) != null ? _b : part.mediaType }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { className: "puck-ai-chat-message-file puck-ai-chat-message-file--document", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+    "a",
+    {
+      href: part.url,
+      target: "_blank",
+      rel: "noopener noreferrer",
+      download: part.filename,
+      children: (_b = part.filename) != null ? _b : part.mediaType
+    }
+  ) });
 }
-function ReasoningMessagePart({ part }) {
+function ReasoningMessagePart({
+  part
+}) {
   const streaming = part.state === "streaming";
   return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("details", { className: "puck-ai-chat-message-reasoning", open: streaming, children: [
     /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("summary", { className: "puck-ai-chat-message-reasoning-summary", children: [
       "Reasoning",
-      streaming ? /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { className: "puck-ai-chat-message-reasoning-streaming", "aria-live": "polite", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Loader, { size: 12 }) }) : null
+      streaming ? /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+        "span",
+        {
+          className: "puck-ai-chat-message-reasoning-streaming",
+          "aria-live": "polite",
+          children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Loader, { size: 12 })
+        }
+      ) : null
     ] }),
     /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { className: "puck-ai-chat-message-reasoning-body", children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react_markdown.default, { children: part.text }) })
   ] });
@@ -657,7 +693,8 @@ function DataMessagePart({ part }) {
     return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { className: "puck-ai-chat-message-data-summary", children: [
       "Tokens: in ",
       (_b = (_a = d.tokenUsage) == null ? void 0 : _a.inputTokens) != null ? _b : "\u2014",
-      " \xB7 out ",
+      " \xB7 out",
+      " ",
       (_d = (_c = d.tokenUsage) == null ? void 0 : _c.outputTokens) != null ? _d : "\u2014",
       d.totalCost !== void 0 ? ` \xB7 cost ${d.totalCost}` : ""
     ] });
@@ -1234,6 +1271,27 @@ function SettingsPanel({
           }
         )
       ] }),
+      /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { className: "puck-ai-settings-row", children: [
+        /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
+          "label",
+          {
+            className: "puck-ai-settings-label",
+            htmlFor: "puck-ai-page-migration-ir",
+            children: [
+              "Page migration IR",
+              /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { className: "puck-ai-settings-hint", children: "Full DOM bundle + structured plan when a source URL is captured (slower; screenshot-only when off)" })
+            ]
+          }
+        ),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+          Toggle,
+          {
+            id: "puck-ai-page-migration-ir",
+            checked: settings.pageMigrationIr,
+            onChange: (v) => onChange({ pageMigrationIr: v })
+          }
+        )
+      ] }),
       /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { className: "puck-ai-settings-row", children: [
         /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
           "label",
@@ -1329,10 +1387,17 @@ function SettingsPanel({
     /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { className: "puck-ai-settings-section", children: [
       /* @__PURE__ */ (0, import_jsx_runtime.jsx)("div", { className: "puck-ai-settings-section-title", children: "Chat" }),
       /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("div", { className: "puck-ai-settings-row puck-ai-settings-row--column", children: [
-        /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("label", { className: "puck-ai-settings-label", htmlFor: "puck-ai-clear-chat", children: [
-          "Clear chat history",
-          /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { className: "puck-ai-settings-hint", children: "Remove all messages and reset the conversation context for this session." })
-        ] }),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
+          "label",
+          {
+            className: "puck-ai-settings-label",
+            htmlFor: "puck-ai-clear-chat",
+            children: [
+              "Clear chat history",
+              /* @__PURE__ */ (0, import_jsx_runtime.jsx)("span", { className: "puck-ai-settings-hint", children: "Remove all messages and reset the conversation context for this session." })
+            ]
+          }
+        ),
         /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
           "button",
           {
@@ -1481,6 +1546,8 @@ function Chat({
         if (currentSettings.thinkingLevel !== "none")
           geminiConfig.thinkingLevel = currentSettings.thinkingLevel;
         if (currentSettings.urlContext) geminiConfig.urlContext = true;
+        if (currentSettings.pageMigrationIr)
+          geminiConfig.pageMigrationIr = true;
         if (currentSettings.googleSearch) geminiConfig.googleSearch = true;
         if (currentSettings.enterpriseWebSearch)
           geminiConfig.enterpriseWebSearch = true;
@@ -1574,8 +1641,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);
     });
   };