@copilotkit/react-core 1.6.0-next.6 → 1.6.0-next.8

package/dist/index.js

@@ -2460,9 +2460,6 @@ function useCoAgent(options) {
   const { addToast } = useToast();
   const lastLoadedThreadId = (0, import_react17.useRef)();
   const lastLoadedState = (0, import_react17.useRef)();
-  const isExternalStateManagement = (options2) => {
-    return "state" in options2 && "setState" in options2;
-  };
   const { name } = options;
   (0, import_react17.useEffect)(() => {
     if ((availableAgents == null ? void 0 : availableAgents.length) && !availableAgents.some((a) => a.name === name)) {
@@ -2471,44 +2468,27 @@ function useCoAgent(options) {
       addToast({ type: "warning", message });
     }
   }, [availableAgents]);
-  const isInternalStateManagementWithInitial = (options2) => {
-    return "initialState" in options2;
-  };
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
   const { coagentStates, coagentStatesRef, setCoagentStatesWithRef, threadId, copilotApiConfig } = context;
   const { appendMessage, runChatCompletion } = useCopilotChat();
-  const getCoagentState = (coagentStates2, name2) => {
-    var _a;
-    if (coagentStates2[name2]) {
-      return coagentStates2[name2];
-    } else {
-      return {
-        name: name2,
-        state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
-        configurable: (_a = options.configurable) != null ? _a : {},
-        running: false,
-        active: false,
-        threadId: void 0,
-        nodeName: void 0,
-        runId: void 0
-      };
-    }
-  };
   const runtimeClient = useCopilotRuntimeClient({
     url: copilotApiConfig.chatApiEndpoint,
     publicApiKey: copilotApiConfig.publicApiKey,
     credentials: copilotApiConfig.credentials
   });
-  const setState = (newState) => {
-    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 setState = (0, import_react17.useCallback)(
+    (newState) => {
+      let coagentState = getCoagentState({ coagentStates, name, options });
+      const updatedState = typeof newState === "function" ? newState(coagentState.state) : newState;
+      setCoagentStatesWithRef(__spreadProps(__spreadValues({}, coagentStatesRef.current), {
+        [name]: __spreadProps(__spreadValues({}, coagentState), {
+          state: updatedState
+        })
+      }));
+    },
+    [coagentStates, name]
+  );
   (0, import_react17.useEffect)(() => {
     const fetchAgentState = () => __async(this, null, function* () {
       var _a, _b, _c, _d;
@@ -2530,7 +2510,6 @@ function useCoAgent(options) {
     });
     void fetchAgentState();
   }, [threadId]);
-  const coagentState = getCoagentState(coagentStates, name);
   (0, import_react17.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
@@ -2548,17 +2527,20 @@ function useCoAgent(options) {
     }),
     [name, context, appendMessage, runChatCompletion]
   );
-  return {
-    name,
-    nodeName: coagentState.nodeName,
-    threadId: coagentState.threadId,
-    running: coagentState.running,
-    state: coagentState.state,
-    setState: isExternalStateManagement(options) ? options.setState : setState,
-    start: () => startAgent(name, context),
-    stop: () => stopAgent(name, context),
-    run: runAgentCallback
-  };
+  return (0, import_react17.useMemo)(() => {
+    const coagentState = getCoagentState({ coagentStates, name, options });
+    return {
+      name,
+      nodeName: coagentState.nodeName,
+      threadId: coagentState.threadId,
+      running: coagentState.running,
+      state: coagentState.state,
+      setState: isExternalStateManagement(options) ? options.setState : setState,
+      start: () => startAgent(name, context),
+      stop: () => stopAgent(name, context),
+      run: runAgentCallback
+    };
+  }, [name, coagentStates, options, setState, runAgentCallback]);
 }
 function startAgent(name, context) {
   const { setAgentSession } = context;
@@ -2614,6 +2596,33 @@ function runAgent(name, context, appendMessage, runChatCompletion, hint) {
     }
   });
 }
+var isExternalStateManagement = (options) => {
+  return "state" in options && "setState" in options;
+};
+var isInternalStateManagementWithInitial = (options) => {
+  return "initialState" in options;
+};
+var getCoagentState = ({
+  coagentStates,
+  name,
+  options
+}) => {
+  var _a;
+  if (coagentStates[name]) {
+    return coagentStates[name];
+  } else {
+    return {
+      name,
+      state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
+      configurable: (_a = options.configurable) != null ? _a : {},
+      running: false,
+      active: false,
+      threadId: void 0,
+      nodeName: void 0,
+      runId: void 0
+    };
+  }
+};
 
 // src/hooks/use-copilot-authenticated-action.ts
 var import_react18 = require("react");