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

package/dist/components/chat/Sidebar.mjs

@@ -1,24 +1,24 @@
 import {
   CopilotSidebar
-} from "../../chunk-MEVD4R3Q.mjs";
-import "../../chunk-AAX6U6YB.mjs";
+} from "../../chunk-G3RCRTQZ.mjs";
+import "../../chunk-4II7NDKP.mjs";
 import "../../chunk-LLOSOTAT.mjs";
-import "../../chunk-SE6DAYSX.mjs";
 import "../../chunk-UC3Y7MWX.mjs";
-import "../../chunk-4V6JWNFZ.mjs";
-import "../../chunk-V7W6IM2V.mjs";
+import "../../chunk-JU7KYJD4.mjs";
+import "../../chunk-PDU5NXDS.mjs";
+import "../../chunk-YQ3D5IQV.mjs";
+import "../../chunk-XSUSSWDS.mjs";
 import "../../chunk-WCPLXRZX.mjs";
-import "../../chunk-I4QG2ZZU.mjs";
-import "../../chunk-H7TM5JE6.mjs";
-import "../../chunk-OTPAZXVR.mjs";
+import "../../chunk-UPTB2MVO.mjs";
+import "../../chunk-SE6DAYSX.mjs";
 import "../../chunk-U6J5DGOE.mjs";
 import "../../chunk-YQFVRDNC.mjs";
-import "../../chunk-2C5JDPHV.mjs";
-import "../../chunk-YQ3D5IQV.mjs";
-import "../../chunk-XSUSSWDS.mjs";
 import "../../chunk-BJPGMY3I.mjs";
-import "../../chunk-UPTB2MVO.mjs";
+import "../../chunk-V7W6IM2V.mjs";
+import "../../chunk-DAHXB2SW.mjs";
+import "../../chunk-H7TM5JE6.mjs";
 import "../../chunk-FZC7X5PK.mjs";
+import "../../chunk-OTPAZXVR.mjs";
 import "../../chunk-54JAUBUJ.mjs";
 import "../../chunk-VOBX4JOA.mjs";
 import "../../chunk-MRXNTQOX.mjs";

package/dist/components/chat/index.js

@@ -994,7 +994,7 @@ var import_react_core = require("@copilotkit/react-core");
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
 var import_jsx_runtime8 = 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_react5.useMemo)(
     () => makeInitialMessages(context.labels.initial),
@@ -1032,8 +1032,8 @@ var Messages = ({ messages, inProgress, children }) => {
       } else if (message instanceof import_runtime_client_gql.TextMessage && message.role == "assistant") {
         return /* @__PURE__ */ (0, import_jsx_runtime8.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: isCurrentMessage && inProgress && !message.content ? context.icons.spinnerIcon : /* @__PURE__ */ (0, import_jsx_runtime8.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_runtime8.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
@@ -1046,18 +1046,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") {
@@ -1075,7 +1075,51 @@ var Messages = ({ messages, inProgress, children }) => {
         } else {
           return /* @__PURE__ */ (0, import_jsx_runtime8.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_runtime8.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_runtime8.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+              context.icons.spinnerIcon,
+              " ",
+              /* @__PURE__ */ (0, import_jsx_runtime8.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_runtime8.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
+            isCurrentMessage && inProgress && context.icons.spinnerIcon,
+            " ",
+            toRender
+          ] }, index);
+        } else {
+          return /* @__PURE__ */ (0, import_jsx_runtime8.jsx)("div", { className: "copilotKitCustomAssistantMessage", children: toRender }, index);
+        }
+      } else if (message instanceof import_runtime_client_gql.ResultMessage && inProgress && isCurrentMessage) {
         return /* @__PURE__ */ (0, import_jsx_runtime8.jsx)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: context.icons.spinnerIcon }, index);
       }
     }),
@@ -1395,7 +1439,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++) {
@@ -1403,7 +1447,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,