@copilotkit/react-core 1.4.8-no-pino-redact.1 → 1.4.8

package/dist/hooks/use-coagent.js

@@ -82,7 +82,7 @@ __export(use_coagent_exports, {
   useCoAgent: () => useCoAgent
 });
 module.exports = __toCommonJS(use_coagent_exports);
-var import_react8 = require("react");
+var import_react9 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -171,20 +171,135 @@ function useCopilotMessagesContext() {
 }
 
 // src/hooks/use-copilot-chat.ts
-var import_react7 = require("react");
+var import_react8 = require("react");
 var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
-var import_react5 = require("react");
+var import_react6 = require("react");
 var import_shared = require("@copilotkit/shared");
 var import_runtime_client_gql2 = require("@copilotkit/runtime-client-gql");
 
 // src/components/toast/toast-provider.tsx
+var import_react4 = require("react");
+
+// src/components/error-boundary/error-utils.tsx
 var import_react3 = require("react");
+
+// src/components/toast/exclamation-mark-icon.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
-var ToastContext = (0, import_react3.createContext)(void 0);
+var ExclamationMarkIcon = ({
+  className,
+  style
+}) => /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(
+  "svg",
+  {
+    xmlns: "http://www.w3.org/2000/svg",
+    width: "24",
+    height: "24",
+    viewBox: "0 0 24 24",
+    fill: "none",
+    stroke: "currentColor",
+    strokeWidth: "2",
+    strokeLinecap: "round",
+    strokeLinejoin: "round",
+    className: `lucide lucide-circle-alert ${className ? className : ""}`,
+    style,
+    children: [
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)("circle", { cx: "12", cy: "12", r: "10" }),
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)("line", { x1: "12", x2: "12", y1: "8", y2: "12" }),
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)("line", { x1: "12", x2: "12.01", y1: "16", y2: "16" })
+    ]
+  }
+);
+
+// src/components/error-boundary/error-utils.tsx
+var import_jsx_runtime2 = 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 code = "extensions" in error ? (_c = error.extensions) == null ? void 0 : _c.code : null;
+    return /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)(
+      "div",
+      {
+        style: {
+          marginTop: idx === 0 ? 0 : 10,
+          marginBottom: 14
+        },
+        children: [
+          /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(ExclamationMarkIcon, { style: { marginBottom: 4 } }),
+          code && /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)(
+            "div",
+            {
+              style: {
+                fontWeight: "600",
+                marginBottom: 4
+              },
+              children: [
+                "Copilot Cloud Error:",
+                " ",
+                /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("span", { style: { fontFamily: "monospace", fontWeight: "normal" }, children: code })
+              ]
+            }
+          ),
+          /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("div", { children: message })
+        ]
+      },
+      idx
+    );
+  });
+  return /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)(
+    "div",
+    {
+      style: {
+        fontSize: "13px",
+        maxWidth: "600px"
+      },
+      children: [
+        errorsToRender,
+        /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("div", { style: { fontSize: "11px", opacity: 0.75 }, children: "NOTE: This error only displays during local development." })
+      ]
+    }
+  );
+}
+function useErrorToast() {
+  const { addToast } = useToast();
+  return (0, import_react3.useCallback)(
+    (error) => {
+      const errorId = error.map((err) => {
+        var _a, _b;
+        const message = "extensions" in err ? ((_b = (_a = err.extensions) == null ? void 0 : _a.originalError) == null ? void 0 : _b.message) || err.message : err.message;
+        const stack = err.stack || "";
+        return btoa(message + stack).slice(0, 32);
+      }).join("|");
+      addToast({
+        type: "error",
+        id: errorId,
+        // Toast libraries typically dedupe by id
+        message: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(ErrorToast, { errors: error })
+      });
+    },
+    [addToast]
+  );
+}
+function useAsyncCallback(callback, deps) {
+  const addErrorToast = useErrorToast();
+  return (0, import_react3.useCallback)((...args) => __async(this, null, function* () {
+    try {
+      return yield callback(...args);
+    } catch (error) {
+      console.error("Error in async callback:", error);
+      addErrorToast([error]);
+      throw error;
+    }
+  }), deps);
+}
+
+// src/components/toast/toast-provider.tsx
+var import_jsx_runtime3 = require("react/jsx-runtime");
+var ToastContext = (0, import_react4.createContext)(void 0);
 function useToast() {
-  const context = (0, import_react3.useContext)(ToastContext);
+  const context = (0, import_react4.useContext)(ToastContext);
   if (!context) {
     throw new Error("useToast must be used within a ToastProvider");
   }
@@ -193,14 +308,17 @@ function useToast() {
 
 // src/hooks/use-copilot-runtime-client.ts
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
-var import_react4 = require("react");
+var import_react5 = require("react");
 var useCopilotRuntimeClient = (options) => {
   const { addGraphQLErrorsToast } = useToast();
-  const runtimeClient = (0, import_react4.useMemo)(() => {
+  const addErrorToast = useErrorToast();
+  const runtimeClient = (0, import_react5.useMemo)(() => {
     return new import_runtime_client_gql.CopilotRuntimeClient(__spreadProps(__spreadValues({}, options), {
       handleGQLErrors: (error) => {
         if (error.graphQLErrors.length) {
           addGraphQLErrorsToast(error.graphQLErrors);
+        } else {
+          addErrorToast([error]);
         }
       }
     }));
@@ -226,12 +344,12 @@ function useChat(options) {
     agentSession,
     setAgentSession
   } = options;
-  const abortControllerRef = (0, import_react5.useRef)();
-  const threadIdRef = (0, import_react5.useRef)(null);
-  const runIdRef = (0, import_react5.useRef)(null);
+  const abortControllerRef = (0, import_react6.useRef)();
+  const threadIdRef = (0, import_react6.useRef)(null);
+  const runIdRef = (0, import_react6.useRef)(null);
   const { addGraphQLErrorsToast } = useToast();
-  const runChatCompletionRef = (0, import_react5.useRef)();
-  const agentSessionRef = (0, import_react5.useRef)(agentSession);
+  const runChatCompletionRef = (0, import_react6.useRef)();
+  const agentSessionRef = (0, import_react6.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 } : {});
@@ -241,210 +359,236 @@ function useChat(options) {
     headers,
     credentials: copilotConfig.credentials
   });
-  const runChatCompletion = (previousMessages) => __async(this, null, function* () {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
-    setIsLoading(true);
-    let newMessages = [
-      new import_runtime_client_gql2.TextMessage({
-        content: "",
-        role: import_runtime_client_gql2.Role.Assistant
-      })
-    ];
-    const abortController = new AbortController();
-    abortControllerRef.current = abortController;
-    setMessages([...previousMessages, ...newMessages]);
-    const systemMessage = makeSystemMessageCallback();
-    const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
-    const stream = runtimeClient.asStream(
-      runtimeClient.generateCopilotResponse({
-        data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
-          frontend: {
-            actions: actions.filter(
-              (action) => action.available !== import_runtime_client_gql2.ActionInputAvailability.Disabled || !action.disabled
-            ).map((action) => {
-              let available = import_runtime_client_gql2.ActionInputAvailability.Enabled;
-              if (action.disabled) {
-                available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
-              } else if (action.available === "disabled") {
-                available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
-              } else if (action.available === "remote") {
-                available = import_runtime_client_gql2.ActionInputAvailability.Remote;
-              }
-              return {
-                name: action.name,
-                description: action.description || "",
-                jsonSchema: JSON.stringify((0, import_shared.actionParametersToJsonSchema)(action.parameters || [])),
-                available
-              };
-            }),
-            url: window.location.href
-          },
-          threadId: threadIdRef.current,
-          runId: runIdRef.current,
-          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: {
-              inputValidationRules: {
-                allowList: copilotConfig.cloud.guardrails.input.restrictToTopic.validTopics,
-                denyList: copilotConfig.cloud.guardrails.input.restrictToTopic.invalidTopics
+  const runChatCompletion = useAsyncCallback(
+    (previousMessages) => __async(this, null, function* () {
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+      setIsLoading(true);
+      let newMessages = [
+        new import_runtime_client_gql2.TextMessage({
+          content: "",
+          role: import_runtime_client_gql2.Role.Assistant
+        })
+      ];
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      setMessages([...previousMessages, ...newMessages]);
+      const systemMessage = makeSystemMessageCallback();
+      const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
+      const stream = runtimeClient.asStream(
+        runtimeClient.generateCopilotResponse({
+          data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
+            frontend: {
+              actions: actions.filter(
+                (action) => action.available !== import_runtime_client_gql2.ActionInputAvailability.Disabled || !action.disabled
+              ).map((action) => {
+                let available = import_runtime_client_gql2.ActionInputAvailability.Enabled;
+                if (action.disabled) {
+                  available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
+                } else if (action.available === "disabled") {
+                  available = import_runtime_client_gql2.ActionInputAvailability.Disabled;
+                } else if (action.available === "remote") {
+                  available = import_runtime_client_gql2.ActionInputAvailability.Remote;
+                }
+                return {
+                  name: action.name,
+                  description: action.description || "",
+                  jsonSchema: JSON.stringify(
+                    (0, import_shared.actionParametersToJsonSchema)(action.parameters || [])
+                  ),
+                  available
+                };
+              }),
+              url: window.location.href
+            },
+            threadId: threadIdRef.current,
+            runId: runIdRef.current,
+            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: {
+                inputValidationRules: {
+                  allowList: copilotConfig.cloud.guardrails.input.restrictToTopic.validTopics,
+                  denyList: copilotConfig.cloud.guardrails.input.restrictToTopic.invalidTopics
+                }
               }
+            } : {})
+          } : {}), {
+            metadata: {
+              requestType: import_runtime_client_gql2.CopilotRequestType.Chat
             }
-          } : {})
-        } : {}), {
-          metadata: {
-            requestType: import_runtime_client_gql2.CopilotRequestType.Chat
+          }), agentSessionRef.current ? {
+            agentSession: agentSessionRef.current
+          } : {}), {
+            agentStates: Object.values(coagentStatesRef.current).map((state) => ({
+              agentName: state.name,
+              state: JSON.stringify(state.state)
+            }))
+          }),
+          properties: copilotConfig.properties,
+          signal: (_d = abortControllerRef.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;
+      const reader = stream.getReader();
+      let actionResults = {};
+      let executedCoAgentStateRenders = [];
+      let followUp = void 0;
+      try {
+        while (true) {
+          let done, value;
+          try {
+            const readResult = yield reader.read();
+            done = readResult.done;
+            value = readResult.value;
+          } catch (readError) {
+            break;
           }
-        }), agentSessionRef.current ? {
-          agentSession: agentSessionRef.current
-        } : {}), {
-          agentStates: Object.values(coagentStatesRef.current).map((state) => ({
-            agentName: state.name,
-            state: JSON.stringify(state.state)
-          }))
-        }),
-        properties: copilotConfig.properties,
-        signal: (_d = abortControllerRef.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;
-    const reader = stream.getReader();
-    let actionResults = {};
-    let executedCoAgentStateRenders = [];
-    let followUp = void 0;
-    try {
-      while (true) {
-        let done, value;
-        try {
-          const readResult = yield reader.read();
-          done = readResult.done;
-          value = readResult.value;
-        } catch (readError) {
-          break;
-        }
-        if (done) {
-          break;
-        }
-        if (!(value == null ? void 0 : value.generateCopilotResponse)) {
-          continue;
-        }
-        threadIdRef.current = value.generateCopilotResponse.threadId || null;
-        runIdRef.current = value.generateCopilotResponse.runId || null;
-        const messages2 = (0, import_runtime_client_gql2.convertGqlOutputToMessages)(
-          (0, import_runtime_client_gql2.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
-        );
-        if (messages2.length === 0) {
-          continue;
-        }
-        newMessages = [];
-        if (((_h = value.generateCopilotResponse.status) == null ? void 0 : _h.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
-          newMessages = [
-            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_gql2.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
-              if (!(message.id in actionResults)) {
+          if (done) {
+            break;
+          }
+          if (!(value == null ? void 0 : value.generateCopilotResponse)) {
+            continue;
+          }
+          threadIdRef.current = value.generateCopilotResponse.threadId || null;
+          runIdRef.current = value.generateCopilotResponse.runId || null;
+          const messages2 = (0, import_runtime_client_gql2.convertGqlOutputToMessages)(
+            (0, import_runtime_client_gql2.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
+          );
+          if (messages2.length === 0) {
+            continue;
+          }
+          newMessages = [];
+          if (((_h = value.generateCopilotResponse.status) == null ? void 0 : _h.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
+            newMessages = [
+              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_gql2.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
+                if (!(message.id in actionResults)) {
+                  if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
+                    break;
+                  }
+                  try {
+                    setMessages([...previousMessages, ...newMessages]);
+                    const action = actions.find((action2) => action2.name === message.name);
+                    if (action) {
+                      followUp = action.followUp;
+                    }
+                    const result = yield onFunctionCall({
+                      messages: previousMessages,
+                      name: message.name,
+                      args: message.arguments
+                    });
+                    actionResults[message.id] = result;
+                  } catch (e) {
+                    actionResults[message.id] = `Failed to execute action ${message.name}`;
+                    console.error(`Failed to execute action ${message.name}: ${e}`);
+                  }
+                }
+                newMessages.push(
+                  new import_runtime_client_gql2.ResultMessage({
+                    result: import_runtime_client_gql2.ResultMessage.encodeResult(actionResults[message.id]),
+                    actionExecutionId: message.id,
+                    actionName: message.name
+                  })
+                );
+              }
+              if (message.isAgentStateMessage() && !message.active && !executedCoAgentStateRenders.includes(message.id) && onCoAgentStateRender) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
                 }
-                try {
-                  setMessages([...previousMessages, ...newMessages]);
-                  const action = actions.find((action2) => action2.name === message.name);
-                  if (action) {
-                    followUp = action.followUp;
-                  }
-                  const result = yield onFunctionCall({
-                    messages: previousMessages,
-                    name: message.name,
-                    args: message.arguments
-                  });
-                  actionResults[message.id] = result;
-                } catch (e) {
-                  actionResults[message.id] = `Failed to execute action ${message.name}`;
-                  console.error(`Failed to execute action ${message.name}: ${e}`);
-                }
+                yield onCoAgentStateRender({
+                  name: message.agentName,
+                  nodeName: message.nodeName,
+                  state: message.state
+                });
+                executedCoAgentStateRenders.push(message.id);
               }
-              newMessages.push(
-                new import_runtime_client_gql2.ResultMessage({
-                  result: import_runtime_client_gql2.ResultMessage.encodeResult(actionResults[message.id]),
-                  actionExecutionId: message.id,
-                  actionName: message.name
-                })
-              );
             }
-            if (message.isAgentStateMessage() && !message.active && !executedCoAgentStateRenders.includes(message.id) && onCoAgentStateRender) {
-              if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
-                break;
+            const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
+            if (lastAgentStateMessage) {
+              setCoagentStatesWithRef((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
+                [lastAgentStateMessage.agentName]: {
+                  name: lastAgentStateMessage.agentName,
+                  state: lastAgentStateMessage.state,
+                  running: lastAgentStateMessage.running,
+                  active: lastAgentStateMessage.active,
+                  threadId: lastAgentStateMessage.threadId,
+                  nodeName: lastAgentStateMessage.nodeName,
+                  runId: lastAgentStateMessage.runId
+                }
+              }));
+              if (lastAgentStateMessage.running) {
+                setAgentSession({
+                  threadId: lastAgentStateMessage.threadId,
+                  agentName: lastAgentStateMessage.agentName,
+                  nodeName: lastAgentStateMessage.nodeName
+                });
+              } else {
+                setAgentSession(null);
               }
-              yield onCoAgentStateRender({
-                name: message.agentName,
-                nodeName: message.nodeName,
-                state: message.state
-              });
-              executedCoAgentStateRenders.push(message.id);
             }
           }
-          const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
-          if (lastAgentStateMessage) {
-            setCoagentStatesWithRef((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
-              [lastAgentStateMessage.agentName]: {
-                name: lastAgentStateMessage.agentName,
-                state: lastAgentStateMessage.state,
-                running: lastAgentStateMessage.running,
-                active: lastAgentStateMessage.active,
-                threadId: lastAgentStateMessage.threadId,
-                nodeName: lastAgentStateMessage.nodeName,
-                runId: lastAgentStateMessage.runId
-              }
-            }));
-            if (lastAgentStateMessage.running) {
-              setAgentSession({
-                threadId: lastAgentStateMessage.threadId,
-                agentName: lastAgentStateMessage.agentName,
-                nodeName: lastAgentStateMessage.nodeName
-              });
-            } else {
-              setAgentSession(null);
-            }
+          if (newMessages.length > 0) {
+            setMessages([...previousMessages, ...newMessages]);
           }
         }
-        if (newMessages.length > 0) {
-          setMessages([...previousMessages, ...newMessages]);
+        if (
+          // if followUp is not explicitly false
+          followUp !== false && // if we have client side results
+          (Object.values(actionResults).length || // or the last message we received is a result
+          newMessages.length && newMessages[newMessages.length - 1].isResultMessage())
+        ) {
+          yield new Promise((resolve) => setTimeout(resolve, 10));
+          return yield runChatCompletionRef.current([...previousMessages, ...newMessages]);
+        } else {
+          return newMessages.slice();
         }
+      } finally {
+        setIsLoading(false);
       }
-      if (
-        // if followUp is not explicitly false
-        followUp !== false && // if we have client side results
-        (Object.values(actionResults).length || // or the last message we received is a result
-        newMessages.length && newMessages[newMessages.length - 1].isResultMessage())
-      ) {
-        yield new Promise((resolve) => setTimeout(resolve, 10));
-        return yield runChatCompletionRef.current([...previousMessages, ...newMessages]);
-      } else {
-        return newMessages.slice();
-      }
-    } finally {
-      setIsLoading(false);
-    }
-  });
+    }),
+    [
+      messages,
+      setMessages,
+      makeSystemMessageCallback,
+      copilotConfig,
+      setIsLoading,
+      initialMessages,
+      isLoading,
+      actions,
+      onFunctionCall,
+      onCoAgentStateRender,
+      setCoagentStatesWithRef,
+      coagentStatesRef,
+      agentSession,
+      setAgentSession
+    ]
+  );
   runChatCompletionRef.current = runChatCompletion;
-  const runChatCompletionAndHandleFunctionCall = (messages2) => __async(this, null, function* () {
-    yield runChatCompletionRef.current(messages2);
-  });
-  const append = (message) => __async(this, null, function* () {
-    if (isLoading) {
-      return;
-    }
-    const newMessages = [...messages, message];
-    setMessages(newMessages);
-    return runChatCompletionAndHandleFunctionCall(newMessages);
-  });
-  const reload = () => __async(this, null, function* () {
+  const runChatCompletionAndHandleFunctionCall = useAsyncCallback(
+    (messages2) => __async(this, null, function* () {
+      yield runChatCompletionRef.current(messages2);
+    }),
+    [messages]
+  );
+  const append = useAsyncCallback(
+    (message) => __async(this, null, function* () {
+      if (isLoading) {
+        return;
+      }
+      const newMessages = [...messages, message];
+      setMessages(newMessages);
+      return runChatCompletionAndHandleFunctionCall(newMessages);
+    }),
+    [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
+  );
+  const reload = useAsyncCallback(() => __async(this, null, function* () {
     if (isLoading || messages.length === 0) {
       return;
     }
@@ -455,7 +599,7 @@ function useChat(options) {
     }
     setMessages(newMessages);
     return runChatCompletionAndHandleFunctionCall(newMessages);
-  });
+  }), [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]);
   const stop = () => {
     var _a;
     (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
@@ -469,10 +613,10 @@ function useChat(options) {
 }
 
 // src/components/copilot-provider/copilotkit.tsx
-var import_react6 = require("react");
+var import_react7 = require("react");
 var import_react_dom = require("react-dom");
 var import_shared2 = require("@copilotkit/shared");
-var import_jsx_runtime2 = require("react/jsx-runtime");
+var import_jsx_runtime4 = require("react/jsx-runtime");
 var defaultCopilotContextCategories = ["global"];
 
 // src/hooks/use-copilot-chat.ts
@@ -498,13 +642,13 @@ function useCopilotChat(_a = {}) {
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
-  const deleteMessage = (0, import_react7.useCallback)(
+  const deleteMessage = (0, import_react8.useCallback)(
     (messageId) => {
       setMessages((prev) => prev.filter((message) => message.id !== messageId));
     },
     [setMessages]
   );
-  const makeSystemMessageCallback = (0, import_react7.useCallback)(() => {
+  const makeSystemMessageCallback = (0, import_react8.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
     return new import_runtime_client_gql3.TextMessage({
@@ -512,7 +656,7 @@ function useCopilotChat(_a = {}) {
       role: import_runtime_client_gql3.Role.System
     });
   }, [getContextString, makeSystemMessage, chatInstructions]);
-  const onCoAgentStateRender = (0, import_react7.useCallback)(
+  const onCoAgentStateRender = useAsyncCallback(
     (args) => __async(this, null, function* () {
       var _a2;
       const { name, nodeName, state } = args;
@@ -547,38 +691,38 @@ function useCopilotChat(_a = {}) {
     setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
-  const latestAppendFunc = (0, import_react7.useCallback)(
-    (message) => {
-      return latestAppend.current(message);
-    },
+  const latestAppendFunc = useAsyncCallback(
+    (message) => __async(this, null, function* () {
+      return yield latestAppend.current(message);
+    }),
     [latestAppend]
   );
   const latestReload = useUpdatedRef(reload);
-  const latestReloadFunc = (0, import_react7.useCallback)(() => {
-    return latestReload.current();
-  }, [latestReload]);
+  const latestReloadFunc = useAsyncCallback(() => __async(this, null, function* () {
+    return yield latestReload.current();
+  }), [latestReload]);
   const latestStop = useUpdatedRef(stop);
-  const latestStopFunc = (0, import_react7.useCallback)(() => {
+  const latestStopFunc = (0, import_react8.useCallback)(() => {
     return latestStop.current();
   }, [latestStop]);
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react7.useCallback)(
+  const latestDeleteFunc = (0, import_react8.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
   const latestSetMessages = useUpdatedRef(setMessages);
-  const latestSetMessagesFunc = (0, import_react7.useCallback)(
+  const latestSetMessagesFunc = (0, import_react8.useCallback)(
     (messages2) => {
       return latestSetMessages.current(messages2);
     },
     [latestSetMessages]
   );
   const latestRunChatCompletion = useUpdatedRef(runChatCompletion);
-  const latestRunChatCompletionFunc = (0, import_react7.useCallback)(() => {
-    return latestRunChatCompletion.current();
-  }, [latestRunChatCompletion]);
+  const latestRunChatCompletionFunc = useAsyncCallback(() => __async(this, null, function* () {
+    return yield latestRunChatCompletion.current();
+  }), [latestRunChatCompletion]);
   return {
     visibleMessages: messages,
     appendMessage: latestAppendFunc,
@@ -591,8 +735,8 @@ function useCopilotChat(_a = {}) {
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react7.useRef)(value);
-  (0, import_react7.useEffect)(() => {
+  const ref = (0, import_react8.useRef)(value);
+  (0, import_react8.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -661,28 +805,29 @@ function useCoAgent(options) {
   };
   const coagentState = getCoagentState(coagentStates, name);
   const state = isExternalStateManagement(options) ? options.state : coagentState.state;
-  (0, import_react8.useEffect)(() => {
+  (0, import_react9.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
     } else if (coagentStates[name] === void 0) {
       setState(options.initialState === void 0 ? {} : options.initialState);
     }
   }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
+  const runAgentCallback = useAsyncCallback(
+    (hint) => __async(this, null, function* () {
+      yield runAgent(name, context, appendMessage, runChatCompletion, hint);
+    }),
+    [name, context, appendMessage, runChatCompletion]
+  );
   return {
     name,
     nodeName: coagentState.nodeName,
-    state,
-    setState,
+    threadId: coagentState.threadId,
     running: coagentState.running,
-    start: () => {
-      startAgent(name, context);
-    },
-    stop: () => {
-      stopAgent(name, context);
-    },
-    run: (hint) => {
-      return runAgent(name, context, appendMessage, runChatCompletion, hint);
-    }
+    state: coagentState.state,
+    setState: isExternalStateManagement(options) ? options.setState : setState,
+    start: () => startAgent(name, context),
+    stop: () => stopAgent(name, context),
+    run: runAgentCallback
   };
 }
 function startAgent(name, context) {