fix: handle escalation task deletion scenario (#654)

Author: cristian-groza

src/uipath_langchain/agent/tools/escalation_tool.py

@@ -210,6 +210,13 @@ async def create_escalation_task():
         if isinstance(result, dict):
             result = TypeAdapter(EscalationToolOutput).validate_python(result)
 
+        if result.is_deleted:
+            return {
+                "action": EscalationAction.END,
+                "output": None,
+                "outcome": "The escalation task was deleted",
+            }
+
         outcome = result.action
         escalation_output = _parse_task_data(
             result.data.model_dump()

tests/agent/tools/test_escalation_tool.py

@@ -297,6 +297,7 @@ async def test_escalation_tool_metadata_has_recipient(
         mock_result = MagicMock()
         mock_result.action = None
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         tool = create_escalation_tool(escalation_resource)
@@ -325,6 +326,7 @@ async def test_escalation_tool_metadata_recipient_none_when_no_recipients(
         mock_result = MagicMock()
         mock_result.action = None
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         tool = create_escalation_tool(escalation_resource_no_recipient)
@@ -349,6 +351,7 @@ async def test_escalation_tool_with_string_task_title(
         mock_result = MagicMock()
         mock_result.action = None
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         # Create resource with string task title
@@ -398,6 +401,7 @@ async def test_escalation_tool_with_text_builder_task_title(
         mock_result = MagicMock()
         mock_result.action = None
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         # Create resource with TEXT_BUILDER task title containing variable token
@@ -455,6 +459,7 @@ async def test_escalation_tool_with_empty_task_title_defaults_to_escalation_task
         mock_result = MagicMock()
         mock_result.action = None
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         # Create resource with empty string task title
@@ -555,6 +560,7 @@ async def test_escalation_tool_result_validation(
         mock_result.assigned_to_user = None
         mock_result.action = "approve"
         mock_result.data = {}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         tool = create_escalation_tool(escalation_resource)
@@ -579,6 +585,7 @@ async def test_escalation_tool_extracts_action_from_result(
         mock_result = MagicMock()
         mock_result.action = "approve"
         mock_result.data = {"approved": True}
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         tool = create_escalation_tool(escalation_resource)
@@ -588,6 +595,31 @@ async def test_escalation_tool_extracts_action_from_result(
 
         assert mock_interrupt.called
 
+    @pytest.mark.asyncio
+    @patch("uipath_langchain.agent.tools.escalation_tool.UiPath")
+    @patch("uipath_langchain.agent.tools.durable_interrupt.decorator.interrupt")
+    async def test_escalation_tool_raises_when_task_is_deleted(
+        self, mock_interrupt, mock_uipath_class, escalation_resource
+    ):
+        """Test that escalation tool raises AgentRuntimeError when task is deleted."""
+        from uipath_langchain.agent.exceptions import AgentRuntimeError
+
+        mock_client = MagicMock()
+        mock_client.tasks.create_async = AsyncMock(return_value=_make_mock_task())
+        mock_uipath_class.return_value = mock_client
+
+        mock_result = MagicMock()
+        mock_result.action = "approve"
+        mock_result.data = {}
+        mock_result.is_deleted = True
+        mock_interrupt.return_value = mock_result
+
+        tool = create_escalation_tool(escalation_resource)
+        call = ToolCall(args={}, id="test-call", name=tool.name)
+
+        with pytest.raises(AgentRuntimeError):
+            await tool.awrapper(tool, call, {})  # type: ignore[attr-defined]
+
     @pytest.mark.asyncio
     @patch("uipath_langchain.agent.tools.escalation_tool.UiPath")
     @patch("uipath_langchain.agent.tools.durable_interrupt.decorator.interrupt")
@@ -787,6 +819,7 @@ async def test_creates_task_then_interrupts_with_wait_escalation(
         mock_result.action = "approve"
         mock_result.data = {}
         mock_result.assigned_to_user = None
+        mock_result.is_deleted = False
         mock_interrupt.return_value = mock_result
 
         tool = create_escalation_tool(escalation_resource)