@copilotkit/react-core 1.9.2-next.9 → 1.9.3-next.0

package/dist/index.js

@@ -184,7 +184,7 @@ var emptyCopilotContext = {
   langGraphInterruptAction: null,
   setLangGraphInterruptAction: () => null,
   removeLangGraphInterruptAction: () => null,
-  onTrace: void 0
+  onError: void 0
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -738,11 +738,11 @@ function CopilotMessages({ children }) {
   const lastLoadedThreadId = (0, import_react6.useRef)();
   const lastLoadedAgentName = (0, import_react6.useRef)();
   const lastLoadedMessages = (0, import_react6.useRef)();
-  const { threadId, agentSession, runtimeClient, showDevConsole, onTrace, copilotApiConfig } = useCopilotContext();
+  const { threadId, agentSession, runtimeClient, showDevConsole, onError, copilotApiConfig } = useCopilotContext();
   const { setBannerError } = useToast();
   const traceUIError = (0, import_react6.useCallback)(
     (error, originalError) => __async(this, null, function* () {
-      if (!onTrace || !copilotApiConfig.publicApiKey)
+      if (!onError || !copilotApiConfig.publicApiKey)
         return;
       try {
         const traceEvent = {
@@ -763,12 +763,12 @@ function CopilotMessages({ children }) {
           },
           error
         };
-        yield onTrace(traceEvent);
+        yield onError(traceEvent);
       } catch (traceError) {
-        console.error("Error in CopilotMessages onTrace handler:", traceError);
+        console.error("Error in CopilotMessages onError handler:", traceError);
       }
     }),
-    [onTrace, copilotApiConfig.publicApiKey, copilotApiConfig.chatApiEndpoint]
+    [onError, copilotApiConfig.publicApiKey, copilotApiConfig.chatApiEndpoint]
   );
   const createStructuredError2 = (gqlError) => {
     const extensions = gqlError.extensions;
@@ -873,7 +873,7 @@ function CopilotMessages({ children }) {
       }
     });
     void fetchMessages();
-  }, [threadId, agentSession == null ? void 0 : agentSession.agentName, runtimeClient]);
+  }, [threadId, agentSession == null ? void 0 : agentSession.agentName]);
   const memoizedChildren = (0, import_react6.useMemo)(() => children, [children]);
   return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(
     CopilotMessagesContext.Provider,
@@ -893,13 +893,13 @@ var import_react7 = require("react");
 var import_shared5 = require("@copilotkit/shared");
 var useCopilotRuntimeClient = (options) => {
   const { setBannerError } = useToast();
-  const _a = options, { showDevConsole, onTrace } = _a, runtimeOptions = __objRest(_a, ["showDevConsole", "onTrace"]);
+  const _a = options, { showDevConsole, onError } = _a, runtimeOptions = __objRest(_a, ["showDevConsole", "onError"]);
   const lastStructuredErrorRef = (0, import_react7.useRef)(null);
   const traceUIError = (error, originalError) => __async(void 0, null, function* () {
-    if (!onTrace || !runtimeOptions.publicApiKey)
+    if (!onError || !runtimeOptions.publicApiKey)
       return;
     try {
-      const traceEvent = {
+      const errorEvent = {
         type: "error",
         timestamp: Date.now(),
         context: {
@@ -917,9 +917,9 @@ var useCopilotRuntimeClient = (options) => {
         },
         error
       };
-      yield onTrace(traceEvent);
-    } catch (traceError) {
-      console.error("Error in onTrace handler:", traceError);
+      yield onError(errorEvent);
+    } catch (error2) {
+      console.error("Error in onError handler:", error2);
     }
   });
   const runtimeClient = (0, import_react7.useMemo)(() => {
@@ -983,7 +983,7 @@ var useCopilotRuntimeClient = (options) => {
         setBannerError(warningError);
       }
     }));
-  }, [runtimeOptions, setBannerError, showDevConsole, onTrace]);
+  }, [runtimeOptions, setBannerError, showDevConsole, onError]);
   return runtimeClient;
 };
 function createStructuredError(gqlError) {
@@ -1899,18 +1899,24 @@ ${nonDocumentStrings}`;
     headers,
     credentials: copilotApiConfig.credentials,
     showDevConsole: (_b = props.showDevConsole) != null ? _b : false,
-    onTrace: props.onTrace
+    onError: props.onError
   });
   const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = (0, import_react10.useState)({});
-  const addChatSuggestionConfiguration = (id, suggestion) => {
-    setChatSuggestionConfiguration((prev) => __spreadProps(__spreadValues({}, prev), { [id]: suggestion }));
-  };
-  const removeChatSuggestionConfiguration = (id) => {
-    setChatSuggestionConfiguration((prev) => {
-      const _a2 = prev, { [id]: _ } = _a2, rest = __objRest(_a2, [__restKey(id)]);
-      return rest;
-    });
-  };
+  const addChatSuggestionConfiguration = (0, import_react10.useCallback)(
+    (id, suggestion) => {
+      setChatSuggestionConfiguration((prev) => __spreadProps(__spreadValues({}, prev), { [id]: suggestion }));
+    },
+    [setChatSuggestionConfiguration]
+  );
+  const removeChatSuggestionConfiguration = (0, import_react10.useCallback)(
+    (id) => {
+      setChatSuggestionConfiguration((prev) => {
+        const _a2 = prev, { [id]: _ } = _a2, rest = __objRest(_a2, [__restKey(id)]);
+        return rest;
+      });
+    },
+    [setChatSuggestionConfiguration]
+  );
   const [availableAgents, setAvailableAgents] = (0, import_react10.useState)([]);
   const [coagentStates, setCoagentStates] = (0, import_react10.useState)({});
   const coagentStatesRef = (0, import_react10.useRef)({});
@@ -1990,6 +1996,41 @@ ${nonDocumentStrings}`;
     setLangGraphInterruptAction(null);
   }, []);
   const memoizedChildren = (0, import_react10.useMemo)(() => children, [children]);
