@copilotkit/react-core 1.4.2 → 1.4.3-pre.0

package/dist/index.js

@@ -152,6 +152,9 @@ var emptyCopilotContext = {
   coagentStates: {},
   setCoagentStates: () => {
   },
+  coagentStatesRef: { current: {} },
+  setCoagentStatesWithRef: () => {
+  },
   agentSession: null,
   setAgentSession: () => {
   }
@@ -946,6 +949,17 @@ ${nonDocumentStrings}`;
     });
   };
   const [coagentStates, setCoagentStates] = (0, import_react8.useState)({});
+  const coagentStatesRef = (0, import_react8.useRef)({});
+  const setCoagentStatesWithRef = (0, import_react8.useCallback)(
+    (value) => {
+      const newValue = typeof value === "function" ? value(coagentStatesRef.current) : value;
+      coagentStatesRef.current = newValue;
+      setCoagentStates((prev) => {
+        return newValue;
+      });
+    },
+    []
+  );
   let initialAgentSession = null;
   if (props.agent) {
     initialAgentSession = {
@@ -983,6 +997,8 @@ ${nonDocumentStrings}`;
         showDevConsole,
         coagentStates,
         setCoagentStates,
+        coagentStatesRef,
+        setCoagentStatesWithRef,
         agentSession,
         setAgentSession,
         runtimeClient
@@ -1038,8 +1054,8 @@ function useChat(options) {
     actions,
     onFunctionCall,
     onCoAgentStateRender,
-    setCoagentStates,
-    coagentStates,
+    setCoagentStatesWithRef,
+    coagentStatesRef,
     agentSession,
     setAgentSession
   } = options;
@@ -1048,8 +1064,6 @@ function useChat(options) {
   const runIdRef = (0, import_react9.useRef)(null);
   const { addGraphQLErrorsToast } = useToast();
   const runChatCompletionRef = (0, import_react9.useRef)();
-  const coagentStatesRef = (0, import_react9.useRef)(coagentStates);
-  coagentStatesRef.current = coagentStates;
   const agentSessionRef = (0, import_react9.useRef)(agentSession);
   agentSessionRef.current = agentSession;
   const publicApiKey = copilotConfig.publicApiKey;
@@ -1210,7 +1224,7 @@ function useChat(options) {
           }
           const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
           if (lastAgentStateMessage) {
-            setCoagentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
+            setCoagentStatesWithRef((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
               [lastAgentStateMessage.agentName]: {
                 name: lastAgentStateMessage.agentName,
                 state: lastAgentStateMessage.state,
@@ -1302,8 +1316,8 @@ function useCopilotChat(_a = {}) {
     setIsLoading,
     chatInstructions,
     actions,
-    coagentStates,
-    setCoagentStates,
+    coagentStatesRef,
+    setCoagentStatesWithRef,
     coAgentStateRenders,
     agentSession,
     setAgentSession
@@ -1353,8 +1367,8 @@ function useCopilotChat(_a = {}) {
     makeSystemMessageCallback,
     isLoading,
     setIsLoading,
-    coagentStates,
-    setCoagentStates,
+    coagentStatesRef,
+    setCoagentStatesWithRef,
     agentSession,
     setAgentSession
   }));
@@ -1590,7 +1604,6 @@ function useCopilotReadable({ description, value, parentId, categories, convert
 
 // src/hooks/use-coagent.ts
 var import_react15 = require("react");
-var import_react_dom2 = require("react-dom");
 function useCoAgent(options) {
   const isExternalStateManagement = (options2) => {
     return "state" in options2 && "setState" in options2;
@@ -1602,7 +1615,7 @@ function useCoAgent(options) {
   const generalContext = useCopilotContext();
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
-  const { coagentStates, setCoagentStates } = context;
+  const { coagentStates, coagentStatesRef, setCoagentStatesWithRef } = context;
   const { appendMessage, runChatCompletion } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
     if (coagentStates2[name2]) {
@@ -1620,17 +1633,13 @@ function useCoAgent(options) {
     }
   };
   const setState = (newState) => {
-    (0, import_react_dom2.flushSync)(() => {
-      setCoagentStates((prevAgentStates) => {
-        let coagentState2 = getCoagentState(prevAgentStates, name);
-        const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
-        return __spreadProps(__spreadValues({}, prevAgentStates), {
-          [name]: __spreadProps(__spreadValues({}, coagentState2), {
-            state: updatedState
-          })
-        });
-      });
-    });
+    let coagentState2 = getCoagentState(coagentStatesRef.current || {}, name);
+    const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
+    setCoagentStatesWithRef(__spreadProps(__spreadValues({}, coagentStatesRef.current), {
+      [name]: __spreadProps(__spreadValues({}, coagentState2), {
+        state: updatedState
+      })
+    }));
   };
   const coagentState = getCoagentState(coagentStates, name);
   const state = isExternalStateManagement(options) ? options.state : coagentState.state;
@@ -1688,7 +1697,7 @@ function runAgent(name, context, appendMessage, runChatCompletion, hint) {
         previousState = message.state;
       }
     }
-    let state = ((_b = (_a = context.coagentStates) == null ? void 0 : _a[name]) == null ? void 0 : _b.state) || {};
+    let state = ((_b = (_a = context.coagentStatesRef.current) == null ? void 0 : _a[name]) == null ? void 0 : _b.state) || {};
     if (hint) {
       const hintMessage = hint({ previousState, currentState: state });
       if (hintMessage) {