@copilotkit/react-ui 1.3.16-mme-lgc-langgraph-package.11 → 1.3.16-mme-reset-chat.10

package/dist/components/chat/Chat.js

@@ -1905,6 +1905,7 @@ function DebugMenuButton({
 }
 
 // src/components/chat/Chat.tsx
+var import_react_core9 = require("@copilotkit/react-core");
 var import_jsx_runtime16 = require("react/jsx-runtime");
 function CopilotChat({
   instructions,
@@ -1912,6 +1913,8 @@ function CopilotChat({
   makeSystemMessage,
   showResponseButton = true,
   onInProgress,
+  onStopGeneration,
+  onReloadMessages,
   Messages: Messages2 = Messages,
   RenderTextMessage: RenderTextMessage2 = RenderTextMessage,
   RenderActionExecutionMessage: RenderActionExecutionMessage2 = RenderActionExecutionMessage,
@@ -1934,7 +1937,13 @@ function CopilotChat({
     sendMessage,
     stopGeneration,
     reloadMessages
-  } = useCopilotChatLogic(makeSystemMessage, onInProgress, onSubmitMessage);
+  } = useCopilotChatLogic(
+    makeSystemMessage,
+    onInProgress,
+    onSubmitMessage,
+    onStopGeneration,
+    onReloadMessages
+  );
   const chatContext = import_react10.default.useContext(ChatContext);
   const isVisible = chatContext ? chatContext.open : true;
   return /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)(WrappedCopilotChat, { icons, labels, className, children: [
@@ -1990,8 +1999,15 @@ function WrappedCopilotChat({
   return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(import_jsx_runtime16.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
   });
@@ -1999,8 +2015,8 @@ var useCopilotChatLogic = (makeSystemMessage, onInProgress, onSubmitMessage) =>
   const suggestionsAbortControllerRef = (0, import_react10.useRef)(null);
   const debounceTimerRef = (0, import_react10.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)();
@@ -2026,7 +2042,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([]);
@@ -2044,6 +2066,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,