fix: properly handle error types in context manager

Author: andreitava-uip

src/uipath/runtime/context.py

@@ -12,6 +12,7 @@
 from uipath.core.tracing import UiPathTraceManager
 
 from uipath.runtime.errors import (
+    UiPathBaseRuntimeError,
     UiPathErrorCategory,
     UiPathErrorCode,
     UiPathErrorContract,
@@ -201,22 +202,19 @@ def __exit__(self, exc_type, exc_val, exc_tb):
                 self.result = UiPathRuntimeResult()
 
             if exc_type:
-                # Create error info from exception
-                match exc_type:
-                    case UiPathFaultedTriggerError():
-                        error_info = UiPathRuntimeError.from_resume_trigger_error(
-                            exc_type
-                        ).error_info
-                    case UiPathRuntimeError():
-                        error_info = exc_val.error_info
-                    case _:
-                        # Generic error
-                        error_info = UiPathErrorContract(
-                            code=f"ERROR_{exc_type.__name__}",
-                            title=f"Runtime error: {exc_type.__name__}",
-                            detail=str(exc_val),
-                            category=UiPathErrorCategory.UNKNOWN,
-                        )
+                if issubclass(exc_type, UiPathFaultedTriggerError):
+                    error_info = UiPathRuntimeError.from_resume_trigger_error(
+                        exc_val
+                    ).error_info
+                elif issubclass(exc_type, UiPathBaseRuntimeError):
+                    error_info = exc_val.error_info
+                else:
+                    error_info = UiPathErrorContract(
+                        code=f"ERROR_{exc_type.__name__}",
+                        title=f"Runtime error: {exc_type.__name__}",
+                        detail=str(exc_val),
+                        category=UiPathErrorCategory.UNKNOWN,
+                    )
 
                 self.result.status = UiPathRuntimeStatus.FAULTED
                 self.result.error = error_info