@copilotkit/react-core 1.4.2-tyler-reset-chat.1 → 1.4.2

package/dist/hooks/use-coagent.js

@@ -79,9 +79,6 @@ var __async = (__this, __arguments, generator) => {
 // src/hooks/use-coagent.ts
 var use_coagent_exports = {};
 __export(use_coagent_exports, {
-  runAgent: () => runAgent,
-  startAgent: () => startAgent,
-  stopAgent: () => stopAgent,
   useCoAgent: () => useCoAgent
 });
 module.exports = __toCommonJS(use_coagent_exports);
@@ -138,15 +135,7 @@ var emptyCopilotContext = {
   },
   agentSession: null,
   setAgentSession: () => {
-  },
-  agentLock: null,
-  threadId: null,
-  setThreadId: () => {
-  },
-  runId: null,
-  setRunId: () => {
-  },
-  chatAbortControllerRef: { current: null }
+  }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -232,23 +221,17 @@ function useChat(options) {
     setCoagentStates,
     coagentStates,
     agentSession,
-    setAgentSession,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   } = options;
+  const abortControllerRef = (0, import_react5.useRef)();
+  const threadIdRef = (0, import_react5.useRef)(null);
+  const runIdRef = (0, import_react5.useRef)(null);
   const { addGraphQLErrorsToast } = useToast();
   const runChatCompletionRef = (0, import_react5.useRef)();
   const coagentStatesRef = (0, import_react5.useRef)(coagentStates);
   coagentStatesRef.current = coagentStates;
   const agentSessionRef = (0, import_react5.useRef)(agentSession);
   agentSessionRef.current = agentSession;
-  const threadIdRef = (0, import_react5.useRef)(threadId);
-  threadIdRef.current = threadId;
-  const runIdRef = (0, import_react5.useRef)(runId);
-  runIdRef.current = runId;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = useCopilotRuntimeClient({
@@ -266,7 +249,8 @@ function useChat(options) {
         role: import_runtime_client_gql2.Role.Assistant
       })
     ];
-    chatAbortControllerRef.current = new AbortController();
+    const abortController = new AbortController();
+    abortControllerRef.current = abortController;
     setMessages([...previousMessages, ...newMessages]);
     const systemMessage = makeSystemMessageCallback();
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
@@ -319,7 +303,7 @@ function useChat(options) {
           }))
         }),
         properties: copilotConfig.properties,
-        signal: (_d = chatAbortControllerRef.current) == null ? void 0 : _d.signal
+        signal: (_d = abortControllerRef.current) == null ? void 0 : _d.signal
       })
     );
     const guardrailsEnabled = ((_g = (_f = (_e = copilotConfig.cloud) == null ? void 0 : _e.guardrails) == null ? void 0 : _f.input) == null ? void 0 : _g.restrictToTopic.enabled) || false;
