except GraphBubbleUp to handle durable_interrupt

Author: JoshParkSJ

src/uipath_langchain/agent/tools/tool_node.py

@@ -7,6 +7,7 @@
 from langchain_core.messages.tool import ToolCall, ToolMessage
 from langchain_core.tools import BaseTool
 from langgraph._internal._runnable import RunnableCallable
+from langgraph.errors import GraphBubbleUp
 from langgraph.types import Command
 from pydantic import BaseModel
 from uipath.platform.resume_triggers import is_no_content_marker
@@ -80,10 +81,10 @@ def _func(self, state: AgentGraphState) -> OutputType:
         if call is None:
             return None
 
-        # HITL: prompt user for approval if tool requires confirmation
+        # prompt user for approval if tool requires confirmation
         confirmation = request_tool_confirmation(call, self.tool)
 
-        # HITL cancelled: user rejected the tool call
+        # user rejected the tool call
         if confirmation is not None and confirmation.cancelled:
             return self._process_result(call, confirmation.cancelled)
 
@@ -96,10 +97,12 @@ def _func(self, state: AgentGraphState) -> OutputType:
             else:
                 result = self.tool.invoke(call)
             output = self._process_result(call, result)
-            # HITL approved: tag result with approved args (and whether they were modified)
+            # HITL approved - apply confirmation metadata to tool result message
             if confirmation is not None:
                 confirmation.annotate_result(output)
             return output
+        except GraphBubbleUp:
+            raise
         except Exception as e:
             if self.handle_tool_errors:
                 return self._process_error_result(call, e)
@@ -110,10 +113,10 @@ async def _afunc(self, state: AgentGraphState) -> OutputType:
         if call is None:
             return None
 
-        # HITL: prompt user for approval if tool requires confirmation
+        # prompt user for approval if tool requires confirmation
         confirmation = request_tool_confirmation(call, self.tool)
 
-        # HITL cancelled: user rejected the tool call
+        # user rejected the tool call
         if confirmation is not None and confirmation.cancelled:
             return self._process_result(call, confirmation.cancelled)
 
@@ -126,10 +129,12 @@ async def _afunc(self, state: AgentGraphState) -> OutputType:
             else:
                 result = await self.tool.ainvoke(call)
             output = self._process_result(call, result)
-            # HITL approved: tag result with approved args (and whether they were modified)
+            # HITL approved - apply confirmation metadata to tool result message
             if confirmation is not None:
                 confirmation.annotate_result(output)
             return output
+        except GraphBubbleUp:
+            raise
         except Exception as e:
             if self.handle_tool_errors:
                 return self._process_error_result(call, e)

src/uipath_langchain/runtime/messages.py

@@ -60,7 +60,7 @@ def __init__(self, runtime_id: str, storage: UiPathRuntimeStorageProtocol | None
         """Initialize the mapper with empty state."""
         self.runtime_id = runtime_id
         self.storage = storage
-        self.confirmation_tool_names: set[str] = set()
+        self.tool_names_requiring_confirmation: set[str] = set()
         self.current_message: AIMessageChunk
         self.seen_message_ids: set[str] = set()
         self._storage_lock = asyncio.Lock()
@@ -393,8 +393,11 @@ async def map_current_message_to_start_tool_call_events(self):
                             self.current_message.id
                         )
 
-                        if tool_call["name"] not in self.confirmation_tool_names:
-                            # if tool requires HITL, we skip start tool call
+                        # if tool requires confirmation, we skip start tool call
+                        if (
+                            tool_call["name"]
+                            not in self.tool_names_requiring_confirmation
+                        ):
                             events.append(
                                 self.map_tool_call_to_tool_call_start_event(
                                     self.current_message.id, tool_call
@@ -434,7 +437,7 @@ async def map_tool_message_to_events(
 
         events: list[UiPathConversationMessageEvent] = []
 
-        # Emit deferred startToolCall for confirmation tools (skipped in Pass 1)
+        # emit startToolCall for tools requiring confirmation after it's approved
         approved_args = message.response_metadata.get(CONVERSATIONAL_APPROVED_TOOL_ARGS)
         if approved_args is not None:
             tool_call = ToolCall(
@@ -683,6 +686,7 @@ def _map_langchain_ai_message_to_uipath_message_data(
             role="assistant",
             content_parts=content_parts,
             tool_calls=uipath_tool_calls,
+            interrupts=[],
         )
 
 

src/uipath_langchain/runtime/runtime.py

@@ -65,7 +65,9 @@ def __init__(
         self.entrypoint: str | None = entrypoint
         self.callbacks: list[BaseCallbackHandler] = callbacks or []
         self.chat = UiPathChatMessagesMapper(self.runtime_id, storage)
-        self.chat.confirmation_tool_names = self._get_confirmation_tool_names()
+        self.chat.tool_names_requiring_confirmation = (
+            self._get_tool_names_requiring_confirmation()
+        )
         self._middleware_node_names: set[str] = self._detect_middleware_nodes()
 
     async def execute(
@@ -488,10 +490,10 @@ def _detect_middleware_nodes(self) -> set[str]:
 
         return middleware_nodes
 
-    def _get_confirmation_tool_names(self) -> set[str]:
+    def _get_tool_names_requiring_confirmation(self) -> set[str]:
         names: set[str] = set()
         for node_name, node_spec in self.graph.nodes.items():
-            # PregelNode.bound -> Runnable, Runnable.tool -> BaseTool (if tool node)
+            # langgraph's processing node.bound -> runnable.tool -> baseTool (if tool node)
             tool = getattr(getattr(node_spec, "bound", None), "tool", None)
             if tool is None:
                 continue

tests/runtime/test_chat_message_mapper.py

@@ -1729,7 +1729,7 @@ async def test_start_tool_call_skipped_for_confirmation_tool(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.confirmation_tool_names = {"confirm_tool"}
+        mapper.tool_names_requiring_confirmation = {"confirm_tool"}
 
         # First chunk starts the message with a confirmation tool call
         first_chunk = AIMessageChunk(
@@ -1758,7 +1758,7 @@ async def test_start_tool_call_emitted_for_non_confirmation_tool(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.confirmation_tool_names = {"other_tool"}
+        mapper.tool_names_requiring_confirmation = {"other_tool"}
 
         first_chunk = AIMessageChunk(
             content="",
@@ -1790,7 +1790,7 @@ async def test_deferred_start_tool_call_emitted_from_tool_message(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {"tc-3": "msg-3"}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.confirmation_tool_names = {"confirm_tool"}
+        mapper.tool_names_requiring_confirmation = {"confirm_tool"}
 
         approved_args = {"query": "approved value"}
         tool_msg = ToolMessage(
@@ -1824,7 +1824,7 @@ async def test_mixed_tools_only_confirmation_deferred(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.confirmation_tool_names = {"confirm_tool"}
+        mapper.tool_names_requiring_confirmation = {"confirm_tool"}
 
         first_chunk = AIMessageChunk(
             content="",