@copilotkit/react-core 1.5.12-next.5 → 1.5.12-next.7

package/dist/index.js

@@ -164,14 +164,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() {
@@ -407,10 +410,30 @@ function useCopilotMessagesContext() {
 }
 
 // src/components/copilot-provider/copilot-messages.tsx
+var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
 var import_jsx_runtime = require("react/jsx-runtime");
 function CopilotMessages(_a) {
   var _b = _a, { children } = _b, props = __objRest(_b, ["children"]);
   const [messages, setMessages] = (0, import_react5.useState)([]);
+  const { threadId, agentSession, runtimeClient } = useCopilotContext();
+  (0, import_react5.useEffect)(() => {
+    if (agentSession == null ? void 0 : agentSession.agentName) {
+      const fetchAgentState = () => __async(this, null, function* () {
+        var _a2, _b2, _c, _d;
+        const result = yield runtimeClient.loadAgentState({
+          threadId,
+          agentName: agentSession.agentName
+        });
+        if ((_b2 = (_a2 = result.data) == null ? void 0 : _a2.loadAgentState) == null ? void 0 : _b2.threadExists) {
+          const messages2 = (0, import_runtime_client_gql.loadMessagesFromJsonRepresentation)(
+            JSON.parse(((_d = (_c = result.data) == null ? void 0 : _c.loadAgentState) == null ? void 0 : _d.messages) || "[]")
+          );
+          setMessages(messages2);
+        }
+      });
+      void fetchAgentState();
+    }
+  }, [threadId, (agentSession == null ? void 0 : agentSession.agentName) !== void 0]);
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
     CopilotMessagesContext.Provider,
     {
@@ -457,11 +480,13 @@ var ExclamationMarkIcon = ({
 );
 
 // src/components/error-boundary/error-utils.tsx
+var import_react_markdown = __toESM(require("react-markdown"));
 var import_jsx_runtime3 = require("react/jsx-runtime");
 function ErrorToast({ errors }) {
   const errorsToRender = errors.map((error, idx) => {
     var _a, _b, _c;
-    const message = "extensions" in error ? (_b = (_a = error.extensions) == null ? void 0 : _a.originalError) == null ? void 0 : _b.message : error.message;
+    const originalError = "extensions" in error ? (_a = error.extensions) == null ? void 0 : _a.originalError : {};
+    const message = (_b = originalError == null ? void 0 : originalError.message) != null ? _b : error.message;
     const code = "extensions" in error ? (_c = error.extensions) == null ? void 0 : _c.code : null;
     return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
       "div",
@@ -486,7 +511,7 @@ function ErrorToast({ errors }) {
               ]
             }
           ),
-          /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { children: message })
+          /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(import_react_markdown.default, { children: message })
         ]
       },
       idx
@@ -678,13 +703,13 @@ function Toast({
 }
 
 // src/hooks/use-copilot-runtime-client.ts
-var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
 var import_react8 = require("react");
 var useCopilotRuntimeClient = (options) => {
   const { addGraphQLErrorsToast } = useToast();
   const addErrorToast = useErrorToast();
   const runtimeClient = (0, import_react8.useMemo)(() => {
-    return new import_runtime_client_gql.CopilotRuntimeClient(__spreadProps(__spreadValues({}, options), {
+    return new import_runtime_client_gql2.CopilotRuntimeClient(__spreadProps(__spreadValues({}, options), {
       handleGQLErrors: (error) => {
         if (error.graphQLErrors.length) {
           addGraphQLErrorsToast(error.graphQLErrors);
@@ -699,8 +724,8 @@ var useCopilotRuntimeClient = (options) => {
 
 // src/utils/extract.ts
 var import_shared3 = require("@copilotkit/shared");
-var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
 var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 function extract(_0) {
   return __async(this, arguments, function* ({
     context,
@@ -710,7 +735,7 @@ function extract(_0) {
     data,
     abortSignal,
     stream,
-    requestType = import_runtime_client_gql2.CopilotRequestType.Task,
+    requestType = import_runtime_client_gql3.CopilotRequestType.Task,
     forwardedParameters
   }) {
     var _a, _b;
@@ -731,13 +756,13 @@ function extract(_0) {
     if (includeReadable) {
       contextString += context.getContextString([], defaultCopilotContextCategories);
     }
-    const systemMessage = new import_runtime_client_gql2.TextMessage({
+    const systemMessage = new import_runtime_client_gql3.TextMessage({
       content: makeSystemMessage(contextString, instructions),
-      role: import_runtime_client_gql2.Role.System
+      role: import_runtime_client_gql3.Role.System
     });
-    const instructionsMessage = new import_runtime_client_gql2.TextMessage({
+    const instructionsMessage = new import_runtime_client_gql3.TextMessage({
       content: makeInstructionsMessage(instructions),
-      role: import_runtime_client_gql2.Role.User
+      role: import_runtime_client_gql3.Role.User
     });
     const response = context.runtimeClient.asStream(
       context.runtimeClient.generateCopilotResponse({
@@ -752,8 +777,8 @@ function extract(_0) {
             ],
             url: window.location.href
           },
-          messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)(
-            includeMessages ? [systemMessage, instructionsMessage, ...(0, import_runtime_client_gql3.filterAgentStateMessages)(messages)] : [systemMessage, instructionsMessage]
+          messages: (0, import_runtime_client_gql4.convertMessagesToGqlInput)(
+            includeMessages ? [systemMessage, instructionsMessage, ...(0, import_runtime_client_gql4.filterAgentStateMessages)(messages)] : [systemMessage, instructionsMessage]
           ),
           metadata: {
             requestType
@@ -778,7 +803,7 @@ function extract(_0) {
       if (abortSignal == null ? void 0 : abortSignal.aborted) {
         throw new Error("Aborted");
       }
-      actionExecutionMessage = (0, import_runtime_client_gql2.convertGqlOutputToMessages)(
+      actionExecutionMessage = (0, import_runtime_client_gql3.convertGqlOutputToMessages)(
         value.generateCopilotResponse.messages
       ).find((msg) => msg.isActionExecutionMessage());
       if (!actionExecutionMessage) {
@@ -899,6 +924,7 @@ function CopilotKitInternal(_a) {
   const [isLoading, setIsLoading] = (0, import_react10.useState)(false);
   const [chatInstructions, setChatInstructions] = (0, import_react10.useState)("");
   const [authStates, setAuthStates] = (0, import_react10.useState)({});
+  const [extensions, setExtensions] = (0, import_react10.useState)({});
   const {
     addElement: addDocument,
     removeElement: removeDocument,
@@ -1083,7 +1109,21 @@ ${nonDocumentStrings}`;
     };
   }
   const [agentSession, setAgentSession] = (0, import_react10.useState)(initialAgentSession);
-  const [threadId, setThreadId] = (0, import_react10.useState)(null);
+  const [internalThreadId, setInternalThreadId] = (0, import_react10.useState)(props.threadId || (0, import_shared4.randomUUID)());
+  const setThreadId = (0, import_react10.useCallback)(
+    (value) => {
+      if (props.threadId) {
+        throw new Error("Cannot call setThreadId() when threadId is provided via props.");
+      }
+      setInternalThreadId(value);
+    },
+    [props.threadId]
+  );
+  (0, import_react10.useEffect)(() => {
+    if (props.threadId !== void 0) {
+      setInternalThreadId(props.threadId);
+    }
+  }, [props.threadId]);
   const [runId, setRunId] = (0, import_react10.useState)(null);
   const chatAbortControllerRef = (0, import_react10.useRef)(null);
   const showDevConsole = props.showDevConsole === void 0 ? "auto" : props.showDevConsole;
@@ -1123,7 +1163,7 @@ ${nonDocumentStrings}`;
         runtimeClient,
         forwardedParameters: props.forwardedParameters || {},
         agentLock: props.agent || null,
-        threadId,
+        threadId: internalThreadId,
         setThreadId,
         runId,
         setRunId,
@@ -1131,7 +1171,9 @@ ${nonDocumentStrings}`;
         availableAgents,
         authConfig: props.authConfig,
         authStates,
-        setAuthStates
+        setAuthStates,
+        extensions,
+        setExtensions
       },
       children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(CopilotMessages, { children })
     }
@@ -1166,27 +1208,27 @@ function entryPointsToFunctionCallHandler(actions) {
 
 // src/hooks/use-copilot-chat.ts
 var import_react12 = require("react");
-var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
 var import_react11 = require("react");
 var import_shared6 = require("@copilotkit/shared");
-var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
 
 // src/types/frontend-action.ts
-var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
 var import_shared5 = require("@copilotkit/shared");
 function processActionsForRuntimeRequest(actions) {
   const filteredActions = actions.filter(
-    (action) => action.available !== import_runtime_client_gql4.ActionInputAvailability.Disabled && action.disabled !== true && action.name !== "*"
+    (action) => action.available !== import_runtime_client_gql5.ActionInputAvailability.Disabled && action.disabled !== true && action.name !== "*"
   ).map((action) => {
-    let available = import_runtime_client_gql4.ActionInputAvailability.Enabled;
+    let available = import_runtime_client_gql5.ActionInputAvailability.Enabled;
     if (action.disabled) {
-      available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
+      available = import_runtime_client_gql5.ActionInputAvailability.Disabled;
     } else if (action.available === "disabled") {
-      available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
+      available = import_runtime_client_gql5.ActionInputAvailability.Disabled;
     } else if (action.available === "remote") {
-      available = import_runtime_client_gql4.ActionInputAvailability.Remote;
+      available = import_runtime_client_gql5.ActionInputAvailability.Remote;
     }
     return {
       name: action.name,
@@ -1220,16 +1262,18 @@ function useChat(options) {
     runId,
     setRunId,
     chatAbortControllerRef,
-    agentLock
+    agentLock,
+    extensions,
+    setExtensions
   } = options;
   const runChatCompletionRef = (0, import_react11.useRef)();
   const addErrorToast = useErrorToast();
   const agentSessionRef = (0, import_react11.useRef)(agentSession);
   agentSessionRef.current = agentSession;
-  const threadIdRef = (0, import_react11.useRef)(threadId);
-  threadIdRef.current = threadId;
   const runIdRef = (0, import_react11.useRef)(runId);
   runIdRef.current = runId;
+  const extensionsRef = (0, import_react11.useRef)(extensions);
+  extensionsRef.current = extensions;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared6.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = useCopilotRuntimeClient({
@@ -1243,9 +1287,9 @@ function useChat(options) {
       var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
       setIsLoading(true);
       let newMessages = [
-        new import_runtime_client_gql5.TextMessage({
+        new import_runtime_client_gql6.TextMessage({
           content: "",
-          role: import_runtime_client_gql5.Role.Assistant
+          role: import_runtime_client_gql6.Role.Assistant
         })
       ];
       chatAbortControllerRef.current = new AbortController();
@@ -1260,9 +1304,10 @@ function useChat(options) {
               actions: processActionsForRuntimeRequest(actions),
               url: window.location.href
             },
-            threadId: threadIdRef.current,
+            threadId,
             runId: runIdRef.current,
-            messages: (0, import_runtime_client_gql5.convertMessagesToGqlInput)((0, import_runtime_client_gql5.filterAgentStateMessages)(messagesWithContext))
+            extensions: extensionsRef.current,
+            messages: (0, import_runtime_client_gql6.convertMessagesToGqlInput)((0, import_runtime_client_gql6.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) ? {
               guardrails: {
@@ -1274,7 +1319,7 @@ function useChat(options) {
             } : {})
           } : {}), {
             metadata: {
-              requestType: import_runtime_client_gql5.CopilotRequestType.Chat
+              requestType: import_runtime_client_gql6.CopilotRequestType.Chat
             }
           }), agentSessionRef.current ? {
             agentSession: agentSessionRef.current
@@ -1314,12 +1359,14 @@ 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_gql6.CopilotRuntimeClient.removeGraphQLTypename(
+            value.generateCopilotResponse.extensions || {}
+          );
           setRunId(runIdRef.current);
-          messages2 = (0, import_runtime_client_gql5.convertGqlOutputToMessages)(
-            (0, import_runtime_client_gql5.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
+          setExtensions(extensionsRef.current);
+          messages2 = (0, import_runtime_client_gql6.convertGqlOutputToMessages)(
+            (0, import_runtime_client_gql6.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
           );
           if (messages2.length === 0) {
             continue;
@@ -1327,8 +1374,8 @@ function useChat(options) {
           newMessages = [];
           if (((_h = value.generateCopilotResponse.status) == null ? void 0 : _h.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
             newMessages = [
-              new import_runtime_client_gql5.TextMessage({
-                role: import_runtime_client_gql5.MessageRole.Assistant,
+              new import_runtime_client_gql6.TextMessage({
+                role: import_runtime_client_gql6.MessageRole.Assistant,
                 content: ((_i = value.generateCopilotResponse.status.details) == null ? void 0 : _i.guardrailsReason) || ""
               })
             ];
@@ -1352,7 +1399,7 @@ function useChat(options) {
             const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
             if (lastAgentStateMessage) {
               if (lastAgentStateMessage.state.messages && lastAgentStateMessage.state.messages.length > 0) {
-                syncedMessages = (0, import_runtime_client_gql5.loadMessagesFromJsonRepresentation)(
+                syncedMessages = (0, import_runtime_client_gql6.loadMessagesFromJsonRepresentation)(
                   lastAgentStateMessage.state.messages
                 );
               }
@@ -1396,7 +1443,7 @@ function useChat(options) {
           const lastMessages = [];
           for (let i = finalMessages.length - 1; i >= 0; i--) {
             const message = finalMessages[i];
-            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql5.MessageStatusCode.Pending) {
+            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql6.MessageStatusCode.Pending) {
               lastMessages.unshift(message);
             } else {
               break;
@@ -1444,9 +1491,9 @@ function useChat(options) {
               finalMessages.splice(
                 messageIndex + 1,
                 0,
-                new import_runtime_client_gql5.ResultMessage({
+                new import_runtime_client_gql6.ResultMessage({
                   id: "result-" + message.id,
-                  result: import_runtime_client_gql5.ResultMessage.encodeResult(
+                  result: import_runtime_client_gql6.ResultMessage.encodeResult(
                     error ? {
                       content: result,
                       error: JSON.parse(
@@ -1603,7 +1650,9 @@ function useCopilotChat(_a = {}) {
     setThreadId,
     runId,
     setRunId,
-    chatAbortControllerRef
+    chatAbortControllerRef,
+    extensions,
+    setExtensions
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -1616,9 +1665,9 @@ function useCopilotChat(_a = {}) {
   const makeSystemMessageCallback = (0, import_react12.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage2 || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql6.TextMessage({
+    return new import_runtime_client_gql7.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql6.Role.System
+      role: import_runtime_client_gql7.Role.System
     });
   }, [getContextString, makeSystemMessage2, chatInstructions]);
   const onCoAgentStateRender = useAsyncCallback(
@@ -1660,7 +1709,9 @@ function useCopilotChat(_a = {}) {
     runId,
     setRunId,
     chatAbortControllerRef,
-    agentLock
+    agentLock,
+    extensions,
+    setExtensions
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = useAsyncCallback(
@@ -1698,7 +1749,6 @@ function useCopilotChat(_a = {}) {
   const reset = (0, import_react12.useCallback)(() => {
     latestStopFunc();
     setMessages([]);
-    setThreadId(null);
     setRunId(null);
     setCoagentStatesWithRef({});
     let initialAgentSession = null;
@@ -1965,7 +2015,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]) {
@@ -1982,6 +2032,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;
@@ -1991,8 +2046,21 @@ function useCoAgent(options) {
       })
     }));
   };
+  (0, import_react17.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_react17.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
@@ -2122,7 +2190,7 @@ function useCopilotAuthenticatedAction(action, dependencies) {
 }
 
 // src/lib/copilot-task.ts
-var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql8 = require("@copilotkit/runtime-client-gql");
 var CopilotTask = class {
   constructor(config) {
     this.instructions = config.instructions;
@@ -2150,12 +2218,12 @@ var CopilotTask = class {
       if (this.includeCopilotReadable) {
         contextString += context.getContextString([], defaultCopilotContextCategories);
       }
-      const systemMessage = new import_runtime_client_gql7.TextMessage({
+      const systemMessage = new import_runtime_client_gql8.TextMessage({
         content: taskSystemMessage(contextString, this.instructions),
-        role: import_runtime_client_gql7.Role.System
+        role: import_runtime_client_gql8.Role.System
       });
       const messages = [systemMessage];
-      const runtimeClient = new import_runtime_client_gql7.CopilotRuntimeClient({
+      const runtimeClient = new import_runtime_client_gql8.CopilotRuntimeClient({
         url: context.copilotApiConfig.chatApiEndpoint,
         publicApiKey: context.copilotApiConfig.publicApiKey,
         headers: context.copilotApiConfig.headers,
@@ -2167,9 +2235,9 @@ var CopilotTask = class {
             actions: processActionsForRuntimeRequest(Object.values(actions)),
             url: window.location.href
           },
-          messages: (0, import_runtime_client_gql7.convertMessagesToGqlInput)((0, import_runtime_client_gql7.filterAgentStateMessages)(messages)),
+          messages: (0, import_runtime_client_gql8.convertMessagesToGqlInput)((0, import_runtime_client_gql8.filterAgentStateMessages)(messages)),
           metadata: {
-            requestType: import_runtime_client_gql7.CopilotRequestType.Task
+            requestType: import_runtime_client_gql8.CopilotRequestType.Task
           },
           forwardedParameters: __spreadProps(__spreadValues({}, (_a = this.forwardedParameters) != null ? _a : {}), {
             toolChoice: "required"
@@ -2178,7 +2246,7 @@ var CopilotTask = class {
         properties: context.copilotApiConfig.properties
       }).toPromise();
       const functionCallHandler = context.getFunctionCallHandler(actions);
-      const functionCalls = (0, import_runtime_client_gql7.convertGqlOutputToMessages)(
+      const functionCalls = (0, import_runtime_client_gql8.convertGqlOutputToMessages)(
         ((_c = (_b = response.data) == null ? void 0 : _b.generateCopilotResponse) == null ? void 0 : _c.messages) || []
       ).filter((m) => m.isActionExecutionMessage());
       for (const functionCall of functionCalls) {