@copilotkit/react-core 1.9.2-next.10 → 1.9.2-next.12

package/dist/hooks/use-coagent.mjs

@@ -3,8 +3,8 @@ import {
   startAgent,
   stopAgent,
   useCoAgent
-} from "../chunk-ADIITPD2.mjs";
-import "../chunk-EXU7GWLC.mjs";
+} from "../chunk-UHEKBL6C.mjs";
+import "../chunk-ENLPADPJ.mjs";
 import "../chunk-WOGURSAL.mjs";
 import "../chunk-SJJNFYGQ.mjs";
 import "../chunk-JDEWNLNP.mjs";
@@ -14,7 +14,7 @@ import "../chunk-PMAFHQ7P.mjs";
 import "../chunk-57K2ZJ5F.mjs";
 import "../chunk-YPSGKPDA.mjs";
 import "../chunk-DCTJZ742.mjs";
-import "../chunk-G7LYGERN.mjs";
+import "../chunk-IF2M6XIO.mjs";
 import "../chunk-4CEQJ2X6.mjs";
 import "../chunk-3OQM3NEK.mjs";
 import "../chunk-O7ARI5CV.mjs";

package/dist/hooks/use-copilot-chat.js

@@ -962,21 +962,35 @@ function useChat(options) {
     [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
   );
   const reload = useAsyncCallback(
-    (messageId) => __async(this, null, function* () {
+    (reloadMessageId) => __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`);
+      const reloadMessageIndex = messages.findIndex((msg) => msg.id === reloadMessageId);
+      if (reloadMessageIndex === -1) {
+        console.warn(`Message with id ${reloadMessageId} 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);
+      const reloadMessageRole = messages[reloadMessageIndex].role;
+      if (reloadMessageRole !== import_runtime_client_gql3.MessageRole.Assistant) {
+        console.warn(`Regenerate cannot be performed on ${reloadMessageRole} role`);
+        return;
       }
-      setMessages(newMessages);
-      return runChatCompletionAndHandleFunctionCall(newMessages);
+      let historyCutoff = [];
+      if (messages.length > 2) {
+        const lastUserMessageBeforeRegenerate = messages.slice(0, reloadMessageIndex).reverse().find(
+          (msg) => (
+            // @ts-expect-error -- message has role
+            msg.role === import_runtime_client_gql3.MessageRole.User
+          )
+        );
+        const indexOfLastUserMessageBeforeRegenerate = messages.findIndex(
+          (msg) => msg.id === lastUserMessageBeforeRegenerate.id
+        );
+        historyCutoff = messages.slice(0, indexOfLastUserMessageBeforeRegenerate + 1);
+      }
+      setMessages(historyCutoff);
+      return runChatCompletionAndHandleFunctionCall(historyCutoff);
     }),
     [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
   );