@copilotkit/react-core 1.3.16-mme-revert-rxjs-changes.10 → 1.3.16-mme-reset-chat.11

package/dist/index.js

@@ -87,6 +87,9 @@ __export(src_exports, {
   CopilotTask: () => CopilotTask,
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
+  runAgent: () => runAgent,
+  startAgent: () => startAgent,
+  stopAgent: () => stopAgent,
   useCoAgent: () => useCoAgent,
   useCoAgentStateRender: () => useCoAgentStateRender,
   useCopilotAction: () => useCopilotAction,
@@ -151,7 +154,15 @@ 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() {
@@ -565,6 +576,9 @@ ${nonDocumentStrings}`;
     };
   }
   const [agentSession, setAgentSession] = (0, import_react6.useState)(initialAgentSession);
+  const [threadId, setThreadId] = (0, import_react6.useState)(null);
+  const [runId, setRunId] = (0, import_react6.useState)(null);
+  const chatAbortControllerRef = (0, import_react6.useRef)(null);
   return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(
     CopilotContext.Provider,
     {
@@ -595,7 +609,13 @@ ${nonDocumentStrings}`;
         coagentStates,
         setCoagentStates,
         agentSession,
-        setAgentSession
+        setAgentSession,
+        agentLock: props.agent || null,
+        threadId,
+        setThreadId,
+        runId,
+        setRunId,
+        chatAbortControllerRef
       },
       children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(CopilotMessages, { children })
     }
@@ -651,16 +671,22 @@ function useChat(options) {
     setCoagentStates,
     coagentStates,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   } = options;
-  const abortControllerRef = (0, import_react7.useRef)();
-  const threadIdRef = (0, import_react7.useRef)(null);
-  const runIdRef = (0, import_react7.useRef)(null);
   const runChatCompletionRef = (0, import_react7.useRef)();
   const coagentStatesRef = (0, import_react7.useRef)(coagentStates);
   coagentStatesRef.current = coagentStates;
   const agentSessionRef = (0, import_react7.useRef)(agentSession);
   agentSessionRef.current = agentSession;
+  const threadIdRef = (0, import_react7.useRef)(threadId);
+  threadIdRef.current = threadId;
+  const runIdRef = (0, import_react7.useRef)(runId);
+  runIdRef.current = runId;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared4.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = new import_runtime_client_gql.CopilotRuntimeClient({
@@ -679,7 +705,7 @@ function useChat(options) {
       })
     ];
     const abortController = new AbortController();
-    abortControllerRef.current = abortController;
+    chatAbortControllerRef.current = abortController;
     setMessages([...previousMessages, ...newMessages]);
     const systemMessage = makeSystemMessageCallback();
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
@@ -719,7 +745,7 @@ function useChat(options) {
           }))
         }),
         properties: copilotConfig.properties,
-        signal: (_d = abortControllerRef.current) == null ? void 0 : _d.signal
+        signal: (_d = chatAbortControllerRef.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;
@@ -738,6 +764,8 @@ 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_gql.convertGqlOutputToMessages)(
           (0, import_runtime_client_gql.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
         );
@@ -866,7 +894,7 @@ function useChat(options) {
   });
   const stop = () => {
     var _a;
-    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
+    (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort();
   };
   return {
     append,
@@ -894,7 +922,13 @@ function useCopilotChat(_a = {}) {
     setCoagentStates,
     coAgentStateRenders,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    agentLock,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -944,7 +978,12 @@ function useCopilotChat(_a = {}) {
     coagentStates,
     setCoagentStates,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = (0, import_react8.useCallback)(
@@ -975,6 +1014,24 @@ function useCopilotChat(_a = {}) {
     },
     [latestSetMessages]
   );
+  const reset = (0, import_react8.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_react8.useCallback)(() => {
+    return latestReset.current();
+  }, [latestReset]);
   return {
     visibleMessages: messages,
     appendMessage: latestAppendFunc,
@@ -982,6 +1039,7 @@ function useCopilotChat(_a = {}) {
     reloadMessages: latestReloadFunc,
     stopGeneration: latestStopFunc,
     deleteMessage: latestDeleteFunc,
+    reset: latestResetFunc,
     isLoading
   };
 }
@@ -1206,7 +1264,11 @@ function useCoAgent(options) {
     } else if (coagentStates[name] === void 0) {
       setState(options.initialState === void 0 ? {} : options.initialState);
     }
-  }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
+  }, [
+    isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0,
+    // reset initial state on reset
+    coagentStates[name] === void 0
+  ]);
   return {
     name,
     nodeName: coagentState.nodeName,
@@ -1234,6 +1296,17 @@ 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}`);
   }
@@ -1512,6 +1585,9 @@ This is not a conversation, so please do not ask questions. Just call the functi
   CopilotTask,
   defaultCopilotContextCategories,
   extract,
+  runAgent,
+  startAgent,
+  stopAgent,
   useCoAgent,
   useCoAgentStateRender,
   useCopilotAction,