@puckeditor/plugin-ai 0.6.1-canary.433ca837 → 0.6.1-canary.45a1a2e4

package/dist/index.css

@@ -1469,7 +1469,7 @@ button._Menu-item_1tu16_55:hover {
 }
 
 /* css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/ChatBody/styles.module.css/#css-module-data */
-._Chat_8hszl_1 {
+._Chat_1njo2_1 {
   --fade-size: 32px;
   --chat-background-color: white;
   display: flex;
@@ -1479,46 +1479,46 @@ button._Menu-item_1tu16_55:hover {
   overflow: hidden;
 }
 @media (min-width: 638px) {
-  ._Chat_8hszl_1 {
+  ._Chat_1njo2_1 {
     max-height: none;
   }
 }
-._Chat-inner_8hszl_17 {
+._Chat-inner_1njo2_17 {
   background-color: var(--chat-background-color);
   display: flex;
   flex-direction: column;
   max-height: 100%;
   position: relative;
 }
-._Chat--hasMessages_8hszl_25 ._Chat-inner_8hszl_17 {
+._Chat--hasMessages_1njo2_25 ._Chat-inner_1njo2_17 {
   padding-top: 0;
 }
-._Chat_8hszl_1:not(._Chat--hasChildren_8hszl_29) ._Chat-inner_8hszl_17 {
+._Chat_1njo2_1:not(._Chat--hasChildren_1njo2_29) ._Chat-inner_1njo2_17 {
   height: 100%;
   padding-top: 0;
 }
-._Chat--hasMessages_8hszl_25 ._Chat-messages_8hszl_34 {
+._Chat--hasMessages_1njo2_25 ._Chat-messages_1njo2_34 {
   display: flex;
   flex-direction: column-reverse;
   padding: 16px;
   overflow: auto;
 }
-._Chat-form_8hszl_41 {
+._Chat-form_1njo2_41 {
   background-color: white;
   padding: 16px;
 }
-._Chat--hideInput_8hszl_46 ._Chat-form_8hszl_41 {
+._Chat--hideInput_1njo2_46 ._Chat-form_1njo2_41 {
   display: none;
 }
-._Chat--hasMessages_8hszl_25 ._Chat-form_8hszl_41 {
+._Chat--hasMessages_1njo2_25 ._Chat-form_1njo2_41 {
   border-top: 1px solid var(--puck-color-grey-09);
 }
 @media (min-width: 638px) {
-  ._Chat-form_8hszl_41 {
+  ._Chat-form_1njo2_41 {
     border-top: 1px solid var(--puck-color-grey-09);
   }
 }
-._Chat-formHelper_8hszl_60 {
+._Chat-formHelper_1njo2_60 {
   align-items: center;
   display: flex;
   color: var(--puck-color-grey-04);
@@ -1527,11 +1527,11 @@ button._Menu-item_1tu16_55:hover {
   gap: 8px;
   padding-bottom: 12px;
 }
-._Chat--hasMessages_8hszl_25 ._Chat-formHelper_8hszl_60 {
+._Chat--hasMessages_1njo2_25 ._Chat-formHelper_1njo2_60 {
   opacity: 0;
   transition: opacity 250ms ease-in;
 }
-._Chat-examplePrompts_8hszl_75 {
+._Chat-examplePrompts_1njo2_75 {
   align-items: center;
   display: flex;
   flex-wrap: nowrap;
@@ -1540,7 +1540,7 @@ button._Menu-item_1tu16_55:hover {
   width: 100%;
   overflow-x: auto;
 }
-._Chat-examplePrompt_8hszl_75 {
+._Chat-examplePrompt_1njo2_75 {
   align-items: center;
   background: white;
   border: 1px solid var(--puck-color-grey-09);
@@ -1553,35 +1553,35 @@ button._Menu-item_1tu16_55:hover {
   padding: 6px 12px;
   text-wrap: nowrap;
 }
-._Chat-examplePrompt_8hszl_75:hover {
+._Chat-examplePrompt_1njo2_75:hover {
   background: var(--puck-color-azure-11);
   border-color: var(--puck-color-azure-08);
   color: var(--puck-color-azure-04);
 }
-._Chat-examplePromptArrow_8hszl_105 {
+._Chat-examplePromptArrow_1njo2_105 {
   display: flex;
   margin-left: auto;
 }
-._Chat-default_8hszl_110 {
+._Chat-default_1njo2_110 {
   overflow-y: auto;
   flex-grow: 1;
 }
-._Chat--hasMessages_8hszl_25 ._Chat-default_8hszl_110 {
+._Chat--hasMessages_1njo2_25 ._Chat-default_1njo2_110 {
   opacity: 0;
   padding: 0;
   pointer-events: none;
   transition: opacity 250ms ease-in;
 }
-._Chat--hasMessages_8hszl_25 {
+._Chat--hasMessages_1njo2_25 {
   background-color: white;
   transition: background-color 250ms ease-in;
 }
-._ChatMessage_8hszl_127 {
+._ChatMessage_1njo2_127 {
   font-size: var(--puck-font-size-xs);
   line-height: var(--line-height-s);
   overflow-wrap: break-word;
 }
-._ChatMessage--userRole_8hszl_133 ._ChatMessage-text_8hszl_133 {
+._ChatMessage--userRole_1njo2_133 ._ChatMessage-text_1njo2_133 {
   background: var(--puck-color-azure-10);
   border-radius: 16px;
   color: var(--puck-color-grey-03);
@@ -1590,70 +1590,70 @@ button._Menu-item_1tu16_55:hover {
   margin-top: 8px;
   padding: 12px 16px;
 }
-._ChatMessage_8hszl_127 ul,
-._ChatMessage_8hszl_127 ol {
+._ChatMessage_1njo2_127 ul,
+._ChatMessage_1njo2_127 ol {
   list-style: decimal;
   margin: 0;
   margin-bottom: 12px;
   margin-top: 12px;
   padding-left: 32px;
 }
-._ChatMessage_8hszl_127 li {
+._ChatMessage_1njo2_127 li {
   margin-bottom: 8px;
   margin-top: 8px;
   padding-left: 8px;
 }
-._ChatMessage_8hszl_127 h1,
-._ChatMessage_8hszl_127 h2,
-._ChatMessage_8hszl_127 h3,
-._ChatMessage_8hszl_127 h4,
-._ChatMessage_8hszl_127 h5,
-._ChatMessage_8hszl_127 h6 {
+._ChatMessage_1njo2_127 h1,
+._ChatMessage_1njo2_127 h2,
+._ChatMessage_1njo2_127 h3,
+._ChatMessage_1njo2_127 h4,
+._ChatMessage_1njo2_127 h5,
+._ChatMessage_1njo2_127 h6 {
   font-weight: 600;
 }
-._ChatMessage_8hszl_127 h1 {
+._ChatMessage_1njo2_127 h1 {
   font-size: var(--puck-font-l);
 }
-._ChatMessage_8hszl_127 h2 {
+._ChatMessage_1njo2_127 h2 {
   font-size: var(--puck-font-m);
 }
-._ChatMessage_8hszl_127 h3 {
+._ChatMessage_1njo2_127 h3 {
   font-size: var(--puck-font-m);
 }
-._ChatMessage_8hszl_127 h4 {
+._ChatMessage_1njo2_127 h4 {
   font-size: var(--puck-font-s);
 }
-._ChatMessage_8hszl_127 h5 {
+._ChatMessage_1njo2_127 h5 {
   font-size: var(--puck-font-xs);
 }
-._ChatMessage_8hszl_127 h6 {
+._ChatMessage_1njo2_127 h6 {
   font-size: var(--puck-font-xm);
 }
-._ChatMessage_8hszl_127 p {
+._ChatMessage_1njo2_127 p {
   margin-bottom: 8px;
   margin-top: 8px;
 }
-._ChatMessage--userRole_8hszl_133 ._ChatMessage-text_8hszl_133 > *:first-of-type {
+._ChatMessage--userRole_1njo2_133 ._ChatMessage-text_1njo2_133 > *:first-of-type {
   margin-top: 0px;
 }
-._ChatMessage--userRole_8hszl_133 ._ChatMessage-text_8hszl_133 > *:last-of-type {
+._ChatMessage--userRole_1njo2_133 ._ChatMessage-text_1njo2_133 > *:last-of-type {
   margin-bottom: 0px;
 }
-._ChatMessage_8hszl_127 a {
+._ChatMessage_1njo2_127 a {
   color: var(--puck-color-azure-04);
   font-weight: 600;
   text-decoration: none;
 }
-._ChatMessage_8hszl_127 a:hover {
+._ChatMessage_1njo2_127 a:hover {
   color: var(--puck-color-azure-05);
 }
-._ChatMessage_8hszl_127 hr {
+._ChatMessage_1njo2_127 hr {
   border-top: none;
   border-bottom-color: var(--puck-color-grey-09);
   margin-bottom: 16px;
   margin-top: 16px;
 }
-._ChatMessage_8hszl_127 blockquote {
+._ChatMessage_1njo2_127 blockquote {
   margin-bottom: 8px;
   margin-top: 8px;
   margin-left: 0;
@@ -1662,7 +1662,7 @@ button._Menu-item_1tu16_55:hover {
   padding-bottom: 2px;
   padding-top: 2px;
 }
-._ChatMessage-data_8hszl_231 {
+._ChatMessage-data_1njo2_231 {
   background: var(--puck-color-grey-12);
   border: 1px solid var(--puck-color-grey-08);
   border-radius: 16px;
@@ -1672,19 +1672,19 @@ button._Menu-item_1tu16_55:hover {
   margin-top: 12px;
   padding: 12px 16px;
 }
-._ChatMessage-dataInner_8hszl_242 {
+._ChatMessage-dataInner_1njo2_242 {
   align-items: center;
   display: flex;
   gap: 8px;
 }
-._ChatMessage-dataIcon_8hszl_248 {
+._ChatMessage-dataIcon_1njo2_248 {
   display: flex;
 }
-._Chat-loader_8hszl_252 {
+._Chat-loader_1njo2_252 {
   padding: 16px;
   padding-top: 0;
 }
-._Chat-error_8hszl_257 {
+._Chat-error_1njo2_257 {
   align-items: center;
   display: flex;
   gap: 8px;
@@ -1692,122 +1692,12 @@ button._Menu-item_1tu16_55:hover {
   margin: 16px;
   margin-top: 0;
 }
-._Chat-errorLabel_8hszl_266 {
+._Chat-errorLabel_1njo2_266 {
   color: var(--puck-color-red-04);
 }
-._Chat-errorAction_8hszl_270 {
+._Chat-errorAction_1njo2_270 {
   margin-left: auto;
 }
-._Chat-planGroup_8hszl_274 {
-  background: var(--puck-color-grey-12);
-  border: 1px solid var(--puck-color-grey-09);
-  border-radius: 16px;
-  margin-block: 8px;
-  margin-bottom: 12px;
-  overflow: hidden;
-}
-._Chat-planGroupHeader_8hszl_283 {
-  color: var(--puck-color-grey-02);
-  background: var(--puck-color-grey-11);
-  padding: 8px 12px;
-  border-bottom: 1px solid var(--puck-color-grey-09);
-  font-weight: 600;
-  font-size: var(--puck-font-size-xxs);
-}
-._Chat-planGroupFooter_8hszl_292 {
-  color: var(--puck-color-grey-02);
-  background: var(--puck-color-grey-11);
-  display: flex;
-  flex-direction: row-reverse;
-  padding: 8px 12px;
-  border-top: 1px solid var(--puck-color-grey-09);
-  font-weight: 600;
-  font-size: var(--puck-font-size-xxs);
-}
-._Chat-planSummary_8hszl_303 {
-  color: var(--puck-color-grey-02);
-  display: flex;
-  justify-content: space-between;
-  gap: 8px;
-  align-items: flex-end;
-}
-._Chat-planSummaryIcon_8hszl_311 {
-  transition: ease-out 150ms;
-}
-._Chat-plan_8hszl_274[open] ._Chat-planSummaryIcon_8hszl_311 {
-  transform: rotate(180deg);
-}
-._Chat-planSummary_8hszl_303:hover {
-  color: var(--puck-color-azure-04);
-  cursor: pointer;
-}
-._Chat-planPath_8hszl_324 {
-  font-weight: 600;
-  font-size: var(--puck-font-size-xxxs);
-  opacity: 0.8;
-}
-._Chat-planHeader_8hszl_330 {
-  display: flex;
-  justify-content: space-between;
-  gap: 8px;
-}
-._Chat-planTitle_8hszl_336 {
-  font-weight: 600;
-  font-size: var(--puck-font-size-xs);
-}
-._Chat-planPrompt_8hszl_341 {
-  color: var(--puck-color-grey-03);
-  font-size: var(--puck-font-size-xxs);
-  margin-top: 4px;
-}
-._Chat-planStatus_8hszl_347 {
-  margin-left: auto;
-}
-._Chat-plan_8hszl_274 {
-  border-bottom: 1px solid var(--puck-color-grey-09);
-  padding: 8px 12px;
-}
-._Chat-plan_8hszl_274:last-of-type {
-  border-bottom: 0;
-}
-._Chat-planSummary_8hszl_303 {
-  list-style: none;
-}
-._Chat-planSummary_8hszl_303::-webkit-details-marker {
-  display: none;
-}
-._Chat-textStatus_8hszl_368 {
-  background-color: var(--puck-color-grey-04);
-  background-image:
-    linear-gradient(
-      110deg,
-      var(--puck-color-grey-04) 0%,
-      var(--puck-color-grey-08) 50%,
-      var(--puck-color-grey-04) 100%);
-  background-size: 40% 100%;
-  background-position: 100% 0;
-  background-repeat: no-repeat;
-  -webkit-background-clip: text;
-  background-clip: text;
-  color: transparent;
-  -webkit-text-fill-color: transparent;
-  animation: _Chat-textStatusShimmer_8hszl_1 1.4s linear infinite;
-  font-weight: 500;
-  margin-top: 4px;
-}
-@keyframes _Chat-textStatusShimmer_8hszl_1 {
-  from {
-    background-position: -100% 100%;
-  }
-}
-@media (prefers-reduced-motion: reduce) {
-  ._Chat-textStatus_8hszl_368 {
-    animation: none;
-    background-image: none;
-    color: var(--puck-color-grey-03);
-    -webkit-text-fill-color: currentColor;
-  }
-}
 
 /* css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/PromptForm/styles.module.css/#css-module-data */
 ._PromptForm_1r4kq_1 {

package/dist/index.js

@@ -250,12 +250,6 @@ var Bot = createLucideIcon("Bot", [
 init_react_import();
 var Check = createLucideIcon("Check", [["path", { d: "M20 6 9 17l-5-5", key: "1gmf2c" }]]);
 
-// ../../node_modules/.pnpm/lucide-react@0.452.0_react@19.1.1/node_modules/lucide-react/dist/esm/icons/chevron-down.js
-init_react_import();
-var ChevronDown = createLucideIcon("ChevronDown", [
-  ["path", { d: "m6 9 6 6 6-6", key: "qrunsl" }]
-]);
-
 // ../../node_modules/.pnpm/lucide-react@0.452.0_react@19.1.1/node_modules/lucide-react/dist/esm/icons/rotate-ccw.js
 init_react_import();
 var RotateCcw = createLucideIcon("RotateCcw", [
@@ -531,7 +525,7 @@ function useChat({
 }
 
 // src/components/Chat/index.tsx
-var import_core3 = require("@puckeditor/core");
+var import_core2 = require("@puckeditor/core");
 var import_ai4 = require("ai");
 var import_react25 = require("react");
 var import_ulid = require("ulid");
@@ -6186,7 +6180,7 @@ var ctx = (0, import_react20.createContext)({});
 
 // css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/ChatBody/styles.module.css#css-module
 init_react_import();
-var styles_module_default8 = { "Chat": "_Chat_8hszl_1", "Chat-inner": "_Chat-inner_8hszl_17", "Chat--hasMessages": "_Chat--hasMessages_8hszl_25", "Chat--hasChildren": "_Chat--hasChildren_8hszl_29", "Chat-messages": "_Chat-messages_8hszl_34", "Chat-form": "_Chat-form_8hszl_41", "Chat--hideInput": "_Chat--hideInput_8hszl_46", "Chat-formHelper": "_Chat-formHelper_8hszl_60", "Chat-examplePrompts": "_Chat-examplePrompts_8hszl_75", "Chat-examplePrompt": "_Chat-examplePrompt_8hszl_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_8hszl_105", "Chat-default": "_Chat-default_8hszl_110", "ChatMessage": "_ChatMessage_8hszl_127", "ChatMessage--userRole": "_ChatMessage--userRole_8hszl_133", "ChatMessage-text": "_ChatMessage-text_8hszl_133", "ChatMessage-data": "_ChatMessage-data_8hszl_231", "ChatMessage-dataInner": "_ChatMessage-dataInner_8hszl_242", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_8hszl_248", "Chat-loader": "_Chat-loader_8hszl_252", "Chat-error": "_Chat-error_8hszl_257", "Chat-errorLabel": "_Chat-errorLabel_8hszl_266", "Chat-errorAction": "_Chat-errorAction_8hszl_270", "Chat-planGroup": "_Chat-planGroup_8hszl_274", "Chat-planGroupHeader": "_Chat-planGroupHeader_8hszl_283", "Chat-planGroupFooter": "_Chat-planGroupFooter_8hszl_292", "Chat-planSummary": "_Chat-planSummary_8hszl_303", "Chat-planSummaryIcon": "_Chat-planSummaryIcon_8hszl_311", "Chat-plan": "_Chat-plan_8hszl_274", "Chat-planPath": "_Chat-planPath_8hszl_324", "Chat-planHeader": "_Chat-planHeader_8hszl_330", "Chat-planTitle": "_Chat-planTitle_8hszl_336", "Chat-planPrompt": "_Chat-planPrompt_8hszl_341", "Chat-planStatus": "_Chat-planStatus_8hszl_347", "Chat-textStatus": "_Chat-textStatus_8hszl_368", "Chat-textStatusShimmer": "_Chat-textStatusShimmer_8hszl_1" };
+var styles_module_default8 = { "Chat": "_Chat_1njo2_1", "Chat-inner": "_Chat-inner_1njo2_17", "Chat--hasMessages": "_Chat--hasMessages_1njo2_25", "Chat--hasChildren": "_Chat--hasChildren_1njo2_29", "Chat-messages": "_Chat-messages_1njo2_34", "Chat-form": "_Chat-form_1njo2_41", "Chat--hideInput": "_Chat--hideInput_1njo2_46", "Chat-formHelper": "_Chat-formHelper_1njo2_60", "Chat-examplePrompts": "_Chat-examplePrompts_1njo2_75", "Chat-examplePrompt": "_Chat-examplePrompt_1njo2_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_1njo2_105", "Chat-default": "_Chat-default_1njo2_110", "ChatMessage": "_ChatMessage_1njo2_127", "ChatMessage--userRole": "_ChatMessage--userRole_1njo2_133", "ChatMessage-text": "_ChatMessage-text_1njo2_133", "ChatMessage-data": "_ChatMessage-data_1njo2_231", "ChatMessage-dataInner": "_ChatMessage-dataInner_1njo2_242", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_1njo2_248", "Chat-loader": "_Chat-loader_1njo2_252", "Chat-error": "_Chat-error_1njo2_257", "Chat-errorLabel": "_Chat-errorLabel_1njo2_266", "Chat-errorAction": "_Chat-errorAction_1njo2_270" };
 
 // src/components/PromptForm/index.tsx
 init_react_import();
@@ -6338,31 +6332,7 @@ var toolStatusContext = (0, import_react22.createContext)({});
 var ToolStatusProvider = toolStatusContext.Provider;
 
 // src/components/ChatBody/index.tsx
-var import_core2 = require("@puckeditor/core");
-var import_plugin_pages = require("@puckeditor/plugin-pages");
 var import_jsx_runtime30 = require("react/jsx-runtime");
-async function* iterateSSE(body) {
-  const reader = body.getReader();
-  const dec = new TextDecoder();
-  let buf = "";
-  while (true) {
-    const { done, value } = await reader.read();
-    if (done) break;
-    buf += dec.decode(value, { stream: true });
-    let idx;
-    while ((idx = buf.indexOf("\n\n")) >= 0) {
-      const chunk = buf.slice(0, idx).trim();
-      buf = buf.slice(idx + 2);
-      if (!chunk) continue;
-      const line = chunk.startsWith("data: ") ? chunk.slice(6) : chunk;
-      try {
-        yield JSON.parse(line);
-      } catch {
-      }
-    }
-  }
-}
-var usePuck = (0, import_core2.createUsePuck)();
 var cx8 = getClassNameFactory2("Chat", styles_module_default8);
 var cxMessage = getClassNameFactory2("ChatMessage", styles_module_default8);
 function ChatBody({
@@ -6437,86 +6407,11 @@ function ChatMessagePart({
   if (part.type === "text") {
     return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cxMessage("text"), children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(ChatMessageWrapper, { role, children: part.text }) });
   }
-  if (part.type === "tool-plan") {
-    return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(PlanTool, { ...part });
-  }
   if (part.type.startsWith("tool-")) {
     return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(PuckTool, { ...part });
   }
   return null;
 }
-function PlanTool({ output }) {
-  const planOutput = output?.plan;
-  const config = usePuck((s2) => s2.config);
-  const history = usePuck((s2) => s2.history);
-  const dispatch = usePuck((s2) => s2.dispatch);
-  const getPuck = (0, import_core2.useGetPuck)();
-  const [loading, setLoading] = (0, import_react23.useState)({});
-  const { setPageVersions, createPage } = (0, import_plugin_pages.usePagesStore)();
-  if (!planOutput) {
-    return /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("textStatus"), children: "Planning" });
-  }
-  return /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)("div", { className: cx8("planGroup"), children: [
-    /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planGroupHeader"), children: "Plan" }),
-    planOutput?.pages?.map((item) => /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)("details", { className: cx8("plan"), children: [
-      /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)("summary", { className: cx8("planSummary"), children: [
-        /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)("div", { className: cx8("planSummaryText"), children: [
-          /* @__PURE__ */ (0, import_jsx_runtime30.jsxs)("div", { className: cx8("planHeader"), children: [
-            /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planTitle"), children: item.shortTitle }),
-            /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planStatus"), children: loading[item.path] && /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(Loader, { size: 18 }) })
-          ] }),
-          /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planPath"), children: item.path })
-        ] }),
-        /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planSummaryIcon"), children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(ChevronDown, {}) })
-      ] }),
-      /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planPrompt"), children: item.pagePrompt })
-    ] }, item.path)),
-    /* @__PURE__ */ (0, import_jsx_runtime30.jsx)("div", { className: cx8("planGroupFooter"), children: /* @__PURE__ */ (0, import_jsx_runtime30.jsx)(
-      Button,
-      {
-        onClick: async () => {
-          const res = await fetch(`/api/puck/plan`, {
-            method: "POST",
-            body: JSON.stringify({ plan: planOutput, config })
-          });
-          if (res.body) {
-            for await (const chunk of iterateSSE(res.body)) {
-              if (chunk.type === "data-plan-progress") {
-                if (chunk.data.data) {
-                  const tempId = Math.random().toString();
-                  setPageVersions(
-                    chunk.data.path,
-                    [
-                      {
-                        id: tempId,
-                        data: chunk.data.data,
-                        versionNumber: 1,
-                        isNew: true
-                      }
-                    ],
-                    chunk.data.path
-                  );
-                  setLoading((s2) => ({ ...s2, [chunk.data.path]: false }));
-                  if (import_plugin_pages.usePagesStore.getState().activePage === chunk.data.path) {
-                    history.setHistories([]);
-                    dispatch({
-                      type: "set",
-                      state: { ...getPuck().appState, data: chunk.data.data }
-                    });
-                  }
-                } else if (chunk.data.path && chunk.data.status === "processing") {
-                  createPage(chunk.data.path);
-                  setLoading((s2) => ({ ...s2, [chunk.data.path]: true }));
-                }
-              }
-            }
-          }
-        },
-        children: "Execute plan"
-      }
-    ) })
-  ] });
-}
 function PuckTool({
   toolCallId,
   output,
@@ -6744,7 +6639,7 @@ var q2 = (0, import_qler.default)();
 var BENCHMARK = false;
 var prefixedUlid = (prefix = "") => `${prefix ? `${prefix}_` : ""}${(0, import_ulid.ulid)()}`;
 var getClassName18 = getClassNameFactory("Chat", styles_module_default);
-var usePuck2 = (0, import_core3.createUsePuck)();
+var usePuck = (0, import_core2.createUsePuck)();
 function Chat2({
   chat,
   host = "/api/puck/chat",
@@ -6752,8 +6647,8 @@ function Chat2({
   scrollTrackingId
 }) {
   const { examplePrompts } = chat ?? {};
-  const puckDispatch = usePuck2((s2) => s2.dispatch);
-  const getPuck = (0, import_core3.useGetPuck)();
+  const puckDispatch = usePuck((s2) => s2.dispatch);
+  const getPuck = (0, import_core2.useGetPuck)();
   const localChatId = (0, import_react25.useRef)("");
   const inputRef = (0, import_react25.useRef)(null);
   const pluginRef = (0, import_react25.useRef)(null);
@@ -6761,29 +6656,39 @@ function Chat2({
   const [toolStatus, setToolStatus] = (0, import_react25.useState)({});
   const uploadScreenshot = (0, import_react25.useCallback)(
     async (width, bucketUrl) => {
-      const iframeDocument = document?.getElementById("preview-frame")?.contentDocument?.documentElement ?? null;
-      if (!iframeDocument) return;
-      const canvas = await (0, import_html2canvas_pro.default)(iframeDocument, {
-        scale: 2,
-        backgroundColor: "#ffffff",
-        width,
-        windowWidth: width,
-        foreignObjectRendering: false,
-        imageTimeout: 3e4,
-        logging: false,
-        allowTaint: false,
-        useCORS: true,
-        scrollY: 0,
-        ignoreElements: (el) => Array.from(el.classList).some(
-          (c2) => c2.startsWith("_DraggableComponent--hover") || c2.startsWith("_ActionBar")
-        )
-      });
-      const image = canvas.toDataURL("image/webp", 0.8);
-      const blob = await (await fetch(image)).blob();
-      await fetch(bucketUrl, {
-        method: "PUT",
-        body: blob
-      });
+      try {
+        const iframeDocument = document?.getElementById("preview-frame")?.contentDocument?.documentElement ?? null;
+        if (!iframeDocument) {
+          throw new Error("Preview frame not found");
+        }
+        const canvas = await (0, import_html2canvas_pro.default)(iframeDocument, {
+          scale: 2,
+          backgroundColor: "#ffffff",
+          width,
+          windowWidth: width,
+          foreignObjectRendering: false,
+          imageTimeout: 3e4,
+          logging: false,
+          allowTaint: false,
+          useCORS: true,
+          scrollY: 0,
+          ignoreElements: (el) => Array.from(el.classList).some(
+            (c2) => c2.startsWith("_DraggableComponent--hover") || c2.startsWith("_ActionBar")
+          )
+        });
+        const image = canvas.toDataURL("image/webp", 0.8);
+        const imageResponse = await fetch(image);
+        const blob = await imageResponse.blob();
+        const uploadResponse = await fetch(bucketUrl, {
+          method: "PUT",
+          body: blob
+        });
+        if (!uploadResponse.ok) {
+          throw new Error(`Upload failed with status ${uploadResponse.status}`);
+        }
+      } catch (e2) {
+        console.error("Failed to upload screenshot", e2);
+      }
     },
     []
   );
@@ -6828,9 +6733,10 @@ function Chat2({
           return;
         }
         case "data-send-screenshot": {
-          dataPart.data.urls.map((obj) => {
+          dataPart.data.urls.forEach((obj) => {
             const [[key, value]] = Object.entries(obj);
             const breakpoint = Number(key);
+            uploadScreenshot(breakpoint, value);
           });
           return;
         }
@@ -6839,7 +6745,7 @@ function Chat2({
           return;
       }
     },
-    [getPuck]
+    [getPuck, puckDispatch, uploadScreenshot]
   );
   const { messages, status, sendMessage, regenerate, setMessages } = useChat({
     generateId: () => prefixedUlid("msg"),
@@ -7194,7 +7100,6 @@ lucide-react/dist/esm/createLucideIcon.js:
 lucide-react/dist/esm/icons/arrow-up.js:
 lucide-react/dist/esm/icons/bot.js:
 lucide-react/dist/esm/icons/check.js:
-lucide-react/dist/esm/icons/chevron-down.js:
 lucide-react/dist/esm/icons/rotate-ccw.js:
 lucide-react/dist/esm/icons/triangle-alert.js:
 lucide-react/dist/esm/lucide-react.js:

package/dist/index.mjs

@@ -238,12 +238,6 @@ var Bot = createLucideIcon("Bot", [
 init_react_import();
 var Check = createLucideIcon("Check", [["path", { d: "M20 6 9 17l-5-5", key: "1gmf2c" }]]);
 
-// ../../node_modules/.pnpm/lucide-react@0.452.0_react@19.1.1/node_modules/lucide-react/dist/esm/icons/chevron-down.js
-init_react_import();
-var ChevronDown = createLucideIcon("ChevronDown", [
-  ["path", { d: "m6 9 6 6 6-6", key: "qrunsl" }]
-]);
-
 // ../../node_modules/.pnpm/lucide-react@0.452.0_react@19.1.1/node_modules/lucide-react/dist/esm/icons/rotate-ccw.js
 init_react_import();
 var RotateCcw = createLucideIcon("RotateCcw", [
@@ -522,13 +516,13 @@ function useChat({
 
 // src/components/Chat/index.tsx
 import {
-  createUsePuck as createUsePuck2,
-  useGetPuck as useGetPuck2
+  createUsePuck,
+  useGetPuck
 } from "@puckeditor/core";
 import {
   DefaultChatTransport
 } from "ai";
-import { useCallback as useCallback4, useEffect as useEffect10, useMemo as useMemo4, useRef as useRef9, useState as useState15 } from "react";
+import { useCallback as useCallback4, useEffect as useEffect10, useMemo as useMemo4, useRef as useRef9, useState as useState14 } from "react";
 import { ulid } from "ulid";
 
 // ../utils/lib/get-class-name-factory.ts
@@ -572,8 +566,7 @@ var styles_module_default = { "Chat": "_Chat_3s08c_1", "Chat-header": "_Chat-hea
 // src/components/ChatBody/index.tsx
 init_react_import();
 import {
-  useContext as useContext3,
-  useState as useState14
+  useContext as useContext3
 } from "react";
 import { useStickToBottom } from "use-stick-to-bottom";
 import Markdown from "react-markdown";
@@ -6184,7 +6177,7 @@ var ctx = createContext2({});
 
 // css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/ChatBody/styles.module.css#css-module
 init_react_import();
-var styles_module_default8 = { "Chat": "_Chat_8hszl_1", "Chat-inner": "_Chat-inner_8hszl_17", "Chat--hasMessages": "_Chat--hasMessages_8hszl_25", "Chat--hasChildren": "_Chat--hasChildren_8hszl_29", "Chat-messages": "_Chat-messages_8hszl_34", "Chat-form": "_Chat-form_8hszl_41", "Chat--hideInput": "_Chat--hideInput_8hszl_46", "Chat-formHelper": "_Chat-formHelper_8hszl_60", "Chat-examplePrompts": "_Chat-examplePrompts_8hszl_75", "Chat-examplePrompt": "_Chat-examplePrompt_8hszl_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_8hszl_105", "Chat-default": "_Chat-default_8hszl_110", "ChatMessage": "_ChatMessage_8hszl_127", "ChatMessage--userRole": "_ChatMessage--userRole_8hszl_133", "ChatMessage-text": "_ChatMessage-text_8hszl_133", "ChatMessage-data": "_ChatMessage-data_8hszl_231", "ChatMessage-dataInner": "_ChatMessage-dataInner_8hszl_242", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_8hszl_248", "Chat-loader": "_Chat-loader_8hszl_252", "Chat-error": "_Chat-error_8hszl_257", "Chat-errorLabel": "_Chat-errorLabel_8hszl_266", "Chat-errorAction": "_Chat-errorAction_8hszl_270", "Chat-planGroup": "_Chat-planGroup_8hszl_274", "Chat-planGroupHeader": "_Chat-planGroupHeader_8hszl_283", "Chat-planGroupFooter": "_Chat-planGroupFooter_8hszl_292", "Chat-planSummary": "_Chat-planSummary_8hszl_303", "Chat-planSummaryIcon": "_Chat-planSummaryIcon_8hszl_311", "Chat-plan": "_Chat-plan_8hszl_274", "Chat-planPath": "_Chat-planPath_8hszl_324", "Chat-planHeader": "_Chat-planHeader_8hszl_330", "Chat-planTitle": "_Chat-planTitle_8hszl_336", "Chat-planPrompt": "_Chat-planPrompt_8hszl_341", "Chat-planStatus": "_Chat-planStatus_8hszl_347", "Chat-textStatus": "_Chat-textStatus_8hszl_368", "Chat-textStatusShimmer": "_Chat-textStatusShimmer_8hszl_1" };
+var styles_module_default8 = { "Chat": "_Chat_1njo2_1", "Chat-inner": "_Chat-inner_1njo2_17", "Chat--hasMessages": "_Chat--hasMessages_1njo2_25", "Chat--hasChildren": "_Chat--hasChildren_1njo2_29", "Chat-messages": "_Chat-messages_1njo2_34", "Chat-form": "_Chat-form_1njo2_41", "Chat--hideInput": "_Chat--hideInput_1njo2_46", "Chat-formHelper": "_Chat-formHelper_1njo2_60", "Chat-examplePrompts": "_Chat-examplePrompts_1njo2_75", "Chat-examplePrompt": "_Chat-examplePrompt_1njo2_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_1njo2_105", "Chat-default": "_Chat-default_1njo2_110", "ChatMessage": "_ChatMessage_1njo2_127", "ChatMessage--userRole": "_ChatMessage--userRole_1njo2_133", "ChatMessage-text": "_ChatMessage-text_1njo2_133", "ChatMessage-data": "_ChatMessage-data_1njo2_231", "ChatMessage-dataInner": "_ChatMessage-dataInner_1njo2_242", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_1njo2_248", "Chat-loader": "_Chat-loader_1njo2_252", "Chat-error": "_Chat-error_1njo2_257", "Chat-errorLabel": "_Chat-errorLabel_1njo2_266", "Chat-errorAction": "_Chat-errorAction_1njo2_270" };
 
 // src/components/PromptForm/index.tsx
 init_react_import();
@@ -6340,31 +6333,7 @@ var toolStatusContext = createContext3({});
 var ToolStatusProvider = toolStatusContext.Provider;
 
 // src/components/ChatBody/index.tsx
-import { createUsePuck, useGetPuck } from "@puckeditor/core";
-import { usePagesStore } from "@puckeditor/plugin-pages";
 import { jsx as jsx30, jsxs as jsxs17 } from "react/jsx-runtime";
-async function* iterateSSE(body) {
-  const reader = body.getReader();
-  const dec = new TextDecoder();
-  let buf = "";
-  while (true) {
-    const { done, value } = await reader.read();
-    if (done) break;
-    buf += dec.decode(value, { stream: true });
-    let idx;
-    while ((idx = buf.indexOf("\n\n")) >= 0) {
-      const chunk = buf.slice(0, idx).trim();
-      buf = buf.slice(idx + 2);
-      if (!chunk) continue;
-      const line = chunk.startsWith("data: ") ? chunk.slice(6) : chunk;
-      try {
-        yield JSON.parse(line);
-      } catch {
-      }
-    }
-  }
-}
-var usePuck = createUsePuck();
 var cx8 = getClassNameFactory2("Chat", styles_module_default8);
 var cxMessage = getClassNameFactory2("ChatMessage", styles_module_default8);
 function ChatBody({
@@ -6439,86 +6408,11 @@ function ChatMessagePart({
   if (part.type === "text") {
     return /* @__PURE__ */ jsx30("div", { className: cxMessage("text"), children: /* @__PURE__ */ jsx30(ChatMessageWrapper, { role, children: part.text }) });
   }
-  if (part.type === "tool-plan") {
-    return /* @__PURE__ */ jsx30(PlanTool, { ...part });
-  }
   if (part.type.startsWith("tool-")) {
     return /* @__PURE__ */ jsx30(PuckTool, { ...part });
   }
   return null;
 }
-function PlanTool({ output }) {
-  const planOutput = output?.plan;
-  const config = usePuck((s2) => s2.config);
-  const history = usePuck((s2) => s2.history);
-  const dispatch = usePuck((s2) => s2.dispatch);
-  const getPuck = useGetPuck();
-  const [loading, setLoading] = useState14({});
-  const { setPageVersions, createPage } = usePagesStore();
-  if (!planOutput) {
-    return /* @__PURE__ */ jsx30("div", { className: cx8("textStatus"), children: "Planning" });
-  }
-  return /* @__PURE__ */ jsxs17("div", { className: cx8("planGroup"), children: [
-    /* @__PURE__ */ jsx30("div", { className: cx8("planGroupHeader"), children: "Plan" }),
-    planOutput?.pages?.map((item) => /* @__PURE__ */ jsxs17("details", { className: cx8("plan"), children: [
-      /* @__PURE__ */ jsxs17("summary", { className: cx8("planSummary"), children: [
-        /* @__PURE__ */ jsxs17("div", { className: cx8("planSummaryText"), children: [
-          /* @__PURE__ */ jsxs17("div", { className: cx8("planHeader"), children: [
-            /* @__PURE__ */ jsx30("div", { className: cx8("planTitle"), children: item.shortTitle }),
-            /* @__PURE__ */ jsx30("div", { className: cx8("planStatus"), children: loading[item.path] && /* @__PURE__ */ jsx30(Loader, { size: 18 }) })
-          ] }),
-          /* @__PURE__ */ jsx30("div", { className: cx8("planPath"), children: item.path })
-        ] }),
-        /* @__PURE__ */ jsx30("div", { className: cx8("planSummaryIcon"), children: /* @__PURE__ */ jsx30(ChevronDown, {}) })
-      ] }),
-      /* @__PURE__ */ jsx30("div", { className: cx8("planPrompt"), children: item.pagePrompt })
-    ] }, item.path)),
-    /* @__PURE__ */ jsx30("div", { className: cx8("planGroupFooter"), children: /* @__PURE__ */ jsx30(
-      Button,
-      {
-        onClick: async () => {
-          const res = await fetch(`/api/puck/plan`, {
-            method: "POST",
-            body: JSON.stringify({ plan: planOutput, config })
-          });
-          if (res.body) {
-            for await (const chunk of iterateSSE(res.body)) {
-              if (chunk.type === "data-plan-progress") {
-                if (chunk.data.data) {
-                  const tempId = Math.random().toString();
-                  setPageVersions(
-                    chunk.data.path,
-                    [
-                      {
-                        id: tempId,
-                        data: chunk.data.data,
-                        versionNumber: 1,
-                        isNew: true
-                      }
-                    ],
-                    chunk.data.path
-                  );
-                  setLoading((s2) => ({ ...s2, [chunk.data.path]: false }));
-                  if (usePagesStore.getState().activePage === chunk.data.path) {
-                    history.setHistories([]);
-                    dispatch({
-                      type: "set",
-                      state: { ...getPuck().appState, data: chunk.data.data }
-                    });
-                  }
-                } else if (chunk.data.path && chunk.data.status === "processing") {
-                  createPage(chunk.data.path);
-                  setLoading((s2) => ({ ...s2, [chunk.data.path]: true }));
-                }
-              }
-            }
-          }
-        },
-        children: "Execute plan"
-      }
-    ) })
-  ] });
-}
 function PuckTool({
   toolCallId,
   output,
@@ -6746,7 +6640,7 @@ var q2 = qler();
 var BENCHMARK = false;
 var prefixedUlid = (prefix = "") => `${prefix ? `${prefix}_` : ""}${ulid()}`;
 var getClassName18 = getClassNameFactory("Chat", styles_module_default);
-var usePuck2 = createUsePuck2();
+var usePuck = createUsePuck();
 function Chat2({
   chat,
   host = "/api/puck/chat",
@@ -6754,38 +6648,48 @@ function Chat2({
   scrollTrackingId
 }) {
   const { examplePrompts } = chat ?? {};
-  const puckDispatch = usePuck2((s2) => s2.dispatch);
-  const getPuck = useGetPuck2();
+  const puckDispatch = usePuck((s2) => s2.dispatch);
+  const getPuck = useGetPuck();
   const localChatId = useRef9("");
   const inputRef = useRef9(null);
   const pluginRef = useRef9(null);
-  const [error, setError] = useState15();
-  const [toolStatus, setToolStatus] = useState15({});
+  const [error, setError] = useState14();
+  const [toolStatus, setToolStatus] = useState14({});
   const uploadScreenshot = useCallback4(
     async (width, bucketUrl) => {
-      const iframeDocument = document?.getElementById("preview-frame")?.contentDocument?.documentElement ?? null;
-      if (!iframeDocument) return;
-      const canvas = await html2canvas(iframeDocument, {
-        scale: 2,
-        backgroundColor: "#ffffff",
-        width,
-        windowWidth: width,
-        foreignObjectRendering: false,
-        imageTimeout: 3e4,
-        logging: false,
-        allowTaint: false,
-        useCORS: true,
-        scrollY: 0,
-        ignoreElements: (el) => Array.from(el.classList).some(
-          (c2) => c2.startsWith("_DraggableComponent--hover") || c2.startsWith("_ActionBar")
-        )
-      });
-      const image = canvas.toDataURL("image/webp", 0.8);
-      const blob = await (await fetch(image)).blob();
-      await fetch(bucketUrl, {
-        method: "PUT",
-        body: blob
-      });
+      try {
+        const iframeDocument = document?.getElementById("preview-frame")?.contentDocument?.documentElement ?? null;
+        if (!iframeDocument) {
+          throw new Error("Preview frame not found");
+        }
+        const canvas = await html2canvas(iframeDocument, {
+          scale: 2,
+          backgroundColor: "#ffffff",
+          width,
+          windowWidth: width,
+          foreignObjectRendering: false,
+          imageTimeout: 3e4,
+          logging: false,
+          allowTaint: false,
+          useCORS: true,
+          scrollY: 0,
+          ignoreElements: (el) => Array.from(el.classList).some(
+            (c2) => c2.startsWith("_DraggableComponent--hover") || c2.startsWith("_ActionBar")
+          )
+        });
+        const image = canvas.toDataURL("image/webp", 0.8);
+        const imageResponse = await fetch(image);
+        const blob = await imageResponse.blob();
+        const uploadResponse = await fetch(bucketUrl, {
+          method: "PUT",
+          body: blob
+        });
+        if (!uploadResponse.ok) {
+          throw new Error(`Upload failed with status ${uploadResponse.status}`);
+        }
+      } catch (e2) {
+        console.error("Failed to upload screenshot", e2);
+      }
     },
     []
   );
@@ -6830,9 +6734,10 @@ function Chat2({
           return;
         }
         case "data-send-screenshot": {
-          dataPart.data.urls.map((obj) => {
+          dataPart.data.urls.forEach((obj) => {
             const [[key, value]] = Object.entries(obj);
             const breakpoint = Number(key);
+            uploadScreenshot(breakpoint, value);
           });
           return;
         }
@@ -6841,7 +6746,7 @@ function Chat2({
           return;
       }
     },
-    [getPuck]
+    [getPuck, puckDispatch, uploadScreenshot]
   );
   const { messages, status, sendMessage, regenerate, setMessages } = useChat({
     generateId: () => prefixedUlid("msg"),
@@ -6913,7 +6818,7 @@ function Chat2({
       }
     }
   });
-  const [forcedStatus, setForcedStatus] = useState15();
+  const [forcedStatus, setForcedStatus] = useState14();
   const resolvedStatus = useMemo4(
     () => forcedStatus ?? status,
     [status, forcedStatus]
@@ -7195,7 +7100,6 @@ lucide-react/dist/esm/createLucideIcon.js:
 lucide-react/dist/esm/icons/arrow-up.js:
 lucide-react/dist/esm/icons/bot.js:
 lucide-react/dist/esm/icons/check.js:
-lucide-react/dist/esm/icons/chevron-down.js:
 lucide-react/dist/esm/icons/rotate-ccw.js:
 lucide-react/dist/esm/icons/triangle-alert.js:
 lucide-react/dist/esm/lucide-react.js:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@puckeditor/plugin-ai",
-  "version": "0.6.1-canary.433ca837",
+  "version": "0.6.1-canary.45a1a2e4",
   "author": "Chris Villa <chris@puckeditor.com>",
   "repository": "puckeditor/puck",
   "bugs": "https://github.com/puckeditor/puck/issues",
@@ -21,7 +21,7 @@
   "scripts": {
     "lint": "eslint \"**/*.ts*\"",
     "build": "rm -rf dist && tsup index.ts",
-    "dev": "tsup index.ts --watch src"
+    "dev": "tsup index.ts --watch"
   },
   "files": [
     "dist"
@@ -40,7 +40,6 @@
     "@puckeditor/ai-types": "workspace:*",
     "@puckeditor/core": "0.21.1",
     "@puckeditor/platform-types": "workspace:*",
-    "@puckeditor/plugin-pages": "workspace:*",
     "@types/jest": "^30.0.0",
     "@types/node": "^24.3.0",
     "@types/react": "^19.1.8",