@copilotkit/react-core 1.3.16-pre.11 → 1.3.16-pre.13

package/dist/hooks/use-make-copilot-document-readable.mjs

@@ -1,7 +1,7 @@
 import {
   useMakeCopilotDocumentReadable
-} from "../chunk-TRXKPXDE.mjs";
-import "../chunk-2MPUQRAY.mjs";
+} from "../chunk-N57PDRVX.mjs";
+import "../chunk-WBBL7MG6.mjs";
 import "../chunk-SKC7AJIV.mjs";
 export {
   useMakeCopilotDocumentReadable

package/dist/index.d.ts

@@ -8,7 +8,7 @@ export { useCoAgentStateRender } from './hooks/use-coagent-state-render.js';
 export { useMakeCopilotDocumentReadable } from './hooks/use-make-copilot-document-readable.js';
 export { UseChatHelpers } from './hooks/use-chat.js';
 export { useCopilotReadable } from './hooks/use-copilot-readable.js';
-export { useCoAgent } from './hooks/use-coagent.js';
+export { HintFunction, runAgent, startAgent, stopAgent, useCoAgent } from './hooks/use-coagent.js';
 export { DocumentPointer } from './types/document-pointer.js';
 export { SystemMessageFunction } from './types/system-message.js';
 export { ActionRenderProps, RenderFunctionStatus } from './types/frontend-action.js';

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({
@@ -678,8 +704,7 @@ function useChat(options) {
         role: import_runtime_client_gql.Role.Assistant
       })
     ];
-    const abortController = new AbortController();
-    abortControllerRef.current = abortController;
+    chatAbortControllerRef.current = new AbortController();
     setMessages([...previousMessages, ...newMessages]);
     const systemMessage = makeSystemMessageCallback();
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
@@ -719,7 +744,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;
@@ -731,6 +756,9 @@ function useChat(options) {
       while (true) {
         const { done, value } = yield reader.read();
         if (done) {
+          if (chatAbortControllerRef.current.signal.aborted) {
+            return newMessages.slice();
+          }
           break;
         }
         if (!(value == null ? void 0 : value.generateCopilotResponse)) {
@@ -738,6 +766,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)
         );
@@ -766,12 +796,22 @@ function useChat(options) {
                   if (action) {
                     followUp = action.followUp;
                   }
-                  const result = yield onFunctionCall({
-                    messages: previousMessages,
-                    name: message.name,
-                    args: message.arguments
-                  });
-                  actionResults[message.id] = result;
+                  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;
+                  }
                 } catch (e) {
                   actionResults[message.id] = `Failed to execute action ${message.name}`;
                   console.error(`Failed to execute action ${message.name}: ${e}`);
@@ -866,7 +906,7 @@ function useChat(options) {
   });
   const stop = () => {
     var _a;
-    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
+    (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort("Stop was called");
   };
   return {
     append,
@@ -894,7 +934,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 +990,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 +1026,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 +1051,7 @@ function useCopilotChat(_a = {}) {
     reloadMessages: latestReloadFunc,
     stopGeneration: latestStopFunc,
     deleteMessage: latestDeleteFunc,
+    reset: latestResetFunc,
     isLoading
   };
 }
@@ -1206,7 +1276,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 +1308,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 +1597,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,