@copilotkit/react-core 1.4.2-tyler-reset-chat.2 → 1.4.3-pre.0

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);
@@ -136,17 +133,12 @@ var emptyCopilotContext = {
   coagentStates: {},
   setCoagentStates: () => {
   },
+  coagentStatesRef: { current: {} },
+  setCoagentStatesWithRef: () => {
+  },
   agentSession: null,
   setAgentSession: () => {
-  },
-  agentLock: null,
-  threadId: null,
-  setThreadId: () => {
-  },
-  runId: null,
-  setRunId: () => {
-  },
-  chatAbortControllerRef: { current: null }
+  }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -229,26 +221,18 @@ function useChat(options) {
     actions,
     onFunctionCall,
     onCoAgentStateRender,
-    setCoagentStates,
-    coagentStates,
+    setCoagentStatesWithRef,
+    coagentStatesRef,
     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 +250,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 +304,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 +323,6 @@ function useChat(options) {
           break;
         }
         if (done) {
-          if (chatAbortControllerRef.current.signal.aborted) {
-            return [];
-          }
           break;
         }
         if (!(value == null ? void 0 : value.generateCopilotResponse)) {
@@ -348,8 +330,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 +358,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}`);
@@ -421,7 +391,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,
@@ -488,12 +458,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)
   };
 }
 
@@ -519,17 +490,11 @@ function useCopilotChat(_a = {}) {
     setIsLoading,
     chatInstructions,
     actions,
-    coagentStates,
-    setCoagentStates,
+    coagentStatesRef,
+    setCoagentStatesWithRef,
     coAgentStateRenders,
     agentSession,
-    setAgentSession,
-    agentLock,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -565,7 +530,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 || [],
@@ -576,15 +541,10 @@ function useCopilotChat(_a = {}) {
     makeSystemMessageCallback,
     isLoading,
     setIsLoading,
-    coagentStates,
-    setCoagentStates,
+    coagentStatesRef,
+    setCoagentStatesWithRef,
     agentSession,
-    setAgentSession,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = (0, import_react7.useCallback)(
@@ -615,32 +575,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
   };
 }
@@ -687,8 +633,8 @@ function useCoAgent(options) {
   const generalContext = useCopilotContext();
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
-  const { coagentStates, setCoagentStates } = context;
-  const { appendMessage } = useCopilotChat();
+  const { coagentStates, coagentStatesRef, setCoagentStatesWithRef } = context;
+  const { appendMessage, runChatCompletion } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
     if (coagentStates2[name2]) {
       return coagentStates2[name2];
@@ -705,15 +651,13 @@ 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
-        })
-      });
-    });
+    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;
@@ -723,11 +667,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 +681,7 @@ function useCoAgent(options) {
       stopAgent(name, context);
     },
     run: (hint) => {
-      return runAgent(name, context, appendMessage, hint);
+      return runAgent(name, context, appendMessage, runChatCompletion, hint);
     }
   };
 }
@@ -755,22 +695,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;
@@ -786,20 +715,21 @@ function runAgent(name, context, appendMessage, 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) {
         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