@copilotkit/react-core 1.3.16-mme-improve-langsmith.1 → 1.3.16-mme-reset-chat.3

package/dist/index.js

@@ -151,6 +151,13 @@ var emptyCopilotContext = {
   },
   agentSession: null,
   setAgentSession: () => {
+  },
+  agentLock: null,
+  threadId: null,
+  setThreadId: () => {
+  },
+  runId: null,
+  setRunId: () => {
   }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
@@ -565,6 +572,8 @@ ${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);
   return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(
     CopilotContext.Provider,
     {
@@ -595,7 +604,12 @@ ${nonDocumentStrings}`;
         coagentStates,
         setCoagentStates,
         agentSession,
-        setAgentSession
+        setAgentSession,
+        agentLock: props.agent || null,
+        threadId,
+        setThreadId,
+        runId,
+        setRunId
       },
       children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(CopilotMessages, { children })
     }
@@ -651,16 +665,22 @@ function useChat(options) {
     setCoagentStates,
     coagentStates,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId
   } = 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({
@@ -738,6 +758,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)
         );
@@ -894,7 +916,12 @@ function useCopilotChat(_a = {}) {
     setCoagentStates,
     coAgentStateRenders,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    agentLock,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -944,7 +971,11 @@ function useCopilotChat(_a = {}) {
     coagentStates,
     setCoagentStates,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = (0, import_react8.useCallback)(
@@ -975,6 +1006,23 @@ function useCopilotChat(_a = {}) {
     },
     [latestSetMessages]
   );
+  const reset = (0, import_react8.useCallback)(() => {
+    latestStopFunc();
+    setMessages([]);
+    setThreadId(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 +1030,7 @@ function useCopilotChat(_a = {}) {
     reloadMessages: latestReloadFunc,
     stopGeneration: latestStopFunc,
     deleteMessage: latestDeleteFunc,
+    reset: latestResetFunc,
     isLoading
   };
 }