@copilotkit/react-core 1.8.4 → 1.8.5-next.1

package/dist/index.js

@@ -2005,18 +2005,25 @@ function useChat(options) {
     }),
     [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
   );
-  const reload = useAsyncCallback(() => __async(this, null, function* () {
-    if (isLoading || messages.length === 0) {
-      return;
-    }
-    let newMessages = [...messages];
-    const lastMessage = messages[messages.length - 1];
-    if (lastMessage.isTextMessage() && lastMessage.role === "assistant") {
-      newMessages = newMessages.slice(0, -1);
-    }
-    setMessages(newMessages);
-    return runChatCompletionAndHandleFunctionCall(newMessages);
-  }), [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]);
+  const reload = useAsyncCallback(
+    (messageId) => __async(this, null, function* () {
+      if (isLoading || messages.length === 0) {
+        return;
+      }
+      const index = messages.findIndex((msg) => msg.id === messageId);
+      if (index === -1) {
+        console.warn(`Message with id ${messageId} not found`);
+        return;
+      }
+      let newMessages = messages.slice(0, index);
+      if (newMessages.length > 0 && newMessages[newMessages.length - 1].isAgentStateMessage()) {
+        newMessages = newMessages.slice(0, newMessages.length - 1);
+      }
+      setMessages(newMessages);
+      return runChatCompletionAndHandleFunctionCall(newMessages);
+    }),
+    [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
+  );
   const stop = () => {
     var _a;
     (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort("Stop was called");
@@ -2208,9 +2215,12 @@ function useCopilotChat(_a = {}) {
     [latestAppend]
   );
   const latestReload = useUpdatedRef(reload);
-  const latestReloadFunc = useAsyncCallback(() => __async(this, null, function* () {
-    return yield latestReload.current();
-  }), [latestReload]);
+  const latestReloadFunc = useAsyncCallback(
+    (messageId) => __async(this, null, function* () {
+      return yield latestReload.current(messageId);
+    }),
+    [latestReload]
+  );
   const latestStop = useUpdatedRef(stop);
   const latestStopFunc = (0, import_react12.useCallback)(() => {
     return latestStop.current();