@copilotkit/react-core 1.5.15-next.3 → 1.5.15-next.5

package/dist/hooks/use-coagent.js

@@ -154,7 +154,10 @@ var emptyCopilotContext = {
   availableAgents: [],
   extensions: {},
   setExtensions: () => {
-  }
+  },
+  langGraphInterruptAction: null,
+  setLangGraphInterruptAction: () => null,
+  removeLangGraphInterruptAction: () => null
 };
 var CopilotContext = import_react.default.createContext(emptyCopilotContext);
 function useCopilotContext() {
@@ -187,7 +190,7 @@ function useCopilotMessagesContext() {
 
 // src/hooks/use-copilot-chat.ts
 var import_react8 = require("react");
-var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
+var import_runtime_client_gql5 = require("@copilotkit/runtime-client-gql");
 
 // src/hooks/use-chat.ts
 var import_react6 = require("react");
@@ -371,6 +374,7 @@ var useCopilotRuntimeClient = (options) => {
 };
 
 // src/hooks/use-chat.ts
+var import_runtime_client_gql4 = require("@copilotkit/runtime-client-gql");
 function useChat(options) {
   const {
     messages,
@@ -394,7 +398,9 @@ function useChat(options) {
     chatAbortControllerRef,
     agentLock,
     extensions,
-    setExtensions
+    setExtensions,
+    langGraphInterruptAction,
+    setLangGraphInterruptAction
   } = options;
   const runChatCompletionRef = (0, import_react6.useRef)();
   const addErrorToast = useErrorToast();
@@ -414,7 +420,7 @@ function useChat(options) {
   });
   const runChatCompletion = useAsyncCallback(
     (previousMessages) => __async(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
       setIsLoading(true);
       let newMessages = [
         new import_runtime_client_gql3.TextMessage({
@@ -437,6 +443,7 @@ function useChat(options) {
             threadId,
             runId: runIdRef.current,
             extensions: extensionsRef.current,
+            metaEvents: composeAndFlushMetaEventsInput([langGraphInterruptAction == null ? void 0 : langGraphInterruptAction.event]),
             messages: (0, import_runtime_client_gql3.convertMessagesToGqlInput)((0, import_runtime_client_gql3.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) ? {
@@ -498,15 +505,22 @@ function useChat(options) {
           messages2 = (0, import_runtime_client_gql3.convertGqlOutputToMessages)(
             (0, import_runtime_client_gql3.filterAdjacentAgentStateMessages)(value.generateCopilotResponse.messages)
           );
+          ((_i = (_h = value.generateCopilotResponse) == null ? void 0 : _h.metaEvents) != null ? _i : []).forEach((ev) => {
+            if (ev.name === "LangGraphInterruptEvent") {
+              setLangGraphInterruptAction({
+                event: (0, import_runtime_client_gql3.langGraphInterruptEvent)(ev)
+              });
+            }
+          });
           if (messages2.length === 0) {
             continue;
           }
           newMessages = [];
-          if (((_h = value.generateCopilotResponse.status) == null ? void 0 : _h.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
+          if (((_j = value.generateCopilotResponse.status) == null ? void 0 : _j.__typename) === "FailedResponseStatus" && value.generateCopilotResponse.status.reason === "GUARDRAILS_VALIDATION_FAILED") {
             newMessages = [
               new import_runtime_client_gql3.TextMessage({
                 role: import_runtime_client_gql3.MessageRole.Assistant,
-                content: ((_i = value.generateCopilotResponse.status.details) == null ? void 0 : _i.guardrailsReason) || ""
+                content: ((_k = value.generateCopilotResponse.status.details) == null ? void 0 : _k.guardrailsReason) || ""
               })
             ];
             setMessages([...previousMessages, ...newMessages]);
@@ -644,11 +658,11 @@ function useChat(options) {
           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)
+          !((_l = chatAbortControllerRef.current) == null ? void 0 : _l.signal.aborted)
         ) {
           yield new Promise((resolve) => setTimeout(resolve, 10));
           return yield runChatCompletionRef.current(finalMessages);
-        } else if ((_k = chatAbortControllerRef.current) == null ? void 0 : _k.signal.aborted) {
+        } else if ((_m = chatAbortControllerRef.current) == null ? void 0 : _m.signal.aborted) {
           const repairedMessages = finalMessages.filter((message, actionExecutionIndex) => {
             if (message.isActionExecutionMessage()) {
               return finalMessages.find(
@@ -659,7 +673,7 @@ function useChat(options) {
           });
           const repairedMessageIds = repairedMessages.map((message) => message.id);
           setMessages(repairedMessages);
-          if ((_l = agentSessionRef.current) == null ? void 0 : _l.nodeName) {
+          if ((_n = agentSessionRef.current) == null ? void 0 : _n.nodeName) {
             setAgentSession({
               threadId: agentSessionRef.current.threadId,
               agentName: agentSessionRef.current.agentName,
@@ -698,6 +712,33 @@ function useChat(options) {
     }),
     [messages]
   );
+  const composeAndFlushMetaEventsInput = (0, import_react6.useCallback)(
+    (metaEvents) => {
+      return metaEvents.reduce((acc, event) => {
+        if (!event)
+          return acc;
+        switch (event.name) {
+          case import_runtime_client_gql4.MetaEventName.LangGraphInterruptEvent:
+            if (event.response) {
+              setLangGraphInterruptAction(null);
+              return [
+                ...acc,
+                {
+                  type: event.type,
+                  name: event.name,
+                  value: event.value,
+                  response: event.response
+                }
+              ];
+            }
+            return acc;
+          default:
+            return acc;
+        }
+      }, []);
+    },
+    [setLangGraphInterruptAction]
+  );
   const append = useAsyncCallback(
     (message, options2) => __async(this, null, function* () {
       var _a;
@@ -789,7 +830,9 @@ function useCopilotChat(_a = {}) {
     setRunId,
     chatAbortControllerRef,
     extensions,
-    setExtensions
+    setExtensions,
+    langGraphInterruptAction,
+    setLangGraphInterruptAction
   } = useCopilotContext();
   const { messages, setMessages } = useCopilotMessagesContext();
   const latestGetContextString = useUpdatedRef(getContextString);
@@ -802,9 +845,9 @@ function useCopilotChat(_a = {}) {
   const makeSystemMessageCallback = (0, import_react8.useCallback)(() => {
     const systemMessageMaker = makeSystemMessage || defaultSystemMessage;
     const contextString = latestGetContextString.current([], defaultCopilotContextCategories);
-    return new import_runtime_client_gql4.TextMessage({
+    return new import_runtime_client_gql5.TextMessage({
       content: systemMessageMaker(contextString, chatInstructions),
-      role: import_runtime_client_gql4.Role.System
+      role: import_runtime_client_gql5.Role.System
     });
   }, [getContextString, makeSystemMessage, chatInstructions]);
   const onCoAgentStateRender = useAsyncCallback(
@@ -848,7 +891,9 @@ function useCopilotChat(_a = {}) {
     chatAbortControllerRef,
     agentLock,
     extensions,
-    setExtensions
+    setExtensions,
+    langGraphInterruptAction,
+    setLangGraphInterruptAction
   }));
   const latestAppend = useUpdatedRef(append);
   const latestAppendFunc = useAsyncCallback(