fix: properly handle error types in context manager (#79)

* fix: properly handle error types in context manager

* chore: bump version

Author: andreitava-uip

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-runtime"
-version = "0.6.2"
+version = "0.6.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"

src/uipath/runtime/context.py

@@ -12,6 +12,7 @@
 from uipath.core.tracing import UiPathTraceManager
 
 from uipath.runtime.errors import (
+    UiPathBaseRuntimeError,
     UiPathErrorCategory,
     UiPathErrorCode,
     UiPathErrorContract,
@@ -200,14 +201,14 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             if self.result is None:
                 self.result = UiPathRuntimeResult()
 
-            if exc_type:
+            if exc_type and exc_val:
                 # Create error info from exception
-                match exc_type:
+                match exc_val:
                     case UiPathFaultedTriggerError():
                         error_info = UiPathRuntimeError.from_resume_trigger_error(
                             exc_type
                         ).error_info
-                    case UiPathRuntimeError():
+                    case UiPathBaseRuntimeError():
                         error_info = exc_val.error_info
                     case _:
                         # Generic error