@puckeditor/plugin-ai 0.1.0-canary.bde85814 → 0.1.0-canary.c433e7ae

package/dist/index.css

@@ -1423,7 +1423,7 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
 }
 
 /* css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/ChatBody/styles.module.css/#css-module-data */
-._Chat_dx6l4_1 {
+._Chat_14j71_1 {
   --fade-size: 32px;
   --chat-background-color: white;
   display: flex;
@@ -1433,46 +1433,46 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   overflow: hidden;
 }
 @media (min-width: 638px) {
-  ._Chat_dx6l4_1 {
+  ._Chat_14j71_1 {
     max-height: none;
   }
 }
-._Chat-inner_dx6l4_17 {
+._Chat-inner_14j71_17 {
   background-color: var(--chat-background-color);
   display: flex;
   flex-direction: column;
   max-height: 100%;
   position: relative;
 }
-._Chat--hasMessages_dx6l4_25 ._Chat-inner_dx6l4_17 {
+._Chat--hasMessages_14j71_25 ._Chat-inner_14j71_17 {
   padding-top: 0;
 }
-._Chat_dx6l4_1:not(._Chat--hasChildren_dx6l4_29) ._Chat-inner_dx6l4_17 {
+._Chat_14j71_1:not(._Chat--hasChildren_14j71_29) ._Chat-inner_14j71_17 {
   height: 100%;
   padding-top: 0;
 }
-._Chat--hasMessages_dx6l4_25 ._Chat-messages_dx6l4_34 {
+._Chat--hasMessages_14j71_25 ._Chat-messages_14j71_34 {
   display: flex;
   flex-direction: column-reverse;
   padding: 16px;
   overflow: auto;
 }
-._Chat-form_dx6l4_41 {
+._Chat-form_14j71_41 {
   background-color: white;
   padding: 16px;
 }
-._Chat--hideInput_dx6l4_46 ._Chat-form_dx6l4_41 {
+._Chat--hideInput_14j71_46 ._Chat-form_14j71_41 {
   display: none;
 }
-._Chat--hasMessages_dx6l4_25 ._Chat-form_dx6l4_41 {
+._Chat--hasMessages_14j71_25 ._Chat-form_14j71_41 {
   border-top: 1px solid var(--puck-color-grey-09);
 }
 @media (min-width: 638px) {
-  ._Chat-form_dx6l4_41 {
+  ._Chat-form_14j71_41 {
     border-top: 1px solid var(--puck-color-grey-09);
   }
 }
-._Chat-formHelper_dx6l4_60 {
+._Chat-formHelper_14j71_60 {
   align-items: center;
   display: flex;
   color: var(--puck-color-grey-04);
@@ -1481,11 +1481,11 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   gap: 8px;
   padding-bottom: 12px;
 }
-._Chat--hasMessages_dx6l4_25 ._Chat-formHelper_dx6l4_60 {
+._Chat--hasMessages_14j71_25 ._Chat-formHelper_14j71_60 {
   opacity: 0;
   transition: opacity 250ms ease-in;
 }
-._Chat-examplePrompts_dx6l4_75 {
+._Chat-examplePrompts_14j71_75 {
   align-items: center;
   display: flex;
   flex-wrap: nowrap;
@@ -1494,7 +1494,7 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   width: 100%;
   overflow-x: auto;
 }
-._Chat-examplePrompt_dx6l4_75 {
+._Chat-examplePrompt_14j71_75 {
   align-items: center;
   background: white;
   border: 1px solid var(--puck-color-grey-09);
@@ -1505,33 +1505,33 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   padding: 6px 12px;
   text-wrap: nowrap;
 }
-._Chat-examplePrompt_dx6l4_75:hover {
+._Chat-examplePrompt_14j71_75:hover {
   background: var(--puck-color-azure-11);
   border-color: var(--puck-color-azure-08);
   color: var(--puck-color-azure-04);
 }
-._Chat-examplePromptArrow_dx6l4_103 {
+._Chat-examplePromptArrow_14j71_103 {
   margin-left: auto;
 }
-._Chat-default_dx6l4_107 {
+._Chat-default_14j71_107 {
   overflow-y: auto;
   flex-grow: 1;
 }
-._Chat--hasMessages_dx6l4_25 ._Chat-default_dx6l4_107 {
+._Chat--hasMessages_14j71_25 ._Chat-default_14j71_107 {
   opacity: 0;
   padding: 0;
   pointer-events: none;
   transition: opacity 250ms ease-in;
 }
-._Chat--hasMessages_dx6l4_25 {
+._Chat--hasMessages_14j71_25 {
   background-color: white;
   transition: background-color 250ms ease-in;
 }
-._ChatMessage_dx6l4_124 {
+._ChatMessage_14j71_124 {
   font-size: var(--puck-font-size-xs);
   line-height: var(--line-height-s);
 }
-._ChatMessage--userRole_dx6l4_129 ._ChatMessage-text_dx6l4_129 {
+._ChatMessage--userRole_14j71_129 ._ChatMessage-text_14j71_129 {
   background: var(--puck-color-azure-10);
   border-radius: 16px;
   color: var(--puck-color-grey-03);
@@ -1540,64 +1540,64 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   margin-top: 8px;
   padding: 12px 16px;
 }
-._ChatMessage_dx6l4_124 ul,
-._ChatMessage_dx6l4_124 ol {
+._ChatMessage_14j71_124 ul,
+._ChatMessage_14j71_124 ol {
   list-style: decimal;
   margin: 0;
   margin-bottom: 12px;
   margin-top: 12px;
   padding-left: 32px;
 }
-._ChatMessage_dx6l4_124 li {
+._ChatMessage_14j71_124 li {
   margin-bottom: 8px;
   margin-top: 8px;
   padding-left: 8px;
 }
-._ChatMessage_dx6l4_124 h1,
-._ChatMessage_dx6l4_124 h2,
-._ChatMessage_dx6l4_124 h3,
-._ChatMessage_dx6l4_124 h4,
-._ChatMessage_dx6l4_124 h5,
-._ChatMessage_dx6l4_124 h6 {
+._ChatMessage_14j71_124 h1,
+._ChatMessage_14j71_124 h2,
+._ChatMessage_14j71_124 h3,
+._ChatMessage_14j71_124 h4,
+._ChatMessage_14j71_124 h5,
+._ChatMessage_14j71_124 h6 {
   font-weight: 600;
 }
-._ChatMessage_dx6l4_124 h1 {
+._ChatMessage_14j71_124 h1 {
   font-size: var(--puck-font-l);
 }
-._ChatMessage_dx6l4_124 h2 {
+._ChatMessage_14j71_124 h2 {
   font-size: var(--puck-font-m);
 }
-._ChatMessage_dx6l4_124 h3 {
+._ChatMessage_14j71_124 h3 {
   font-size: var(--puck-font-m);
 }
-._ChatMessage_dx6l4_124 h4 {
+._ChatMessage_14j71_124 h4 {
   font-size: var(--puck-font-s);
 }
-._ChatMessage_dx6l4_124 h5 {
+._ChatMessage_14j71_124 h5 {
   font-size: var(--puck-font-xs);
 }
-._ChatMessage_dx6l4_124 h6 {
+._ChatMessage_14j71_124 h6 {
   font-size: var(--puck-font-xm);
 }
-._ChatMessage_dx6l4_124 p {
+._ChatMessage_14j71_124 p {
   margin-bottom: 8px;
   margin-top: 8px;
 }
-._ChatMessage_dx6l4_124 a {
+._ChatMessage_14j71_124 a {
   color: var(--puck-color-azure-04);
   font-weight: 600;
   text-decoration: none;
 }
-._ChatMessage_dx6l4_124 a:hover {
+._ChatMessage_14j71_124 a:hover {
   color: var(--puck-color-azure-05);
 }
-._ChatMessage_dx6l4_124 hr {
+._ChatMessage_14j71_124 hr {
   border-top: none;
   border-bottom-color: var(--puck-color-grey-09);
   margin-bottom: 16px;
   margin-top: 16px;
 }
-._ChatMessage_dx6l4_124 blockquote {
+._ChatMessage_14j71_124 blockquote {
   margin-bottom: 8px;
   margin-top: 8px;
   margin-left: 0;
@@ -1606,7 +1606,7 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   padding-bottom: 2px;
   padding-top: 2px;
 }
-._ChatMessage-data_dx6l4_219 {
+._ChatMessage-data_14j71_219 {
   background: var(--puck-color-grey-12);
   border: 1px solid var(--puck-color-grey-08);
   border-radius: 16px;
@@ -1616,18 +1616,32 @@ a._Avatar_meij0_1:hover ._Avatar-name_meij0_41 {
   margin-top: 12px;
   padding: 12px 16px;
 }
-._ChatMessage-dataInner_dx6l4_230 {
+._ChatMessage-dataInner_14j71_230 {
   align-items: center;
   display: flex;
   gap: 8px;
 }
-._ChatMessage-dataIcon_dx6l4_236 {
+._ChatMessage-dataIcon_14j71_236 {
   display: flex;
 }
-._Chat-loader_dx6l4_240 {
+._Chat-loader_14j71_240 {
   padding: 16px;
   padding-top: 0;
 }
+._Chat-error_14j71_245 {
+  align-items: center;
+  display: flex;
+  gap: 8px;
+  font-size: var(--puck-font-size-xxs);
+  margin: 16px;
+  margin-top: 0;
+}
+._Chat-errorLabel_14j71_254 {
+  color: var(--puck-color-red-04);
+}
+._Chat-errorAction_14j71_258 {
+  margin-left: auto;
+}
 
 /* css-module:/home/runner/work/puck-platform/puck-platform/packages/plugin-ai/src/components/PromptForm/styles.module.css/#css-module-data */
 ._PromptForm_136g1_1 {

package/dist/index.d.mts

@@ -32,7 +32,9 @@ type ChatProps = {
     inputRef?: RefObject<HTMLTextAreaElement | null>;
     status?: ChatStatus;
     messages?: PuckMessage[];
+    error?: string;
+    handleRetry?: () => void;
 };
-declare function ChatBody({ children, examplePrompts, handleSubmit, hideInput, inputRef, messages, status, }: ChatProps): react_jsx_runtime.JSX.Element;
+declare function ChatBody({ children, examplePrompts, handleSubmit, hideInput, inputRef, messages, status, error, handleRetry, }: ChatProps): react_jsx_runtime.JSX.Element;
 
 export { ChatBody, createAIPlugin as default };

package/dist/index.d.ts

@@ -32,7 +32,9 @@ type ChatProps = {
     inputRef?: RefObject<HTMLTextAreaElement | null>;
     status?: ChatStatus;
     messages?: PuckMessage[];
+    error?: string;
+    handleRetry?: () => void;
 };
-declare function ChatBody({ children, examplePrompts, handleSubmit, hideInput, inputRef, messages, status, }: ChatProps): react_jsx_runtime.JSX.Element;
+declare function ChatBody({ children, examplePrompts, handleSubmit, hideInput, inputRef, messages, status, error, handleRetry, }: ChatProps): react_jsx_runtime.JSX.Element;
 
 export { ChatBody, createAIPlugin as default };

package/dist/index.js

@@ -262,6 +262,13 @@ 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/rotate-ccw.js
+init_react_import();
+var RotateCcw = createLucideIcon("RotateCcw", [
+  ["path", { d: "M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8", key: "1357e3" }],
+  ["path", { d: "M3 3v5h5", key: "1xhq8a" }]
+]);
+
 // src/components/Chat/index.tsx
 init_react_import();
 var import_react20 = require("@ai-sdk/react");
@@ -606,6 +613,57 @@ var IconButton_module_default = { "IconButton": "_IconButton_thegm_1", "IconButt
 // ../platform-client/components/IconButton/IconButton.tsx
 var import_jsx_runtime13 = require("react/jsx-runtime");
 var getClassName10 = get_class_name_factory_default("IconButton", IconButton_module_default);
+var IconButton = ({
+  children,
+  href,
+  onClick,
+  variant = "primary",
+  type,
+  disabled,
+  tabIndex,
+  newTab,
+  fullWidth,
+  title,
+  dark
+}) => {
+  const [loading, setLoading] = (0, import_react9.useState)(false);
+  const ElementType = href ? "a" : onClick ? "button" : "div";
+  const el = /* @__PURE__ */ (0, import_jsx_runtime13.jsxs)(
+    ElementType,
+    {
+      className: getClassName10({
+        primary: variant === "primary",
+        secondary: variant === "secondary",
+        disabled,
+        fullWidth,
+        dark
+      }),
+      onClick: (e) => {
+        if (!onClick) return;
+        setLoading(true);
+        Promise.resolve(onClick(e)).then(() => {
+          setLoading(false);
+        });
+      },
+      type,
+      disabled: disabled || loading,
+      tabIndex,
+      target: newTab ? "_blank" : void 0,
+      rel: newTab ? "noreferrer" : void 0,
+      href,
+      title,
+      children: [
+        /* @__PURE__ */ (0, import_jsx_runtime13.jsx)("span", { className: getClassName10("title"), children: title }),
+        children,
+        loading && /* @__PURE__ */ (0, import_jsx_runtime13.jsxs)(import_jsx_runtime13.Fragment, { children: [
+          "\xA0\xA0",
+          /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(Loader, { size: 14 })
+        ] })
+      ]
+    }
+  );
+  return el;
+};
 
 // ../platform-client/components/InfoBar/index.tsx
 init_react_import();
@@ -936,7 +994,7 @@ var import_react_markdown = __toESM(require("react-markdown"));
 
 // 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_default4 = { "Chat": "_Chat_dx6l4_1", "Chat-inner": "_Chat-inner_dx6l4_17", "Chat--hasMessages": "_Chat--hasMessages_dx6l4_25", "Chat--hasChildren": "_Chat--hasChildren_dx6l4_29", "Chat-messages": "_Chat-messages_dx6l4_34", "Chat-form": "_Chat-form_dx6l4_41", "Chat--hideInput": "_Chat--hideInput_dx6l4_46", "Chat-formHelper": "_Chat-formHelper_dx6l4_60", "Chat-examplePrompts": "_Chat-examplePrompts_dx6l4_75", "Chat-examplePrompt": "_Chat-examplePrompt_dx6l4_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_dx6l4_103", "Chat-default": "_Chat-default_dx6l4_107", "ChatMessage": "_ChatMessage_dx6l4_124", "ChatMessage--userRole": "_ChatMessage--userRole_dx6l4_129", "ChatMessage-text": "_ChatMessage-text_dx6l4_129", "ChatMessage-data": "_ChatMessage-data_dx6l4_219", "ChatMessage-dataInner": "_ChatMessage-dataInner_dx6l4_230", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_dx6l4_236", "Chat-loader": "_Chat-loader_dx6l4_240" };
+var styles_module_default4 = { "Chat": "_Chat_14j71_1", "Chat-inner": "_Chat-inner_14j71_17", "Chat--hasMessages": "_Chat--hasMessages_14j71_25", "Chat--hasChildren": "_Chat--hasChildren_14j71_29", "Chat-messages": "_Chat-messages_14j71_34", "Chat-form": "_Chat-form_14j71_41", "Chat--hideInput": "_Chat--hideInput_14j71_46", "Chat-formHelper": "_Chat-formHelper_14j71_60", "Chat-examplePrompts": "_Chat-examplePrompts_14j71_75", "Chat-examplePrompt": "_Chat-examplePrompt_14j71_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_14j71_103", "Chat-default": "_Chat-default_14j71_107", "ChatMessage": "_ChatMessage_14j71_124", "ChatMessage--userRole": "_ChatMessage--userRole_14j71_129", "ChatMessage-text": "_ChatMessage-text_14j71_129", "ChatMessage-data": "_ChatMessage-data_14j71_219", "ChatMessage-dataInner": "_ChatMessage-dataInner_14j71_230", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_14j71_236", "Chat-loader": "_Chat-loader_14j71_240", "Chat-error": "_Chat-error_14j71_245", "Chat-errorLabel": "_Chat-errorLabel_14j71_254", "Chat-errorAction": "_Chat-errorAction_14j71_258" };
 
 // src/components/PromptForm/index.tsx
 init_react_import();
@@ -1060,7 +1118,9 @@ function ChatBody({
   hideInput,
   inputRef,
   messages = [],
-  status
+  status,
+  error,
+  handleRetry
 }) {
   const { scrollRef, contentRef } = (0, import_use_stick_to_bottom.useStickToBottom)();
   const hasMessages = messages && messages.length > 0;
@@ -1068,7 +1128,11 @@ function ChatBody({
     children ? /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("default"), children }) : null,
     /* @__PURE__ */ (0, import_jsx_runtime33.jsxs)("div", { className: cx5("inner"), ref: scrollRef, children: [
       /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("messages"), ref: contentRef, children: messages.toReversed().map((message) => /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(ChatMessage, { message }, message.id)) }),
-      status === "submitted" && /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("loader"), children: /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(Loader, {}) }),
+      status === "submitted" && /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("loader"), children: /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(Loader, { size: 14 }) }),
+      error && /* @__PURE__ */ (0, import_jsx_runtime33.jsxs)("div", { className: cx5("error"), children: [
+        /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("errorLabel"), children: "Something went wrong." }),
+        handleRetry && /* @__PURE__ */ (0, import_jsx_runtime33.jsx)("div", { className: cx5("errorAction"), children: /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(IconButton, { title: "Retry", onClick: handleRetry, children: /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(RotateCcw, { size: "14" }) }) })
+      ] }),
       /* @__PURE__ */ (0, import_jsx_runtime33.jsxs)("div", { className: cx5("form"), children: [
         /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(
           PromptForm,
@@ -1176,8 +1240,9 @@ function Chat({
   const [localChatId, setLocalChatId] = (0, import_react21.useState)(chatId);
   const inputRef = (0, import_react21.useRef)(null);
   const pluginRef = (0, import_react21.useRef)(null);
+  const [error, setError] = (0, import_react21.useState)();
   const [toolStatus, setToolStatus] = (0, import_react21.useState)({});
-  const { messages, status, sendMessage } = (0, import_react20.useChat)({
+  const { messages, status, sendMessage, regenerate } = (0, import_react20.useChat)({
     generateId: () => prefixedUlid("msg"),
     messages: initialMessages,
     transport: new import_ai.DefaultChatTransport({
@@ -1353,6 +1418,7 @@ function Chat({
       if (BENCHMARK) {
         console.timeEnd("chat");
       }
+      setError(e.message);
     },
     onFinish: () => {
       puckDispatch({
@@ -1370,6 +1436,7 @@ function Chat({
     if (!text) {
       return;
     }
+    setError("");
     sendMessage(
       {
         text
@@ -1405,6 +1472,11 @@ function Chat({
         inputRef,
         status,
         examplePrompts: examplePrompts == null ? void 0 : examplePrompts.map(({ label, href }) => /* @__PURE__ */ (0, import_jsx_runtime34.jsx)(ExamplePrompt, { label, href }, label)),
+        error,
+        handleRetry: () => {
+          setError("");
+          regenerate();
+        },
         children: /* @__PURE__ */ (0, import_jsx_runtime34.jsx)(
           Placeholder,
           {
@@ -1504,6 +1576,7 @@ lucide-react/dist/esm/icons/arrow-right.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/rotate-ccw.js:
 lucide-react/dist/esm/lucide-react.js:
   (**
    * @license lucide-react v0.452.0 - ISC

package/dist/index.mjs

@@ -249,6 +249,13 @@ 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/rotate-ccw.js
+init_react_import();
+var RotateCcw = createLucideIcon("RotateCcw", [
+  ["path", { d: "M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8", key: "1357e3" }],
+  ["path", { d: "M3 3v5h5", key: "1xhq8a" }]
+]);
+
 // src/components/Chat/index.tsx
 init_react_import();
 import { useChat } from "@ai-sdk/react";
@@ -593,6 +600,57 @@ var IconButton_module_default = { "IconButton": "_IconButton_thegm_1", "IconButt
 // ../platform-client/components/IconButton/IconButton.tsx
 import { Fragment as Fragment2, jsx as jsx13, jsxs as jsxs8 } from "react/jsx-runtime";
 var getClassName10 = get_class_name_factory_default("IconButton", IconButton_module_default);
+var IconButton = ({
+  children,
+  href,
+  onClick,
+  variant = "primary",
+  type,
+  disabled,
+  tabIndex,
+  newTab,
+  fullWidth,
+  title,
+  dark
+}) => {
+  const [loading, setLoading] = useState4(false);
+  const ElementType = href ? "a" : onClick ? "button" : "div";
+  const el = /* @__PURE__ */ jsxs8(
+    ElementType,
+    {
+      className: getClassName10({
+        primary: variant === "primary",
+        secondary: variant === "secondary",
+        disabled,
+        fullWidth,
+        dark
+      }),
+      onClick: (e) => {
+        if (!onClick) return;
+        setLoading(true);
+        Promise.resolve(onClick(e)).then(() => {
+          setLoading(false);
+        });
+      },
+      type,
+      disabled: disabled || loading,
+      tabIndex,
+      target: newTab ? "_blank" : void 0,
+      rel: newTab ? "noreferrer" : void 0,
+      href,
+      title,
+      children: [
+        /* @__PURE__ */ jsx13("span", { className: getClassName10("title"), children: title }),
+        children,
+        loading && /* @__PURE__ */ jsxs8(Fragment2, { children: [
+          "\xA0\xA0",
+          /* @__PURE__ */ jsx13(Loader, { size: 14 })
+        ] })
+      ]
+    }
+  );
+  return el;
+};
 
 // ../platform-client/components/InfoBar/index.tsx
 init_react_import();
@@ -762,7 +820,7 @@ init_react_import();
 var PuckHeader_module_default = { "PuckHeader": "_PuckHeader_19h0f_1", "PuckHeader-left": "_PuckHeader-left_19h0f_16", "PuckHeader-center": "_PuckHeader-center_19h0f_22", "PuckHeader-right": "_PuckHeader-right_19h0f_27", "PuckHeader-pageInput": "_PuckHeader-pageInput_19h0f_34", "PuckHeader-pageInputActions": "_PuckHeader-pageInputActions_19h0f_50", "PuckHeader-input": "_PuckHeader-input_19h0f_56", "PuckHeader-historyActions": "_PuckHeader-historyActions_19h0f_69" };
 
 // ../platform-client/components/PuckHeader/PuckHeader.tsx
-import { IconButton, usePuck } from "@measured/puck";
+import { IconButton as IconButton2, usePuck } from "@measured/puck";
 
 // ../platform-client/context/editor.tsx
 init_react_import();
@@ -923,7 +981,7 @@ import Markdown from "react-markdown";
 
 // 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_default4 = { "Chat": "_Chat_dx6l4_1", "Chat-inner": "_Chat-inner_dx6l4_17", "Chat--hasMessages": "_Chat--hasMessages_dx6l4_25", "Chat--hasChildren": "_Chat--hasChildren_dx6l4_29", "Chat-messages": "_Chat-messages_dx6l4_34", "Chat-form": "_Chat-form_dx6l4_41", "Chat--hideInput": "_Chat--hideInput_dx6l4_46", "Chat-formHelper": "_Chat-formHelper_dx6l4_60", "Chat-examplePrompts": "_Chat-examplePrompts_dx6l4_75", "Chat-examplePrompt": "_Chat-examplePrompt_dx6l4_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_dx6l4_103", "Chat-default": "_Chat-default_dx6l4_107", "ChatMessage": "_ChatMessage_dx6l4_124", "ChatMessage--userRole": "_ChatMessage--userRole_dx6l4_129", "ChatMessage-text": "_ChatMessage-text_dx6l4_129", "ChatMessage-data": "_ChatMessage-data_dx6l4_219", "ChatMessage-dataInner": "_ChatMessage-dataInner_dx6l4_230", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_dx6l4_236", "Chat-loader": "_Chat-loader_dx6l4_240" };
+var styles_module_default4 = { "Chat": "_Chat_14j71_1", "Chat-inner": "_Chat-inner_14j71_17", "Chat--hasMessages": "_Chat--hasMessages_14j71_25", "Chat--hasChildren": "_Chat--hasChildren_14j71_29", "Chat-messages": "_Chat-messages_14j71_34", "Chat-form": "_Chat-form_14j71_41", "Chat--hideInput": "_Chat--hideInput_14j71_46", "Chat-formHelper": "_Chat-formHelper_14j71_60", "Chat-examplePrompts": "_Chat-examplePrompts_14j71_75", "Chat-examplePrompt": "_Chat-examplePrompt_14j71_75", "Chat-examplePromptArrow": "_Chat-examplePromptArrow_14j71_103", "Chat-default": "_Chat-default_14j71_107", "ChatMessage": "_ChatMessage_14j71_124", "ChatMessage--userRole": "_ChatMessage--userRole_14j71_129", "ChatMessage-text": "_ChatMessage-text_14j71_129", "ChatMessage-data": "_ChatMessage-data_14j71_219", "ChatMessage-dataInner": "_ChatMessage-dataInner_14j71_230", "ChatMessage-dataIcon": "_ChatMessage-dataIcon_14j71_236", "Chat-loader": "_Chat-loader_14j71_240", "Chat-error": "_Chat-error_14j71_245", "Chat-errorLabel": "_Chat-errorLabel_14j71_254", "Chat-errorAction": "_Chat-errorAction_14j71_258" };
 
 // src/components/PromptForm/index.tsx
 init_react_import();
@@ -1051,7 +1109,9 @@ function ChatBody({
   hideInput,
   inputRef,
   messages = [],
-  status
+  status,
+  error,
+  handleRetry
 }) {
   const { scrollRef, contentRef } = useStickToBottom();
   const hasMessages = messages && messages.length > 0;
@@ -1059,7 +1119,11 @@ function ChatBody({
     children ? /* @__PURE__ */ jsx33("div", { className: cx5("default"), children }) : null,
     /* @__PURE__ */ jsxs21("div", { className: cx5("inner"), ref: scrollRef, children: [
       /* @__PURE__ */ jsx33("div", { className: cx5("messages"), ref: contentRef, children: messages.toReversed().map((message) => /* @__PURE__ */ jsx33(ChatMessage, { message }, message.id)) }),
-      status === "submitted" && /* @__PURE__ */ jsx33("div", { className: cx5("loader"), children: /* @__PURE__ */ jsx33(Loader, {}) }),
+      status === "submitted" && /* @__PURE__ */ jsx33("div", { className: cx5("loader"), children: /* @__PURE__ */ jsx33(Loader, { size: 14 }) }),
+      error && /* @__PURE__ */ jsxs21("div", { className: cx5("error"), children: [
+        /* @__PURE__ */ jsx33("div", { className: cx5("errorLabel"), children: "Something went wrong." }),
+        handleRetry && /* @__PURE__ */ jsx33("div", { className: cx5("errorAction"), children: /* @__PURE__ */ jsx33(IconButton, { title: "Retry", onClick: handleRetry, children: /* @__PURE__ */ jsx33(RotateCcw, { size: "14" }) }) })
+      ] }),
       /* @__PURE__ */ jsxs21("div", { className: cx5("form"), children: [
         /* @__PURE__ */ jsx33(
           PromptForm,
@@ -1167,8 +1231,9 @@ function Chat({
   const [localChatId, setLocalChatId] = useState12(chatId);
   const inputRef = useRef3(null);
   const pluginRef = useRef3(null);
+  const [error, setError] = useState12();
   const [toolStatus, setToolStatus] = useState12({});
-  const { messages, status, sendMessage } = useChat({
+  const { messages, status, sendMessage, regenerate } = useChat({
     generateId: () => prefixedUlid("msg"),
     messages: initialMessages,
     transport: new DefaultChatTransport({
@@ -1344,6 +1409,7 @@ function Chat({
       if (BENCHMARK) {
         console.timeEnd("chat");
       }
+      setError(e.message);
     },
     onFinish: () => {
       puckDispatch({
@@ -1361,6 +1427,7 @@ function Chat({
     if (!text) {
       return;
     }
+    setError("");
     sendMessage(
       {
         text
@@ -1396,6 +1463,11 @@ function Chat({
         inputRef,
         status,
         examplePrompts: examplePrompts == null ? void 0 : examplePrompts.map(({ label, href }) => /* @__PURE__ */ jsx34(ExamplePrompt, { label, href }, label)),
+        error,
+        handleRetry: () => {
+          setError("");
+          regenerate();
+        },
         children: /* @__PURE__ */ jsx34(
           Placeholder,
           {
@@ -1495,6 +1567,7 @@ lucide-react/dist/esm/icons/arrow-right.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/rotate-ccw.js:
 lucide-react/dist/esm/lucide-react.js:
   (**
    * @license lucide-react v0.452.0 - ISC

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@puckeditor/plugin-ai",
-  "version": "0.1.0-canary.bde85814",
+  "version": "0.1.0-canary.c433e7ae",
   "author": "Chris Villa <chris@puckeditor.com>",
   "repository": "puckeditor/puck",
   "bugs": "https://github.com/puckeditor/puck/issues",