@copilotkit/react-core 1.5.12-next.6 → 1.5.12

package/dist/hooks/use-coagent.js

@@ -144,14 +144,17 @@ var emptyCopilotContext = {
   },
   forwardedParameters: {},
   agentLock: null,
-  threadId: null,
+  threadId: "",
   setThreadId: () => {
   },
   runId: null,
   setRunId: () => {
   },
   chatAbortControllerRef: { current: null },
-  availableAgents: []
+  availableAgents: [],
+  extensions: {},
+  setExtensions: () => {
+  }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -389,16 +392,18 @@ function useChat(options) {
     runId,
     setRunId,
     chatAbortControllerRef,
-    agentLock
+    agentLock,
+    extensions,
+    setExtensions
   } = options;
   const runChatCompletionRef = (0, import_react6.useRef)();
   const addErrorToast = useErrorToast();
   const agentSessionRef = (0, import_react6.useRef)(agentSession);
   agentSessionRef.current = agentSession;
-  const threadIdRef = (0, import_react6.useRef)(threadId);
-  threadIdRef.current = threadId;
   const runIdRef = (0, import_react6.useRef)(runId);
   runIdRef.current = runId;
+  const extensionsRef = (0, import_react6.useRef)(extensions);
+  extensionsRef.current = extensions;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared2.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = useCopilotRuntimeClient({
@@ -429,8 +434,9 @@ function useChat(options) {
               actions: processActionsForRuntimeRequest(actions),
               url: window.location.href
             },
-            threadId: threadIdRef.current,
+            threadId,
             runId: runIdRef.current,
+            extensions: extensionsRef.current,
             messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)((0, import_runtime_client_gql3.filterAgentStateMessages)(messagesWithContext))
           }, copilotConfig.cloud ? {
             cloud: __spreadValues({}, ((_c = (_b = (_a = copilotConfig.cloud.guardrails) == null ? void 0 : _a.input) == null ? void 0 : _b.restrictToTopic) == null ? void 0 : _c.enabled) ? {
@@ -483,10 +489,12 @@ function useChat(options) {
           if (!(value == null ? void 0 : value.generateCopilotResponse)) {
             continue;
           }
-          threadIdRef.current = value.generateCopilotResponse.threadId || null;
           runIdRef.current = value.generateCopilotResponse.runId || null;
-          setThreadId(threadIdRef.current);
+          extensionsRef.current = import_runtime_client_gql3.CopilotRuntimeClient.removeGraphQLTypename(
+            value.generateCopilotResponse.extensions || {}
+          );
           setRunId(runIdRef.current);
+          setExtensions(extensionsRef.current);
           messages2 = (0, import_runtime_client_gql3.convertGqlOutputToMessages)(
             (0, import_runtime_client_gql3.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
           );
@@ -779,7 +787,9 @@ function useCopilotChat(_a = {}) {
     setThreadId,
     runId,
     setRunId,
-    chatAbortControllerRef
+    chatAbortControllerRef,
+    extensions,
+    setExtensions
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -836,7 +846,9 @@ function useCopilotChat(_a = {}) {
     runId,
     setRunId,
     chatAbortControllerRef,
-    agentLock
+    agentLock,
+    extensions,
+    setExtensions
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = useAsyncCallback(
@@ -874,7 +886,6 @@ function useCopilotChat(_a = {}) {
   const reset = (0, import_react8.useCallback)(() => {
     latestStopFunc();
     setMessages([]);
-    setThreadId(null);
     setRunId(null);
     setCoagentStatesWithRef({});
     let initialAgentSession = null;
@@ -963,7 +974,7 @@ function useCoAgent(options) {
   };
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
-  const { coagentStates, coagentStatesRef, setCoagentStatesWithRef } = context;
+  const { coagentStates, coagentStatesRef, setCoagentStatesWithRef, threadId, copilotApiConfig } = context;
   const { appendMessage, runChatCompletion } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
     if (coagentStates2[name2]) {
@@ -980,6 +991,11 @@ function useCoAgent(options) {
       };
     }
   };
+  const runtimeClient = useCopilotRuntimeClient({
+    url: copilotApiConfig.chatApiEndpoint,
+    publicApiKey: copilotApiConfig.publicApiKey,
+    credentials: copilotApiConfig.credentials
+  });
   const setState = (newState) => {
     let coagentState2 = getCoagentState(coagentStatesRef.current || {}, name);
     const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
@@ -989,8 +1005,21 @@ function useCoAgent(options) {
       })
     }));
   };
+  (0, import_react9.useEffect)(() => {
+    const fetchAgentState = () => __async(this, null, function* () {
+      var _a, _b, _c, _d, _e, _f, _g, _h;
+      const result = yield runtimeClient.loadAgentState({
+        threadId,
+        agentName: name
+      });
+      if (((_b = (_a = result.data) == null ? void 0 : _a.loadAgentState) == null ? void 0 : _b.threadExists) && ((_d = (_c = result.data) == null ? void 0 : _c.loadAgentState) == null ? void 0 : _d.state) && ((_f = (_e = result.data) == null ? void 0 : _e.loadAgentState) == null ? void 0 : _f.state) != "{}") {
+        const fetchedState = JSON.parse((_h = (_g = result.data) == null ? void 0 : _g.loadAgentState) == null ? void 0 : _h.state);
+        isExternalStateManagement(options) ? options.setState(fetchedState) : setState(fetchedState);
+      }
+    });
+    void fetchAgentState();
+  }, [threadId]);
   const coagentState = getCoagentState(coagentStates, name);
-  const state = isExternalStateManagement(options) ? options.state : coagentState.state;
   (0, import_react9.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);