fix: move interrupt offset to hitl

Author: JoshParkSJ

src/uipath_langchain/agent/tools/tool_node.py

@@ -22,7 +22,6 @@
     extract_current_tool_call_index,
     find_latest_ai_message,
 )
-from uipath_langchain.agent.tools.durable_interrupt import add_interrupt_offset
 from uipath_langchain.chat.hitl import request_conversational_tool_confirmation
 
 # the type safety can be improved with generics
@@ -90,7 +89,6 @@ def _func(self, state: AgentGraphState) -> OutputType:
             if conversational_confirmation.cancelled:
                 # tool confirmation rejected
                 return self._process_result(call, conversational_confirmation.cancelled)
-            add_interrupt_offset()  # HITL consumed 1 interrupt slot
 
         try:
             if self.wrapper:
@@ -128,7 +126,6 @@ async def _afunc(self, state: AgentGraphState) -> OutputType:
             if conversational_confirmation.cancelled:
                 # tool confirmation rejected
                 return self._process_result(call, conversational_confirmation.cancelled)
-            add_interrupt_offset()  # HITL consumed 1 interrupt slot
 
         try:
             if self.awrapper:

src/uipath_langchain/chat/hitl.py

@@ -12,6 +12,8 @@
     UiPathConversationToolCallConfirmationValue,
 )
 
+from uipath_langchain.agent.tools.durable_interrupt import add_interrupt_offset
+
 CANCELLED_MESSAGE = "Cancelled by user"
 
 CONVERSATIONAL_APPROVED_TOOL_ARGS = "conversational_approved_tool_args"
@@ -127,6 +129,9 @@ def request_approval(
             input_value=tool_args,
         )
     )
+    # Workaround for langgraph#6792 — remove when subgraph @task + interrupt()
+    # checkpoint caching is fixed upstream
+    add_interrupt_offset()
 
     # The resume payload from CAS has shape:
     #   {"type": "uipath_cas_tool_call_confirmation",
@@ -145,6 +150,7 @@ def request_approval(
     )
 
 
+# for conversational low code agents
 def request_conversational_tool_confirmation(
     call: ToolCall, tool: BaseTool
 ) -> ConfirmationResult | None:
@@ -176,6 +182,7 @@ def request_conversational_tool_confirmation(
     )
 
 
+# for conversational coded agents
 def requires_approval(
     func: Callable[..., Any] | None = None,
     *,