@copilotkit/react-ui 1.3.16-pre.11 → 1.3.16-pre.12

package/dist/components/chat/Modal.mjs

@@ -1,10 +1,10 @@
 import {
   CopilotModal
-} from "../../chunk-H5CXJBR5.mjs";
+} from "../../chunk-OG2XHQQM.mjs";
 import "../../chunk-YAGE7RCE.mjs";
 import "../../chunk-VEC45H6Q.mjs";
 import "../../chunk-RQNJNK2W.mjs";
-import "../../chunk-ORHE7FYT.mjs";
+import "../../chunk-2BAND4IT.mjs";
 import "../../chunk-V7W6IM2V.mjs";
 import "../../chunk-OMTPWC7T.mjs";
 import "../../chunk-ZABXREBH.mjs";

package/dist/components/chat/Popup.js

@@ -2092,6 +2092,7 @@ function DebugMenuButton({
 }
 
 // src/components/chat/Chat.tsx
+var import_react_core9 = require("@copilotkit/react-core");
 var import_jsx_runtime19 = require("react/jsx-runtime");
 function CopilotChat({
   instructions,
@@ -2099,6 +2100,8 @@ function CopilotChat({
   makeSystemMessage,
   showResponseButton = true,
   onInProgress,
+  onStopGeneration,
+  onReloadMessages,
   Messages: Messages2 = Messages,
   RenderTextMessage: RenderTextMessage2 = RenderTextMessage,
   RenderActionExecutionMessage: RenderActionExecutionMessage2 = RenderActionExecutionMessage,
@@ -2121,7 +2124,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_runtime19.jsxs)(WrappedCopilotChat, { icons, labels, className, children: [
@@ -2177,8 +2186,15 @@ function WrappedCopilotChat({
   return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(import_jsx_runtime19.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,
+    isLoading
+  } = (0, import_react_core8.useCopilotChat)({
     id: (0, import_shared3.randomId)(),
     makeSystemMessage
   });
@@ -2186,8 +2202,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)();
@@ -2213,7 +2229,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([]);
@@ -2231,6 +2253,79 @@ 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, 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,
@@ -2250,6 +2345,8 @@ var CopilotModal = ({
   hitEscapeToClose = true,
   onSetOpen,
   onSubmitMessage,
+  onStopGeneration,
+  onReloadMessages,
   shortcut = "/",
   icons,
   labels,
@@ -2287,6 +2384,8 @@ var CopilotModal = ({
               {
                 instructions,
                 onSubmitMessage,
+                onStopGeneration,
+                onReloadMessages,
                 makeSystemMessage,
                 showResponseButton,
                 onInProgress,