@copilotkit/react-core 1.6.0-next.10 → 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,8 +379,6 @@ var useCopilotRuntimeClient = (options) => {
 };
 
 // src/hooks/use-chat.ts
-var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
-var import_shared3 = require("@copilotkit/shared");
 function useChat(options) {
   const {
     messages,
@@ -429,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"
@@ -523,16 +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;
-              eventValue = (0, import_shared3.parseJson)(eventValue, eventValue);
+              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)(
@@ -613,7 +611,7 @@ function useChat(options) {
             setMessages([...previousMessages, ...newMessages]);
           }
         }
-        const finalMessages = constructFinalMessages(
+        let finalMessages = constructFinalMessages(
           [...syncedMessages, ...interruptMessages],
           previousMessages,
           newMessages
@@ -623,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;
@@ -631,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
               );
             }
           }
@@ -754,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;
@@ -830,11 +820,72 @@ 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");
 var import_react_dom = require("react-dom");
-var import_shared4 = require("@copilotkit/shared");
+var import_shared3 = require("@copilotkit/shared");
 var import_jsx_runtime4 = require("react/jsx-runtime");
 var defaultCopilotContextCategories = ["global"];
 
@@ -881,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(
@@ -1035,7 +1086,7 @@ ${additionalInstructions}` : "");
 }
 
 // src/hooks/use-coagent.ts
-var import_shared5 = require("@copilotkit/shared");
+var import_shared4 = require("@copilotkit/shared");
 function useCoAgent(options) {
   const generalContext = useCopilotContext();
   const { availableAgents } = generalContext;
@@ -1086,7 +1137,7 @@ 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 = (0, import_shared5.parseJson)(newState, {});
+        const fetchedState = (0, import_shared4.parseJson)(newState, {});
         isExternalStateManagement(options) ? options.setState(fetchedState) : setState(fetchedState);
       }
     });