@copilotkit/react-ui 1.8.4 → 1.8.5-next.1

package/dist/components/chat/Popup.js

@@ -1454,276 +1454,11 @@ var Input = ({ inProgress, onSend, isVisible = false, onStop }) => {
   ] }) });
 };
 
-// src/components/chat/messages/RenderTextMessage.tsx
+// src/components/chat/messages/UserMessage.tsx
 var import_jsx_runtime13 = require("react/jsx-runtime");
-function RenderTextMessage(props) {
-  const {
-    message,
-    inProgress,
-    index,
-    isCurrentMessage,
-    UserMessage: UserMessage2,
-    AssistantMessage: AssistantMessage2,
-    onRegenerate,
-    onCopy,
-    onThumbsUp,
-    onThumbsDown
-  } = props;
-  if (message.isTextMessage()) {
-    if (message.role === "user") {
-      return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(
-        UserMessage2,
-        {
-          "data-message-role": "user",
-          message: message.content,
-          rawData: message
-        },
-        index
-      );
-    } else if (message.role == "assistant") {
-      return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(
-        AssistantMessage2,
-        {
-          "data-message-role": "assistant",
-          message: message.content,
-          rawData: message,
-          isLoading: inProgress && isCurrentMessage && !message.content,
-          isGenerating: inProgress && isCurrentMessage && !!message.content,
-          isCurrentMessage,
-          onRegenerate,
-          onCopy,
-          onThumbsUp,
-          onThumbsDown
-        },
-        index
-      );
-    }
-  }
-}
-
-// src/components/chat/messages/RenderActionExecutionMessage.tsx
-var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
-var import_react_core6 = require("@copilotkit/react-core");
-var import_jsx_runtime14 = require("react/jsx-runtime");
-function RenderActionExecutionMessage(props) {
-  const { chatComponentsCache } = (0, import_react_core6.useCopilotContext)();
-  const { message, inProgress, index, isCurrentMessage, actionResult, AssistantMessage: AssistantMessage2 } = props;
-  if (message.isActionExecutionMessage()) {
-    if (chatComponentsCache.current !== null && (chatComponentsCache.current.actions[message.name] || chatComponentsCache.current.actions["*"])) {
-      const render = chatComponentsCache.current.actions[message.name] || chatComponentsCache.current.actions["*"];
-      if (typeof render === "string") {
-        if (isCurrentMessage && inProgress) {
-          return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
-            AssistantMessage2,
-            {
-              rawData: message,
-              "data-message-role": "assistant",
-              isLoading: false,
-              isGenerating: true,
-              message: render
-            },
-            index
-          );
-        } else {
-          return null;
-        }
-      } else {
-        const args = message.arguments;
-        let status = "inProgress";
-        if (actionResult !== void 0) {
-          status = "complete";
-        } else if (message.status.code !== import_runtime_client_gql2.MessageStatusCode.Pending) {
-          status = "executing";
-        }
-        try {
-          const toRender = render({
-            status,
-            args,
-            result: actionResult,
-            name: message.name
-          });
-          if (!toRender && status === "complete") {
-            return null;
-          }
-          if (typeof toRender === "string") {
-            return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
-              AssistantMessage2,
-              {
-                rawData: message,
-                "data-message-role": "assistant",
-                isLoading: false,
-                isGenerating: false,
-                message: toRender
-              },
-              index
-            );
-          } else {
-            return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
-              AssistantMessage2,
-              {
-                rawData: message,
-                "data-message-role": "action-render",
-                isLoading: false,
-                isGenerating: false,
-                subComponent: toRender
-              },
-              index
-            );
-          }
-        } catch (e) {
-          console.error(`Error executing render function for action ${message.name}: ${e}`);
-          return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
-            AssistantMessage2,
-            {
-              rawData: message,
-              "data-message-role": "assistant",
-              isLoading: false,
-              isGenerating: false,
-              subComponent: /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)("div", { className: "copilotKitMessage copilotKitAssistantMessage", children: [
-                /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)("b", { children: [
-                  "\u274C Error executing render function for action ",
-                  message.name,
-                  ":"
-                ] }),
-                /* @__PURE__ */ (0, import_jsx_runtime14.jsx)("pre", { children: e instanceof Error ? e.message : String(e) })
-              ] })
-            },
-            index
-          );
-        }
-      }
-    } else if (!inProgress || !isCurrentMessage) {
-      return null;
-    } else {
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
-        AssistantMessage2,
-        {
-          rawData: message,
-          "data-message-role": "assistant",
-          isLoading: true,
-          isGenerating: true
-        },
-        index
-      );
-    }
-  }
-}
-
-// src/components/chat/messages/RenderResultMessage.tsx
-var import_jsx_runtime15 = require("react/jsx-runtime");
-function RenderResultMessage(props) {
-  const { message, inProgress, index, isCurrentMessage, AssistantMessage: AssistantMessage2 } = props;
-  if (message.isResultMessage() && inProgress && isCurrentMessage) {
-    return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
-      AssistantMessage2,
-      {
-        "data-message-role": "assistant",
-        rawData: message,
-        isLoading: true,
-        isGenerating: true
-      },
-      index
-    );
-  } else {
-    return null;
-  }
-}
-
-// src/components/chat/messages/RenderAgentStateMessage.tsx
-var import_react_core7 = require("@copilotkit/react-core");
-var import_jsx_runtime16 = require("react/jsx-runtime");
-function RenderAgentStateMessage(props) {
-  const { chatComponentsCache } = (0, import_react_core7.useCopilotContext)();
-  const { message, inProgress, index, isCurrentMessage, AssistantMessage: AssistantMessage2 } = props;
-  if (message.isAgentStateMessage()) {
-    let render;
-    if (chatComponentsCache.current !== null) {
-      render = chatComponentsCache.current.coAgentStateRenders[`${message.agentName}-${message.nodeName}`] || chatComponentsCache.current.coAgentStateRenders[`${message.agentName}-global`];
-    }
-    if (render) {
-      if (typeof render === "string") {
-        if (isCurrentMessage && inProgress) {
-          return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
-            AssistantMessage2,
-            {
-              rawData: message,
-              message: render,
-              "data-message-role": "assistant",
-              isLoading: true,
-              isGenerating: true
-            },
-            index
-          );
-        } else {
-          return null;
-        }
-      } else {
-        const state = message.state;
-        let status = message.active ? "inProgress" : "complete";
-        const toRender = render({
-          status,
-          state,
-          nodeName: message.nodeName
-        });
-        if (!toRender && status === "complete") {
-          return null;
-        }
-        if (!toRender && isCurrentMessage && inProgress) {
-          return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
-            AssistantMessage2,
-            {
-              "data-message-role": "assistant",
-              rawData: message,
-              isLoading: true,
-              isGenerating: true
-            },
-            index
-          );
-        } else if (!toRender) {
-          return null;
-        }
-        if (typeof toRender === "string") {
-          return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
-            AssistantMessage2,
-            {
-              rawData: message,
-              message: toRender,
-              isLoading: true,
-              isGenerating: true,
-              "data-message-role": "assistant"
-            },
-            index
-          );
-        } else {
-          return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
-            AssistantMessage2,
-            {
-              rawData: message,
-              "data-message-role": "agent-state-render",
-              isLoading: false,
-              isGenerating: false,
-              subComponent: toRender
-            },
-            index
-          );
-        }
-      }
-    } else if (!inProgress || !isCurrentMessage) {
-      return null;
-    } else {
-      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
-        AssistantMessage2,
-        {
-          rawData: message,
-          isLoading: true,
-          isGenerating: true,
-          "data-message-role": "assistant"
-        },
-        index
-      );
-    }
-  }
-}
+var UserMessage = (props) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)("div", { className: "copilotKitMessage copilotKitUserMessage", children: props.message });
+};
 
 // src/components/chat/Markdown.tsx
 var import_react11 = require("react");
