@copilotkit/react-core 1.5.1-next.1 → 1.5.1-next.3

package/dist/index.js

@@ -87,10 +87,7 @@ __export(src_exports, {
   CopilotTask: () => CopilotTask,
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
-  runAgent: () => runAgent,
   shouldShowDevConsole: () => shouldShowDevConsole,
-  startAgent: () => startAgent,
-  stopAgent: () => stopAgent,
   useCoAgent: () => useCoAgent,
   useCoAgentStateRender: () => useCoAgentStateRender,
   useCopilotAction: () => useCopilotAction,
@@ -104,7 +101,7 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/components/copilot-provider/copilotkit.tsx
-var import_react10 = require("react");
+var import_react8 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -155,21 +152,9 @@ var emptyCopilotContext = {
   coagentStates: {},
   setCoagentStates: () => {
   },
-  coagentStatesRef: { current: {} },
-  setCoagentStatesWithRef: () => {
-  },
   agentSession: null,
   setAgentSession: () => {
-  },
-  forwardedParameters: {},
-  agentLock: null,
-  threadId: null,
-  setThreadId: () => {
-  },
-  runId: null,
-  setRunId: () => {
-  },
-  chatAbortControllerRef: { current: null }
+  }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -423,9 +408,6 @@ function CopilotMessages(_a) {
 }
 
 // src/components/toast/toast-provider.tsx
-var import_react7 = require("react");
-
-// src/components/error-boundary/error-utils.tsx
 var import_react6 = require("react");
 
 // src/components/toast/exclamation-mark-icon.tsx
@@ -455,94 +437,11 @@ var ExclamationMarkIcon = ({
   }
 );
 
-// src/components/error-boundary/error-utils.tsx
-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 code = "extensions" in error ? (_c = error.extensions) == null ? void 0 : _c.code : null;
-    return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
-      "div",
-      {
-        style: {
-          marginTop: idx === 0 ? 0 : 10,
-          marginBottom: 14
-        },
-        children: [
-          /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(ExclamationMarkIcon, { style: { marginBottom: 4 } }),
-          code && /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
-            "div",
-            {
-              style: {
-                fontWeight: "600",
-                marginBottom: 4
-              },
-              children: [
-                "Copilot Cloud Error:",
-                " ",
-                /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("span", { style: { fontFamily: "monospace", fontWeight: "normal" }, children: code })
-              ]
-            }
-          ),
-          /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { children: message })
-        ]
-      },
-      idx
-    );
-  });
-  return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
-    "div",
-    {
-      style: {
-        fontSize: "13px",
-        maxWidth: "600px"
-      },
-      children: [
-        errorsToRender,
-        /* @__PURE__ */ (0, import_jsx_runtime3.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_react6.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_runtime3.jsx)(ErrorToast, { errors: error })
-      });
-    },
-    [addToast]
-  );
-}
-function useAsyncCallback(callback, deps) {
-  const addErrorToast = useErrorToast();
-  return (0, import_react6.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_runtime4 = require("react/jsx-runtime");
-var ToastContext = (0, import_react7.createContext)(void 0);
+var import_jsx_runtime3 = require("react/jsx-runtime");
+var ToastContext = (0, import_react6.createContext)(void 0);
 function useToast() {
-  const context = (0, import_react7.useContext)(ToastContext);
+  const context = (0, import_react6.useContext)(ToastContext);
   if (!context) {
     throw new Error("useToast must be used within a ToastProvider");
   }
@@ -552,28 +451,68 @@ function ToastProvider({
   enabled,
   children
 }) {
-  const [toasts, setToasts] = (0, import_react7.useState)([]);
-  const addToast = (0, import_react7.useCallback)((toast) => {
-    var _a;
-    const id = (_a = toast.id) != null ? _a : Math.random().toString(36).substring(2, 9);
-    setToasts((currentToasts) => {
-      if (currentToasts.find((toast2) => toast2.id === id))
-        return currentToasts;
-      return [...currentToasts, __spreadProps(__spreadValues({}, toast), { id })];
-    });
+  const [toasts, setToasts] = (0, import_react6.useState)([]);
+  const addToast = (0, import_react6.useCallback)((toast) => {
+    const id = Math.random().toString(36).substring(2, 9);
+    setToasts((currentToasts) => [...currentToasts, __spreadProps(__spreadValues({}, toast), { id })]);
     if (toast.duration) {
       setTimeout(() => {
         removeToast(id);
       }, toast.duration);
     }
   }, []);
-  const addGraphQLErrorsToast = (0, import_react7.useCallback)((errors) => {
+  const addGraphQLErrorsToast = (0, import_react6.useCallback)((errors) => {
+    const errorsToRender = errors.map((error, idx) => {
+      var _a;
+      const message = error.message;
+      const code = (_a = error.extensions) == null ? void 0 : _a.code;
+      return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
+        "div",
+        {
+          style: {
+            marginTop: idx === 0 ? 0 : 10,
+            marginBottom: 14
+          },
+          children: [
+            /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(ExclamationMarkIcon, { style: { marginBottom: 4 } }),
+            code && /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
+              "div",
+              {
+                style: {
+                  fontWeight: "600",
+                  marginBottom: 4
+                },
+                children: [
+                  "Copilot Cloud Error:",
+                  " ",
+                  /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("span", { style: { fontFamily: "monospace", fontWeight: "normal" }, children: code })
+                ]
+              }
+            ),
+            /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { children: message })
+          ]
+        },
+        idx
+      );
+    });
     addToast({
       type: "error",
-      message: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ErrorToast, { errors })
+      message: /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
+        "div",
+        {
+          style: {
+            fontSize: "13px",
+            maxWidth: "600px"
+          },
+          children: [
+            errorsToRender,
+            /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { style: { fontSize: "11px", opacity: 0.75 }, children: "NOTE: This is a Copilot Cloud error, and it only displays during local development." })
+          ]
+        }
+      )
     });
   }, []);
-  const removeToast = (0, import_react7.useCallback)((id) => {
+  const removeToast = (0, import_react6.useCallback)((id) => {
     setToasts((currentToasts) => currentToasts.filter((toast) => toast.id !== id));
   }, []);
   const value = {
@@ -583,8 +522,8 @@ function ToastProvider({
     removeToast,
     enabled
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime4.jsxs)(ToastContext.Provider, { value, children: [
-    /* @__PURE__ */ (0, import_jsx_runtime4.jsxs)(
+  return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(ToastContext.Provider, { value, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
       "div",
       {
         style: {
@@ -598,7 +537,7 @@ function ToastProvider({
           gap: "0.5rem"
         },
         children: [
-          toasts.length > 1 && /* @__PURE__ */ (0, import_jsx_runtime4.jsx)("div", { style: { textAlign: "right" }, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
+          toasts.length > 1 && /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { style: { textAlign: "right" }, children: /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(
             "button",
             {
               onClick: () => setToasts([]),
@@ -613,7 +552,7 @@ function ToastProvider({
               children: "Close All"
             }
           ) }),
-          toasts.map((toast) => /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
+          toasts.map((toast) => /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(
             Toast,
             {
               message: toast.message,
@@ -639,7 +578,7 @@ function Toast({
     warning: "#eab308",
     error: "#ef4444"
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime4.jsxs)(
+  return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
     "div",
     {
       style: {
@@ -652,8 +591,8 @@ function Toast({
         minWidth: "200px"
       },
       children: [
-        /* @__PURE__ */ (0, import_jsx_runtime4.jsx)("div", { children: message }),
-        /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
+        /* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { children: message }),
+        /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(
           "button",
           {
             onClick: onClose,
@@ -678,17 +617,14 @@ function Toast({
 
 // src/hooks/use-copilot-runtime-client.ts
 var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
-var import_react8 = require("react");
+var import_react7 = require("react");
 var useCopilotRuntimeClient = (options) => {
   const { addGraphQLErrorsToast } = useToast();
-  const addErrorToast = useErrorToast();
-  const runtimeClient = (0, import_react8.useMemo)(() => {
+  const runtimeClient = (0, import_react7.useMemo)(() => {
     return new import_runtime_client_gql.CopilotRuntimeClient(__spreadProps(__spreadValues({}, options), {
       handleGQLErrors: (error) => {
         if (error.graphQLErrors.length) {
           addGraphQLErrorsToast(error.graphQLErrors);
-        } else {
-          addErrorToast([error]);
         }
       }
     }));
@@ -709,8 +645,7 @@ function extract(_0) {
     data,
     abortSignal,
     stream,
-    requestType = import_runtime_client_gql2.CopilotRequestType.Task,
-    forwardedParameters
+    requestType = import_runtime_client_gql2.CopilotRequestType.Task
   }) {
     var _a, _b;
     const { messages } = context;
@@ -757,10 +692,10 @@ function extract(_0) {
           metadata: {
             requestType
           },
-          forwardedParameters: __spreadProps(__spreadValues({}, forwardedParameters != null ? forwardedParameters : {}), {
+          forwardedParameters: {
             toolChoice: "function",
             toolChoiceFunctionName: action.name
-          })
+          }
         },
         properties: context.copilotApiConfig.properties,
         signal: abortSignal
@@ -844,41 +779,13 @@ function getHostname() {
   return "";
 }
 
-// src/components/error-boundary/error-boundary.tsx
-var import_react9 = __toESM(require("react"));
-var import_jsx_runtime5 = require("react/jsx-runtime");
-var CopilotErrorBoundary = class extends import_react9.default.Component {
-  constructor(props) {
-    super(props);
-    this.state = { hasError: false };
-  }
-  static getDerivedStateFromError(error) {
-    return { hasError: true, error };
-  }
-  componentDidCatch(error, errorInfo) {
-    console.error("CopilotKit Error:", error, errorInfo);
-  }
-  render() {
-    return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(ErrorToast2, { error: this.state.error, children: this.props.children });
-  }
-};
-function ErrorToast2({ error, children }) {
-  const addErrorToast = useErrorToast();
-  (0, import_react9.useEffect)(() => {
-    if (error) {
-      addErrorToast([error]);
-    }
-  }, [error, addErrorToast]);
-  return children;
-}
-
 // src/components/copilot-provider/copilotkit.tsx
-var import_jsx_runtime6 = require("react/jsx-runtime");
+var import_jsx_runtime4 = require("react/jsx-runtime");
 function CopilotKit(_a) {
   var _b = _a, { children } = _b, props = __objRest(_b, ["children"]);
   const showDevConsole = props.showDevConsole === void 0 ? "auto" : props.showDevConsole;
   const enabled = shouldShowDevConsole(showDevConsole);
-  return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(ToastProvider, { enabled, children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(CopilotErrorBoundary, { children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(CopilotKitInternal, __spreadProps(__spreadValues({}, props), { children })) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ToastProvider, { enabled, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(CopilotKitInternal, __spreadProps(__spreadValues({}, props), { children })) });
 }
 function CopilotKitInternal(_a) {
   var _b = _a, { children } = _b, props = __objRest(_b, ["children"]);
@@ -888,49 +795,49 @@ function CopilotKitInternal(_a) {
     );
   }
   const chatApiEndpoint = props.runtimeUrl || import_shared4.COPILOT_CLOUD_CHAT_URL;
-  const [actions, setActions] = (0, import_react10.useState)({});
-  const [coAgentStateRenders, setCoAgentStateRenders] = (0, import_react10.useState)({});
-  const chatComponentsCache = (0, import_react10.useRef)({
+  const [actions, setActions] = (0, import_react8.useState)({});
+  const [coAgentStateRenders, setCoAgentStateRenders] = (0, import_react8.useState)({});
+  const chatComponentsCache = (0, import_react8.useRef)({
     actions: {},
     coAgentStateRenders: {}
   });
   const { addElement, removeElement, printTree } = use_tree_default();
-  const [isLoading, setIsLoading] = (0, import_react10.useState)(false);
-  const [chatInstructions, setChatInstructions] = (0, import_react10.useState)("");
+  const [isLoading, setIsLoading] = (0, import_react8.useState)(false);
+  const [chatInstructions, setChatInstructions] = (0, import_react8.useState)("");
   const {
     addElement: addDocument,
     removeElement: removeDocument,
     allElements: allDocuments
   } = use_flat_category_store_default();
-  const setAction = (0, import_react10.useCallback)((id, action) => {
+  const setAction = (0, import_react8.useCallback)((id, action) => {
     setActions((prevPoints) => {
       return __spreadProps(__spreadValues({}, prevPoints), {
         [id]: action
       });
     });
   }, []);
-  const removeAction = (0, import_react10.useCallback)((id) => {
+  const removeAction = (0, import_react8.useCallback)((id) => {
     setActions((prevPoints) => {
       const newPoints = __spreadValues({}, prevPoints);
       delete newPoints[id];
       return newPoints;
     });
   }, []);
-  const setCoAgentStateRender = (0, import_react10.useCallback)((id, stateRender) => {
+  const setCoAgentStateRender = (0, import_react8.useCallback)((id, stateRender) => {
     setCoAgentStateRenders((prevPoints) => {
       return __spreadProps(__spreadValues({}, prevPoints), {
         [id]: stateRender
       });
     });
   }, []);
-  const removeCoAgentStateRender = (0, import_react10.useCallback)((id) => {
+  const removeCoAgentStateRender = (0, import_react8.useCallback)((id) => {
     setCoAgentStateRenders((prevPoints) => {
       const newPoints = __spreadValues({}, prevPoints);
       delete newPoints[id];
       return newPoints;
     });
   }, []);
-  const getContextString = (0, import_react10.useCallback)(
+  const getContextString = (0, import_react8.useCallback)(
     (documents, categories) => {
       const documentsString = documents.map((document) => {
         return `${document.name} (${document.sourceApplication}):
@@ -943,37 +850,37 @@ ${nonDocumentStrings}`;
     },
     [printTree]
   );
-  const addContext = (0, import_react10.useCallback)(
+  const addContext = (0, import_react8.useCallback)(
     (context, parentId, categories = defaultCopilotContextCategories) => {
       return addElement(context, categories, parentId);
     },
     [addElement]
   );
-  const removeContext = (0, import_react10.useCallback)(
+  const removeContext = (0, import_react8.useCallback)(
     (id) => {
       removeElement(id);
     },
     [removeElement]
   );
-  const getFunctionCallHandler = (0, import_react10.useCallback)(
+  const getFunctionCallHandler = (0, import_react8.useCallback)(
     (customEntryPoints) => {
       return entryPointsToFunctionCallHandler(Object.values(customEntryPoints || actions));
     },
     [actions]
   );
-  const getDocumentsContext = (0, import_react10.useCallback)(
+  const getDocumentsContext = (0, import_react8.useCallback)(
     (categories) => {
       return allDocuments(categories);
     },
     [allDocuments]
   );
-  const addDocumentContext = (0, import_react10.useCallback)(
+  const addDocumentContext = (0, import_react8.useCallback)(
     (documentPointer, categories = defaultCopilotContextCategories) => {
       return addDocument(documentPointer, categories);
     },
     [addDocument]
   );
-  const removeDocumentContext = (0, import_react10.useCallback)(
+  const removeDocumentContext = (0, import_react8.useCallback)(
     (documentId) => {
       removeDocument(documentId);
     },
@@ -986,7 +893,7 @@ ${nonDocumentStrings}`;
       );
     }
   }
-  const copilotApiConfig = (0, import_react10.useMemo)(() => {
+  const copilotApiConfig = (0, import_react8.useMemo)(() => {
     var _a2, _b2;
     let cloud = void 0;
     if (props.publicApiKey) {
@@ -1028,7 +935,7 @@ ${nonDocumentStrings}`;
     headers,
     credentials: copilotApiConfig.credentials
   });
-  const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = (0, import_react10.useState)({});
+  const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = (0, import_react8.useState)({});
   const addChatSuggestionConfiguration = (id, suggestion) => {
     setChatSuggestionConfiguration((prev) => __spreadProps(__spreadValues({}, prev), { [id]: suggestion }));
   };
@@ -1038,30 +945,16 @@ ${nonDocumentStrings}`;
       return rest;
     });
   };
-  const [coagentStates, setCoagentStates] = (0, import_react10.useState)({});
-  const coagentStatesRef = (0, import_react10.useRef)({});
-  const setCoagentStatesWithRef = (0, import_react10.useCallback)(
-    (value) => {
-      const newValue = typeof value === "function" ? value(coagentStatesRef.current) : value;
-      coagentStatesRef.current = newValue;
-      setCoagentStates((prev) => {
-        return newValue;
-      });
-    },
-    []
-  );
+  const [coagentStates, setCoagentStates] = (0, import_react8.useState)({});
   let initialAgentSession = null;
   if (props.agent) {
     initialAgentSession = {
       agentName: props.agent
     };
   }
-  const [agentSession, setAgentSession] = (0, import_react10.useState)(initialAgentSession);
-  const [threadId, setThreadId] = (0, import_react10.useState)(null);
-  const [runId, setRunId] = (0, import_react10.useState)(null);
-  const chatAbortControllerRef = (0, import_react10.useRef)(null);
+  const [agentSession, setAgentSession] = (0, import_react8.useState)(initialAgentSession);
   const showDevConsole = props.showDevConsole === void 0 ? "auto" : props.showDevConsole;
-  return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
     CopilotContext.Provider,
     {
       value: {
@@ -1090,20 +983,11 @@ ${nonDocumentStrings}`;
         showDevConsole,
         coagentStates,
         setCoagentStates,
-        coagentStatesRef,
-        setCoagentStatesWithRef,
         agentSession,
         setAgentSession,
-        runtimeClient,
-        forwardedParameters: props.forwardedParameters || {},
-        agentLock: props.agent || null,
-        threadId,
-        setThreadId,
-        runId,
-        setRunId,
-        chatAbortControllerRef
+        runtimeClient
       },
-      children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(CopilotMessages, { children })
+      children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(CopilotMessages, { children })
     }
   );
 }
@@ -1135,40 +1019,13 @@ function entryPointsToFunctionCallHandler(actions) {
 }
 
 // src/hooks/use-copilot-chat.ts
-var import_react12 = require("react");
-var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
-
-// src/hooks/use-chat.ts
-var import_react11 = require("react");
-var import_shared6 = require("@copilotkit/shared");
+var import_react10 = require("react");
 var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
 
-// src/types/frontend-action.ts
-var import_runtime_client_gql4 = 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 !== "*"
-  ).map((action) => {
-    let available = import_runtime_client_gql4.ActionInputAvailability.Enabled;
-    if (action.disabled) {
-      available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
-    } else if (action.available === "disabled") {
-      available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
-    } else if (action.available === "remote") {
-      available = import_runtime_client_gql4.ActionInputAvailability.Remote;
-    }
-    return {
-      name: action.name,
-      description: action.description || "",
-      jsonSchema: JSON.stringify((0, import_shared5.actionParametersToJsonSchema)(action.parameters || [])),
-      available
-    };
-  });
-  return filteredActions;
-}
-
 // src/hooks/use-chat.ts
+var import_react9 = require("react");
+var import_shared5 = require("@copilotkit/shared");
+var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 function useChat(options) {
   const {
     messages,
@@ -1181,309 +1038,232 @@ function useChat(options) {
     actions,
     onFunctionCall,
     onCoAgentStateRender,
-    setCoagentStatesWithRef,
-    coagentStatesRef,
+    setCoagentStates,
+    coagentStates,
     agentSession,
-    setAgentSession,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   } = options;
+  const abortControllerRef = (0, import_react9.useRef)();
+  const threadIdRef = (0, import_react9.useRef)(null);
+  const runIdRef = (0, import_react9.useRef)(null);
   const { addGraphQLErrorsToast } = useToast();
-  const runChatCompletionRef = (0, import_react11.useRef)();
-  const agentSessionRef = (0, import_react11.useRef)(agentSession);
+  const runChatCompletionRef = (0, import_react9.useRef)();
+  const coagentStatesRef = (0, import_react9.useRef)(coagentStates);
+  coagentStatesRef.current = coagentStates;
+  const agentSessionRef = (0, import_react9.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 publicApiKey = copilotConfig.publicApiKey;
-  const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared6.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
+  const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared5.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = useCopilotRuntimeClient({
     url: copilotConfig.chatApiEndpoint,
     publicApiKey: copilotConfig.publicApiKey,
     headers,
     credentials: copilotConfig.credentials
   });
-  const runChatCompletion = useAsyncCallback(
-    (previousMessages) => __async(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
-      setIsLoading(true);
-      let newMessages = [
-        new import_runtime_client_gql5.TextMessage({
-          content: "",
-          role: import_runtime_client_gql5.Role.Assistant
-        })
-      ];
-      chatAbortControllerRef.current = new AbortController();
-      setMessages([...previousMessages, ...newMessages]);
-      const systemMessage = makeSystemMessageCallback();
-      const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
-      const isAgentRun = agentSessionRef.current !== null;
-      const stream = runtimeClient.asStream(
-        runtimeClient.generateCopilotResponse({
-          data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
-            frontend: {
-              actions: processActionsForRuntimeRequest(actions),
-              url: window.location.href
-            },
-            threadId: threadIdRef.current,
-            runId: runIdRef.current,
-            messages: (0, import_runtime_client_gql5.convertMessagesToGqlInput)((0, import_runtime_client_gql5.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 = (previousMessages) => __async(this, null, function* () {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    setIsLoading(true);
+    let newMessages = [
+      new import_runtime_client_gql4.TextMessage({
+        content: "",
+        role: import_runtime_client_gql4.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_gql4.ActionInputAvailability.Disabled || !action.disabled
+            ).map((action) => {
+              let available = import_runtime_client_gql4.ActionInputAvailability.Enabled;
+              if (action.disabled) {
+                available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
+              } else if (action.available === "disabled") {
+                available = import_runtime_client_gql4.ActionInputAvailability.Disabled;
+              } else if (action.available === "remote") {
+                available = import_runtime_client_gql4.ActionInputAvailability.Remote;
+              }
+              return {
+                name: action.name,
+                description: action.description || "",
+                jsonSchema: JSON.stringify((0, import_shared5.actionParametersToJsonSchema)(action.parameters || [])),
+                available
+              };
+            }),
+            url: window.location.href
+          },
+          threadId: threadIdRef.current,
+          runId: runIdRef.current,
+          messages: (0, import_runtime_client_gql4.convertMessagesToGqlInput)((0, import_runtime_client_gql4.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_gql5.CopilotRequestType.Chat
-            }
-          }), agentSessionRef.current ? {
-            agentSession: agentSessionRef.current
-          } : {}), {
-            agentStates: Object.values(coagentStatesRef.current).map((state) => ({
-              agentName: state.name,
-              state: JSON.stringify(state.state)
-            })),
-            forwardedParameters: options.forwardedParameters || {}
-          }),
-          properties: copilotConfig.properties,
-          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;
-      const reader = stream.getReader();
-      let executedCoAgentStateRenders = [];
-      let followUp = void 0;
-      let messages2 = [];
-      let syncedMessages = [];
-      try {
-        while (true) {
-          let done, value;
-          try {
-            const readResult = yield reader.read();
-            done = readResult.done;
-            value = readResult.value;
-          } catch (readError) {
-            break;
-          }
-          if (done) {
-            if (chatAbortControllerRef.current.signal.aborted) {
-              return [];
             }
-            break;
-          }
-          if (!(value == null ? void 0 : value.generateCopilotResponse)) {
-            continue;
-          }
-          threadIdRef.current = value.generateCopilotResponse.threadId || null;
-          runIdRef.current = value.generateCopilotResponse.runId || null;
-          setThreadId(threadIdRef.current);
-          setRunId(runIdRef.current);
-          messages2 = (0, import_runtime_client_gql5.convertGqlOutputToMessages)(
-            (0, import_runtime_client_gql5.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
-          );
-          if (messages2.length === 0) {
-            continue;
+          } : {})
+        } : {}), {
+          metadata: {
+            requestType: import_runtime_client_gql4.CopilotRequestType.Chat
           }
-          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,
-                content: ((_i = value.generateCopilotResponse.status.details) == null ? void 0 : _i.guardrailsReason) || ""
-              })
-            ];
-            setMessages([...previousMessages, ...newMessages]);
-            break;
-          } else {
-            newMessages = [...messages2];
-            for (const message of messages2) {
-              if (message.isAgentStateMessage() && !message.active && !executedCoAgentStateRenders.includes(message.id) && onCoAgentStateRender) {
+        }), 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_gql4.convertGqlOutputToMessages)(
+          (0, import_runtime_client_gql4.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_gql4.TextMessage({
+              role: import_runtime_client_gql4.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_gql4.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
+              if (!(message.id in actionResults)) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
                 }
-                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) {
-              if (lastAgentStateMessage.state.messages && lastAgentStateMessage.state.messages.length > 0) {
-                syncedMessages = (0, import_runtime_client_gql5.loadMessagesFromJsonRepresentation)(
-                  lastAgentStateMessage.state.messages
-                );
-              }
-              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]);
-          }
-        }
-        const finalMessages = constructFinalMessages(syncedMessages, previousMessages, newMessages);
-        let didExecuteAction = false;
-        if (onFunctionCall) {
-          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) {
-              lastMessages.unshift(message);
-            } else {
-              break;
-            }
-          }
-          for (const message of lastMessages) {
-            setMessages(finalMessages);
-            const action = actions.find((action2) => action2.name === message.name);
-            if (action) {
-              followUp = action.followUp;
-              let result;
-              try {
-                result = yield Promise.race([
-                  onFunctionCall({
+                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
-                  }),
-                  new Promise(
-                    (resolve) => {
-                      var _a2;
-                      return (_a2 = chatAbortControllerRef.current) == null ? void 0 : _a2.signal.addEventListener(
-                        "abort",
-                        () => resolve("Operation was aborted by the user")
-                      );
-                    }
-                  ),
-                  // if the user stopped generation, we also abort consecutive actions
-                  new Promise((resolve) => {
-                    var _a2;
-                    if ((_a2 = chatAbortControllerRef.current) == null ? void 0 : _a2.signal.aborted) {
-                      resolve("Operation was aborted by the user");
-                    }
-                  })
-                ]);
-              } catch (e) {
-                result = `Failed to execute action ${message.name}`;
-                console.error(`Failed to execute action ${message.name}: ${e}`);
+                  });
+                  actionResults[message.id] = result;
+                } catch (e) {
+                  actionResults[message.id] = `Failed to execute action ${message.name}`;
+                  console.error(`Failed to execute action ${message.name}: ${e}`);
+                }
               }
-              didExecuteAction = true;
-              const messageIndex = finalMessages.findIndex((msg) => msg.id === message.id);
-              finalMessages.splice(
-                messageIndex + 1,
-                0,
-                new import_runtime_client_gql5.ResultMessage({
-                  id: "result-" + message.id,
-                  result: import_runtime_client_gql5.ResultMessage.encodeResult(result),
+              newMessages.push(
+                new import_runtime_client_gql4.ResultMessage({
+                  result: import_runtime_client_gql4.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;
+              }
+              yield onCoAgentStateRender({
+                name: message.agentName,
+                nodeName: message.nodeName,
+                state: message.state
+              });
+              executedCoAgentStateRenders.push(message.id);
+            }
           }
-          setMessages(finalMessages);
-        }
-        if (
-          // if followUp is not explicitly false
-          followUp !== false && // and we executed an action
-          (didExecuteAction || // the last message is a server side result
-          !isAgentRun && finalMessages.length && finalMessages[finalMessages.length - 1].isResultMessage()) && // the user did not stop generation
-          !((_j = chatAbortControllerRef.current) == null ? void 0 : _j.signal.aborted)
-        ) {
-          yield new Promise((resolve) => setTimeout(resolve, 10));
-          return yield runChatCompletionRef.current(finalMessages);
-        } else if ((_k = chatAbortControllerRef.current) == null ? void 0 : _k.signal.aborted) {
-          const repairedMessages = finalMessages.filter((message, actionExecutionIndex) => {
-            if (message.isActionExecutionMessage()) {
-              return finalMessages.find(
-                (msg, resultIndex) => msg.isResultMessage() && msg.actionExecutionId === message.id && resultIndex === actionExecutionIndex + 1
-              );
+          const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
+          if (lastAgentStateMessage) {
+            setCoagentStates((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);
             }
-            return true;
-          });
-          const repairedMessageIds = repairedMessages.map((message) => message.id);
-          setMessages(repairedMessages);
-          if ((_l = agentSessionRef.current) == null ? void 0 : _l.nodeName) {
-            setAgentSession({
-              threadId: agentSessionRef.current.threadId,
-              agentName: agentSessionRef.current.agentName,
-              nodeName: "__end__"
-            });
           }
-          return newMessages.filter((message) => repairedMessageIds.includes(message.id));
-        } else {
-          return newMessages.slice();
         }
-      } finally {
-        setIsLoading(false);
+        if (newMessages.length > 0) {
+          setMessages([...previousMessages, ...newMessages]);
+        }
       }
-    }),
-    [
-      messages,
-      setMessages,
-      makeSystemMessageCallback,
-      copilotConfig,
-      setIsLoading,
-      initialMessages,
-      isLoading,
-      actions,
-      onFunctionCall,
-      onCoAgentStateRender,
-      setCoagentStatesWithRef,
-      coagentStatesRef,
-      agentSession,
-      setAgentSession
-    ]
-  );
-  runChatCompletionRef.current = runChatCompletion;
-  const runChatCompletionAndHandleFunctionCall = useAsyncCallback(
-    (messages2) => __async(this, null, function* () {
-      yield runChatCompletionRef.current(messages2);
-    }),
-    [messages]
-  );
-  const append = useAsyncCallback(
-    (message) => __async(this, null, function* () {
-      if (isLoading) {
-        return;
+      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();
       }
-      const newMessages = [...messages, message];
-      setMessages(newMessages);
-      return runChatCompletionAndHandleFunctionCall(newMessages);
-    }),
-    [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]
-  );
-  const reload = useAsyncCallback(() => __async(this, null, function* () {
+    } finally {
+      setIsLoading(false);
+    }
+  });
+  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* () {
     if (isLoading || messages.length === 0) {
       return;
     }
@@ -1494,35 +1274,17 @@ function useChat(options) {
     }
     setMessages(newMessages);
     return runChatCompletionAndHandleFunctionCall(newMessages);
-  }), [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]);
+  });
   const stop = () => {
     var _a;
-    (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort("Stop was called");
+    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
   };
   return {
     append,
     reload,
-    stop,
-    runChatCompletion: () => runChatCompletionRef.current(messages)
+    stop
   };
 }
-function constructFinalMessages(syncedMessages, previousMessages, newMessages) {
-  const finalMessages = syncedMessages.length > 0 ? [...syncedMessages] : [...previousMessages, ...newMessages];
-  if (syncedMessages.length > 0) {
-    const messagesWithAgentState = [...previousMessages, ...newMessages];
-    let previousMessageId = void 0;
-    for (const message of messagesWithAgentState) {
-      if (message.isAgentStateMessage()) {
-        const index = finalMessages.findIndex((msg) => msg.id === previousMessageId);
-        if (index !== -1) {
-          finalMessages.splice(index + 1, 0, message);
-        }
-      }
-      previousMessageId = message.id;
-    }
-  }
-  return finalMessages;
-}
 
 // src/hooks/use-copilot-chat.ts
 function useCopilotChat(_a = {}) {
@@ -1539,36 +1301,29 @@ function useCopilotChat(_a = {}) {
     setIsLoading,
     chatInstructions,
     actions,
-    coagentStatesRef,
-    setCoagentStatesWithRef,
+    coagentStates,
+    setCoagentStates,
     coAgentStateRenders,
     agentSession,
-    setAgentSession,
-    forwardedParameters,
-    agentLock,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
-  const deleteMessage = (0, import_react12.useCallback)(
+  const deleteMessage = (0, import_react10.useCallback)(
     (messageId) => {
       setMessages((prev) => prev.filter((message) => message.id !== messageId));
     },
     [setMessages]
   );
-  const makeSystemMessageCallback = (0, import_react12.useCallback)(() => {
+  const makeSystemMessageCallback = (0, import_react10.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage2 || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql6.TextMessage({
+    return new import_runtime_client_gql5.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql6.Role.System
+      role: import_runtime_client_gql5.Role.System
     });
   }, [getContextString, makeSystemMessage2, chatInstructions]);
-  const onCoAgentStateRender = useAsyncCallback(
+  const onCoAgentStateRender = (0, import_react10.useCallback)(
     (args) => __async(this, null, function* () {
       var _a2;
       const { name, nodeName, state } = args;
@@ -1586,7 +1341,7 @@ function useCopilotChat(_a = {}) {
     }),
     [coAgentStateRenders]
   );
-  const { append, reload, stop, runChatCompletion } = useChat(__spreadProps(__spreadValues({}, options), {
+  const { append, reload, stop } = useChat(__spreadProps(__spreadValues({}, options), {
     actions: Object.values(actions),
     copilotConfig: copilotApiConfig,
     initialMessages: options.initialMessages || [],
@@ -1597,90 +1352,53 @@ function useCopilotChat(_a = {}) {
     makeSystemMessageCallback,
     isLoading,
     setIsLoading,
-    coagentStatesRef,
-    setCoagentStatesWithRef,
+    coagentStates,
+    setCoagentStates,
     agentSession,
-    setAgentSession,
-    forwardedParameters,
-    threadId,
-    setThreadId,
-    runId,
-    setRunId,
-    chatAbortControllerRef
+    setAgentSession
   }));
   const latestAppend = useUpdatedRef(append);
-  const latestAppendFunc = useAsyncCallback(
-    (message) => __async(this, null, function* () {
-      return yield latestAppend.current(message);
-    }),
+  const latestAppendFunc = (0, import_react10.useCallback)(
+    (message) => {
+      return latestAppend.current(message);
+    },
     [latestAppend]
   );
   const latestReload = useUpdatedRef(reload);
-  const latestReloadFunc = useAsyncCallback(() => __async(this, null, function* () {
-    return yield latestReload.current();
-  }), [latestReload]);
+  const latestReloadFunc = (0, import_react10.useCallback)(() => {
+    return latestReload.current();
+  }, [latestReload]);
   const latestStop = useUpdatedRef(stop);
-  const latestStopFunc = (0, import_react12.useCallback)(() => {
+  const latestStopFunc = (0, import_react10.useCallback)(() => {
     return latestStop.current();
   }, [latestStop]);
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react12.useCallback)(
+  const latestDeleteFunc = (0, import_react10.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
   const latestSetMessages = useUpdatedRef(setMessages);
-  const latestSetMessagesFunc = (0, import_react12.useCallback)(
+  const latestSetMessagesFunc = (0, import_react10.useCallback)(
     (messages2) => {
       return latestSetMessages.current(messages2);
     },
     [latestSetMessages]
   );
-  const latestRunChatCompletion = useUpdatedRef(runChatCompletion);
-  const latestRunChatCompletionFunc = useAsyncCallback(() => __async(this, null, function* () {
-    return yield latestRunChatCompletion.current();
-  }), [latestRunChatCompletion]);
-  const reset = (0, import_react12.useCallback)(() => {
-    latestStopFunc();
-    setMessages([]);
-    setThreadId(null);
-    setRunId(null);
-    setCoagentStatesWithRef({});
-    let initialAgentSession = null;
-    if (agentLock) {
-      initialAgentSession = {
-        agentName: agentLock
-      };
-    }
-    setAgentSession(initialAgentSession);
-  }, [
-    latestStopFunc,
-    setMessages,
-    setThreadId,
-    setCoagentStatesWithRef,
-    setAgentSession,
-    agentLock
-  ]);
-  const latestReset = useUpdatedRef(reset);
-  const latestResetFunc = (0, import_react12.useCallback)(() => {
-    return latestReset.current();
-  }, [latestReset]);
   return {
     visibleMessages: messages,
     appendMessage: latestAppendFunc,
     setMessages: latestSetMessagesFunc,
     reloadMessages: latestReloadFunc,
     stopGeneration: latestStopFunc,
-    reset: latestResetFunc,
     deleteMessage: latestDeleteFunc,
-    runChatCompletion: latestRunChatCompletionFunc,
     isLoading
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react12.useRef)(value);
-  (0, import_react12.useEffect)(() => {
+  const ref = (0, import_react10.useRef)(value);
+  (0, import_react10.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -1710,22 +1428,18 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-copilot-action.ts
-var import_shared7 = require("@copilotkit/shared");
-var import_react13 = require("react");
+var import_shared6 = require("@copilotkit/shared");
+var import_react11 = require("react");
 function useCopilotAction(action, dependencies) {
   const { setAction, removeAction, actions, chatComponentsCache } = useCopilotContext();
-  const idRef = (0, import_react13.useRef)((0, import_shared7.randomId)());
-  const renderAndWaitRef = (0, import_react13.useRef)(null);
+  const idRef = (0, import_react11.useRef)((0, import_shared6.randomId)());
+  const renderAndWaitRef = (0, import_react11.useRef)(null);
   action = __spreadValues({}, action);
-  if (
-    // renderAndWaitForResponse is not available for catch all actions
-    isFrontendAction(action) && // check if renderAndWaitForResponse is set
-    (action.renderAndWait || action.renderAndWaitForResponse)
-  ) {
+  if (action.renderAndWait || action.renderAndWaitForResponse) {
     const renderAndWait = action.renderAndWait || action.renderAndWaitForResponse;
     action.renderAndWait = void 0;
     action.renderAndWaitForResponse = void 0;
-    action.handler = useAsyncCallback(() => __async(this, null, function* () {
+    action.handler = () => __async(this, null, function* () {
       let resolve;
       let reject;
       const promise = new Promise((resolvePromise, rejectPromise) => {
@@ -1734,18 +1448,14 @@ function useCopilotAction(action, dependencies) {
       });
       renderAndWaitRef.current = { promise, resolve, reject };
       return yield promise;
-    }), []);
+    });
     action.render = (props) => {
-      let status = props.status;
-      if (props.status === "executing" && !renderAndWaitRef.current) {
-        status = "inProgress";
-      }
       const waitProps = {
-        status,
+        status: props.status,
         args: props.args,
         result: props.result,
-        handler: status === "executing" ? renderAndWaitRef.current.resolve : void 0,
-        respond: status === "executing" ? renderAndWaitRef.current.resolve : void 0
+        handler: props.status === "executing" ? renderAndWaitRef.current.resolve : void 0,
+        respond: props.status === "executing" ? renderAndWaitRef.current.resolve : void 0
       };
       const isNoArgsRenderWait = (_fn) => {
         var _a;
@@ -1758,14 +1468,12 @@ function useCopilotAction(action, dependencies) {
           return renderAndWait(waitProps);
         }
       }
-      return (0, import_react13.createElement)(import_react13.Fragment);
+      return (0, import_react11.createElement)(import_react11.Fragment);
     };
   }
   if (dependencies === void 0) {
     if (actions[idRef.current]) {
-      if (isFrontendAction(action)) {
-        actions[idRef.current].handler = action.handler;
-      }
+      actions[idRef.current].handler = action.handler;
       if (typeof action.render === "function") {
         if (chatComponentsCache.current !== null) {
           chatComponentsCache.current.actions[action.name] = action.render;
@@ -1773,7 +1481,7 @@ function useCopilotAction(action, dependencies) {
       }
     }
   }
-  (0, import_react13.useEffect)(() => {
+  (0, import_react11.useEffect)(() => {
     setAction(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
       chatComponentsCache.current.actions[action.name] = action.render;
@@ -1784,34 +1492,31 @@ function useCopilotAction(action, dependencies) {
   }, [
     setAction,
     removeAction,
-    isFrontendAction(action) ? action.description : void 0,
+    action.description,
     action.name,
-    isFrontendAction(action) ? action.disabled : void 0,
-    isFrontendAction(action) ? action.available : void 0,
+    action.disabled,
+    action.available,
     // This should be faster than deep equality checking
     // In addition, all major JS engines guarantee the order of object keys
-    JSON.stringify(isFrontendAction(action) ? action.parameters : []),
+    JSON.stringify(action.parameters),
     // include render only if it's a string
     typeof action.render === "string" ? action.render : void 0,
     // dependencies set by the developer
     ...dependencies || []
   ]);
 }
-function isFrontendAction(action) {
-  return action.name !== "*";
-}
 
 // src/hooks/use-coagent-state-render.ts
-var import_react14 = require("react");
-var import_shared8 = require("@copilotkit/shared");
+var import_react12 = require("react");
+var import_shared7 = require("@copilotkit/shared");
 function useCoAgentStateRender(action, dependencies) {
   const {
     setCoAgentStateRender,
     removeCoAgentStateRender,
     coAgentStateRenders,
     chatComponentsCache
-  } = (0, import_react14.useContext)(CopilotContext);
-  const idRef = (0, import_react14.useRef)((0, import_shared8.randomId)());
+  } = (0, import_react12.useContext)(CopilotContext);
+  const idRef = (0, import_react12.useRef)((0, import_shared7.randomId)());
   const key = `${action.name}-${action.nodeName || "global"}`;
   if (dependencies === void 0) {
     if (coAgentStateRenders[idRef.current]) {
@@ -1823,7 +1528,7 @@ function useCoAgentStateRender(action, dependencies) {
       }
     }
   }
-  (0, import_react14.useEffect)(() => {
+  (0, import_react12.useEffect)(() => {
     setCoAgentStateRender(idRef.current, action);
     if (chatComponentsCache.current !== null && action.render !== void 0) {
       chatComponentsCache.current.coAgentStateRenders[key] = action.render;
@@ -1843,11 +1548,11 @@ function useCoAgentStateRender(action, dependencies) {
 }
 
 // src/hooks/use-make-copilot-document-readable.ts
-var import_react15 = require("react");
+var import_react13 = require("react");
 function useMakeCopilotDocumentReadable(document, categories, dependencies = []) {
   const { addDocumentContext, removeDocumentContext } = useCopilotContext();
-  const idRef = (0, import_react15.useRef)();
-  (0, import_react15.useEffect)(() => {
+  const idRef = (0, import_react13.useRef)();
+  (0, import_react13.useEffect)(() => {
     const id = addDocumentContext(document, categories);
     idRef.current = id;
     return () => {
@@ -1858,16 +1563,16 @@ function useMakeCopilotDocumentReadable(document, categories, dependencies = [])
 }
 
 // src/hooks/use-copilot-readable.ts
-var import_react16 = require("react");
+var import_react14 = require("react");
 function convertToJSON(description, value) {
   return `${description}: ${typeof value === "string" ? value : JSON.stringify(value)}`;
 }
 function useCopilotReadable({ description, value, parentId, categories, convert }, dependencies) {
   const { addContext, removeContext } = useCopilotContext();
-  const idRef = (0, import_react16.useRef)();
+  const idRef = (0, import_react14.useRef)();
   convert = convert || convertToJSON;
   const information = convert(description, value);
-  (0, import_react16.useEffect)(() => {
+  (0, import_react14.useEffect)(() => {
     const id = addContext(information, parentId, categories);
     idRef.current = id;
     return () => {
@@ -1878,7 +1583,7 @@ function useCopilotReadable({ description, value, parentId, categories, convert
 }
 
 // src/hooks/use-coagent.ts
-var import_react17 = require("react");
+var import_react15 = require("react");
 function useCoAgent(options) {
   const isExternalStateManagement = (options2) => {
     return "state" in options2 && "setState" in options2;
@@ -1890,8 +1595,8 @@ function useCoAgent(options) {
   const generalContext = useCopilotContext();
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
-  const { coagentStates, coagentStatesRef, setCoagentStatesWithRef } = context;
-  const { appendMessage, runChatCompletion } = useCopilotChat();
+  const { coagentStates, setCoagentStates } = context;
+  const { appendMessage } = useCopilotChat();
   const getCoagentState = (coagentStates2, name2) => {
     if (coagentStates2[name2]) {
       return coagentStates2[name2];
@@ -1908,43 +1613,40 @@ function useCoAgent(options) {
     }
   };
   const setState = (newState) => {
-    let coagentState2 = getCoagentState(coagentStatesRef.current || {}, name);
-    const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
-    setCoagentStatesWithRef(__spreadProps(__spreadValues({}, coagentStatesRef.current), {
-      [name]: __spreadProps(__spreadValues({}, coagentState2), {
-        state: updatedState
-      })
-    }));
+    setCoagentStates((prevAgentStates) => {
+      let coagentState2 = getCoagentState(prevAgentStates, name);
+      const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
+      return __spreadProps(__spreadValues({}, prevAgentStates), {
+        [name]: __spreadProps(__spreadValues({}, coagentState2), {
+          state: updatedState
+        })
+      });
+    });
   };
   const coagentState = getCoagentState(coagentStates, name);
   const state = isExternalStateManagement(options) ? options.state : coagentState.state;
-  (0, import_react17.useEffect)(() => {
+  (0, import_react15.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,
-    // reset initialstate on reset
-    coagentStates[name] === void 0
-  ]);
-  const runAgentCallback = useAsyncCallback(
-    (hint) => __async(this, null, function* () {
-      yield runAgent(name, context, appendMessage, runChatCompletion, hint);
-    }),
-    [name, context, appendMessage, runChatCompletion]
-  );
+  }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
   return {
     name,
     nodeName: coagentState.nodeName,
-    threadId: coagentState.threadId,
+    state,
+    setState,
     running: coagentState.running,
-    state: coagentState.state,
-    setState: isExternalStateManagement(options) ? options.setState : setState,
-    start: () => startAgent(name, context),
-    stop: () => stopAgent(name, context),
-    run: runAgentCallback
+    start: () => {
+      startAgent(name, context);
+    },
+    stop: () => {
+      stopAgent(name, context);
+    },
+    run: (hint) => {
+      return runAgent(name, context, appendMessage, hint);
+    }
   };
 }
 function startAgent(name, context) {
@@ -1957,22 +1659,11 @@ 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}`);
   }
 }
-function runAgent(name, context, appendMessage, runChatCompletion, hint) {
+function runAgent(name, context, appendMessage, hint) {
   return __async(this, null, function* () {
     var _a, _b;
     const { agentSession, setAgentSession } = context;
@@ -1988,29 +1679,25 @@ function runAgent(name, context, appendMessage, runChatCompletion, hint) {
         previousState = message.state;
       }
     }
-    let state = ((_b = (_a = context.coagentStatesRef.current) == null ? void 0 : _a[name]) == null ? void 0 : _b.state) || {};
+    let state = ((_b = (_a = context.coagentStates) == null ? void 0 : _a[name]) == null ? void 0 : _b.state) || {};
     if (hint) {
       const hintMessage = hint({ previousState, currentState: state });
       if (hintMessage) {
         yield appendMessage(hintMessage);
-      } else {
-        yield runChatCompletion();
       }
-    } else {
-      yield runChatCompletion();
     }
   });
 }
 
 // src/lib/copilot-task.ts
-var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
+var import_shared8 = require("@copilotkit/shared");
 var CopilotTask = class {
   constructor(config) {
     this.instructions = config.instructions;
     this.actions = config.actions || [];
     this.includeCopilotReadable = config.includeCopilotReadable !== false;
     this.includeCopilotActions = config.includeCopilotActions !== false;
-    this.forwardedParameters = config.forwardedParameters;
   }
   /**
    * Run the task.
@@ -2019,7 +1706,7 @@ var CopilotTask = class {
    */
   run(context, data) {
     return __async(this, null, function* () {
-      var _a, _b, _c;
+      var _a, _b;
       const actions = this.includeCopilotActions ? Object.assign({}, context.actions) : {};
       for (const fn of this.actions) {
         actions[fn.name] = fn;
@@ -2031,12 +1718,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_gql6.TextMessage({
         content: taskSystemMessage(contextString, this.instructions),
-        role: import_runtime_client_gql7.Role.System
+        role: import_runtime_client_gql6.Role.System
       });
       const messages = [systemMessage];
-      const runtimeClient = new import_runtime_client_gql7.CopilotRuntimeClient({
+      const runtimeClient = new import_runtime_client_gql6.CopilotRuntimeClient({
         url: context.copilotApiConfig.chatApiEndpoint,
         publicApiKey: context.copilotApiConfig.publicApiKey,
         headers: context.copilotApiConfig.headers,
@@ -2045,22 +1732,26 @@ var CopilotTask = class {
       const response = yield runtimeClient.generateCopilotResponse({
         data: {
           frontend: {
-            actions: processActionsForRuntimeRequest(Object.values(actions)),
+            actions: Object.values(actions).map((action) => ({
+              name: action.name,
+              description: action.description || "",
+              jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters || []))
+            })),
             url: window.location.href
           },
-          messages: (0, import_runtime_client_gql7.convertMessagesToGqlInput)((0, import_runtime_client_gql7.filterAgentStateMessages)(messages)),
+          messages: (0, import_runtime_client_gql6.convertMessagesToGqlInput)((0, import_runtime_client_gql6.filterAgentStateMessages)(messages)),
           metadata: {
-            requestType: import_runtime_client_gql7.CopilotRequestType.Task
+            requestType: import_runtime_client_gql6.CopilotRequestType.Task
           },
-          forwardedParameters: __spreadProps(__spreadValues({}, (_a = this.forwardedParameters) != null ? _a : {}), {
+          forwardedParameters: {
             toolChoice: "required"
-          })
+          }
         },
         properties: context.copilotApiConfig.properties
       }).toPromise();
       const functionCallHandler = context.getFunctionCallHandler(actions);
-      const functionCalls = (0, import_runtime_client_gql7.convertGqlOutputToMessages)(
-        ((_c = (_b = response.data) == null ? void 0 : _b.generateCopilotResponse) == null ? void 0 : _c.messages) || []
+      const functionCalls = (0, import_runtime_client_gql6.convertGqlOutputToMessages)(
+        ((_b = (_a = response.data) == null ? void 0 : _a.generateCopilotResponse) == null ? void 0 : _b.messages) || []
       ).filter((m) => m.isActionExecutionMessage());
       for (const functionCall of functionCalls) {
         yield functionCallHandler({
@@ -2105,10 +1796,7 @@ ${instructions}
   CopilotTask,
   defaultCopilotContextCategories,
   extract,
-  runAgent,
   shouldShowDevConsole,
-  startAgent,
-  stopAgent,
   useCoAgent,
   useCoAgentStateRender,
   useCopilotAction,