@copilotkit/react-core 1.4.1-pre.5 → 1.4.1-pre.6

package/dist/hooks/use-coagent.js

@@ -82,7 +82,7 @@ __export(use_coagent_exports, {
   useCoAgent: () => useCoAgent
 });
 module.exports = __toCommonJS(use_coagent_exports);
-var import_react6 = require("react");
+var import_react8 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -112,6 +112,7 @@ var emptyCopilotContext = {
   addDocumentContext: () => returnAndThrowInDebug(""),
   removeDocumentContext: () => {
   },
+  runtimeClient: {},
   copilotApiConfig: new class {
     get chatApiEndpoint() {
       throw new Error("Remember to wrap your app in a `<CopilotKit> {...} </CopilotKit>` !!!");
@@ -167,13 +168,44 @@ function useCopilotMessagesContext() {
 }
 
 // src/hooks/use-copilot-chat.ts
+var import_react7 = require("react");
+var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
+
+// src/hooks/use-chat.ts
 var import_react5 = require("react");
+var import_shared = require("@copilotkit/shared");
 var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
 
-// src/hooks/use-chat.ts
+// src/components/toast/toast-provider.tsx
 var import_react3 = require("react");
-var import_shared = require("@copilotkit/shared");
+var import_jsx_runtime = require("react/jsx-runtime");
+var ToastContext = (0, import_react3.createContext)(void 0);
+function useToast() {
+  const context = (0, import_react3.useContext)(ToastContext);
+  if (!context) {
+    throw new Error("useToast must be used within a ToastProvider");
+  }
+  return context;
+}
+
+// src/hooks/use-copilot-runtime-client.ts
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
+var import_react4 = require("react");
+var useCopilotRuntimeClient = (options) => {
+  const { addGraphQLErrorsToast } = useToast();
+  const runtimeClient = (0, import_react4.useMemo)(() => {
+    return new import_runtime_client_gql.CopilotRuntimeClient(__spreadProps(__spreadValues({}, options), {
+      handleGQLErrors: (error) => {
+        if (error.graphQLErrors.length) {
+          addGraphQLErrorsToast(error.graphQLErrors);
+        }
+      }
+    }));
+  }, [options, addGraphQLErrorsToast]);
+  return runtimeClient;
+};
+
+// src/hooks/use-chat.ts
 function useChat(options) {
   const {
     messages,
@@ -191,17 +223,18 @@ function useChat(options) {
     agentSession,
     setAgentSession
   } = options;
-  const abortControllerRef = (0, import_react3.useRef)();
-  const threadIdRef = (0, import_react3.useRef)(null);
-  const runIdRef = (0, import_react3.useRef)(null);
-  const runChatCompletionRef = (0, import_react3.useRef)();
-  const coagentStatesRef = (0, import_react3.useRef)(coagentStates);
+  const abortControllerRef = (0, import_react5.useRef)();
+  const threadIdRef = (0, import_react5.useRef)(null);
+  const runIdRef = (0, import_react5.useRef)(null);
+  const { addGraphQLErrorsToast } = useToast();
+  const runChatCompletionRef = (0, import_react5.useRef)();
+  const coagentStatesRef = (0, import_react5.useRef)(coagentStates);
   coagentStatesRef.current = coagentStates;
-  const agentSessionRef = (0, import_react3.useRef)(agentSession);
+  const agentSessionRef = (0, import_react5.useRef)(agentSession);
   agentSessionRef.current = agentSession;
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
-  const runtimeClient = new import_runtime_client_gql.CopilotRuntimeClient({
+  const runtimeClient = useCopilotRuntimeClient({
     url: copilotConfig.chatApiEndpoint,
     publicApiKey: copilotConfig.publicApiKey,
     headers,
@@ -211,9 +244,9 @@ function useChat(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i;
     setIsLoading(true);
     let newMessages = [
-      new import_runtime_client_gql.TextMessage({
+      new import_runtime_client_gql2.TextMessage({
         content: "",
-        role: import_runtime_client_gql.Role.Assistant
+        role: import_runtime_client_gql2.Role.Assistant
       })
     ];
     const abortController = new AbortController();
@@ -221,20 +254,20 @@ function useChat(options) {
     setMessages([...previousMessages, ...newMessages]);
     const systemMessage = makeSystemMessageCallback();
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
-    const stream = import_runtime_client_gql.CopilotRuntimeClient.asStream(
+    const stream = runtimeClient.asStream(
       runtimeClient.generateCopilotResponse({
         data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
           frontend: {
             actions: actions.filter(
-              (action) => action.available !== import_runtime_client_gql.ActionInputAvailability.Disabled || !action.disabled
+              (action) => action.available !== import_runtime_client_gql2.ActionInputAvailability.Disabled || !action.disabled
             ).map((action) => {
-              let available = import_runtime_client_gql.ActionInputAvailability.Enabled;
+              let available = import_runtime_client_gql2.ActionInputAvailability.Enabled;
               if (action.disabled) {
-                available = import_runtime_client_gql.ActionInputAvailability.Disabled;
+                available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
               } else if (action.available === "disabled") {
-                available = import_runtime_client_gql.ActionInputAvailability.Disabled;
+                available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
               } else if (action.available === "remote") {
-                available = import_runtime_client_gql.ActionInputAvailability.Remote;
+                available = import_runtime_client_gql2.ActionInputAvailability.Remote;
               }
               return {
                 name: action.name,
@@ -247,7 +280,7 @@ function useChat(options) {
           },
           threadId: threadIdRef.current,
           runId: runIdRef.current,
-          messages: (0, import_runtime_client_gql.convertMessagesToGqlInput)((0, import_runtime_client_gql.filterAgentStateMessages)(messagesWithContext))
+          messages: (0, import_runtime_client_gql2.convertMessagesToGqlInput)((0, import_runtime_client_gql2.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: {
@@ -259,7 +292,7 @@ function useChat(options) {
           } : {})
         } : {}), {
           metadata: {
-            requestType: import_runtime_client_gql.CopilotRequestType.Chat
+            requestType: import_runtime_client_gql2.CopilotRequestType.Chat
           }
         }), agentSessionRef.current ? {
           agentSession: agentSessionRef.current
@@ -280,7 +313,14 @@ function useChat(options) {
     let followUp = void 0;
     try {
       while (true) {
-        const { done, value } = yield reader.read();
+        let done, value;
+        try {
+          const readResult = yield reader.read();
+          done = readResult.done;
+          value = readResult.value;
+        } catch (readError) {
+          break;
+        }
         if (done) {
           break;
         }
@@ -289,8 +329,8 @@ function useChat(options) {
         }
         threadIdRef.current = value.generateCopilotResponse.threadId || null;
         runIdRef.current = value.generateCopilotResponse.runId || null;
-        const messages2 = (0, import_runtime_client_gql.convertGqlOutputToMessages)(
-          (0, import_runtime_client_gql.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
+        const messages2 = (0, import_runtime_client_gql2.convertGqlOutputToMessages)(
+          (0, import_runtime_client_gql2.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
         );
         if (messages2.length === 0) {
           continue;
@@ -298,15 +338,15 @@ 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_gql.TextMessage({
-              role: import_runtime_client_gql.MessageRole.Assistant,
+            new import_runtime_client_gql2.TextMessage({
+              role: import_runtime_client_gql2.MessageRole.Assistant,
               content: ((_i = value.generateCopilotResponse.status.details) == null ? void 0 : _i.guardrailsReason) || ""
             })
           ];
         } else {
           for (const message of messages2) {
             newMessages.push(message);
-            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
+            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql2.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
               if (!(message.id in actionResults)) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
@@ -329,8 +369,8 @@ function useChat(options) {
                 }
               }
               newMessages.push(
-                new import_runtime_client_gql.ResultMessage({
-                  result: import_runtime_client_gql.ResultMessage.encodeResult(actionResults[message.id]),
+                new import_runtime_client_gql2.ResultMessage({
+                  result: import_runtime_client_gql2.ResultMessage.encodeResult(actionResults[message.id]),
                   actionExecutionId: message.id,
                   actionName: message.name
                 })
@@ -427,10 +467,10 @@ function useChat(options) {
 }
 
 // src/components/copilot-provider/copilotkit.tsx
-var import_react4 = require("react");
+var import_react6 = require("react");
 var import_react_dom = require("react-dom");
 var import_shared2 = require("@copilotkit/shared");
-var import_jsx_runtime = require("react/jsx-runtime");
+var import_jsx_runtime2 = require("react/jsx-runtime");
 var defaultCopilotContextCategories = ["global"];
 
 // src/hooks/use-copilot-chat.ts
@@ -456,21 +496,21 @@ function useCopilotChat(_a = {}) {
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
-  const deleteMessage = (0, import_react5.useCallback)(
+  const deleteMessage = (0, import_react7.useCallback)(
     (messageId) => {
       setMessages((prev) => prev.filter((message) => message.id !== messageId));
     },
     [setMessages]
   );
-  const makeSystemMessageCallback = (0, import_react5.useCallback)(() => {
+  const makeSystemMessageCallback = (0, import_react7.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql2.TextMessage({
+    return new import_runtime_client_gql3.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql2.Role.System
+      role: import_runtime_client_gql3.Role.System
     });
   }, [getContextString, makeSystemMessage, chatInstructions]);
-  const onCoAgentStateRender = (0, import_react5.useCallback)(
+  const onCoAgentStateRender = (0, import_react7.useCallback)(
     (args) => __async(this, null, function* () {
       var _a2;
       const { name, nodeName, state } = args;
@@ -505,29 +545,29 @@ function useCopilotChat(_a = {}) {
     setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
-  const latestAppendFunc = (0, import_react5.useCallback)(
+  const latestAppendFunc = (0, import_react7.useCallback)(
     (message) => {
       return latestAppend.current(message);
     },
     [latestAppend]
   );
   const latestReload = useUpdatedRef(reload);
-  const latestReloadFunc = (0, import_react5.useCallback)(() => {
+  const latestReloadFunc = (0, import_react7.useCallback)(() => {
     return latestReload.current();
   }, [latestReload]);
   const latestStop = useUpdatedRef(stop);
-  const latestStopFunc = (0, import_react5.useCallback)(() => {
+  const latestStopFunc = (0, import_react7.useCallback)(() => {
     return latestStop.current();
   }, [latestStop]);
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react5.useCallback)(
+  const latestDeleteFunc = (0, import_react7.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
   const latestSetMessages = useUpdatedRef(setMessages);
-  const latestSetMessagesFunc = (0, import_react5.useCallback)(
+  const latestSetMessagesFunc = (0, import_react7.useCallback)(
     (messages2) => {
       return latestSetMessages.current(messages2);
     },
@@ -544,8 +584,8 @@ function useCopilotChat(_a = {}) {
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react5.useRef)(value);
-  (0, import_react5.useEffect)(() => {
+  const ref = (0, import_react7.useRef)(value);
+  (0, import_react7.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -616,7 +656,7 @@ function useCoAgent(options) {
   };
   const coagentState = getCoagentState(coagentStates, name);
   const state = isExternalStateManagement(options) ? options.state : coagentState.state;
-  (0, import_react6.useEffect)(() => {
+  (0, import_react8.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
     } else if (coagentStates[name] === void 0) {