@copilotkit/react-core 1.4.8 → 1.5.0-coagents-v0-3.0

package/dist/index.js

@@ -87,7 +87,10 @@ __export(src_exports, {
   CopilotTask: () => CopilotTask,
   defaultCopilotContextCategories: () => defaultCopilotContextCategories,
   extract: () => extract,
+  runAgent: () => runAgent,
   shouldShowDevConsole: () => shouldShowDevConsole,
+  startAgent: () => startAgent,
+  stopAgent: () => stopAgent,
   useCoAgent: () => useCoAgent,
   useCoAgentStateRender: () => useCoAgentStateRender,
   useCopilotAction: () => useCopilotAction,
@@ -157,7 +160,15 @@ var emptyCopilotContext = {
   },
   agentSession: null,
   setAgentSession: () => {
-  }
+  },
+  agentLock: null,
+  threadId: null,
+  setThreadId: () => {
+  },
+  runId: null,
+  setRunId: () => {
+  },
+  chatAbortControllerRef: { current: null }
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -1044,6 +1055,9 @@ ${nonDocumentStrings}`;
     };
   }
   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 showDevConsole = props.showDevConsole === void 0 ? "auto" : props.showDevConsole;
   return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
     CopilotContext.Provider,
@@ -1078,7 +1092,13 @@ ${nonDocumentStrings}`;
         setCoagentStatesWithRef,
         agentSession,
         setAgentSession,
-        runtimeClient
+        runtimeClient,
+        agentLock: props.agent || null,
+        threadId,
+        setThreadId,
+        runId,
+        setRunId,
+        chatAbortControllerRef
       },
       children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(CopilotMessages, { children })
     }
@@ -1113,12 +1133,39 @@ function entryPointsToFunctionCallHandler(actions) {
 
 // src/hooks/use-copilot-chat.ts
 var import_react12 = require("react");
-var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
 var import_react11 = require("react");
-var import_shared5 = require("@copilotkit/shared");
+var import_shared6 = require("@copilotkit/shared");
+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
 function useChat(options) {
   const {
     messages,
@@ -1134,17 +1181,23 @@ function useChat(options) {
     setCoagentStatesWithRef,
     coagentStatesRef,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   } = options;
-  const abortControllerRef = (0, import_react11.useRef)();
-  const threadIdRef = (0, import_react11.useRef)(null);
-  const runIdRef = (0, import_react11.useRef)(null);
   const { addGraphQLErrorsToast } = useToast();
   const runChatCompletionRef = (0, import_react11.useRef)();
   const agentSessionRef = (0, import_react11.useRef)(agentSession);
   agentSessionRef.current = agentSession;
+  const threadIdRef = (0, import_react11.useRef)(threadId);
+  threadIdRef.current = threadId;
+  const runIdRef = (0, import_react11.useRef)(runId);
+  runIdRef.current = runId;
   const publicApiKey = copilotConfig.publicApiKey;
-  const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared5.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
+  const headers = __spreadValues(__spreadValues({}, copilotConfig.headers || {}), publicApiKey ? { [import_shared6.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: publicApiKey } : {});
   const runtimeClient = useCopilotRuntimeClient({
     url: copilotConfig.chatApiEndpoint,
     publicApiKey: copilotConfig.publicApiKey,
@@ -1153,48 +1206,29 @@ function useChat(options) {
   });
   const runChatCompletion = useAsyncCallback(
     (previousMessages) => __async(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
       setIsLoading(true);
       let newMessages = [
-        new import_runtime_client_gql4.TextMessage({
+        new import_runtime_client_gql5.TextMessage({
           content: "",
-          role: import_runtime_client_gql4.Role.Assistant
+          role: import_runtime_client_gql5.Role.Assistant
         })
       ];
-      const abortController = new AbortController();
-      abortControllerRef.current = abortController;
+      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: 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
-                };
-              }),
+              actions: processActionsForRuntimeRequest(actions),
               url: window.location.href
             },
             threadId: threadIdRef.current,
             runId: runIdRef.current,
-            messages: (0, import_runtime_client_gql4.convertMessagesToGqlInput)((0, import_runtime_client_gql4.filterAgentStateMessages)(messagesWithContext))
+            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: {
@@ -1206,7 +1240,7 @@ function useChat(options) {
             } : {})
           } : {}), {
             metadata: {
-              requestType: import_runtime_client_gql4.CopilotRequestType.Chat
+              requestType: import_runtime_client_gql5.CopilotRequestType.Chat
             }
           }), agentSessionRef.current ? {
             agentSession: agentSessionRef.current
@@ -1217,14 +1251,15 @@ function useChat(options) {
             }))
           }),
           properties: copilotConfig.properties,
-          signal: (_d = abortControllerRef.current) == null ? void 0 : _d.signal
+          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 actionResults = {};
       let executedCoAgentStateRenders = [];
       let followUp = void 0;
+      let messages2 = [];
+      let syncedMessages = [];
       try {
         while (true) {
           let done, value;
@@ -1236,6 +1271,9 @@ function useChat(options) {
             break;
           }
           if (done) {
+            if (chatAbortControllerRef.current.signal.aborted) {
+              return [];
+            }
             break;
           }
           if (!(value == null ? void 0 : value.generateCopilotResponse)) {
@@ -1243,8 +1281,10 @@ function useChat(options) {
           }
           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)
+          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;
@@ -1252,44 +1292,16 @@ function useChat(options) {
           newMessages = [];
           if (((_h = value.generateCopilotResponse.status) == null ? void 0 : _h.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
             newMessages = [
-              new import_runtime_client_gql4.TextMessage({
-                role: import_runtime_client_gql4.MessageRole.Assistant,
+              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) {
-              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;
-                  }
-                  try {
-                    setMessages([...previousMessages, ...newMessages]);
-                    const action = actions.find((action2) => action2.name === message.name);
-                    if (action) {
-                      followUp = action.followUp;
-                    }
-                    const result = yield onFunctionCall({
-                      messages: previousMessages,
-                      name: message.name,
-                      args: message.arguments
-                    });
-                    actionResults[message.id] = result;
-                  } catch (e) {
-                    actionResults[message.id] = `Failed to execute action ${message.name}`;
-                    console.error(`Failed to execute action ${message.name}: ${e}`);
-                  }
-                }
-                newMessages.push(
-                  new import_runtime_client_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;
@@ -1304,6 +1316,11 @@ function useChat(options) {
             }
             const lastAgentStateMessage = [...messages2].reverse().find((message) => message.isAgentStateMessage());
             if (lastAgentStateMessage) {
+              if (lastAgentStateMessage.state.messages && lastAgentStateMessage.state.messages.length > 0) {
+                syncedMessages = (0, import_runtime_client_gql5.loadMessagesFromJsonRepresentation)(
+                  lastAgentStateMessage.state.messages
+                );
+              }
               setCoagentStatesWithRef((prevAgentStates) => __spreadProps(__spreadValues({}, prevAgentStates), {
                 [lastAgentStateMessage.agentName]: {
                   name: lastAgentStateMessage.agentName,
@@ -1330,14 +1347,96 @@ function useChat(options) {
             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({
+                    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}`);
+              }
+              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),
+                  actionExecutionId: message.id,
+                  actionName: message.name
+                })
+              );
+            }
+          }
+          setMessages(finalMessages);
+        }
         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())
+          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([...previousMessages, ...newMessages]);
+          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
+              );
+            }
+            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();
         }
@@ -1394,7 +1493,7 @@ function useChat(options) {
   }), [isLoading, messages, setMessages, runChatCompletionAndHandleFunctionCall]);
   const stop = () => {
     var _a;
-    (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
+    (_a = chatAbortControllerRef.current) == null ? void 0 : _a.abort("Stop was called");
   };
   return {
     append,
@@ -1403,6 +1502,23 @@ function useChat(options) {
     runChatCompletion: () => runChatCompletionRef.current(messages)
   };
 }
+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 = {}) {
@@ -1423,7 +1539,13 @@ function useCopilotChat(_a = {}) {
     setCoagentStatesWithRef,
     coAgentStateRenders,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    agentLock,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -1436,9 +1558,9 @@ function useCopilotChat(_a = {}) {
   const makeSystemMessageCallback = (0, import_react12.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage2 || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql5.TextMessage({
+    return new import_runtime_client_gql6.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql5.Role.System
+      role: import_runtime_client_gql6.Role.System
     });
   }, [getContextString, makeSystemMessage2, chatInstructions]);
   const onCoAgentStateRender = useAsyncCallback(
@@ -1473,7 +1595,12 @@ function useCopilotChat(_a = {}) {
     coagentStatesRef,
     setCoagentStatesWithRef,
     agentSession,
-    setAgentSession
+    setAgentSession,
+    threadId,
+    setThreadId,
+    runId,
+    setRunId,
+    chatAbortControllerRef
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = useAsyncCallback(
@@ -1508,12 +1635,38 @@ function useCopilotChat(_a = {}) {
   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
@@ -1551,14 +1704,18 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-copilot-action.ts
-var import_shared6 = require("@copilotkit/shared");
+var import_shared7 = require("@copilotkit/shared");
 var import_react13 = require("react");
 function useCopilotAction(action, dependencies) {
   const { setAction, removeAction, actions, chatComponentsCache } = useCopilotContext();
-  const idRef = (0, import_react13.useRef)((0, import_shared6.randomId)());
+  const idRef = (0, import_react13.useRef)((0, import_shared7.randomId)());
   const renderAndWaitRef = (0, import_react13.useRef)(null);
   action = __spreadValues({}, action);
-  if (action.renderAndWait || action.renderAndWaitForResponse) {
+  if (
+    // renderAndWaitForResponse is not available for catch all actions
+    isFrontendAction(action) && // check if renderAndWaitForResponse is set
+    (action.renderAndWait || action.renderAndWaitForResponse)
+  ) {
     const renderAndWait = action.renderAndWait || action.renderAndWaitForResponse;
     action.renderAndWait = void 0;
     action.renderAndWaitForResponse = void 0;
@@ -1596,7 +1753,9 @@ function useCopilotAction(action, dependencies) {
   }
   if (dependencies === void 0) {
     if (actions[idRef.current]) {
-      actions[idRef.current].handler = action.handler;
+      if (isFrontendAction(action)) {
+        actions[idRef.current].handler = action.handler;
+      }
       if (typeof action.render === "function") {
         if (chatComponentsCache.current !== null) {
           chatComponentsCache.current.actions[action.name] = action.render;
@@ -1615,23 +1774,26 @@ function useCopilotAction(action, dependencies) {
   }, [
     setAction,
     removeAction,
-    action.description,
+    isFrontendAction(action) ? action.description : void 0,
     action.name,
-    action.disabled,
-    action.available,
+    isFrontendAction(action) ? action.disabled : void 0,
+    isFrontendAction(action) ? action.available : void 0,
     // This should be faster than deep equality checking
     // In addition, all major JS engines guarantee the order of object keys
-    JSON.stringify(action.parameters),
+    JSON.stringify(isFrontendAction(action) ? 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_shared7 = require("@copilotkit/shared");
+var import_shared8 = require("@copilotkit/shared");
 function useCoAgentStateRender(action, dependencies) {
   const {
     setCoAgentStateRender,
@@ -1639,7 +1801,7 @@ function useCoAgentStateRender(action, dependencies) {
     coAgentStateRenders,
     chatComponentsCache
   } = (0, import_react14.useContext)(CopilotContext);
-  const idRef = (0, import_react14.useRef)((0, import_shared7.randomId)());
+  const idRef = (0, import_react14.useRef)((0, import_shared8.randomId)());
   const key = `${action.name}-${action.nodeName || "global"}`;
   if (dependencies === void 0) {
     if (coAgentStateRenders[idRef.current]) {
@@ -1752,7 +1914,11 @@ function useCoAgent(options) {
     } else if (coagentStates[name] === void 0) {
       setState(options.initialState === void 0 ? {} : options.initialState);
     }
-  }, [isExternalStateManagement(options) ? JSON.stringify(options.state) : void 0]);
+  }, [
+    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);
@@ -1781,6 +1947,17 @@ 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}`);
   }
@@ -1816,8 +1993,7 @@ function runAgent(name, context, appendMessage, runChatCompletion, hint) {
 }
 
 // src/lib/copilot-task.ts
-var import_runtime_client_gql6 = require("@copilotkit/runtime-client-gql");
-var import_shared8 = require("@copilotkit/shared");
+var import_runtime_client_gql7 = require("@copilotkit/runtime-client-gql");
 var CopilotTask = class {
   constructor(config) {
     this.instructions = config.instructions;
@@ -1844,12 +2020,12 @@ var CopilotTask = class {
       if (this.includeCopilotReadable) {
         contextString += context.getContextString([], defaultCopilotContextCategories);
       }
-      const systemMessage = new import_runtime_client_gql6.TextMessage({
+      const systemMessage = new import_runtime_client_gql7.TextMessage({
         content: taskSystemMessage(contextString, this.instructions),
-        role: import_runtime_client_gql6.Role.System
+        role: import_runtime_client_gql7.Role.System
       });
       const messages = [systemMessage];
-      const runtimeClient = new import_runtime_client_gql6.CopilotRuntimeClient({
+      const runtimeClient = new import_runtime_client_gql7.CopilotRuntimeClient({
         url: context.copilotApiConfig.chatApiEndpoint,
         publicApiKey: context.copilotApiConfig.publicApiKey,
         headers: context.copilotApiConfig.headers,
@@ -1858,16 +2034,12 @@ var CopilotTask = class {
       const response = yield runtimeClient.generateCopilotResponse({
         data: {
           frontend: {
-            actions: Object.values(actions).map((action) => ({
-              name: action.name,
-              description: action.description || "",
-              jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters || []))
-            })),
+            actions: processActionsForRuntimeRequest(Object.values(actions)),
             url: window.location.href
           },
-          messages: (0, import_runtime_client_gql6.convertMessagesToGqlInput)((0, import_runtime_client_gql6.filterAgentStateMessages)(messages)),
+          messages: (0, import_runtime_client_gql7.convertMessagesToGqlInput)((0, import_runtime_client_gql7.filterAgentStateMessages)(messages)),
           metadata: {
-            requestType: import_runtime_client_gql6.CopilotRequestType.Task
+            requestType: import_runtime_client_gql7.CopilotRequestType.Task
           },
           forwardedParameters: {
             toolChoice: "required"
@@ -1876,7 +2048,7 @@ var CopilotTask = class {
         properties: context.copilotApiConfig.properties
       }).toPromise();
       const functionCallHandler = context.getFunctionCallHandler(actions);
-      const functionCalls = (0, import_runtime_client_gql6.convertGqlOutputToMessages)(
+      const functionCalls = (0, import_runtime_client_gql7.convertGqlOutputToMessages)(
         ((_b = (_a = response.data) == null ? void 0 : _a.generateCopilotResponse) == null ? void 0 : _b.messages) || []
       ).filter((m) => m.isActionExecutionMessage());
       for (const functionCall of functionCalls) {
@@ -1922,7 +2094,10 @@ ${instructions}
   CopilotTask,
   defaultCopilotContextCategories,
   extract,
+  runAgent,
   shouldShowDevConsole,
+  startAgent,
+  stopAgent,
   useCoAgent,
   useCoAgentStateRender,
   useCopilotAction,