@copilotkit/react-ui 1.1.1-feat-runtime-remote-actions.2 → 1.1.1-feat-runtime-remote-actions.4

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # ui
 
+## 1.1.1-feat-runtime-remote-actions.4
+
+### Patch Changes
+
+- Show agent state in the chat window
+- Updated dependencies
+  - @copilotkit/runtime-client-gql@1.1.1-feat-runtime-remote-actions.4
+  - @copilotkit/react-core@1.1.1-feat-runtime-remote-actions.4
+  - @copilotkit/shared@1.1.1-feat-runtime-remote-actions.4
+
+## 1.1.1-feat-runtime-remote-actions.3
+
+### Patch Changes
+
+- stream intermediate state
+- Updated dependencies
+  - @copilotkit/runtime-client-gql@1.1.1-feat-runtime-remote-actions.3
+  - @copilotkit/react-core@1.1.1-feat-runtime-remote-actions.3
+  - @copilotkit/shared@1.1.1-feat-runtime-remote-actions.3
+
 ## 1.1.1-feat-runtime-remote-actions.2
 
 ### Patch Changes

package/dist/{chunk-AAX6U6YB.mjs → chunk-4II7NDKP.mjs}

@@ -1,24 +1,24 @@
 import {
   Window
 } from "./chunk-LLOSOTAT.mjs";
-import {
-  Header
-} from "./chunk-SE6DAYSX.mjs";
 import {
   Button
 } from "./chunk-UC3Y7MWX.mjs";
 import {
   CopilotChat
-} from "./chunk-4V6JWNFZ.mjs";
-import {
-  Input
-} from "./chunk-U6J5DGOE.mjs";
+} from "./chunk-JU7KYJD4.mjs";
 import {
   Messages
-} from "./chunk-2C5JDPHV.mjs";
+} from "./chunk-PDU5NXDS.mjs";
 import {
   ResponseButton
 } from "./chunk-XSUSSWDS.mjs";
+import {
+  Header
+} from "./chunk-SE6DAYSX.mjs";
+import {
+  Input
+} from "./chunk-U6J5DGOE.mjs";
 import {
   ChatContextProvider
 } from "./chunk-BJPGMY3I.mjs";