@@ -1756,7 +1491,7 @@ function useCopyToClipboard({ timeout = 2e3 }) {
 }
 
 // src/components/chat/CodeBlock.tsx
-var import_jsx_runtime17 = require("react/jsx-runtime");
+var import_jsx_runtime14 = require("react/jsx-runtime");
 var programmingLanguages = {
   javascript: ".js",
   python: ".py",
@@ -1819,15 +1554,15 @@ var CodeBlock = (0, import_react10.memo)(({ language, value }) => {
       return;
     copyToClipboard(value);
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime17.jsxs)("div", { className: "copilotKitCodeBlock", children: [
-    /* @__PURE__ */ (0, import_jsx_runtime17.jsxs)("div", { className: "copilotKitCodeBlockToolbar", children: [
-      /* @__PURE__ */ (0, import_jsx_runtime17.jsx)("span", { className: "copilotKitCodeBlockToolbarLanguage", children: language }),
-      /* @__PURE__ */ (0, import_jsx_runtime17.jsxs)("div", { className: "copilotKitCodeBlockToolbarButtons", children: [
-        /* @__PURE__ */ (0, import_jsx_runtime17.jsx)("button", { className: "copilotKitCodeBlockToolbarButton", onClick: downloadAsFile, children: DownloadIcon }),
-        /* @__PURE__ */ (0, import_jsx_runtime17.jsx)("button", { className: "copilotKitCodeBlockToolbarButton", onClick: onCopy, children: isCopied ? CheckIcon : CopyIcon })
+  return /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)("div", { className: "copilotKitCodeBlock", children: [
+    /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)("div", { className: "copilotKitCodeBlockToolbar", children: [
+      /* @__PURE__ */ (0, import_jsx_runtime14.jsx)("span", { className: "copilotKitCodeBlockToolbarLanguage", children: language }),
+      /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)("div", { className: "copilotKitCodeBlockToolbarButtons", children: [
+        /* @__PURE__ */ (0, import_jsx_runtime14.jsx)("button", { className: "copilotKitCodeBlockToolbarButton", onClick: downloadAsFile, children: DownloadIcon }),
+        /* @__PURE__ */ (0, import_jsx_runtime14.jsx)("button", { className: "copilotKitCodeBlockToolbarButton", onClick: onCopy, children: isCopied ? CheckIcon : CopyIcon })
       ] })
     ] }),
-    /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(
+    /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
       import_react_syntax_highlighter.Prism,
       {
         language,
@@ -2130,18 +1865,18 @@ var highlightStyle = {
 // src/components/chat/Markdown.tsx
 var import_remark_gfm = __toESM(require("remark-gfm"));
 var import_remark_math = __toESM(require("remark-math"));
-var import_jsx_runtime18 = require("react/jsx-runtime");
+var import_jsx_runtime15 = require("react/jsx-runtime");
 var MemoizedReactMarkdown = (0, import_react11.memo)(
   import_react_markdown.default,
   (prevProps, nextProps) => prevProps.children === nextProps.children && prevProps.className === nextProps.className
 );
 var Markdown = ({ content }) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)("div", { className: "copilotKitMarkdown", children: /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(MemoizedReactMarkdown, { components, remarkPlugins: [import_remark_gfm.default, import_remark_math.default], children: content }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("div", { className: "copilotKitMarkdown", children: /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(MemoizedReactMarkdown, { components, remarkPlugins: [import_remark_gfm.default, import_remark_math.default], children: content }) });
 };
 var components = {
   a(_a) {
     var _b = _a, { children } = _b, props = __objRest(_b, ["children"]);
-    return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
       "a",
       __spreadProps(__spreadValues({
         style: { color: "blue", textDecoration: "underline" }
@@ -2156,7 +1891,7 @@ var components = {
     var _d = _c, { children, className, inline } = _d, props = __objRest(_d, ["children", "className", "inline"]);
     if (children.length) {
       if (children[0] == "\u258D") {
-        return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+        return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
           "span",
           {
             style: {
@@ -2171,9 +1906,9 @@ var components = {
     }
     const match = /language-(\w+)/.exec(className || "");
     if (inline) {
-      return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)("code", __spreadProps(__spreadValues({ className }, props), { children }));
+      return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("code", __spreadProps(__spreadValues({ className }, props), { children }));
     }
-    return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
       CodeBlock,
       __spreadValues({
         language: match && match[1] || "",
@@ -2186,7 +1921,7 @@ var components = {
 
 // src/components/chat/messages/AssistantMessage.tsx
 var import_react12 = require("react");
-var import_jsx_runtime19 = require("react/jsx-runtime");
+var import_jsx_runtime16 = require("react/jsx-runtime");
 var AssistantMessage = (props) => {
   const { icons, labels } = useChatContext();
   const {
@@ -2227,17 +1962,17 @@ var AssistantMessage = (props) => {
       onThumbsDown(message);
     }
   };
-  const LoadingIcon = () => /* @__PURE__ */ (0, import_jsx_runtime19.jsx)("span", { children: icons.activityIcon });
-  return /* @__PURE__ */ (0, import_jsx_runtime19.jsxs)(import_jsx_runtime19.Fragment, { children: [
-    (message || isLoading) && /* @__PURE__ */ (0, import_jsx_runtime19.jsxs)("div", { className: "copilotKitMessage copilotKitAssistantMessage", children: [
-      message && /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(Markdown, { content: message || "" }),
-      isLoading && /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(LoadingIcon, {}),
-      message && !isLoading && /* @__PURE__ */ (0, import_jsx_runtime19.jsxs)(
+  const LoadingIcon = () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)("span", { children: icons.activityIcon });
+  return /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)(import_jsx_runtime16.Fragment, { children: [
+    (message || isLoading) && /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)("div", { className: "copilotKitMessage copilotKitAssistantMessage", children: [
+      message && /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Markdown, { content: message || "" }),
+      isLoading && /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(LoadingIcon, {}),
+      message && !isLoading && /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)(
         "div",
         {
           className: `copilotKitMessageControls ${isCurrentMessage ? "currentMessage" : ""}`,
           children: [
-            /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+            /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
               "button",
               {
                 className: "copilotKitMessageControlButton",
@@ -2247,17 +1982,17 @@ var AssistantMessage = (props) => {
                 children: icons.regenerateIcon
               }
             ),
-            /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+            /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
               "button",
               {
                 className: "copilotKitMessageControlButton",
                 onClick: handleCopy,
                 "aria-label": labels.copyToClipboard,
                 title: labels.copyToClipboard,
-                children: copied ? /* @__PURE__ */ (0, import_jsx_runtime19.jsx)("span", { style: { fontSize: "10px", fontWeight: "bold" }, children: "\u2713" }) : icons.copyIcon
+                children: copied ? /* @__PURE__ */ (0, import_jsx_runtime16.jsx)("span", { style: { fontSize: "10px", fontWeight: "bold" }, children: "\u2713" }) : icons.copyIcon
               }
             ),
-            onThumbsUp && /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+            onThumbsUp && /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
               "button",
               {
                 className: "copilotKitMessageControlButton",
@@ -2267,7 +2002,7 @@ var AssistantMessage = (props) => {
                 children: icons.thumbsUpIcon
               }
             ),
-            onThumbsDown && /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+            onThumbsDown && /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
               "button",
               {
                 className: "copilotKitMessageControlButton",
@@ -2281,15 +2016,300 @@ var AssistantMessage = (props) => {
         }
       )
     ] }),
-    /* @__PURE__ */ (0, import_jsx_runtime19.jsx)("div", { style: { marginBottom: "0.5rem" }, children: subComponent })
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)("div", { style: { marginBottom: "0.5rem" }, children: subComponent })
   ] });
 };
 
-// src/components/chat/messages/UserMessage.tsx
+// src/components/chat/messages/RenderTextMessage.tsx
+var import_jsx_runtime17 = require("react/jsx-runtime");
+function RenderTextMessage(_a) {
+  var _b = _a, {
+    UserMessage: UserMessage2 = UserMessage,
+    AssistantMessage: AssistantMessage2 = AssistantMessage
+  } = _b, props = __objRest(_b, [
+    "UserMessage",
+    "AssistantMessage"
+  ]);
+  const {
+    message,
+    inProgress,
+    index,
+    isCurrentMessage,
+    onRegenerate,
+    onCopy,
+    onThumbsUp,
+    onThumbsDown
+  } = props;
+  if (message.isTextMessage()) {
+    if (message.role === "user") {
+      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(
+        UserMessage2,
+        {
+          "data-message-role": "user",
+          message: message.content,
+          rawData: message
+        },
+        index
+      );
+    } else if (message.role == "assistant") {
+      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(
+        AssistantMessage2,
+        {
+          "data-message-role": "assistant",
+          message: message.content,
+          rawData: message,
+          isLoading: inProgress && isCurrentMessage && !message.content,
+          isGenerating: inProgress && isCurrentMessage && !!message.content,
+          isCurrentMessage,
+          onRegenerate: () => onRegenerate == null ? void 0 : onRegenerate(message.id),
+          onCopy,
+          onThumbsUp,
+          onThumbsDown
+        },
+        index
+      );
+    }
+  }
+}
+
+// src/components/chat/messages/RenderActionExecutionMessage.tsx
+var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
+var import_react_core6 = require("@copilotkit/react-core");
+var import_jsx_runtime18 = require("react/jsx-runtime");
+function RenderActionExecutionMessage(_a) {
+  var _b = _a, {
+    AssistantMessage: AssistantMessage2 = AssistantMessage
+  } = _b, props = __objRest(_b, [
+    "AssistantMessage"
+  ]);
+  const { chatComponentsCache } = (0, import_react_core6.useCopilotContext)();
+  const { message, inProgress, index, isCurrentMessage, actionResult } = props;
+  if (message.isActionExecutionMessage()) {
+    if (chatComponentsCache.current !== null && (chatComponentsCache.current.actions[message.name] || chatComponentsCache.current.actions["*"])) {
+      const render = chatComponentsCache.current.actions[message.name] || chatComponentsCache.current.actions["*"];
+      if (typeof render === "string") {
+        if (isCurrentMessage && inProgress) {
+          return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+            AssistantMessage2,
+            {
+              rawData: message,
+              "data-message-role": "assistant",
+              isLoading: false,
+              isGenerating: true,
+              message: render
+            },
+            index
+          );
+        } else {
+          return null;
+        }
+      } else {
+        const args = message.arguments;
+        let status = "inProgress";
+        if (actionResult !== void 0) {
+          status = "complete";
+        } else if (message.status.code !== import_runtime_client_gql2.MessageStatusCode.Pending) {
+          status = "executing";
+        }
+        try {
+          const toRender = render({
+            status,
+            args,
+            result: actionResult,
+            name: message.name
+          });
+          if (!toRender && status === "complete") {
+            return null;
+          }
+          if (typeof toRender === "string") {
+            return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+              AssistantMessage2,
+              {
+                rawData: message,
+                "data-message-role": "assistant",
+                isLoading: false,
+                isGenerating: false,
+                message: toRender
+              },
+              index
+            );
+          } else {
+            return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+              AssistantMessage2,
+              {
+                rawData: message,
+                "data-message-role": "action-render",
+                isLoading: false,
+                isGenerating: false,
+                subComponent: toRender
+              },
+              index
+            );
+          }
+        } catch (e) {
+          console.error(`Error executing render function for action ${message.name}: ${e}`);
+          return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+            AssistantMessage2,
+            {
+              rawData: message,
+              "data-message-role": "assistant",
+              isLoading: false,
+              isGenerating: false,
+              subComponent: /* @__PURE__ */ (0, import_jsx_runtime18.jsxs)("div", { className: "copilotKitMessage copilotKitAssistantMessage", children: [
+                /* @__PURE__ */ (0, import_jsx_runtime18.jsxs)("b", { children: [
+                  "\u274C Error executing render function for action ",
+                  message.name,
+                  ":"
+                ] }),
+                /* @__PURE__ */ (0, import_jsx_runtime18.jsx)("pre", { children: e instanceof Error ? e.message : String(e) })
+              ] })
+            },
+            index
+          );
+        }
+      }
+    } else if (!inProgress || !isCurrentMessage) {
+      return null;
+    } else {
+      return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+        AssistantMessage2,
+        {
+          rawData: message,
+          "data-message-role": "assistant",
+          isLoading: true,
+          isGenerating: true
+        },
+        index
+      );
+    }
+  }
+}
+
+// src/components/chat/messages/RenderResultMessage.tsx
+var import_jsx_runtime19 = require("react/jsx-runtime");
+function RenderResultMessage(_a) {
+  var _b = _a, {
+    AssistantMessage: AssistantMessage2 = AssistantMessage
+  } = _b, props = __objRest(_b, [
+    "AssistantMessage"
+  ]);
+  const { message, inProgress, index, isCurrentMessage } = props;
+  if (message.isResultMessage() && inProgress && isCurrentMessage) {
+    return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+      AssistantMessage2,
+      {
+        "data-message-role": "assistant",
+        rawData: message,
+        isLoading: true,
+        isGenerating: true
+      },
+      index
+    );
+  } else {
+    return null;
+  }
+}
+
+// src/components/chat/messages/RenderAgentStateMessage.tsx
+var import_react_core7 = require("@copilotkit/react-core");
 var import_jsx_runtime20 = require("react/jsx-runtime");
-var UserMessage = (props) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)("div", { className: "copilotKitMessage copilotKitUserMessage", children: props.message });
-};
+function RenderAgentStateMessage(_a) {
+  var _b = _a, {
+    AssistantMessage: AssistantMessage2 = AssistantMessage
+  } = _b, props = __objRest(_b, [
+    "AssistantMessage"
+  ]);
+  const { chatComponentsCache } = (0, import_react_core7.useCopilotContext)();
+  const { message, inProgress, index, isCurrentMessage } = props;
+  if (message.isAgentStateMessage()) {
+    let render;
+    if (chatComponentsCache.current !== null) {
+      render = chatComponentsCache.current.coAgentStateRenders[`${message.agentName}-${message.nodeName}`] || chatComponentsCache.current.coAgentStateRenders[`${message.agentName}-global`];
+    }
+    if (render) {
+      if (typeof render === "string") {
+        if (isCurrentMessage && inProgress) {
+          return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+            AssistantMessage2,
+            {
+              rawData: message,
+              message: render,
+              "data-message-role": "assistant",
+              isLoading: true,
+              isGenerating: true
+            },
+            index
+          );
+        } else {
+          return null;
+        }
+      } else {
+        const state = message.state;
+        let status = message.active ? "inProgress" : "complete";
+        const toRender = render({
+          status,
+          state,
+          nodeName: message.nodeName
+        });
+        if (!toRender && status === "complete") {
+          return null;
+        }
+        if (!toRender && isCurrentMessage && inProgress) {
+          return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+            AssistantMessage2,
+            {
+              "data-message-role": "assistant",
+              rawData: message,
+              isLoading: true,
+              isGenerating: true
+            },
+            index
+          );
+        } else if (!toRender) {
+          return null;
+        }
+        if (typeof toRender === "string") {
+          return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+            AssistantMessage2,
+            {
+              rawData: message,
+              message: toRender,
+              isLoading: true,
+              isGenerating: true,
+              "data-message-role": "assistant"
+            },
+            index
+          );
+        } else {
+          return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+            AssistantMessage2,
+            {
+              rawData: message,
+              "data-message-role": "agent-state-render",
+              isLoading: false,
+              isGenerating: false,
+              subComponent: toRender
+            },
+            index
+          );
+        }
+      }
+    } else if (!inProgress || !isCurrentMessage) {
+      return null;
+    } else {
+      return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+        AssistantMessage2,
+        {
+          rawData: message,
+          isLoading: true,
+          isGenerating: true,
+          "data-message-role": "assistant"
+        },
+        index
+      );
+    }
+  }
+}
 
 // src/components/chat/Suggestion.tsx
 var import_react_core8 = require("@copilotkit/react-core");
@@ -2441,11 +2461,11 @@ function CopilotChat({
   );
   const chatContext = import_react13.default.useContext(ChatContext);
   const isVisible = chatContext ? chatContext.open : true;
-  const handleRegenerate = () => {
+  const handleRegenerate = (messageId) => {
     if (onRegenerate) {
-      onRegenerate();
+      onRegenerate(messageId);
     }
-    reloadMessages();
+    reloadMessages(messageId);
   };
   const handleCopy = (message) => {
     if (onCopy) {
@@ -2637,7 +2657,7 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onS
       defaultStopGeneration();
     }
   }
-  function reloadMessages() {
+  function reloadMessages(messageId) {
     if (onReloadMessages) {
       onReloadMessages({
         messages,
@@ -2647,10 +2667,11 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onS
         restartCurrentAgent,
         stopCurrentAgent,
         runCurrentAgent,
-        setCurrentAgentState
+        setCurrentAgentState,
+        messageId
       });
     } else {
-      defaultReloadMessages();
+      defaultReloadMessages(messageId);
     }
   }
   return {