Revert "Remove AsyncioActorExit"

This reverts commit 516bf49. Signed-off-by: Chi-Sheng Liu <[email protected]>
ray-project · Mar 4, 2025 · b845581 · b845581
1 parent 08fbd01
commit b845581
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/doc/source/ray-core/api/exceptions.rst b/doc/source/ray-core/api/exceptions.rst
@@ -15,6 +15,7 @@ Exceptions
     ray.exceptions.ActorDiedError
     ray.exceptions.ActorUnschedulableError
     ray.exceptions.ActorUnavailableError
+    ray.exceptions.AsyncioActorExit
     ray.exceptions.LocalRayletDiedError
     ray.exceptions.WorkerCrashedError
     ray.exceptions.TaskPlacementGroupRemoved

diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx
@@ -180,6 +180,7 @@ from ray.exceptions import (
     OutOfDiskError,
     GetTimeoutError,
     TaskCancelledError,
+    AsyncioActorExit,
     PendingCallsLimitExceeded,
     RpcError,
     ObjectRefStreamEndOfStreamError,
@@ -1957,6 +1958,8 @@ cdef void execute_task(
                         (ray._private.worker.global_worker
                          .debugger_breakpoint) = b""
                     task_exception = False
+                except AsyncioActorExit as e:
+                    exit_current_actor_if_asyncio()
                 except Exception as e:
                     is_retryable_error[0] = determine_if_retryable(
                                     should_retry_exceptions,

diff --git a/python/ray/exceptions.py b/python/ray/exceptions.py
@@ -777,6 +777,13 @@ class PlasmaObjectNotAvailable(RayError):
     pass
 
 
+@PublicAPI
+class AsyncioActorExit(RayError):
+    """Raised when an asyncio actor intentionally exits via exit_actor()."""
+
+    pass
+
+
 @PublicAPI
 class RuntimeEnvSetupError(RayError):
     """Raised when a runtime environment fails to be set up.
@@ -909,6 +916,7 @@ class RayCgraphCapacityExceeded(RaySystemError):
     ObjectReconstructionFailedLineageEvictedError,
     OwnerDiedError,
     GetTimeoutError,
+    AsyncioActorExit,
     RuntimeEnvSetupError,
     TaskPlacementGroupRemoved,
     ActorPlacementGroupRemoved,

diff --git a/python/ray/includes/common.pxi b/python/ray/includes/common.pxi
@@ -26,6 +26,7 @@ from ray.exceptions import (
     OutOfDiskError,
     GetTimeoutError,
     TaskCancelledError,
+    AsyncioActorExit,
     PendingCallsLimitExceeded,
     RpcError,
     ObjectRefStreamEndOfStreamError,