@@ -90,4 +90,4 @@ var CopilotModal = ({
 export {
   CopilotModal
 };
-//# sourceMappingURL=chunk-AAX6U6YB.mjs.map
+//# sourceMappingURL=chunk-4II7NDKP.mjs.map

package/dist/{chunk-I4QG2ZZU.mjs → chunk-DAHXB2SW.mjs}

@@ -5,6 +5,9 @@ import {
   logReadables,
   shouldShowDevConsole
 } from "./chunk-H7TM5JE6.mjs";
+import {
+  SmallSpinnerIcon
+} from "./chunk-FZC7X5PK.mjs";
 import {
   CheckIcon,
   ChevronDownIcon,
@@ -12,9 +15,6 @@ import {
   ExclamationMarkIcon,
   ExclamationMarkTriangleIcon
 } from "./chunk-OTPAZXVR.mjs";
-import {
-  SmallSpinnerIcon
-} from "./chunk-FZC7X5PK.mjs";
 
 // src/components/dev-console/console.tsx
 import { useCopilotContext } from "@copilotkit/react-core";
@@ -217,4 +217,4 @@ export {
   CopilotDevConsole,
   DebugMenuButton
 };
-//# sourceMappingURL=chunk-I4QG2ZZU.mjs.map
+//# sourceMappingURL=chunk-DAHXB2SW.mjs.map

package/dist/{chunk-MEVD4R3Q.mjs → chunk-G3RCRTQZ.mjs}

@@ -1,6 +1,6 @@
 import {
   CopilotModal
-} from "./chunk-AAX6U6YB.mjs";
+} from "./chunk-4II7NDKP.mjs";
 import {
   __spreadProps,
   __spreadValues
@@ -27,4 +27,4 @@ function CopilotSidebar(props) {
 export {
   CopilotSidebar
 };
-//# sourceMappingURL=chunk-MEVD4R3Q.mjs.map
+//# sourceMappingURL=chunk-G3RCRTQZ.mjs.map

package/dist/{chunk-4V6JWNFZ.mjs → chunk-JU7KYJD4.mjs}

@@ -1,23 +1,23 @@
+import {
+  Messages
+} from "./chunk-PDU5NXDS.mjs";
+import {
+  ResponseButton
+} from "./chunk-XSUSSWDS.mjs";
 import {
   Suggestion,
   reloadSuggestions
 } from "./chunk-WCPLXRZX.mjs";
-import {
-  CopilotDevConsole
-} from "./chunk-I4QG2ZZU.mjs";
 import {
   Input
 } from "./chunk-U6J5DGOE.mjs";
-import {
-  Messages
-} from "./chunk-2C5JDPHV.mjs";
-import {
-  ResponseButton
-} from "./chunk-XSUSSWDS.mjs";
 import {
   ChatContext,
   ChatContextProvider
 } from "./chunk-BJPGMY3I.mjs";
+import {
+  CopilotDevConsole
+} from "./chunk-DAHXB2SW.mjs";
 import {
   __async
 } from "./chunk-MRXNTQOX.mjs";
@@ -164,4 +164,4 @@ export {
   WrappedCopilotChat,
   useCopilotChatLogic
 };
-//# sourceMappingURL=chunk-4V6JWNFZ.mjs.map
+//# sourceMappingURL=chunk-JU7KYJD4.mjs.map

package/dist/{chunk-3K4EOAP3.mjs → chunk-LKY2XFK5.mjs}

@@ -1,6 +1,6 @@
 import {
   CopilotModal
-} from "./chunk-AAX6U6YB.mjs";
+} from "./chunk-4II7NDKP.mjs";
 import {
   __spreadProps,
   __spreadValues
@@ -18,4 +18,4 @@ function CopilotPopup(props) {
 export {
   CopilotPopup
 };
-//# sourceMappingURL=chunk-3K4EOAP3.mjs.map
+//# sourceMappingURL=chunk-LKY2XFK5.mjs.map

package/dist/{chunk-2C5JDPHV.mjs → chunk-PDU5NXDS.mjs}

@@ -18,7 +18,7 @@ import {
 } from "@copilotkit/runtime-client-gql";
 import { jsx, jsxs } from "react/jsx-runtime";
 var Messages = ({ messages, inProgress, children }) => {
-  const { chatComponentsCache } = useCopilotContext();
+  const { chatComponentsCache, chatUI } = useCopilotContext();
   const context = useChatContext();
   const initialMessages = useMemo(
     () => makeInitialMessages(context.labels.initial),
@@ -56,8 +56,8 @@ var Messages = ({ messages, inProgress, children }) => {
       } else if (message instanceof TextMessage && message.role == "assistant") {
         return /* @__PURE__ */ jsx("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: isCurrentMessage && inProgress && !message.content ? context.icons.spinnerIcon : /* @__PURE__ */ jsx(Markdown, { content: message.content }) }, index);
       } else if (message instanceof ActionExecutionMessage) {
-        if (chatComponentsCache.current !== null && chatComponentsCache.current[message.name]) {
-          const render = chatComponentsCache.current[message.name];
+        if (chatComponentsCache.current !== null && chatComponentsCache.current.actions[message.name]) {
+          const render = chatComponentsCache.current.actions[message.name];
           if (typeof render === "string") {
             if (isCurrentMessage && inProgress) {
               return /* @__PURE__ */ jsxs("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
@@ -70,18 +70,18 @@ var Messages = ({ messages, inProgress, children }) => {
             }
           } else {
             const args = message.arguments;
-            let status = "inProgress";
+            let status2 = "inProgress";
             if (functionResults[message.id] !== void 0) {
-              status = "complete";
+              status2 = "complete";
             } else if (message.status.code !== MessageStatusCode.Pending) {
-              status = "executing";
+              status2 = "executing";
             }
             const toRender = render({
-              status,
+              status: status2,
               args,
               result: functionResults[message.id]
             });
-            if (!toRender && status === "complete") {
+            if (!toRender && status2 === "complete") {
               return null;
             }
             if (typeof toRender === "string") {
@@ -99,7 +99,51 @@ var Messages = ({ messages, inProgress, children }) => {
         } else {
           return /* @__PURE__ */ jsx("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
         }
-      } else if ((message instanceof AgentStateMessage || message instanceof ResultMessage) && inProgress && isCurrentMessage) {
+      } else if (message instanceof AgentStateMessage) {
+        let currentChatUI = chatUI.find(
+          (ui) => ui.agentName === message.agentName && ui.nodeName === message.nodeName
+        );
+        if (!currentChatUI) {
+          currentChatUI = chatUI.find(
+            (ui) => ui.agentName === message.agentName && ui.nodeName === void 0
+          );
+        }
+        if (!currentChatUI) {
+          if (inProgress && isCurrentMessage) {
+            return /* @__PURE__ */ jsx("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
+          } else {
+            return null;
+          }
+        }
+        if (typeof currentChatUI.render === "string") {
+          if (isCurrentMessage && inProgress) {
+            return /* @__PURE__ */ jsxs("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+              context.icons.spinnerIcon,
+              " ",
+              /* @__PURE__ */ jsx("span", { className: "inProgressLabel", children: currentChatUI.render })
+            ] }, index);
+          } else {
+            return null;
+          }
+        }
+        const toRender = currentChatUI.render({
+          state: message.state,
+          agentName: message.agentName,
+          nodeName: message.nodeName
+        });
+        if (!toRender && status === "complete") {
+          return null;
+        }
+        if (typeof toRender === "string") {
+          return /* @__PURE__ */ jsxs("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+            isCurrentMessage && inProgress && context.icons.spinnerIcon,
+            " ",
+            toRender
+          ] }, index);
+        } else {
+          return /* @__PURE__ */ jsx("div", { className: "copilotKitCustomAssistantMessage", children: toRender }, index);
+        }
+      } else if (message instanceof ResultMessage && inProgress && isCurrentMessage) {
         return /* @__PURE__ */ jsx("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
       }
     }),
@@ -126,4 +170,4 @@ function makeInitialMessages(initial) {
 export {
   Messages
 };
-//# sourceMappingURL=chunk-2C5JDPHV.mjs.map
+//# sourceMappingURL=chunk-PDU5NXDS.mjs.map

package/dist/chunk-PDU5NXDS.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/components/chat/Messages.tsx"],"sourcesContent":["import React, { useEffect, useMemo } from \"react\";\nimport { MessagesProps } from \"./props\";\nimport { useChatContext } from \"./ChatContext\";\nimport { Markdown } from \"./Markdown\";\nimport { RenderFunctionStatus, useCopilotContext } from \"@copilotkit/react-core\";\nimport {\n  MessageStatusCode,\n  ActionExecutionMessage,\n  Message,\n  ResultMessage,\n  TextMessage,\n  Role,\n  AgentStateMessage,\n} from \"@copilotkit/runtime-client-gql\";\n\nexport const Messages = ({ messages, inProgress, children }: MessagesProps) => {\n  const { chatComponentsCache, chatUI } = useCopilotContext();\n\n  const context = useChatContext();\n  const initialMessages = useMemo(\n    () => makeInitialMessages(context.labels.initial),\n    [context.labels.initial],\n  );\n  messages = [...initialMessages, ...messages];\n\n  const functionResults: Record<string, string> = {};\n\n  for (let i = 0; i < messages.length; i++) {\n    if (messages[i] instanceof ActionExecutionMessage) {\n      const id = messages[i].id;\n      const resultMessage: ResultMessage | undefined = messages.find(\n        (message) => message instanceof ResultMessage && message.actionExecutionId === id,\n      ) as ResultMessage | undefined;\n\n      if (resultMessage) {\n        functionResults[id] = ResultMessage.decodeResult(resultMessage.result || \"\");\n      }\n    }\n  }\n\n  const messagesEndRef = React.useRef<HTMLDivElement>(null);\n\n  const scrollToBottom = () => {\n    if (messagesEndRef.current) {\n      messagesEndRef.current.scrollIntoView({\n        behavior: \"auto\",\n      });\n    }\n  };\n\n  useEffect(() => {\n    scrollToBottom();\n  }, [messages]);\n\n  return (\n    <div className=\"copilotKitMessages\">\n      {messages.map((message, index) => {\n        const isCurrentMessage = index === messages.length - 1;\n\n        if (message instanceof TextMessage && message.role === \"user\") {\n          return (\n            <div key={index} className=\"copilotKitMessage copilotKitUserMessage\">\n              {message.content}\n            </div>\n          );\n        } else if (message instanceof TextMessage && message.role == \"assistant\") {\n          return (\n            <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n              {isCurrentMessage && inProgress && !message.content ? (\n                context.icons.spinnerIcon\n              ) : (\n                <Markdown content={message.content} />\n              )}\n            </div>\n          );\n        } else if (message instanceof ActionExecutionMessage) {\n          if (\n            chatComponentsCache.current !== null &&\n            chatComponentsCache.current.actions[message.name]\n          ) {\n            const render = chatComponentsCache.current.actions[message.name];\n            // render a static string\n            if (typeof render === \"string\") {\n              // when render is static, we show it only when in progress\n              if (isCurrentMessage && inProgress) {\n                return (\n                  <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                    {context.icons.spinnerIcon} <span className=\"inProgressLabel\">{render}</span>\n                  </div>\n                );\n              }\n              // Done - silent by default to avoid a series of \"done\" messages\n              else {\n                return null;\n              }\n            }\n            // render is a function\n            else {\n              const args = message.arguments;\n\n              let status: RenderFunctionStatus = \"inProgress\";\n\n              if (functionResults[message.id] !== undefined) {\n                status = \"complete\";\n              } else if (message.status.code !== MessageStatusCode.Pending) {\n                status = \"executing\";\n              }\n\n              const toRender = render({\n                status: status as any,\n                args,\n                result: functionResults[message.id],\n              });\n\n              // No result and complete: stay silent\n              if (!toRender && status === \"complete\") {\n                return null;\n              }\n\n              if (typeof toRender === \"string\") {\n                return (\n                  <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                    {isCurrentMessage && inProgress && context.icons.spinnerIcon} {toRender}\n                  </div>\n                );\n              } else {\n                return (\n                  <div key={index} className=\"copilotKitCustomAssistantMessage\">\n                    {toRender}\n                  </div>\n                );\n              }\n            }\n          }\n          // No render function found- show the default message\n          else if (!inProgress || !isCurrentMessage) {\n            // Done - silent by default to avoid a series of \"done\" messages\n            return null;\n          } else {\n            // In progress\n            return (\n              <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                {context.icons.spinnerIcon}\n              </div>\n            );\n          }\n        } else if (message instanceof AgentStateMessage) {\n          let currentChatUI = chatUI.find(\n            (ui) => ui.agentName === message.agentName && ui.nodeName === message.nodeName,\n          );\n\n          if (!currentChatUI) {\n            currentChatUI = chatUI.find(\n              (ui) => ui.agentName === message.agentName && ui.nodeName === undefined,\n            );\n          }\n\n          if (!currentChatUI) {\n            if (inProgress && isCurrentMessage) {\n              return (\n                <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                  {context.icons.spinnerIcon}\n                </div>\n              );\n            } else {\n              return null;\n            }\n          }\n\n          if (typeof currentChatUI.render === \"string\") {\n            // when render is static, we show it only when in progress\n            if (isCurrentMessage && inProgress) {\n              return (\n                <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                  {context.icons.spinnerIcon}{\" \"}\n                  <span className=\"inProgressLabel\">{currentChatUI.render}</span>\n                </div>\n              );\n            }\n            // Done - silent by default to avoid a series of \"done\" messages\n            else {\n              return null;\n            }\n          }\n\n          const toRender = currentChatUI.render({\n            state: message.state,\n            agentName: message.agentName,\n            nodeName: message.nodeName,\n          });\n\n          // No result and complete: stay silent\n          if (!toRender && status === \"complete\") {\n            return null;\n          }\n\n          if (typeof toRender === \"string\") {\n            return (\n              <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n                {isCurrentMessage && inProgress && context.icons.spinnerIcon} {toRender}\n              </div>\n            );\n          } else {\n            return (\n              <div key={index} className=\"copilotKitCustomAssistantMessage\">\n                {toRender}\n              </div>\n            );\n          }\n        } else if (message instanceof ResultMessage && inProgress && isCurrentMessage) {\n          return (\n            <div key={index} className={`copilotKitMessage copilotKitAssistantMessage`}>\n              {context.icons.spinnerIcon}\n            </div>\n          );\n        }\n      })}\n      <footer ref={messagesEndRef}>{children}</footer>\n    </div>\n  );\n};\n\nfunction makeInitialMessages(initial?: string | string[]): Message[] {\n  let initialArray: string[] = [];\n  if (initial) {\n    if (Array.isArray(initial)) {\n      initialArray.push(...initial);\n    } else {\n      initialArray.push(initial);\n    }\n  }\n\n  return initialArray.map(\n    (message) =>\n      new TextMessage({\n        role: Role.Assistant,\n        content: message,\n      }),\n  );\n}\n"],"mappings":";;;;;;;;AAAA,OAAO,SAAS,WAAW,eAAe;AAI1C,SAA+B,yBAAyB;AACxD;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAgDK,cAyBM,YAzBN;AA9CL,IAAM,WAAW,CAAC,EAAE,UAAU,YAAY,SAAS,MAAqB;AAC7E,QAAM,EAAE,qBAAqB,OAAO,IAAI,kBAAkB;AAE1D,QAAM,UAAU,eAAe;AAC/B,QAAM,kBAAkB;AAAA,IACtB,MAAM,oBAAoB,QAAQ,OAAO,OAAO;AAAA,IAChD,CAAC,QAAQ,OAAO,OAAO;AAAA,EACzB;AACA,aAAW,CAAC,GAAG,iBAAiB,GAAG,QAAQ;AAE3C,QAAM,kBAA0C,CAAC;AAEjD,WAAS,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;AACxC,QAAI,SAAS,CAAC,aAAa,wBAAwB;AACjD,YAAM,KAAK,SAAS,CAAC,EAAE;AACvB,YAAM,gBAA2C,SAAS;AAAA,QACxD,CAAC,YAAY,mBAAmB,iBAAiB,QAAQ,sBAAsB;AAAA,MACjF;AAEA,UAAI,eAAe;AACjB,wBAAgB,EAAE,IAAI,cAAc,aAAa,cAAc,UAAU,EAAE;AAAA,MAC7E;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,MAAM,OAAuB,IAAI;AAExD,QAAM,iBAAiB,MAAM;AAC3B,QAAI,eAAe,SAAS;AAC1B,qBAAe,QAAQ,eAAe;AAAA,QACpC,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,EACF;AAEA,YAAU,MAAM;AACd,mBAAe;AAAA,EACjB,GAAG,CAAC,QAAQ,CAAC;AAEb,SACE,qBAAC,SAAI,WAAU,sBACZ;AAAA,aAAS,IAAI,CAAC,SAAS,UAAU;AAChC,YAAM,mBAAmB,UAAU,SAAS,SAAS;AAErD,UAAI,mBAAmB,eAAe,QAAQ,SAAS,QAAQ;AAC7D,eACE,oBAAC,SAAgB,WAAU,2CACxB,kBAAQ,WADD,KAEV;AAAA,MAEJ,WAAW,mBAAmB,eAAe,QAAQ,QAAQ,aAAa;AACxE,eACE,oBAAC,SAAgB,WAAW,gDACzB,8BAAoB,cAAc,CAAC,QAAQ,UAC1C,QAAQ,MAAM,cAEd,oBAAC,YAAS,SAAS,QAAQ,SAAS,KAJ9B,KAMV;AAAA,MAEJ,WAAW,mBAAmB,wBAAwB;AACpD,YACE,oBAAoB,YAAY,QAChC,oBAAoB,QAAQ,QAAQ,QAAQ,IAAI,GAChD;AACA,gBAAM,SAAS,oBAAoB,QAAQ,QAAQ,QAAQ,IAAI;AAE/D,cAAI,OAAO,WAAW,UAAU;AAE9B,gBAAI,oBAAoB,YAAY;AAClC,qBACE,qBAAC,SAAgB,WAAW,gDACzB;AAAA,wBAAQ,MAAM;AAAA,gBAAY;AAAA,gBAAC,oBAAC,UAAK,WAAU,mBAAmB,kBAAO;AAAA,mBAD9D,KAEV;AAAA,YAEJ,OAEK;AACH,qBAAO;AAAA,YACT;AAAA,UACF,OAEK;AACH,kBAAM,OAAO,QAAQ;AAErB,gBAAIA,UAA+B;AAEnC,gBAAI,gBAAgB,QAAQ,EAAE,MAAM,QAAW;AAC7C,cAAAA,UAAS;AAAA,YACX,WAAW,QAAQ,OAAO,SAAS,kBAAkB,SAAS;AAC5D,cAAAA,UAAS;AAAA,YACX;AAEA,kBAAM,WAAW,OAAO;AAAA,cACtB,QAAQA;AAAA,cACR;AAAA,cACA,QAAQ,gBAAgB,QAAQ,EAAE;AAAA,YACpC,CAAC;AAGD,gBAAI,CAAC,YAAYA,YAAW,YAAY;AACtC,qBAAO;AAAA,YACT;AAEA,gBAAI,OAAO,aAAa,UAAU;AAChC,qBACE,qBAAC,SAAgB,WAAW,gDACzB;AAAA,oCAAoB,cAAc,QAAQ,MAAM;AAAA,gBAAY;AAAA,gBAAE;AAAA,mBADvD,KAEV;AAAA,YAEJ,OAAO;AACL,qBACE,oBAAC,SAAgB,WAAU,oCACxB,sBADO,KAEV;AAAA,YAEJ;AAAA,UACF;AAAA,QACF,WAES,CAAC,cAAc,CAAC,kBAAkB;AAEzC,iBAAO;AAAA,QACT,OAAO;AAEL,iBACE,oBAAC,SAAgB,WAAW,gDACzB,kBAAQ,MAAM,eADP,KAEV;AAAA,QAEJ;AAAA,MACF,WAAW,mBAAmB,mBAAmB;AAC/C,YAAI,gBAAgB,OAAO;AAAA,UACzB,CAAC,OAAO,GAAG,cAAc,QAAQ,aAAa,GAAG,aAAa,QAAQ;AAAA,QACxE;AAEA,YAAI,CAAC,eAAe;AAClB,0BAAgB,OAAO;AAAA,YACrB,CAAC,OAAO,GAAG,cAAc,QAAQ,aAAa,GAAG,aAAa;AAAA,UAChE;AAAA,QACF;AAEA,YAAI,CAAC,eAAe;AAClB,cAAI,cAAc,kBAAkB;AAClC,mBACE,oBAAC,SAAgB,WAAW,gDACzB,kBAAQ,MAAM,eADP,KAEV;AAAA,UAEJ,OAAO;AACL,mBAAO;AAAA,UACT;AAAA,QACF;AAEA,YAAI,OAAO,cAAc,WAAW,UAAU;AAE5C,cAAI,oBAAoB,YAAY;AAClC,mBACE,qBAAC,SAAgB,WAAW,gDACzB;AAAA,sBAAQ,MAAM;AAAA,cAAa;AAAA,cAC5B,oBAAC,UAAK,WAAU,mBAAmB,wBAAc,QAAO;AAAA,iBAFhD,KAGV;AAAA,UAEJ,OAEK;AACH,mBAAO;AAAA,UACT;AAAA,QACF;AAEA,cAAM,WAAW,cAAc,OAAO;AAAA,UACpC,OAAO,QAAQ;AAAA,UACf,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAGD,YAAI,CAAC,YAAY,WAAW,YAAY;AACtC,iBAAO;AAAA,QACT;AAEA,YAAI,OAAO,aAAa,UAAU;AAChC,iBACE,qBAAC,SAAgB,WAAW,gDACzB;AAAA,gCAAoB,cAAc,QAAQ,MAAM;AAAA,YAAY;AAAA,YAAE;AAAA,eADvD,KAEV;AAAA,QAEJ,OAAO;AACL,iBACE,oBAAC,SAAgB,WAAU,oCACxB,sBADO,KAEV;AAAA,QAEJ;AAAA,MACF,WAAW,mBAAmB,iBAAiB,cAAc,kBAAkB;AAC7E,eACE,oBAAC,SAAgB,WAAW,gDACzB,kBAAQ,MAAM,eADP,KAEV;AAAA,MAEJ;AAAA,IACF,CAAC;AAAA,IACD,oBAAC,YAAO,KAAK,gBAAiB,UAAS;AAAA,KACzC;AAEJ;AAEA,SAAS,oBAAoB,SAAwC;AACnE,MAAI,eAAyB,CAAC;AAC9B,MAAI,SAAS;AACX,QAAI,MAAM,QAAQ,OAAO,GAAG;AAC1B,mBAAa,KAAK,GAAG,OAAO;AAAA,IAC9B,OAAO;AACL,mBAAa,KAAK,OAAO;AAAA,IAC3B;AAAA,EACF;AAEA,SAAO,aAAa;AAAA,IAClB,CAAC,YACC,IAAI,YAAY;AAAA,MACd,MAAM,KAAK;AAAA,MACX,SAAS;AAAA,IACX,CAAC;AAAA,EACL;AACF;","names":["status"]}

package/dist/components/chat/Chat.js

@@ -852,7 +852,7 @@ var import_react_core = require("@copilotkit/react-core");
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
 var import_jsx_runtime5 = require("react/jsx-runtime");
 var Messages = ({ messages, inProgress, children }) => {
-  const { chatComponentsCache } = (0, import_react_core.useCopilotContext)();
+  const { chatComponentsCache, chatUI } = (0, import_react_core.useCopilotContext)();
   const context = useChatContext();
   const initialMessages = (0, import_react4.useMemo)(
     () => makeInitialMessages(context.labels.initial),
@@ -890,8 +890,8 @@ var Messages = ({ messages, inProgress, children }) => {
       } else if (message instanceof import_runtime_client_gql.TextMessage && message.role == "assistant") {
         return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: isCurrentMessage && inProgress && !message.content ? context.icons.spinnerIcon : /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(Markdown, { content: message.content }) }, index);
       } else if (message instanceof import_runtime_client_gql.ActionExecutionMessage) {
-        if (chatComponentsCache.current !== null && chatComponentsCache.current[message.name]) {
-          const render = chatComponentsCache.current[message.name];
+        if (chatComponentsCache.current !== null && chatComponentsCache.current.actions[message.name]) {
+          const render = chatComponentsCache.current.actions[message.name];
           if (typeof render === "string") {
             if (isCurrentMessage && inProgress) {
               return /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
@@ -904,18 +904,18 @@ var Messages = ({ messages, inProgress, children }) => {
             }
           } else {
             const args = message.arguments;
-            let status = "inProgress";
+            let status2 = "inProgress";
             if (functionResults[message.id] !== void 0) {
-              status = "complete";
+              status2 = "complete";
             } else if (message.status.code !== import_runtime_client_gql.MessageStatusCode.Pending) {
-              status = "executing";
+              status2 = "executing";
             }
             const toRender = render({
-              status,
+              status: status2,
               args,
               result: functionResults[message.id]
             });
-            if (!toRender && status === "complete") {
+            if (!toRender && status2 === "complete") {
               return null;
             }
             if (typeof toRender === "string") {
@@ -933,7 +933,51 @@ var Messages = ({ messages, inProgress, children }) => {
         } else {
           return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
         }
-      } else if ((message instanceof import_runtime_client_gql.AgentStateMessage || message instanceof import_runtime_client_gql.ResultMessage) && inProgress && isCurrentMessage) {
+      } else if (message instanceof import_runtime_client_gql.AgentStateMessage) {
+        let currentChatUI = chatUI.find(
+          (ui) => ui.agentName === message.agentName && ui.nodeName === message.nodeName
+        );
+        if (!currentChatUI) {
+          currentChatUI = chatUI.find(
+            (ui) => ui.agentName === message.agentName && ui.nodeName === void 0
+          );
+        }
+        if (!currentChatUI) {
+          if (inProgress && isCurrentMessage) {
+            return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
+          } else {
+            return null;
+          }
+        }
+        if (typeof currentChatUI.render === "string") {
+          if (isCurrentMessage && inProgress) {
+            return /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+              context.icons.spinnerIcon,
+              " ",
+              /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("span", { className: "inProgressLabel", children: currentChatUI.render })
+            ] }, index);
+          } else {
+            return null;
+          }
+        }
+        const toRender = currentChatUI.render({
+          state: message.state,
+          agentName: message.agentName,
+          nodeName: message.nodeName
+        });
+        if (!toRender && status === "complete") {
+          return null;
+        }
+        if (typeof toRender === "string") {
+          return /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+            isCurrentMessage && inProgress && context.icons.spinnerIcon,
+            " ",
+            toRender
+          ] }, index);
+        } else {
+          return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("div", { className: "copilotKitCustomAssistantMessage", children: toRender }, index);
+        }
+      } else if (message instanceof import_runtime_client_gql.ResultMessage && inProgress && isCurrentMessage) {
         return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
       }
     }),
@@ -1253,7 +1297,7 @@ var reloadSuggestions = (context, chatSuggestionConfiguration, setCurrentSuggest
           readable: true
         },
         abortSignal: abortController == null ? void 0 : abortController.signal,
-        stream: ({ status, args }) => {
+        stream: ({ status: status2, args }) => {
           const suggestions = args.suggestions || [];
           const newSuggestions = [];
           for (let i = 0; i < suggestions.length; i++) {
@@ -1261,7 +1305,7 @@ var reloadSuggestions = (context, chatSuggestionConfiguration, setCurrentSuggest
               break;
             }
             const { title, message } = suggestions[i];
-            const partial = i == suggestions.length - 1 && status !== "complete";
+            const partial = i == suggestions.length - 1 && status2 !== "complete";
             newSuggestions.push({
               title,
               message,