@copilotkit/react-core 1.1.1-feat-runtime-remote-actions.4 → 1.1.1-feat-runtime-remote-actions.5

package/dist/index.js

@@ -86,10 +86,10 @@ __export(src_exports, {
   CopilotTask: () => CopilotTask,
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
-  useAgentState: () => useAgentState,
+  useCoagent: () => useCoagent,
+  useCoagentAction: () => useCoagentAction,
   useCopilotAction: () => useCopilotAction,
   useCopilotChat: () => useCopilotChat,
-  useCopilotChatUI: () => useCopilotChatUI,
   useCopilotContext: () => useCopilotContext,
   useCopilotReadable: () => useCopilotReadable,
   useMakeCopilotDocumentReadable: () => useMakeCopilotDocumentReadable
@@ -107,7 +107,12 @@ var emptyCopilotContext = {
   },
   removeAction: () => {
   },
-  chatComponentsCache: { current: { actions: {}, chatUI: {} } },
+  coagentActions: {},
+  setCoagentAction: () => {
+  },
+  removeCoagentAction: () => {
+  },
+  chatComponentsCache: { current: { actions: {}, coagentActions: {} } },
   getContextString: (documents, categories) => returnAndThrowInDebug(""),
   addContext: () => "",
   removeContext: () => {
@@ -141,11 +146,11 @@ var emptyCopilotContext = {
   removeChatSuggestionConfiguration: () => {
   },
   showDevConsole: "auto",
-  agentStates: {},
-  setAgentStates: () => {
+  coagentStates: {},
+  setCoagentStates: () => {
   },
-  chatUI: [],
-  setChatUI: () => {
+  agentSession: null,
+  setAgentSession: () => {
   }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
@@ -370,7 +375,11 @@ function CopilotKit(_a) {
   }
   const chatApiEndpoint = props.runtimeUrl || import_shared3.COPILOT_CLOUD_CHAT_URL;
   const [actions, setActions] = (0, import_react4.useState)({});
-  const chatComponentsCache = (0, import_react4.useRef)({ actions: {}, chatUI: {} });
+  const [coagentActions, setCoagentActions] = (0, import_react4.useState)({});
+  const chatComponentsCache = (0, import_react4.useRef)({
+    actions: {},
+    coagentActions: {}
+  });
   const { addElement, removeElement, printTree } = use_tree_default();
   const [messages, setMessages] = (0, import_react4.useState)([]);
   const [isLoading, setIsLoading] = (0, import_react4.useState)(false);
@@ -394,6 +403,20 @@ function CopilotKit(_a) {
       return newPoints;
     });
   }, []);
+  const setCoagentAction = (0, import_react4.useCallback)((id, action) => {
+    setCoagentActions((prevPoints) => {
+      return __spreadProps(__spreadValues({}, prevPoints), {
+        [id]: action
+      });
+    });
+  }, []);
+  const removeCoagentAction = (0, import_react4.useCallback)((id) => {
+    setCoagentActions((prevPoints) => {
+      const newPoints = __spreadValues({}, prevPoints);
+      delete newPoints[id];
+      return newPoints;
+    });
+  }, []);
   const getContextString = (0, import_react4.useCallback)(
     (documents, categories) => {
       const documentsString = documents.map((document) => {
@@ -484,8 +507,14 @@ ${nonDocumentStrings}`;
       return rest;
     });
   };
-  const [agentStates, setAgentStates] = (0, import_react4.useState)({});
-  const [chatUI, setChatUI] = (0, import_react4.useState)([]);
+  const [coagentStates, setCoagentStates] = (0, import_react4.useState)({});
+  let initialAgentSession = null;
+  if (props.agent) {
+    initialAgentSession = {
+      agentName: props.agent
+    };
+  }
+  const [agentSession, setAgentSession] = (0, import_react4.useState)(initialAgentSession);
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
     CopilotContext.Provider,
     {
@@ -495,6 +524,9 @@ ${nonDocumentStrings}`;
         getFunctionCallHandler,
         setAction,
         removeAction,
+        coagentActions,
+        setCoagentAction,
+        removeCoagentAction,
         getContextString,
         addContext,
         removeContext,
@@ -512,10 +544,10 @@ ${nonDocumentStrings}`;
         chatInstructions,
         setChatInstructions,
         showDevConsole: props.showDevConsole === void 0 ? "auto" : props.showDevConsole,
-        agentStates,
-        setAgentStates,
-        chatUI,
-        setChatUI
+        coagentStates,
+        setCoagentStates,
+        agentSession,
+        setAgentSession
       },
       children
     }
@@ -533,8 +565,9 @@ function entryPointsToFunctionCallHandler(actions) {
     if (action) {
       yield new Promise((resolve, reject) => {
         (0, import_react_dom.flushSync)(() => __async(this, null, function* () {
+          var _a;
           try {
-            result = yield action.handler(args);
+            result = yield (_a = action.handler) == null ? void 0 : _a.call(action, args);
             resolve();
           } catch (error) {
             reject(error);
@@ -566,12 +599,15 @@ function useChat(options) {
     isLoading,
     actions,
     onFunctionCall,
-    setAgentStates
+    onCoagentAction,
+    setCoagentStates,
+    coagentStates,
+    agentSession,
+    setAgentSession
   } = options;
   const abortControllerRef = (0, import_react5.useRef)();
   const threadIdRef = (0, import_react5.useRef)(null);
   const runIdRef = (0, import_react5.useRef)(null);
-  const agentSessionRef = (0, import_react5.useRef)(null);
   const publicApiKey = copilotConfig.publicApiKey;
   const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared4.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = new import_runtime_client_gql.CopilotRuntimeClient({
@@ -596,7 +632,7 @@ function useChat(options) {
     const messagesWithContext = [systemMessage, ...initialMessages || [], ...previousMessages];
     const stream = import_runtime_client_gql.CopilotRuntimeClient.asStream(
       runtimeClient.generateCopilotResponse({
-        data: __spreadValues(__spreadProps(__spreadValues({
+        data: __spreadProps(__spreadValues(__spreadProps(__spreadValues({
           frontend: {
             actions: actions.map((action) => ({
               name: action.name,
@@ -620,20 +656,22 @@ function useChat(options) {
           metadata: {
             requestType: import_runtime_client_gql.CopilotRequestType.Chat
           }
-        }), agentSessionRef.current ? {
-          agentSession: {
-            threadId: agentSessionRef.current.threadId,
-            agentName: agentSessionRef.current.agentName,
-            nodeName: agentSessionRef.current.nodeName
-          }
-        } : {}),
+        }), agentSession ? {
+          agentSession
+        } : {}), {
+          agentStates: Object.values(coagentStates).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 results = {};
+    let actionResults = {};
+    let executedCoagentActions = [];
     try {
       while (true) {
         const { done, value } = yield reader.read();
@@ -662,25 +700,28 @@ function useChat(options) {
             newMessages.push(message);
             if (message instanceof import_runtime_client_gql.AgentStateMessage) {
               if (message.running) {
-                setAgentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
-                  [message.agentName]: message
+                setCoagentStates((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
+                  [message.agentName]: {
+                    name: message.agentName,
+                    state: message.state,
+                    running: message.running,
+                    active: message.active,
+                    threadId: message.threadId,
+                    nodeName: message.nodeName,
+                    runId: message.runId
+                  }
                 }));
-                agentSessionRef.current = {
+                setAgentSession({
                   threadId: message.threadId,
                   agentName: message.agentName,
                   nodeName: message.nodeName
-                };
-              } else {
-                setAgentStates((prevAgentStates) => {
-                  const newAgentStates = __spreadValues({}, prevAgentStates);
-                  delete newAgentStates[message.agentName];
-                  return newAgentStates;
                 });
-                agentSessionRef.current = null;
+              } else {
+                setAgentSession(null);
               }
             }
             if (message instanceof import_runtime_client_gql.ActionExecutionMessage && message.status.code !== import_runtime_client_gql.MessageStatusCode.Pending && message.scope === "client" && onFunctionCall) {
-              if (!(message.id in results)) {
+              if (!(message.id in actionResults)) {
                 if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
                   break;
                 }
@@ -689,26 +730,39 @@ function useChat(options) {
                   name: message.name,
                   args: message.arguments
                 });
-                results[message.id] = result;
+                actionResults[message.id] = result;
               }
               newMessages.push(
                 new import_runtime_client_gql.ResultMessage({
-                  result: import_runtime_client_gql.ResultMessage.encodeResult(results[message.id]),
+                  result: import_runtime_client_gql.ResultMessage.encodeResult(actionResults[message.id]),
                   actionExecutionId: message.id,
                   actionName: message.name
                 })
               );
             }
+            if (message instanceof import_runtime_client_gql.AgentStateMessage && !message.active && !executedCoagentActions.includes(message.id) && onCoagentAction) {
+              if (guardrailsEnabled && value.generateCopilotResponse.status === void 0) {
+                break;
+              }
+              yield onCoagentAction({
+                name: message.agentName,
+                nodeName: message.nodeName,
+                state: message.state
+              });
+              executedCoagentActions.push(message.id);
+            }
           }
         }
         if (newMessages.length > 0) {
           const filteredMessages = [...previousMessages, ...newMessages].reduce(
             (acc, message) => {
-              if (message instanceof import_runtime_client_gql.AgentStateMessage && // Check if the current message is an AgentStateMessage
-              acc.length > 0 && // Ensure there is at least one message in the accumulator
-              acc[acc.length - 1] instanceof import_runtime_client_gql.AgentStateMessage && // Check if the last message in the accumulator is also an AgentStateMessage
-              acc[acc.length - 1].agentName === message.agentName && // Check if the agentName is the same
-              acc[acc.length - 1].nodeName === message.nodeName) {
+              if (
+                // If the current message is an AgentStateMessage
+                message instanceof import_runtime_client_gql.AgentStateMessage && // And there is at least one message in the accumulator
+                acc.length > 0 && // And the last message in the accumulator is also an AgentStateMessage
+                acc[acc.length - 1] instanceof import_runtime_client_gql.AgentStateMessage && // And the agentName, nodeName, and runId are the same
+                acc[acc.length - 1].agentName === message.agentName && acc[acc.length - 1].nodeName === message.nodeName && acc[acc.length - 1].runId === message.runId
+              ) {
                 acc[acc.length - 1] = message;
               } else {
                 acc.push(message);
@@ -722,7 +776,7 @@ function useChat(options) {
       }
       if (
         // if we have client side results
-        Object.values(results).length || // or the last message we received is a result
+        Object.values(actionResults).length || // or the last message we received is a result
         newMessages.length && newMessages[newMessages.length - 1] instanceof import_runtime_client_gql.ResultMessage
       ) {
         yield new Promise((resolve) => setTimeout(resolve, 10));
@@ -785,7 +839,11 @@ function useCopilotChat(_a = {}) {
     setIsLoading,
     chatInstructions,
     actions,
-    setAgentStates
+    coagentStates,
+    setCoagentStates,
+    coagentActions,
+    agentSession,
+    setAgentSession
   } = (0, import_react6.useContext)(CopilotContext);
   const latestGetContextString = useUpdatedRef(getContextString);
   const deleteMessage = (0, import_react6.useCallback)(
@@ -802,17 +860,39 @@ function useCopilotChat(_a = {}) {
       role: import_runtime_client_gql2.Role.System
     });
   }, [getContextString, makeSystemMessage2, chatInstructions]);
+  const onCoagentAction = (0, import_react6.useCallback)(
+    (args) => __async(this, null, function* () {
+      var _a2;
+      const { name, nodeName, state } = args;
+      let action = Object.values(coagentActions).find(
+        (action2) => action2.name === name && action2.nodeName === nodeName
+      );
+      if (!action) {
+        action = Object.values(coagentActions).find(
+          (action2) => action2.name === name && !action2.nodeName
+        );
+      }
+      if (action) {
+        yield (_a2 = action.handler) == null ? void 0 : _a2.call(action, { state, nodeName });
+      }
+    }),
+    [coagentActions]
+  );
   const { append, reload, stop } = useChat(__spreadProps(__spreadValues({}, options), {
     actions: Object.values(actions),
     copilotConfig: copilotApiConfig,
     initialMessages: options.initialMessages || [],
     onFunctionCall: getFunctionCallHandler(),
+    onCoagentAction,
     messages,
     setMessages,
     makeSystemMessageCallback,
     isLoading,
     setIsLoading,
-    setAgentStates
+    coagentStates,
+    setCoagentStates,
+    agentSession,
+    setAgentSession
   }));
   return {
     visibleMessages: messages,
@@ -894,12 +974,48 @@ function useCopilotAction(action, dependencies) {
   ]);
 }
 
-// src/hooks/use-make-copilot-document-readable.ts
+// src/hooks/use-coagent-action.ts
 var import_react8 = require("react");
-function useMakeCopilotDocumentReadable(document, categories, dependencies = []) {
-  const { addDocumentContext, removeDocumentContext } = (0, import_react8.useContext)(CopilotContext);
-  const idRef = (0, import_react8.useRef)();
+var import_shared6 = require("@copilotkit/shared");
+function useCoagentAction(action, dependencies) {
+  const { setCoagentAction, removeCoagentAction, coagentActions, chatComponentsCache } = (0, import_react8.useContext)(CopilotContext);
+  const idRef = (0, import_react8.useRef)((0, import_shared6.randomId)());
+  const key = `${action.name}-${action.nodeName || "global"}`;
+  if (dependencies === void 0) {
+    if (coagentActions[idRef.current]) {
+      coagentActions[idRef.current].handler = action.handler;
+      if (typeof action.render === "function") {
+        if (chatComponentsCache.current !== null) {
+          chatComponentsCache.current.coagentActions[key] = action.render;
+        }
+      }
+    }
+  }
   (0, import_react8.useEffect)(() => {
+    setCoagentAction(idRef.current, action);
+    if (chatComponentsCache.current !== null && action.render !== void 0) {
+      chatComponentsCache.current.coagentActions[key] = action.render;
+    }
+    return () => {
+      removeCoagentAction(idRef.current);
+    };
+  }, [
+    setCoagentAction,
+    removeCoagentAction,
+    action.name,
+    // include render only if it's a string
+    typeof action.render === "string" ? action.render : void 0,
+    // dependencies set by the developer
+    ...dependencies || []
+  ]);
+}
+
+// src/hooks/use-make-copilot-document-readable.ts
+var import_react9 = require("react");
+function useMakeCopilotDocumentReadable(document, categories, dependencies = []) {
+  const { addDocumentContext, removeDocumentContext } = (0, import_react9.useContext)(CopilotContext);
+  const idRef = (0, import_react9.useRef)();
+  (0, import_react9.useEffect)(() => {
     const id = addDocumentContext(document, categories);
     idRef.current = id;
     return () => {
@@ -910,16 +1026,16 @@ function useMakeCopilotDocumentReadable(document, categories, dependencies = [])
 }
 
 // src/hooks/use-copilot-readable.ts
-var import_react9 = require("react");
+var import_react10 = 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 } = (0, import_react9.useContext)(CopilotContext);
-  const idRef = (0, import_react9.useRef)();
+  const { addContext, removeContext } = (0, import_react10.useContext)(CopilotContext);
+  const idRef = (0, import_react10.useRef)();
   convert = convert || convertToJSON;
   const information = convert(description, value);
-  (0, import_react9.useEffect)(() => {
+  (0, import_react10.useEffect)(() => {
     const id = addContext(information, parentId, categories);
     idRef.current = id;
     return () => {
@@ -929,56 +1045,84 @@ function useCopilotReadable({ description, value, parentId, categories, convert
   return idRef.current;
 }
 
-// src/hooks/use-agent-state.ts
-var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
-function useAgentState(agentName) {
-  const { agentStates, setAgentStates } = useCopilotContext();
-  const { visibleMessages, setMessages } = useCopilotChat();
-  const setSpecificAgentState = (newState) => {
-    setAgentStates((prevAgentStates) => {
-      const currentState2 = prevAgentStates[agentName];
-      if (!currentState2) {
-        throw new Error(`Agent state ${agentName} not found`);
-      }
-      const updatedState = typeof newState === "function" ? newState(currentState2) : newState;
-      const lastAgentStateMessage = [...visibleMessages].reverse().find((message) => message instanceof import_runtime_client_gql3.AgentStateMessage);
-      const agentStateMessage = new import_runtime_client_gql3.AgentStateMessage({
-        agentName,
-        nodeName: lastAgentStateMessage.nodeName,
-        state: updatedState,
-        running: lastAgentStateMessage.running,
-        threadId: lastAgentStateMessage.threadId,
-        role: lastAgentStateMessage.role
-      });
-      setMessages([...visibleMessages, agentStateMessage]);
+// src/hooks/use-coagent.ts
+var import_react11 = require("react");
+function useCoagent(options) {
+  const isExternalStateManagement = (options2) => {
+    return "state" in options2 && "setState" in options2;
+  };
+  const { name } = options;
+  const isInternalStateManagementWithInitial = (options2) => {
+    return "initialState" in options2;
+  };
+  const { coagentStates, setCoagentStates } = useCopilotContext();
+  const getCoagentState = (coagentStates2, name2) => {
+    if (coagentStates2[name2]) {
+      return coagentStates2[name2];
+    } else {
+      return {
+        name: name2,
+        state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
+        running: false,
+        active: false,
+        threadId: void 0,
+        nodeName: void 0,
+        runId: void 0
+      };
+    }
+  };
+  const setState = (newState) => {
+    setCoagentStates((prevAgentStates) => {
+      let coagentState2 = getCoagentState(prevAgentStates, name);
+      const updatedState = typeof newState === "function" ? newState(coagentState2.state) : newState;
       return __spreadProps(__spreadValues({}, prevAgentStates), {
-        [agentName]: agentStateMessage
+        [name]: __spreadProps(__spreadValues({}, coagentState2), {
+          state: updatedState
+        })
       });
     });
   };
-  const currentState = agentStates[agentName];
-  return [
-    currentState == null ? void 0 : currentState.state,
-    setSpecificAgentState,
-    currentState ? { nodeName: currentState.nodeName } : void 0
-  ];
+  const coagentState = getCoagentState(coagentStates, name);
+  const state = isExternalStateManagement(options) ? options.state : coagentState.state;
+  (0, import_react11.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]);
+  return {
+    name,
+    nodeName: coagentState.nodeName,
+    state,
+    setState,
+    running: coagentState.running,
+    start: () => {
+      startAgent(name);
+    },
+    stop: () => {
+      stopAgent(name);
+    }
+  };
 }
-
-// src/hooks/use-copilot-chat-ui.ts
-var import_react10 = require("react");
-function useCopilotChatUI(chatUI, dependencies) {
-  const { chatUI: chatUIFromContext, setChatUI } = useCopilotContext();
-  (0, import_react10.useEffect)(() => {
-    setChatUI([...chatUIFromContext, chatUI]);
-    return () => {
-      setChatUI(chatUIFromContext.filter((ui) => ui !== chatUI));
-    };
-  }, [chatUI.agentName, chatUI.nodeName, setChatUI]);
+function startAgent(name) {
+  const { setAgentSession } = useCopilotContext();
+  setAgentSession({
+    agentName: name
+  });
+}
+function stopAgent(name) {
+  const { agentSession, setAgentSession } = useCopilotContext();
+  if (agentSession && agentSession.agentName === name) {
+    setAgentSession(null);
+  } else {
+    console.warn(`No agent session found for ${name}`);
+  }
 }
 
 // src/lib/copilot-task.ts
-var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
-var import_shared6 = require("@copilotkit/shared");
+var import_runtime_client_gql3 = require("@copilotkit/runtime-client-gql");
+var import_shared7 = require("@copilotkit/shared");
 var CopilotTask = class {
   constructor(config) {
     this.instructions = config.instructions;
@@ -1005,12 +1149,12 @@ var CopilotTask = class {
       if (this.includeCopilotReadable) {
         contextString += context.getContextString([], defaultCopilotContextCategories);
       }
-      const systemMessage = new import_runtime_client_gql4.TextMessage({
+      const systemMessage = new import_runtime_client_gql3.TextMessage({
         content: taskSystemMessage(contextString, this.instructions),
-        role: import_runtime_client_gql4.Role.System
+        role: import_runtime_client_gql3.Role.System
       });
       const messages = [systemMessage];
-      const runtimeClient = new import_runtime_client_gql4.CopilotRuntimeClient({
+      const runtimeClient = new import_runtime_client_gql3.CopilotRuntimeClient({
         url: context.copilotApiConfig.chatApiEndpoint,
         publicApiKey: context.copilotApiConfig.publicApiKey,
         headers: context.copilotApiConfig.headers,
@@ -1022,12 +1166,12 @@ var CopilotTask = class {
             actions: Object.values(actions).map((action) => ({
               name: action.name,
               description: action.description || "",
-              jsonSchema: JSON.stringify((0, import_shared6.actionParametersToJsonSchema)(action.parameters || []))
+              jsonSchema: JSON.stringify((0, import_shared7.actionParametersToJsonSchema)(action.parameters || []))
             }))
           },
-          messages: (0, import_runtime_client_gql4.convertMessagesToGqlInput)(messages),
+          messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)(messages),
           metadata: {
-            requestType: import_runtime_client_gql4.CopilotRequestType.Task
+            requestType: import_runtime_client_gql3.CopilotRequestType.Task
           },
           forwardedParameters: {
             toolChoice: "required"
@@ -1036,9 +1180,9 @@ var CopilotTask = class {
         properties: context.copilotApiConfig.properties
       }).toPromise();
       const functionCallHandler = context.getFunctionCallHandler(actions);
-      const functionCalls = (0, import_runtime_client_gql4.convertGqlOutputToMessages)(
+      const functionCalls = (0, import_runtime_client_gql3.convertGqlOutputToMessages)(
         ((_b = (_a = response.data) == null ? void 0 : _a.generateCopilotResponse) == null ? void 0 : _b.messages) || []
-      ).filter((m) => m instanceof import_runtime_client_gql4.ActionExecutionMessage);
+      ).filter((m) => m instanceof import_runtime_client_gql3.ActionExecutionMessage);
       for (const functionCall of functionCalls) {
         yield functionCallHandler({
           messages,
@@ -1076,10 +1220,10 @@ ${instructions}
 }
 
 // src/utils/extract.ts
-var import_shared7 = require("@copilotkit/shared");
+var import_shared8 = require("@copilotkit/shared");
+var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
 var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
-var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
 function extract(_0) {
   return __async(this, arguments, function* ({
     context,
@@ -1089,7 +1233,7 @@ function extract(_0) {
     data,
     abortSignal,
     stream,
-    requestType = import_runtime_client_gql5.CopilotRequestType.Task
+    requestType = import_runtime_client_gql4.CopilotRequestType.Task
   }) {
     var _a, _b;
     const { messages } = context;
@@ -1108,18 +1252,18 @@ function extract(_0) {
     if (includeReadable) {
       contextString += context.getContextString([], defaultCopilotContextCategories);
     }
-    const systemMessage = new import_runtime_client_gql5.TextMessage({
+    const systemMessage = new import_runtime_client_gql4.TextMessage({
       content: makeSystemMessage(contextString, instructions),
-      role: import_runtime_client_gql5.Role.System
+      role: import_runtime_client_gql4.Role.System
     });
-    const headers = __spreadValues(__spreadValues({}, context.copilotApiConfig.headers || {}), context.copilotApiConfig.publicApiKey ? { [import_shared7.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: context.copilotApiConfig.publicApiKey } : {});
-    const runtimeClient = new import_runtime_client_gql6.CopilotRuntimeClient({
+    const headers = __spreadValues(__spreadValues({}, context.copilotApiConfig.headers || {}), context.copilotApiConfig.publicApiKey ? { [import_shared8.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: context.copilotApiConfig.publicApiKey } : {});
+    const runtimeClient = new import_runtime_client_gql5.CopilotRuntimeClient({
       url: context.copilotApiConfig.chatApiEndpoint,
       publicApiKey: context.copilotApiConfig.publicApiKey,
       headers,
       credentials: context.copilotApiConfig.credentials
     });
-    const response = import_runtime_client_gql6.CopilotRuntimeClient.asStream(
+    const response = import_runtime_client_gql5.CopilotRuntimeClient.asStream(
       runtimeClient.generateCopilotResponse({
         data: {
           frontend: {
@@ -1127,11 +1271,11 @@ function extract(_0) {
               {
                 name: action.name,
                 description: action.description || "",
-                jsonSchema: JSON.stringify((0, import_shared7.actionParametersToJsonSchema)(action.parameters || []))
+                jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters || []))
               }
             ]
           },
-          messages: (0, import_runtime_client_gql7.convertMessagesToGqlInput)(
+          messages: (0, import_runtime_client_gql6.convertMessagesToGqlInput)(
             includeMessages ? [systemMessage, ...messages] : [systemMessage]
           ),
           metadata: {
@@ -1157,9 +1301,9 @@ function extract(_0) {
       if (abortSignal == null ? void 0 : abortSignal.aborted) {
         throw new Error("Aborted");
       }
-      actionExecutionMessage = (0, import_runtime_client_gql5.convertGqlOutputToMessages)(
+      actionExecutionMessage = (0, import_runtime_client_gql4.convertGqlOutputToMessages)(
         value.generateCopilotResponse.messages
-      ).find((msg) => msg instanceof import_runtime_client_gql5.ActionExecutionMessage);
+      ).find((msg) => msg instanceof import_runtime_client_gql4.ActionExecutionMessage);
       if (!actionExecutionMessage) {
         continue;
       }
@@ -1213,10 +1357,10 @@ Any additional messages provided are for providing context only and should not b
   CopilotTask,
   defaultCopilotContextCategories,
   extract,
-  useAgentState,
+  useCoagent,
+  useCoagentAction,
   useCopilotAction,
   useCopilotChat,
-  useCopilotChatUI,
   useCopilotContext,
   useCopilotReadable,
   useMakeCopilotDocumentReadable