@copilotkit/react-ui 1.4.8 → 1.5.0-coagents-v0-3.0

package/dist/components/chat/Chat.js

@@ -805,9 +805,9 @@ var import_react6 = require("react");
 var import_react_syntax_highlighter = require("react-syntax-highlighter");
 
 // src/hooks/use-copy-to-clipboard.tsx
-var React5 = __toESM(require("react"));
+var React4 = __toESM(require("react"));
 function useCopyToClipboard({ timeout = 2e3 }) {
-  const [isCopied, setIsCopied] = React5.useState(false);
+  const [isCopied, setIsCopied] = React4.useState(false);
   const copyToClipboard = (value) => {
     var _a;
     if (typeof window === "undefined" || !((_a = navigator.clipboard) == null ? void 0 : _a.writeText)) {
@@ -1303,8 +1303,8 @@ function RenderActionExecutionMessage(props) {
   const { chatComponentsCache } = (0, import_react_core3.useCopilotContext)();
   const { icons } = useChatContext();
   if (message.isActionExecutionMessage()) {
-    if (chatComponentsCache.current !== null && chatComponentsCache.current.actions[message.name]) {
-      const render = chatComponentsCache.current.actions[message.name];
+    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_runtime10.jsxs)("div", { className: `copilotKitMessage copilotKitAssistantMessage`, children: [
@@ -1327,7 +1327,8 @@ function RenderActionExecutionMessage(props) {
           const toRender = render({
             status,
             args,
-            result: actionResult
+            result: actionResult,
+            name: message.name
           });
           if (!toRender && status === "complete") {
             return null;
@@ -1641,7 +1642,7 @@ function logMessages(context) {
         type: "ActionExecutionMessage",
         role: void 0,
         name: message.name,
-        scope: message.scope,
+        scope: message.parentMessageId,
         content: message.arguments
       };
     } else if (message.isResultMessage()) {
@@ -2167,6 +2168,7 @@ function DebugMenuButton({
 }
 
 // src/components/chat/Chat.tsx
+var import_react_core9 = require("@copilotkit/react-core");
 var import_jsx_runtime18 = require("react/jsx-runtime");
 function CopilotChat({
   instructions,
@@ -2174,6 +2176,8 @@ function CopilotChat({
   makeSystemMessage,
   showResponseButton = true,
   onInProgress,
+  onStopGeneration,
+  onReloadMessages,
   Messages: Messages2 = Messages,
   RenderTextMessage: RenderTextMessage2 = RenderTextMessage,
   RenderActionExecutionMessage: RenderActionExecutionMessage2 = RenderActionExecutionMessage,
@@ -2196,7 +2200,13 @@ function CopilotChat({
     sendMessage,
     stopGeneration,
     reloadMessages
-  } = useCopilotChatLogic(makeSystemMessage, onInProgress, onSubmitMessage);
+  } = useCopilotChatLogic(
+    makeSystemMessage,
+    onInProgress,
+    onSubmitMessage,
+    onStopGeneration,
+    onReloadMessages
+  );
   const chatContext = import_react11.default.useContext(ChatContext);
   const isVisible = chatContext ? chatContext.open : true;
   return /* @__PURE__ */ (0, import_jsx_runtime18.jsxs)(WrappedCopilotChat, { icons, labels, className, children: [
@@ -2252,8 +2262,16 @@ function WrappedCopilotChat({
   return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(import_jsx_runtime18.Fragment, { children });
 }
 var SUGGESTIONS_DEBOUNCE_TIMEOUT = 1e3;
-var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage) => {
-  const { visibleMessages, appendMessage, reloadMessages, stopGeneration, isLoading } = (0, import_react_core8.useCopilotChat)({
+var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage, onStopGeneration, onReloadMessages) => {
+  var _a;
+  const {
+    visibleMessages,
+    appendMessage,
+    reloadMessages: defaultReloadMessages,
+    stopGeneration: defaultStopGeneration,
+    runChatCompletion,
+    isLoading
+  } = (0, import_react_core8.useCopilotChat)({
     id: (0, import_shared3.randomId)(),
     makeSystemMessage
   });
@@ -2261,8 +2279,8 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage) =>
   const suggestionsAbortControllerRef = (0, import_react11.useRef)(null);
   const debounceTimerRef = (0, import_react11.useRef)();
   const abortSuggestions = () => {
-    var _a;
-    (_a = suggestionsAbortControllerRef.current) == null ? void 0 : _a.abort();
+    var _a2;
+    (_a2 = suggestionsAbortControllerRef.current) == null ? void 0 : _a2.abort();
     suggestionsAbortControllerRef.current = null;
   };
   const generalContext = (0, import_react_core8.useCopilotContext)();
@@ -2288,7 +2306,13 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage) =>
     return () => {
       clearTimeout(debounceTimerRef.current);
     };
-  }, [isLoading, context.chatSuggestionConfiguration]);
+  }, [
+    isLoading,
+    context.chatSuggestionConfiguration,
+    // hackish way to trigger suggestions reload on reset, but better than moving suggestions to the
+    // global context
+    visibleMessages.length == 0
+  ]);
   const sendMessage = (messageContent) => __async(void 0, null, function* () {
     abortSuggestions();
     setCurrentSuggestions([]);
@@ -2306,6 +2330,85 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage) =>
     appendMessage(message);
     return message;
   });
+  const messages = visibleMessages;
+  const { setMessages } = messagesContext;
+  const currentAgentName = (_a = generalContext.agentSession) == null ? void 0 : _a.agentName;
+  const restartCurrentAgent = (hint) => __async(void 0, null, function* () {
+    if (generalContext.agentSession) {
+      generalContext.setAgentSession(__spreadProps(__spreadValues({}, generalContext.agentSession), {
+        nodeName: void 0,
+        threadId: void 0
+      }));
+      generalContext.setCoagentStates((prevAgentStates) => {
+        return __spreadProps(__spreadValues({}, prevAgentStates), {
+          [generalContext.agentSession.agentName]: __spreadProps(__spreadValues({}, prevAgentStates[generalContext.agentSession.agentName]), {
+            threadId: void 0,
+            nodeName: void 0,
+            runId: void 0
+          })
+        });
+      });
+    }
+  });
+  const runCurrentAgent = (hint) => __async(void 0, null, function* () {
+    if (generalContext.agentSession) {
+      yield (0, import_react_core9.runAgent)(
+        generalContext.agentSession.agentName,
+        context,
+        appendMessage,
+        runChatCompletion,
+        hint
+      );
+    }
+  });
+  const stopCurrentAgent = () => {
+    if (generalContext.agentSession) {
+      (0, import_react_core9.stopAgent)(generalContext.agentSession.agentName, context);
+    }
+  };
+  const setCurrentAgentState = (state) => {
+    if (generalContext.agentSession) {
+      generalContext.setCoagentStates((prevAgentStates) => {
+        return __spreadProps(__spreadValues({}, prevAgentStates), {
+          [generalContext.agentSession.agentName]: {
+            state
+          }
+        });
+      });
+    }
+  };
+  function stopGeneration() {
+    if (onStopGeneration) {
+      onStopGeneration({
+        messages,
+        setMessages,
+        stopGeneration: defaultStopGeneration,
+        currentAgentName,
+        restartCurrentAgent,
+        stopCurrentAgent,
+        runCurrentAgent,
+        setCurrentAgentState
+      });
+    } else {
+      defaultStopGeneration();
+    }
+  }
+  function reloadMessages() {
+    if (onReloadMessages) {
+      onReloadMessages({
+        messages,
+        setMessages,
+        stopGeneration: defaultStopGeneration,
+        currentAgentName,
+        restartCurrentAgent,
+        stopCurrentAgent,
+        runCurrentAgent,
+        setCurrentAgentState
+      });
+    } else {
+      defaultReloadMessages();
+    }
+  }
   return {
     visibleMessages,
     isLoading,