@@ -338,9 +322,6 @@ function useChat(options) {
           break;
         }
         if (done) {
-          if (chatAbortControllerRef.current.signal.aborted) {
-            return newMessages.slice();
-          }
           break;
         }
         if (!(value == null ? void 0 : value.generateCopilotResponse)) {
@@ -348,8 +329,6 @@ function useChat(options) {
         }
         threadIdRef.current = value.generateCopilotResponse.threadId || null;
         runIdRef.current = value.generateCopilotResponse.runId || null;
-        setThreadId(threadIdRef.current);
-        setRunId(runIdRef.current);
         const messages2 = (0, import_runtime_client_gql2.convertGqlOutputToMessages)(
           (0, import_runtime_client_gql2.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
         );
@@ -378,22 +357,12 @@ function useChat(options) {
                   if (action) {
                     followUp = action.followUp;
                   }
-                  const result = yield Promise.race([
-                    onFunctionCall({
-                      messages: previousMessages,
-                      name: message.name,
-                      args: message.arguments
-                    }),
-                    new Promise((_, reject) => {
-                      var _a2;
-                      return (_a2 = chatAbortControllerRef.current) == null ? void 0 : _a2.signal.addEventListener("abort", () => reject(new Error("Operation was aborted")));
-                    })
-                  ]);
-                  if (chatAbortControllerRef.current.signal.aborted) {
-                    actionResults[message.id] = "";
-                  } else {
-                    actionResults[message.id] = result;
-                  }
+                  const result = yield onFunctionCall({
+                    messages: previousMessages,
+                    name: message.name,
+                    args: message.arguments
+                  });
+                  actionResults[message.id] = result;
                 } catch (e) {
                   actionResults[message.id] = `Failed to execute action ${message.name}`;
                   console.error(`Failed to execute action ${message.name}: ${e}`);
@@ -488,12 +457,13 @@ function useChat(options) {
   });
   const stop = () => {
     var _a;
-    (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort("Stop was called");
+    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
   };
   return {
     append,
     reload,
-    stop
+    stop,
+    runChatCompletion: () => runChatCompletionRef.current(messages)
   };
 }
 
@@ -523,13 +493,7 @@ function useCopilotChat(_a = {}) {
     setCoagentStates,
     coAgentStateRenders,
     agentSession,
-    setAgentSession,
-    agentLock,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -565,7 +529,7 @@ function useCopilotChat(_a = {}) {
     }),
     [coAgentStateRenders]
   );
-  const { append, reload, stop } = useChat(__spreadProps(__spreadValues({}, options), {
+  const { append, reload, stop, runChatCompletion } = useChat(__spreadProps(__spreadValues({}, options), {
     actions: Object.values(actions),
     copilotConfig: copilotApiConfig,
     initialMessages: options.initialMessages || [],
@@ -579,12 +543,7 @@ function useCopilotChat(_a = {}) {
     coagentStates,
     setCoagentStates,
     agentSession,
-    setAgentSession,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = (0, import_react7.useCallback)(
@@ -615,32 +574,18 @@ function useCopilotChat(_a = {}) {
     },
     [latestSetMessages]
   );
-  const reset = (0, import_react7.useCallback)(() => {
-    latestStopFunc();
-    setMessages([]);
-    setThreadId(null);
-    setRunId(null);
-    setCoagentStates({});
-    let initialAgentSession = null;
-    if (agentLock) {
-      initialAgentSession = {
-        agentName: agentLock
-      };
-    }
-    setAgentSession(initialAgentSession);
-  }, [latestStopFunc, setMessages, setThreadId, setCoagentStates, setAgentSession, agentLock]);
-  const latestReset = useUpdatedRef(reset);
-  const latestResetFunc = (0, import_react7.useCallback)(() => {
-    return latestReset.current();
-  }, [latestReset]);
+  const latestRunChatCompletion = useUpdatedRef(runChatCompletion);
+  const latestRunChatCompletionFunc = (0, import_react7.useCallback)(() => {
+    return latestRunChatCompletion.current();
+  }, [latestRunChatCompletion]);
   return {
     visibleMessages: messages,
     appendMessage: latestAppendFunc,
     setMessages: latestSetMessagesFunc,
     reloadMessages: latestReloadFunc,
     stopGeneration: latestStopFunc,
-    reset: latestResetFunc,
     deleteMessage: latestDeleteFunc,
+    runChatCompletion: latestRunChatCompletionFunc,
     isLoading
   };
 }
@@ -676,6 +621,7 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-coagent.ts
+var import_react_dom2 = require("react-dom");
 function useCoAgent(options) {
   const isExternalStateManagement = (options2) => {
     return "state" in options2 && "setState" in options2;
@@ -688,7 +634,7 @@ function useCoAgent(options) {
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
   const { coagentStates, setCoagentStates } = context;
-  const { appendMessage } = useCopilotChat();
+  const { appendMessage, runChatCompletion } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
     if (coagentStates2[name2]) {
       return coagentStates2[name2];
@@ -705,13 +651,15 @@ function useCoAgent(options) {
     }
   };
   const setState = (newState) => {
-    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
-        })
+    (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
+          })
+        });
       });
     });
   };
@@ -723,11 +671,7 @@ function useCoAgent(options) {
     } else if (coagentStates[name] === void 0) {
       setState(options.initialState === void 0 ? {} : options.initialState);
     }
-  }, [
-    isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0,
-    // reset initialstate on reset
-    coagentStates[name] === void 0
-  ]);
+  }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
   return {
     name,
     nodeName: coagentState.nodeName,
@@ -741,7 +685,7 @@ function useCoAgent(options) {
       stopAgent(name, context);
     },
     run: (hint) => {
-      return runAgent(name, context, appendMessage, hint);
+      return runAgent(name, context, appendMessage, runChatCompletion, hint);
     }
   };
 }
@@ -755,22 +699,11 @@ function stopAgent(name, context) {
   const { agentSession, setAgentSession } = context;
   if (agentSession && agentSession.agentName === name) {
     setAgentSession(null);
-    context.setCoagentStates((prevAgentStates) => {
-      return __spreadProps(__spreadValues({}, prevAgentStates), {
-        [name]: __spreadProps(__spreadValues({}, prevAgentStates[name]), {
-          running: false,
-          active: false,
-          threadId: void 0,
-          nodeName: void 0,
-          runId: void 0
-        })
-      });
-    });
   } else {
     console.warn(`No agent session found for ${name}`);
   }
 }
-function runAgent(name, context, appendMessage, hint) {
+function runAgent(name, context, appendMessage, runChatCompletion, hint) {
   return __async(this, null, function* () {
     var _a, _b;
     const { agentSession, setAgentSession } = context;
@@ -791,15 +724,16 @@ function runAgent(name, context, appendMessage, hint) {
       const hintMessage = hint({ previousState, currentState: state });
       if (hintMessage) {
         yield appendMessage(hintMessage);
+      } else {
+        yield runChatCompletion();
       }
+    } else {
+      yield runChatCompletion();
     }
   });
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  runAgent,
-  startAgent,
-  stopAgent,
   useCoAgent
 });
 //# sourceMappingURL=use-coagent.js.map