feat: support resumable trigger batch deletion (#973)

Author: radu-mocanu

pyproject.toml

@@ -1,14 +1,14 @@
 [project]
 name = "uipath-langchain"
-version = "0.13.25"
+version = "0.14.0"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-    "uipath>=2.12.7, <2.13.0",
+    "uipath>=2.13.0, <2.14.0",
     "uipath-core>=0.5.28, <0.6.0",
-    "uipath-platform>=0.1.91, <0.2.0",
-    "uipath-runtime>=0.11.7, <0.12.0",
+    "uipath-platform>=0.2.0, <0.3.0",
+    "uipath-runtime>=0.12.1, <0.13.0",
     "langgraph>=1.1.8, <2.0.0",
     "langchain-core>=1.2.27, <2.0.0",
     "langgraph-checkpoint-sqlite>=3.0.3, <4.0.0",
@@ -169,4 +169,4 @@ jsonschema-pydantic-converter = false
 name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
-explicit = true
+explicit = true

src/uipath_langchain/runtime/storage.py

@@ -133,18 +133,25 @@ async def delete_trigger(
         self, runtime_id: str, trigger: UiPathResumeTrigger
     ) -> None:
         """Delete resume trigger from storage."""
+        await self.delete_triggers(runtime_id, [trigger])
+
+    async def delete_triggers(
+        self, runtime_id: str, triggers: list[UiPathResumeTrigger]
+    ) -> None:
+        """Delete resume triggers from storage."""
         await self._ensure_table()
+        interrupt_ids = [trigger.interrupt_id for trigger in triggers]
+        if not interrupt_ids:
+            return
 
         async with self.memory.lock, self.memory.conn.cursor() as cur:
             await cur.execute(
                 f"""
                 DELETE FROM {self.rs_table_name}
-                WHERE runtime_id = ? AND interrupt_id = ?
+                WHERE runtime_id = ?
+                AND interrupt_id IN ({",".join("?" for _ in interrupt_ids)})
                 """,
-                (
-                    runtime_id,
-                    trigger.interrupt_id,
-                ),
+                (runtime_id, *interrupt_ids),
             )
             await self.memory.conn.commit()
 

testcases/dev-console/pyproject.toml

@@ -8,7 +8,7 @@ dependencies = [
     "langchain-core>=0.3.34",
     "python-dotenv>=1.0.1",
     "uipath-langchain",
-    "uipath-dev>=0.0.12",
+    "uipath-dev>=0.0.84",
     "pydantic>=2.10.6",
     "typing-extensions>=4.12.2",
     "pexpect>=4.9.0",

tests/runtime/test_resumable.py

@@ -40,6 +40,10 @@ async def create_trigger(self, suspend_value: Any) -> UiPathResumeTrigger:
         )
         return trigger
 
+    async def create_triggers(self, suspend_value: Any) -> list[UiPathResumeTrigger]:
+        """Create triggers from suspend value."""
+        return [await self.create_trigger(suspend_value)]
+
     async def read_trigger(self, trigger: UiPathResumeTrigger) -> Any:
         """Read trigger and return mock response.
 
@@ -79,6 +83,10 @@ async def create_trigger(self, suspend_value: Any) -> UiPathResumeTrigger:
         )
         return trigger
 
+    async def create_triggers(self, suspend_value: Any) -> list[UiPathResumeTrigger]:
+        """Create triggers from suspend value."""
+        return [await self.create_trigger(suspend_value)]
+
     async def read_trigger(self, trigger: UiPathResumeTrigger) -> Any:
         """Read trigger and return immediate response."""
         assert trigger.payload is not None
@@ -716,6 +724,10 @@ async def create_trigger(self, suspend_value: Any) -> UiPathResumeTrigger:
         )
         return trigger
 
+    async def create_triggers(self, suspend_value: Any) -> list[UiPathResumeTrigger]:
+        """Create triggers from suspend value."""
+        return [await self.create_trigger(suspend_value)]
+
     async def read_trigger(self, trigger: UiPathResumeTrigger) -> Any:
         """Read trigger and return mock response based on call pattern."""
         self.call_count += 1

tests/runtime/test_storage.py

@@ -193,6 +193,68 @@ async def test_delete_trigger(self, storage: SqliteResumableStorage):
         assert len(triggers) == 1
         assert triggers[0].interrupt_id == "interrupt2"
 
+    @pytest.mark.asyncio
+    async def test_delete_triggers(self, storage: SqliteResumableStorage):
+        """Test deleting sibling triggers for the same interrupt."""
+        api_trigger = UiPathResumeTrigger(
+            interrupt_id="interrupt1",
+            trigger_type=UiPathResumeTriggerType.API,
+            trigger_name=UiPathResumeTriggerName.API,
+            payload="api-payload",
+        )
+        timer_trigger = UiPathResumeTrigger(
+            interrupt_id="interrupt1",
+            trigger_type=UiPathResumeTriggerType.TIMER,
+            trigger_name=UiPathResumeTriggerName.TIMER,
+            payload="timer-payload",
+        )
+        unrelated_trigger = UiPathResumeTrigger(
+            interrupt_id="interrupt2",
+            trigger_type=UiPathResumeTriggerType.TASK,
+            trigger_name=UiPathResumeTriggerName.TASK,
+            payload="unrelated-payload",
+        )
+        other_runtime_trigger = UiPathResumeTrigger(
+            interrupt_id="interrupt1",
+            trigger_type=UiPathResumeTriggerType.API,
+            trigger_name=UiPathResumeTriggerName.API,
+            payload="other-runtime-payload",
+        )
+        await storage.save_triggers(
+            "runtime1", [api_trigger, timer_trigger, unrelated_trigger]
+        )
+        await storage.save_triggers("runtime2", [other_runtime_trigger])
+
+        await storage.delete_triggers("runtime1", [api_trigger, timer_trigger])
+
+        triggers = await storage.get_triggers("runtime1")
+        assert triggers is not None
+        assert [trigger.interrupt_id for trigger in triggers] == ["interrupt2"]
+        assert triggers[0].payload == "unrelated-payload"
+
+        other_triggers = await storage.get_triggers("runtime2")
+        assert other_triggers is not None
+        assert len(other_triggers) == 1
+        assert other_triggers[0].payload == "other-runtime-payload"
+
+    @pytest.mark.asyncio
+    async def test_delete_triggers_empty_list(self, storage: SqliteResumableStorage):
+        """Test deleting an empty trigger list leaves storage unchanged."""
+        trigger = UiPathResumeTrigger(
+            interrupt_id="interrupt1",
+            trigger_type=UiPathResumeTriggerType.API,
+            trigger_name=UiPathResumeTriggerName.API,
+            payload="payload1",
+        )
+        await storage.save_triggers("runtime1", [trigger])
+
+        await storage.delete_triggers("runtime1", [])
+
+        triggers = await storage.get_triggers("runtime1")
+        assert triggers is not None
+        assert len(triggers) == 1
+        assert triggers[0].interrupt_id == "interrupt1"
+
     @pytest.mark.asyncio
     async def test_get_nonexistent_triggers_returns_none(
         self, storage: SqliteResumableStorage