chore: add missing runtime status

Author: radu-mocanu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-runtime"
-version = "0.1.2"
+version = "0.1.3"
 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.5, <0.1.0",
+  "uipath-core==0.0.7.dev1000110062",
 ]
 classifiers = [
   "Intended Audience :: Developers",
@@ -106,3 +106,6 @@ name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
+
+[tool.uv.sources]
+uipath-core = { index = "testpypi" }

src/uipath/runtime/debug/runtime.py

@@ -4,6 +4,8 @@
 import logging
 from typing import Any, AsyncGenerator, cast
 
+from uipath.core.exceptions import UiPathResumeTriggerException, UiPathResumeTriggerPendingException
+
 from uipath.runtime.base import (
     UiPathExecuteOptions,
     UiPathRuntimeProtocol,
@@ -15,6 +17,8 @@
     UiPathDebugBridgeProtocol,
     UiPathDebugQuitError,
 )
+from uipath.runtime.errors import UiPathRuntimeError, UiPathErrorCode
+from uipath.runtime.errors.exception import handle_runtime_exceptions
 from uipath.runtime.events import (
     UiPathRuntimeEvent,
     UiPathRuntimeStateEvent,
@@ -54,6 +58,7 @@ def __init__(
             raise ValueError("trigger_poll_interval must be >= 0")
         self.trigger_poll_interval = trigger_poll_interval
 
+    @handle_runtime_exceptions
     async def execute(
         self,
         input: dict[str, Any] | None = None,
@@ -71,6 +76,7 @@ async def execute(
             else UiPathRuntimeResult(status=UiPathRuntimeStatus.SUCCESSFUL)
         )
 
+    @handle_runtime_exceptions
     async def stream(
         self,
         input: dict[str, Any] | None = None,
@@ -271,9 +277,14 @@ async def _poll_trigger(
 
                 await self._wait_with_quit_check()
 
-            except UiPathDebugQuitError:
+            except (UiPathDebugQuitError):
                 raise
-            except Exception as e:
+
+            except UiPathResumeTriggerException as e:
+                if not isinstance(e, UiPathResumeTriggerPendingException):
+                    raise UiPathRuntimeError.from_resume_trigger_exception(e)
+
+                # trigger still pending
                 await self.debug_bridge.emit_state_update(
                     UiPathRuntimeStateEvent(
                         node_name="<polling>",
@@ -284,7 +295,7 @@ async def _poll_trigger(
                     )
                 )
 
-                await self._wait_with_quit_check()
+            await self._wait_with_quit_check()
 
     async def _wait_with_quit_check(self) -> None:
         """Wait for specified seconds, but allow quit command to interrupt.

src/uipath/runtime/errors/codes.py

@@ -24,6 +24,7 @@ class UiPathErrorCode(str, Enum):
     INPUT_INVALID_JSON = "INPUT_INVALID_JSON"
 
     # HITL (Human-In-The-Loop) related errors
+    RESUME_TRIGGER_ERROR = "RESUME_TRIGGER_ERROR"
     INVOKED_PROCESS_FAILURE = "INVOKED_PROCESS_FAILURE"
     CREATE_RESUME_TRIGGER_ERROR = "CREATE_RESUME_TRIGGER_ERROR"
     RETRIEVE_RESUME_TRIGGER_ERROR = "RETRIEVE_PAYLOAD_ERROR"

src/uipath/runtime/errors/exception.py

@@ -2,7 +2,10 @@
 
 import sys
 import traceback
-from typing import Any
+from functools import wraps
+from typing import Any, Callable
+
+from uipath.core.exceptions import UiPathResumeTriggerException
 
 from uipath.runtime.errors.codes import UiPathErrorCode
 from uipath.runtime.errors.contract import UiPathErrorCategory, UiPathErrorContract
@@ -99,3 +102,21 @@ def __init__(
             prefix=prefix,
             include_traceback=include_traceback,
         )
+
+    @classmethod
+    def from_resume_trigger_exception(cls, exc: UiPathResumeTriggerException) -> "UiPathRuntimeError":
+        return cls(
+            code=UiPathErrorCode.RESUME_TRIGGER_ERROR,
+            title="Resume trigger error",
+            detail=exc.message,
+        )
+
+def handle_runtime_exceptions(func: Callable[..., Any]) -> Callable[..., Any]:
+    @wraps(func)
+    async def wrapper(*args, **kwargs):
+        try:
+            return await func(*args, **kwargs)
+        except UiPathResumeTriggerException as e:
+            raise UiPathRuntimeError.from_resume_trigger_exception(e)
+
+    return wrapper

src/uipath/runtime/result.py

@@ -17,7 +17,6 @@ class UiPathRuntimeStatus(str, Enum):
     FAULTED = "faulted"
     SUSPENDED = "suspended"
 
-
 class UiPathRuntimeResult(UiPathRuntimeEvent):
     """Result of an execution with status and optional error information."""
 

src/uipath/runtime/resumable/runtime.py

@@ -3,12 +3,16 @@
 import logging
 from typing import Any, AsyncGenerator
 
+from uipath.core.exceptions import UiPathResumeTriggerException
+
 from uipath.runtime.base import (
     UiPathExecuteOptions,
     UiPathRuntimeProtocol,
     UiPathStreamOptions,
 )
 from uipath.runtime.debug import UiPathBreakpointResult
+from uipath.runtime.errors import UiPathRuntimeError
+from uipath.runtime.errors.exception import handle_runtime_exceptions
 from uipath.runtime.events import UiPathRuntimeEvent
 from uipath.runtime.result import UiPathRuntimeResult, UiPathRuntimeStatus
 from uipath.runtime.resumable.protocols import (
@@ -47,6 +51,7 @@ def __init__(
         self.storage = storage
         self.trigger_manager = trigger_manager
 
+    @handle_runtime_exceptions
     async def execute(
         self,
         input: dict[str, Any] | None = None,
@@ -66,11 +71,15 @@ async def execute(
             input = await self._restore_resume_input(input)
 
         # Execute the delegate
-        result = await self.delegate.execute(input, options=options)
+        try:
+            result = await self.delegate.execute(input, options=options)
+        except UiPathResumeTriggerException as e:
+            raise UiPathRuntimeError.from_resume_trigger_exception(e)
 
         # If suspended, create and persist trigger
         return await self._handle_suspension(result)
 
+    @handle_runtime_exceptions
     async def stream(
         self,
         input: dict[str, Any] | None = None,