@copilotkit/react-core 1.6.0-next.1 → 1.6.0-next.11

package/dist/hooks/use-coagent.js

@@ -190,7 +190,7 @@ function useCopilotMessagesContext() {
 
 // src/hooks/use-copilot-chat.ts
 var import_react8 = require("react");
-var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
 var import_react6 = require("react");
@@ -202,7 +202,7 @@ var import_runtime_client_gql = require("@copilotkit/runtime-client-gql");
 var import_shared = require("@copilotkit/shared");
 function processActionsForRuntimeRequest(actions) {
   const filteredActions = actions.filter(
-    (action) => action.available !== import_runtime_client_gql.ActionInputAvailability.Disabled && action.disabled !== true && action.name !== "*"
+    (action) => action.available !== import_runtime_client_gql.ActionInputAvailability.Disabled && action.disabled !== true && action.name !== "*" && action.available != "frontend" && !action.pairedAction
   ).map((action) => {
     let available = import_runtime_client_gql.ActionInputAvailability.Enabled;
     if (action.disabled) {
@@ -379,7 +379,6 @@ var useCopilotRuntimeClient = (options) => {
 };
 
 // src/hooks/use-chat.ts
-var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 function useChat(options) {
   const {
     messages,
@@ -428,7 +427,7 @@ function useChat(options) {
       var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
       setIsLoading(true);
       const interruptEvent = langGraphInterruptAction == null ? void 0 : langGraphInterruptAction.event;
-      if ((interruptEvent == null ? void 0 : interruptEvent.name) === import_runtime_client_gql4.MetaEventName.LangGraphInterruptEvent && (interruptEvent == null ? void 0 : interruptEvent.value) && !(interruptEvent == null ? void 0 : interruptEvent.response) && agentSessionRef.current) {
+      if ((interruptEvent == null ? void 0 : interruptEvent.name) === import_runtime_client_gql3.MetaEventName.LangGraphInterruptEvent && (interruptEvent == null ? void 0 : interruptEvent.value) && !(interruptEvent == null ? void 0 : interruptEvent.response) && agentSessionRef.current) {
         addErrorToast([
           new Error(
             "A message was sent while interrupt is active. This will cause failure on the agent side"
@@ -522,19 +521,16 @@ function useChat(options) {
           setExtensions(extensionsRef.current);
           let rawMessagesResponse = value.generateCopilotResponse.messages;
           ((_i = (_h = value.generateCopilotResponse) == null ? void 0 : _h.metaEvents) != null ? _i : []).forEach((ev) => {
-            if (ev.name === import_runtime_client_gql4.MetaEventName.LangGraphInterruptEvent) {
+            if (ev.name === import_runtime_client_gql3.MetaEventName.LangGraphInterruptEvent) {
               let eventValue = (0, import_runtime_client_gql3.langGraphInterruptEvent)(ev).value;
-              try {
-                eventValue = JSON.parse(eventValue);
-              } catch (e) {
-              }
+              eventValue = (0, import_shared2.parseJson)(eventValue, eventValue);
               setLangGraphInterruptAction({
                 event: __spreadProps(__spreadValues({}, (0, import_runtime_client_gql3.langGraphInterruptEvent)(ev)), {
                   value: eventValue
                 })
               });
             }
-            if (ev.name === import_runtime_client_gql4.MetaEventName.CopilotKitLangGraphInterruptEvent) {
+            if (ev.name === import_runtime_client_gql3.MetaEventName.CopilotKitLangGraphInterruptEvent) {
               const data = ev.data;
               rawMessagesResponse = [...rawMessagesResponse, ...data.messages];
               interruptMessages = (0, import_runtime_client_gql3.convertGqlOutputToMessages)(
@@ -615,7 +611,7 @@ function useChat(options) {
             setMessages([...previousMessages, ...newMessages]);
           }
         }
-        const finalMessages = constructFinalMessages(
+        let finalMessages = constructFinalMessages(
           [...syncedMessages, ...interruptMessages],
           previousMessages,
           newMessages
@@ -625,7 +621,7 @@ function useChat(options) {
           const lastMessages = [];
           for (let i = finalMessages.length - 1; i >= 0; i--) {
             const message = finalMessages[i];
-            if (message.isActionExecutionMessage() && message.status.code !== import_runtime_client_gql3.MessageStatusCode.Pending) {
+            if ((message.isActionExecutionMessage() || message.isResultMessage()) && message.status.code !== import_runtime_client_gql3.MessageStatusCode.Pending) {
               lastMessages.unshift(message);
             } else {
               break;
@@ -633,59 +629,51 @@ function useChat(options) {
           }
           for (const message of lastMessages) {
             setMessages(finalMessages);
-            const action = actions.find((action2) => action2.name === message.name);
-            if (action) {
-              followUp = action.followUp;
-              let result;
-              let error = null;
-              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) {
-                error = e;
-                addErrorToast([error]);
-                result = `Failed to execute action ${message.name}. ${error.message}`;
-                console.error(`Failed to execute action ${message.name}: ${error}`);
-              }
+            const action = actions.find(
+              (action2) => action2.name === message.name
+            );
+            const currentResultMessagePairedFeAction = message.isResultMessage() ? getPairedFeAction(actions, message) : null;
+            const executeActionFromMessage = (action2, message2) => __async(this, null, function* () {
+              followUp = action2 == null ? void 0 : action2.followUp;
+              const resultMessage = yield executeAction({
+                onFunctionCall,
+                previousMessages,
+                message: message2,
+                chatAbortControllerRef,
+                onError: (error) => {
+                  addErrorToast([error]);
+                  console.error(`Failed to execute action ${message2.name}: ${error}`);
+                }
+              });
               didExecuteAction = true;
-              const messageIndex = finalMessages.findIndex((msg) => msg.id === message.id);
-              finalMessages.splice(
-                messageIndex + 1,
-                0,
-                new import_runtime_client_gql3.ResultMessage({
-                  id: "result-" + message.id,
-                  result: import_runtime_client_gql3.ResultMessage.encodeResult(
-                    error ? {
-                      content: result,
-                      error: JSON.parse(
-                        JSON.stringify(error, Object.getOwnPropertyNames(error))
-                      )
-                    } : result
-                  ),
-                  actionExecutionId: message.id,
-                  actionName: message.name
-                })
+              const messageIndex = finalMessages.findIndex((msg) => msg.id === message2.id);
+              finalMessages.splice(messageIndex + 1, 0, resultMessage);
+              return resultMessage;
+            });
+            if (action && message.isActionExecutionMessage()) {
+              const resultMessage = yield executeActionFromMessage(action, message);
+              const pairedFeAction = getPairedFeAction(actions, resultMessage);
+              if (pairedFeAction) {
+                const newExecutionMessage = new import_runtime_client_gql3.ActionExecutionMessage({
+                  name: pairedFeAction.name,
+                  arguments: (0, import_shared2.parseJson)(resultMessage.result, resultMessage.result),
+                  status: message.status,
+                  createdAt: message.createdAt,
+                  parentMessageId: message.parentMessageId
+                });
+                yield executeActionFromMessage(pairedFeAction, newExecutionMessage);
+              }
+            } else if (message.isResultMessage() && currentResultMessagePairedFeAction) {
+              const newExecutionMessage = new import_runtime_client_gql3.ActionExecutionMessage({
+                name: currentResultMessagePairedFeAction.name,
+                arguments: (0, import_shared2.parseJson)(message.result, message.result),
+                status: message.status,
+                createdAt: message.createdAt
+              });
+              finalMessages.push(newExecutionMessage);
+              yield executeActionFromMessage(
+                currentResultMessagePairedFeAction,
+                newExecutionMessage
               );
             }
           }
@@ -756,7 +744,7 @@ function useChat(options) {
         if (!event)
           return acc;
         switch (event.name) {
-          case import_runtime_client_gql4.MetaEventName.LangGraphInterruptEvent:
+          case import_runtime_client_gql3.MetaEventName.LangGraphInterruptEvent:
             if (event.response) {
               setLangGraphInterruptAction(null);
               const value = event.value;
@@ -832,6 +820,67 @@ function constructFinalMessages(syncedMessages, previousMessages, newMessages) {
   }
   return finalMessages;
 }
+function executeAction(_0) {
+  return __async(this, arguments, function* ({
+    onFunctionCall,
+    previousMessages,
+    message,
+    chatAbortControllerRef,
+    onError
+  }) {
+    let result;
+    let error = null;
+    try {
+      result = yield Promise.race([
+        onFunctionCall({
+          messages: previousMessages,
+          name: message.name,
+          args: message.arguments
+        }),
+        new Promise(
+          (resolve) => {
+            var _a;
+            return (_a = chatAbortControllerRef.current) == null ? void 0 : _a.signal.addEventListener(
+              "abort",
+              () => resolve("Operation was aborted by the user")
+            );
+          }
+        ),
+        // if the user stopped generation, we also abort consecutive actions
+        new Promise((resolve) => {
+          var _a;
+          if ((_a = chatAbortControllerRef.current) == null ? void 0 : _a.signal.aborted) {
+            resolve("Operation was aborted by the user");
+          }
+        })
+      ]);
+    } catch (e) {
+      onError(e);
+    }
+    return new import_runtime_client_gql3.ResultMessage({
+      id: "result-" + message.id,
+      result: import_runtime_client_gql3.ResultMessage.encodeResult(
+        error ? {
+          content: result,
+          error: JSON.parse(JSON.stringify(error, Object.getOwnPropertyNames(error)))
+        } : result
+      ),
+      actionExecutionId: message.id,
+      actionName: message.name
+    });
+  });
+}
+function getPairedFeAction(actions, message) {
+  let actionName = null;
+  if (message.isActionExecutionMessage()) {
+    actionName = message.name;
+  } else if (message.isResultMessage()) {
+    actionName = message.actionName;
+  }
+  return actions.find(
+    (action) => action.name === actionName && action.available === "frontend" || action.pairedAction === actionName
+  );
+}
 
 // src/components/copilot-provider/copilotkit.tsx
 var import_react7 = require("react");
@@ -883,9 +932,9 @@ function useCopilotChat(_a = {}) {
   const makeSystemMessageCallback = (0, import_react8.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql5.TextMessage({
+    return new import_runtime_client_gql4.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql5.Role.System
+      role: import_runtime_client_gql4.Role.System
     });
   }, [getContextString, makeSystemMessage, chatInstructions]);
   const onCoAgentStateRender = useAsyncCallback(
@@ -1037,15 +1086,13 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-coagent.ts
+var import_shared4 = require("@copilotkit/shared");
 function useCoAgent(options) {
   const generalContext = useCopilotContext();
   const { availableAgents } = generalContext;
   const { addToast } = useToast();
   const lastLoadedThreadId = (0, import_react9.useRef)();
   const lastLoadedState = (0, import_react9.useRef)();
-  const isExternalStateManagement = (options2) => {
-    return "state" in options2 && "setState" in options2;
-  };
   const { name } = options;
   (0, import_react9.useEffect)(() => {
     if ((availableAgents == null ? void 0 : availableAgents.length) && !availableAgents.some((a) => a.name === name)) {
@@ -1054,44 +1101,27 @@ function useCoAgent(options) {
       addToast({ type: "warning", message });
     }
   }, [availableAgents]);
-  const isInternalStateManagementWithInitial = (options2) => {
-    return "initialState" in options2;
-  };
   const messagesContext = useCopilotMessagesContext();
   const context = __spreadValues(__spreadValues({}, generalContext), messagesContext);
   const { coagentStates, coagentStatesRef, setCoagentStatesWithRef, threadId, copilotApiConfig } = context;
   const { appendMessage, runChatCompletion } = useCopilotChat();
-  const getCoagentState = (coagentStates2, name2) => {
-    var _a;
-    if (coagentStates2[name2]) {
-      return coagentStates2[name2];
-    } else {
-      return {
-        name: name2,
-        state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
-        configurable: (_a = options.configurable) != null ? _a : {},
-        running: false,
-        active: false,
-        threadId: void 0,
-        nodeName: void 0,
-        runId: void 0
-      };
-    }
-  };
   const runtimeClient = useCopilotRuntimeClient({
     url: copilotApiConfig.chatApiEndpoint,
     publicApiKey: copilotApiConfig.publicApiKey,
     credentials: copilotApiConfig.credentials
   });
-  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
-      })
-    }));
-  };
+  const setState = (0, import_react9.useCallback)(
+    (newState) => {
+      let coagentState = getCoagentState({ coagentStates, name, options });
+      const updatedState = typeof newState === "function" ? newState(coagentState.state) : newState;
+      setCoagentStatesWithRef(__spreadProps(__spreadValues({}, coagentStatesRef.current), {
+        [name]: __spreadProps(__spreadValues({}, coagentState), {
+          state: updatedState
+        })
+      }));
+    },
+    [coagentStates, name]
+  );
   (0, import_react9.useEffect)(() => {
     const fetchAgentState = () => __async(this, null, function* () {
       var _a, _b, _c, _d;
@@ -1107,13 +1137,12 @@ function useCoAgent(options) {
       if (((_d = (_c = result.data) == null ? void 0 : _c.loadAgentState) == null ? void 0 : _d.threadExists) && newState && newState != "{}") {
         lastLoadedState.current = newState;
         lastLoadedThreadId.current = threadId;
-        const fetchedState = JSON.parse(newState);
+        const fetchedState = (0, import_shared4.parseJson)(newState, {});
         isExternalStateManagement(options) ? options.setState(fetchedState) : setState(fetchedState);
       }
     });
     void fetchAgentState();
   }, [threadId]);
-  const coagentState = getCoagentState(coagentStates, name);
   (0, import_react9.useEffect)(() => {
     if (isExternalStateManagement(options)) {
       setState(options.state);
@@ -1131,17 +1160,20 @@ function useCoAgent(options) {
     }),
     [name, context, appendMessage, runChatCompletion]
   );
-  return {
-    name,
-    nodeName: coagentState.nodeName,
-    threadId: coagentState.threadId,
-    running: coagentState.running,
-    state: coagentState.state,
-    setState: isExternalStateManagement(options) ? options.setState : setState,
-    start: () => startAgent(name, context),
-    stop: () => stopAgent(name, context),
-    run: runAgentCallback
-  };
+  return (0, import_react9.useMemo)(() => {
+    const coagentState = getCoagentState({ coagentStates, name, options });
+    return {
+      name,
+      nodeName: coagentState.nodeName,
+      threadId: coagentState.threadId,
+      running: coagentState.running,
+      state: coagentState.state,
+      setState: isExternalStateManagement(options) ? options.setState : setState,
+      start: () => startAgent(name, context),
+      stop: () => stopAgent(name, context),
+      run: runAgentCallback
+    };
+  }, [name, coagentStates, options, setState, runAgentCallback]);
 }
 function startAgent(name, context) {
   const { setAgentSession } = context;
@@ -1197,6 +1229,33 @@ function runAgent(name, context, appendMessage, runChatCompletion, hint) {
     }
   });
 }
+var isExternalStateManagement = (options) => {
+  return "state" in options && "setState" in options;
+};
+var isInternalStateManagementWithInitial = (options) => {
+  return "initialState" in options;
+};
+var getCoagentState = ({
+  coagentStates,
+  name,
+  options
+}) => {
+  var _a;
+  if (coagentStates[name]) {
+    return coagentStates[name];
+  } else {
+    return {
+      name,
+      state: isInternalStateManagementWithInitial(options) ? options.initialState : {},
+      configurable: (_a = options.configurable) != null ? _a : {},
+      running: false,
+      active: false,
+      threadId: void 0,
+      nodeName: void 0,
+      runId: void 0
+    };
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   runAgent,