+  const agentLock = (0, import_react10.useMemo)(() => {
+    var _a2;
+    return (_a2 = props.agent) != null ? _a2 : null;
+  }, [props.agent]);
+  const forwardedParameters = (0, import_react10.useMemo)(
+    () => {
+      var _a2;
+      return (_a2 = props.forwardedParameters) != null ? _a2 : {};
+    },
+    [props.forwardedParameters]
+  );
+  const updateExtensions = (0, import_react10.useCallback)(
+    (newExtensions) => {
+      setExtensions((prev) => {
+        const resolved = typeof newExtensions === "function" ? newExtensions(prev) : newExtensions;
+        const isSameLength = Object.keys(resolved).length === Object.keys(prev).length;
+        const isEqual = isSameLength && // @ts-ignore
+        Object.entries(resolved).every(([key, value]) => prev[key] === value);
+        return isEqual ? prev : resolved;
+      });
+    },
+    [setExtensions]
+  );
+  const updateAuthStates = (0, import_react10.useCallback)(
+    (newAuthStates) => {
+      setAuthStates((prev) => {
+        const resolved = typeof newAuthStates === "function" ? newAuthStates(prev) : newAuthStates;
+        const isSameLength = Object.keys(resolved).length === Object.keys(prev).length;
+        const isEqual = isSameLength && // @ts-ignore
+        Object.entries(resolved).every(([key, value]) => prev[key] === value);
+        return isEqual ? prev : resolved;
+      });
+    },
+    [setAuthStates]
+  );
   return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
     CopilotContext.Provider,
     {
@@ -2027,8 +2068,8 @@ ${nonDocumentStrings}`;
         agentSession,
         setAgentSession,
         runtimeClient,
-        forwardedParameters: props.forwardedParameters || {},
-        agentLock: props.agent || null,
+        forwardedParameters,
+        agentLock,
         threadId: internalThreadId,
         setThreadId,
         runId,
@@ -2037,13 +2078,13 @@ ${nonDocumentStrings}`;
         availableAgents,
         authConfig_c: props.authConfig_c,
         authStates_c: authStates,
-        setAuthStates_c: setAuthStates,
+        setAuthStates_c: updateAuthStates,
         extensions,
-        setExtensions,
+        setExtensions: updateExtensions,
         langGraphInterruptAction,
         setLangGraphInterruptAction,
         removeLangGraphInterruptAction,
-        onTrace: props.onTrace
+        onError: props.onError
       },
       children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(CopilotMessages, { children: memoizedChildren })
     }
@@ -2096,6 +2137,7 @@ var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
 var import_react11 = require("react");
+var import_react_dom2 = require("react-dom");
 var import_shared12 = require("@copilotkit/shared");
 var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
 
