chore: debug runtime, check for any resumable runtime in the delegates chain

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-runtime"
-version = "0.8.0"
+version = "0.8.1"
 description = "Runtime abstractions and interfaces for building agents and automation scripts in the UiPath ecosystem"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/runtime/debug/runtime.py

@@ -179,7 +179,7 @@ async def _stream_and_debug(
 
                         # Check if this is a suspended execution that needs polling
                         if (
-                            isinstance(self.delegate, UiPathResumableRuntime)
+                            (resumable_runtime := self.get_resumable_runtime())
                             and self.trigger_poll_interval > 0
                             and final_result.status == UiPathRuntimeStatus.SUSPENDED
                             and final_result.trigger
@@ -204,7 +204,7 @@ async def _stream_and_debug(
                                 else:
                                     trigger_data = await self._poll_trigger(
                                         final_result.trigger,
-                                        self.delegate.trigger_manager,
+                                        resumable_runtime.trigger_manager,
                                     )
                                 resume_data = {interrupt_id: trigger_data}
                             except UiPathDebugQuitError:
@@ -233,6 +233,15 @@ async def _stream_and_debug(
                 elif isinstance(event, UiPathRuntimeStateEvent):
                     await self.debug_bridge.emit_state_update(event)
 
+    def get_resumable_runtime(self) -> UiPathResumableRuntime | None:
+        """Get the delegate resumable runtime, if exists."""
+        current_runtime: UiPathRuntimeProtocol = self
+        while hasattr(current_runtime, "delegate"):
+            current_runtime = current_runtime.delegate
+            if isinstance(current_runtime, UiPathResumableRuntime):
+                return current_runtime
+        return None
+
     async def get_schema(self) -> UiPathRuntimeSchema:
         """Passthrough schema for the delegate."""
         return await self.delegate.get_schema()