chore: import upstream snapshot with attribution

python/ray/serve/_private/deployment_node.py

@@ -0,0 +1,56 @@
+from typing import Any, Dict, List, Optional, Tuple
+
+from ray.dag import DAGNode
+from ray.dag.format_utils import get_dag_node_str
+from ray.serve.deployment import Deployment
+from ray.serve.handle import DeploymentHandle
+
+
+class DeploymentNode(DAGNode):
+    """Represents a deployment node in a DAG authored Ray DAG API."""
+
+    def __init__(
+        self,
+        # For serve structured deployment, deployment body can be import path
+        # to the class or function instead.
+        deployment: Deployment,
+        app_name: str,
+        deployment_init_args: Tuple[Any],
+        deployment_init_kwargs: Dict[str, Any],
+        ray_actor_options: Dict[str, Any],
+        other_args_to_resolve: Optional[Dict[str, Any]] = None,
+    ):
+        # Assign instance variables in base class constructor.
+        super().__init__(
+            deployment_init_args,
+            deployment_init_kwargs,
+            ray_actor_options,
+            other_args_to_resolve=other_args_to_resolve,
+        )
+        self._app_name = app_name
+        self._deployment = deployment
+        self._deployment_handle = DeploymentHandle(
+            self._deployment.name, self._app_name
+        )
+
+    def _copy_impl(
+        self,
+        new_args: List[Any],
+        new_kwargs: Dict[str, Any],
+        new_options: Dict[str, Any],
+        new_other_args_to_resolve: Dict[str, Any],
+    ):
+        return DeploymentNode(
+            self._deployment,
+            self._app_name,
+            new_args,
+            new_kwargs,
+            new_options,
+            other_args_to_resolve=new_other_args_to_resolve,
+        )
+
+    def __str__(self) -> str:
+        return get_dag_node_str(self, str(self._deployment))
+
+    def get_deployment_name(self):
+        return self._deployment.name