@@ -2155,9 +2197,9 @@ function useChat(options) {
   const runChatCompletionRef = (0, import_react11.useRef)();
   const addErrorToast = useErrorToast();
   const { setBannerError } = useToast();
-  const { onTrace } = useCopilotContext();
+  const { onError } = useCopilotContext();
   const traceUIError = (error, originalError) => __async(this, null, function* () {
-    if (!onTrace || !(copilotConfig == null ? void 0 : copilotConfig.publicApiKey))
+    if (!onError || !(copilotConfig == null ? void 0 : copilotConfig.publicApiKey))
       return;
     try {
       const traceEvent = {
@@ -2178,9 +2220,9 @@ function useChat(options) {
         },
         error
       };
-      yield onTrace(traceEvent);
+      yield onError(traceEvent);
     } catch (traceError) {
-      console.error("Error in use-chat onTrace handler:", traceError);
+      console.error("Error in use-chat onError handler:", traceError);
     }
   });
   const agentSessionRef = (0, import_react11.useRef)(agentSession);
@@ -2402,17 +2444,22 @@ function useChat(options) {
                   lastAgentStateMessage.state.messages
                 );
               }
-              setCoagentStatesWithRef((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
-                [lastAgentStateMessage.agentName]: {
-                  name: lastAgentStateMessage.agentName,
-                  state: lastAgentStateMessage.state,
-                  running: lastAgentStateMessage.running,
-                  active: lastAgentStateMessage.active,
-                  threadId: lastAgentStateMessage.threadId,
-                  nodeName: lastAgentStateMessage.nodeName,
-                  runId: lastAgentStateMessage.runId
-                }
-              }));
+              setCoagentStatesWithRef((prevAgentStates) => {
+                var _a2;
+                return __spreadProps(__spreadValues({}, prevAgentStates), {
+                  [lastAgentStateMessage.agentName]: {
+                    name: lastAgentStateMessage.agentName,
+                    state: lastAgentStateMessage.state,
+                    running: lastAgentStateMessage.running,
+                    active: lastAgentStateMessage.active,
+                    threadId: lastAgentStateMessage.threadId,
+                    nodeName: lastAgentStateMessage.nodeName,
+                    runId: lastAgentStateMessage.runId,
+                    // Preserve existing config from previous state
+                    config: (_a2 = prevAgentStates[lastAgentStateMessage.agentName]) == null ? void 0 : _a2.config
+                  }
+                });
+              });
               if (lastAgentStateMessage.running) {
                 setAgentSession({
                   threadId: lastAgentStateMessage.threadId,
@@ -2442,6 +2489,39 @@ function useChat(options) {
           newMessages
         );
         let didExecuteAction = false;
+        const executeActionFromMessage = (currentAction, actionMessage) => __async(this, null, function* () {
+          var _a2;
+          const isInterruptAction = interruptMessages.find((m) => m.id === actionMessage.id);
+          followUp = (_a2 = currentAction == null ? void 0 : currentAction.followUp) != null ? _a2 : !isInterruptAction;
+          if (currentAction == null ? void 0 : currentAction._setActivatingMessageId) {
+            currentAction._setActivatingMessageId(actionMessage.id);
+          }
+          const resultMessage = yield executeAction({
+            onFunctionCall,
+            message: actionMessage,
+            chatAbortControllerRef,
+            onError: (error) => {
+              addErrorToast([error]);
+              console.error(`Failed to execute action ${actionMessage.name}: ${error}`);
+            },
+            setMessages,
+            getFinalMessages: () => finalMessages,
+            isRenderAndWait: (currentAction == null ? void 0 : currentAction._isRenderAndWait) || false
+          });
+          didExecuteAction = true;
+          const messageIndex = finalMessages.findIndex((msg) => msg.id === actionMessage.id);
+          finalMessages.splice(messageIndex + 1, 0, resultMessage);
+          if (currentAction == null ? void 0 : currentAction._isRenderAndWait) {
+            const messagesForImmediateUpdate = [...finalMessages];
+            (0, import_react_dom2.flushSync)(() => {
+              setMessages(messagesForImmediateUpdate);
+            });
+          }
+          if (currentAction == null ? void 0 : currentAction._setActivatingMessageId) {
+            currentAction._setActivatingMessageId(null);
+          }
+          return resultMessage;
+        });
         if (onFunctionCall) {
           const lastMessages = [];
           for (let i = finalMessages.length - 1; i >= 0; i--) {
@@ -2458,37 +2538,28 @@ function useChat(options) {
               (action2) => action2.name === message.name
             );
             const currentResultMessagePairedFeAction = message.isResultMessage() ? getPairedFeAction(actions, message) : null;
-            const executeActionFromMessage = (action2, message2) => __async(this, null, function* () {
-              var _a2;
-              const isInterruptAction = interruptMessages.find((m) => m.id === message2.id);
-              followUp = (_a2 = action2 == null ? void 0 : action2.followUp) != null ? _a2 : !isInterruptAction;
-              const resultMessage = yield executeAction({
-                onFunctionCall,
-                previousMessages,
-                message: message2,
-                chatAbortControllerRef,
-                onError: (error) => {
-                  addErrorToast([error]);
-                  console.error(`Failed to execute action ${message2.name}: ${error}`);
-                }
-              });
-              didExecuteAction = true;
-              const messageIndex = finalMessages.findIndex((msg) => msg.id === message2.id);
-              finalMessages.splice(messageIndex + 1, 0, resultMessage);
-              return resultMessage;
-            });
             if (action && message.isActionExecutionMessage()) {
-              const resultMessage = yield executeActionFromMessage(action, message);
-              const pairedFeAction = getPairedFeAction(actions, resultMessage);
-              if (pairedFeAction) {
-                const newExecutionMessage = new import_runtime_client_gql6.ActionExecutionMessage({
-                  name: pairedFeAction.name,
-                  arguments: (0, import_shared12.parseJson)(resultMessage.result, resultMessage.result),
-                  status: message.status,
-                  createdAt: message.createdAt,
-                  parentMessageId: message.parentMessageId
-                });
-                yield executeActionFromMessage(pairedFeAction, newExecutionMessage);
+              const isRenderAndWaitAction = (action == null ? void 0 : action._isRenderAndWait) || false;
+              const alreadyProcessed = isRenderAndWaitAction && finalMessages.some(
+                (fm) => fm.isResultMessage() && fm.actionExecutionId === message.id
+              );
+              if (alreadyProcessed) {
+              } else {
+                const resultMessage = yield executeActionFromMessage(
+                  action,
+                  message
+                );
+                const pairedFeAction = getPairedFeAction(actions, resultMessage);
+                if (pairedFeAction) {
+                  const newExecutionMessage = new import_runtime_client_gql6.ActionExecutionMessage({
+                    name: pairedFeAction.name,
+                    arguments: (0, import_shared12.parseJson)(resultMessage.result, resultMessage.result),
+                    status: message.status,
+                    createdAt: message.createdAt,
+                    parentMessageId: message.parentMessageId
+                  });
+                  yield executeActionFromMessage(pairedFeAction, newExecutionMessage);
+                }
               }
             } else if (message.isResultMessage() && currentResultMessagePairedFeAction) {
               const newExecutionMessage = new import_runtime_client_gql6.ActionExecutionMessage({
@@ -2506,13 +2577,9 @@ function useChat(options) {
           }
           setMessages(finalMessages);
         }
-        if (
-          // if followUp is not explicitly false
-          followUp !== false && // and we executed an action
-          (didExecuteAction || // the last message is a server side result
-          !isAgentRun && finalMessages.length && finalMessages[finalMessages.length - 1].isResultMessage()) && // the user did not stop generation
-          !((_r = chatAbortControllerRef.current) == null ? void 0 : _r.signal.aborted)
-        ) {
+        if (followUp !== false && (didExecuteAction || // the last message is a server side result
+        !isAgentRun && finalMessages.length && finalMessages[finalMessages.length - 1].isResultMessage()) && // the user did not stop generation
+        !((_r = chatAbortControllerRef.current) == null ? void 0 : _r.signal.aborted)) {
           yield new Promise((resolve) => setTimeout(resolve, 10));
           return yield runChatCompletionRef.current(finalMessages);
         } else if ((_s = chatAbortControllerRef.current) == null ? void 0 : _s.signal.aborted) {
@@ -2620,21 +2687,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_gql6.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_gql6.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]
   );
@@ -2669,20 +2750,31 @@ function constructFinalMessages(syncedMessages, previousMessages, newMessages) {
 function executeAction(_0) {
   return __async(this, arguments, function* ({
     onFunctionCall,
-    previousMessages,
     message,
     chatAbortControllerRef,
-    onError
+    onError,
+    setMessages,
+    getFinalMessages,
+    isRenderAndWait
   }) {
     let result;
     let error = null;
+    const currentMessagesForHandler = getFinalMessages();
+    const handlerReturnedPromise = onFunctionCall({
+      messages: currentMessagesForHandler,
+      name: message.name,
+      args: message.arguments
+    });
+    if (isRenderAndWait) {
+      const currentMessagesForRender = getFinalMessages();
+      (0, import_react_dom2.flushSync)(() => {
+        setMessages([...currentMessagesForRender]);
+      });
+    }
     try {
       result = yield Promise.race([
-        onFunctionCall({
-          messages: previousMessages,
-          name: message.name,
-          args: message.arguments
-        }),
+        handlerReturnedPromise,
+        // Await the promise returned by the handler
         new Promise(
           (resolve) => {
             var _a;
@@ -2949,6 +3041,7 @@ function useCopilotAction(action, dependencies) {
   const { setAction, removeAction, actions, chatComponentsCache } = useCopilotContext();
   const idRef = (0, import_react13.useRef)((0, import_shared13.randomId)());
   const renderAndWaitRef = (0, import_react13.useRef)(null);
+  const activatingMessageIdRef = (0, import_react13.useRef)(null);
   const { addToast } = useToast();
   action = __spreadValues({}, action);
   if (
@@ -2956,30 +3049,48 @@ function useCopilotAction(action, dependencies) {
     isFrontendAction(action) && // check if renderAndWaitForResponse is set
     (action.renderAndWait || action.renderAndWaitForResponse)
   ) {
+    action._isRenderAndWait = true;
     const renderAndWait = action.renderAndWait || action.renderAndWaitForResponse;
     action.renderAndWait = void 0;
     action.renderAndWaitForResponse = void 0;
+    action._setActivatingMessageId = (id) => {
+      activatingMessageIdRef.current = id;
+    };
     action.handler = useAsyncCallback(() => __async(this, null, function* () {
+      const currentActivatingId = activatingMessageIdRef.current;
       let resolve;
       let reject;
       const promise = new Promise((resolvePromise, rejectPromise) => {
         resolve = resolvePromise;
         reject = rejectPromise;
       });
-      renderAndWaitRef.current = { promise, resolve, reject };
-      return yield promise;
+      renderAndWaitRef.current = {
+        promise,
+        resolve,
+        reject,
+        messageId: currentActivatingId
+      };
+      const result = yield promise;
+      return result;
     }), []);
     action.render = (props) => {
+      const currentRenderMessageId = props.messageId;
       let status = props.status;
-      if (props.status === "executing" && !renderAndWaitRef.current) {
-        status = "inProgress";
+      if (props.status === "executing") {
+        if (!renderAndWaitRef.current || !renderAndWaitRef.current.promise) {
+          status = "inProgress";
+        } else if (renderAndWaitRef.current.messageId !== currentRenderMessageId && activatingMessageIdRef.current !== currentRenderMessageId) {
+          status = "inProgress";
+        }
       }
       const waitProps = {
         status,
         args: props.args,
         result: props.result,
-        handler: status === "executing" ? renderAndWaitRef.current.resolve : void 0,
-        respond: status === "executing" ? renderAndWaitRef.current.resolve : void 0
+        // handler and respond should only be provided if this is the truly active instance
+        // and its promise infrastructure is ready.
+        handler: status === "executing" && renderAndWaitRef.current && renderAndWaitRef.current.messageId === currentRenderMessageId ? renderAndWaitRef.current.resolve : void 0,
+        respond: status === "executing" && renderAndWaitRef.current && renderAndWaitRef.current.messageId === currentRenderMessageId ? renderAndWaitRef.current.resolve : void 0
       };
       const isNoArgsRenderWait = (_fn) => {
         var _a;
@@ -3248,6 +3359,32 @@ function useCoAgent(options) {
     // reset initialstate on reset
     coagentStates[name] === void 0
   ]);
+  (0, import_react17.useEffect)(() => {
+    const newConfig = options.config ? options.config : options.configurable ? { configurable: options.configurable } : void 0;
+    if (newConfig === void 0)
+      return;
+    setCoagentStatesWithRef((prev) => {
+      var _a;
+      const existing = (_a = prev[name]) != null ? _a : {
+        name,
+        state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
+        config: {},
+        running: false,
+        active: false,
+        threadId: void 0,
+        nodeName: void 0,
+        runId: void 0
+      };
+      if (JSON.stringify(existing.config) === JSON.stringify(newConfig)) {
+        return prev;
+      }
+      return __spreadProps(__spreadValues({}, prev), {
+        [name]: __spreadProps(__spreadValues({}, existing), {
+          config: newConfig
+        })
+      });
+    });
+  }, [JSON.stringify(options.config), JSON.stringify(options.configurable)]);
   const runAgentCallback = useAsyncCallback(
     (hint) => __async(this, null, function* () {
       yield runAgent(name, context, appendMessage, runChatCompletion, hint);