Merge pull request #41 from UiPath/fix/resumable_output

fix: debugger emit suspend/resumed

Author: cristipufu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-runtime"
-version = "0.1.3"
+version = "0.2.0"
 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"
 dependencies = [
-  "uipath-core>=0.0.9, <0.1.0",
+  "uipath-core>=0.1.0, <0.2.0",
 ]
 classifiers = [
   "Intended Audience :: Developers",

src/uipath/runtime/debug/bridge.py

@@ -37,6 +37,16 @@ async def emit_breakpoint_hit(
         """Notify debugger that a breakpoint was hit."""
         ...
 
+    async def emit_execution_suspended(
+        self, runtime_result: UiPathRuntimeResult
+    ) -> None:
+        """Notify debugger that the execution has been suspended."""
+        ...
+
+    async def emit_execution_resumed(self, resume_data: Any) -> None:
+        """Notify debugger that the execution has resumed."""
+        ...
+
     async def emit_execution_completed(
         self,
         runtime_result: UiPathRuntimeResult,

src/uipath/runtime/debug/runtime.py

@@ -27,7 +27,10 @@
 )
 from uipath.runtime.resumable.protocols import UiPathResumeTriggerReaderProtocol
 from uipath.runtime.resumable.runtime import UiPathResumableRuntime
-from uipath.runtime.resumable.trigger import UiPathResumeTrigger
+from uipath.runtime.resumable.trigger import (
+    UiPathResumeTrigger,
+    UiPathResumeTriggerType,
+)
 from uipath.runtime.schema import UiPathRuntimeSchema
 
 logger = logging.getLogger(__name__)
@@ -181,20 +184,24 @@ async def _stream_and_debug(
                             and final_result.status == UiPathRuntimeStatus.SUSPENDED
                             and final_result.trigger
                         ):
-                            state_event = UiPathRuntimeStateEvent(
-                                node_name="<suspended>",
-                                payload={
-                                    "status": "suspended",
-                                    "trigger": final_result.trigger.model_dump(),
-                                },
+                            await self.debug_bridge.emit_execution_suspended(
+                                final_result
                             )
-                            await self.debug_bridge.emit_state_update(state_event)
 
                             resume_data: dict[str, Any] | None = None
                             try:
-                                resume_data = await self._poll_trigger(
-                                    final_result.trigger, self.delegate.trigger_manager
-                                )
+                                if (
+                                    final_result.trigger.trigger_type
+                                    == UiPathResumeTriggerType.API
+                                ):
+                                    resume_data = (
+                                        await self.debug_bridge.wait_for_resume()
+                                    )
+                                else:
+                                    resume_data = await self._poll_trigger(
+                                        final_result.trigger,
+                                        self.delegate.trigger_manager,
+                                    )
                             except UiPathDebugQuitError:
                                 final_result = UiPathRuntimeResult(
                                     status=UiPathRuntimeStatus.SUCCESSFUL,
@@ -203,14 +210,9 @@ async def _stream_and_debug(
                                 execution_completed = True
 
                             if resume_data is not None:
-                                resumed_event = UiPathRuntimeStateEvent(
-                                    node_name="<resumed>",
-                                    payload={
-                                        "status": "resumed",
-                                        "data": resume_data,
-                                    },
+                                await self.debug_bridge.emit_execution_resumed(
+                                    resume_data
                                 )
-                                await self.debug_bridge.emit_state_update(resumed_event)
 
                                 # Continue with resumed execution
                                 current_input = resume_data

src/uipath/runtime/resumable/runtime.py

@@ -111,7 +111,7 @@ async def _restore_resume_input(
             Input to use for resume, either provided or from storage
         """
         # If user provided explicit input, use it
-        if input:
+        if input is not None:
             return input
 
         # Otherwise, fetch from storage
@@ -146,6 +146,7 @@ async def _handle_suspension(
 
         suspended_result = UiPathRuntimeResult(
             status=UiPathRuntimeStatus.SUSPENDED,
+            output=result.output,
         )
 
         if result.output: