anyscale 0.26.8__py3-none-any.whl → 0.26.9__py3-none-any.whl

anyscale/_private/anyscale_client/anyscale_client.py

@@ -995,13 +995,22 @@ class AnyscaleClient(AnyscaleClientInterface):
 
     @handle_api_exceptions
     def get_service(
-        self, name: str, *, cloud: Optional[str], project: Optional[str]
+        self,
+        name: str,
+        *,
+        cloud: Optional[str],
+        project: Optional[str],
+        include_archived: bool = False,
     ) -> Optional[ServiceModel]:
         # TODO(edoakes): this endpoint is very slow and there's no reason we should need
         # to use this complex list endpoint just to fetch a service by name.
         paging_token = None
         cloud_id = self.get_cloud_id(cloud_name=cloud)
         project_id = self.get_project_id(parent_cloud_id=cloud_id, name=project)
+        archive_status = (
+            ArchiveStatus.ALL if include_archived else ArchiveStatus.NOT_ARCHIVED
+        )
+
         service: Optional[ServiceModel] = None
         while True:
             resp = self._external_api_client.list_services(
@@ -1009,6 +1018,7 @@ class AnyscaleClient(AnyscaleClientInterface):
                 name=name,
                 count=self.LIST_ENDPOINT_COUNT,
                 paging_token=paging_token,
+                archive_status=archive_status,
             )
             for result in resp.results:
                 if result.name == name:
@@ -1121,6 +1131,12 @@ class AnyscaleClient(AnyscaleClientInterface):
         result: ServiceModel = self._external_api_client.archive_service(service_id)
         return result
 
+    @handle_api_exceptions
+    def delete_service(self, service_id: str):
+        self._internal_api_client.delete_service_api_v2_services_v2_service_id_delete(
+            service_id
+        )
+
     @handle_api_exceptions
     def submit_job(self, model: CreateInternalProductionJob) -> InternalProductionJob:
         job: InternalProductionJob = self._internal_api_client.create_job_api_v2_decorated_ha_jobs_create_post(

anyscale/_private/anyscale_client/common.py

@@ -49,7 +49,7 @@ from anyscale.utils.workspace_notification import WorkspaceNotification
 # Maybe just make it part of the release process to update it, or fetch the
 # default builds and get the latest one. The best thing to do is probably
 # to populate this in the backend.
-DEFAULT_RAY_VERSION = "2.44.0"  # RAY_RELEASE_UPDATE: update to latest version.
+DEFAULT_RAY_VERSION = "2.44.1"  # RAY_RELEASE_UPDATE: update to latest version.
 DEFAULT_PYTHON_VERSION = "py311"
 RUNTIME_ENV_PACKAGE_FORMAT = "pkg_{content_hash}.zip"
 
@@ -297,7 +297,12 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def get_service(
-        self, name: str, *, cloud: Optional[str], project: Optional[str]
+        self,
+        name: str,
+        *,
+        cloud: Optional[str],
+        project: Optional[str],
+        include_archived=False,
     ) -> Optional[ServiceModel]:
         """Get a service by name. Filter by cloud and project.
 
@@ -371,6 +376,11 @@ class AnyscaleClientInterface(ABC):
         """Mark the service to be archived asynchronously."""
         raise NotImplementedError
 
+    @abstractmethod
+    def delete_service(self, service_id: str):
+        """Mark the service to be deleted asynchronously."""
+        raise NotImplementedError
+
     @abstractmethod
     def submit_job(self, model: CreateInternalProductionJob) -> InternalProductionJob:
         """Submit the job to run."""

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -145,6 +145,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._workspace_dependency_tracking_enabled: bool = False
         self._services: Dict[str, ServiceModel] = {}
         self._archived_services: Dict[str, ServiceModel] = {}
+        self._deleted_services: Dict[str, ServiceModel] = {}
         self._jobs: Dict[str, ProductionJob] = {}
         self._job_runs: Dict[str, List[APIJobRun]] = defaultdict(list)
         self._project_to_id: Dict[Optional[str] : Dict[Optional[str], str]] = {}
@@ -601,7 +602,12 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._services[model.id] = model
 
     def get_service(
-        self, name: str, *, cloud: Optional[str], project: Optional[str]
+        self,
+        name: str,
+        *,
+        cloud: Optional[str],
+        project: Optional[str],
+        include_archived=False,
     ) -> Optional[ServiceModel]:
         cloud_id = self.get_cloud_id(cloud_name=cloud)
         cloud_project_dict = self._project_to_id.get(cloud_id, None)
@@ -614,6 +620,13 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             ):
                 return service
 
+        if include_archived:
+            for service in self._archived_services.values():
+                if service.name == name and (
+                    project_id is None or service.project_id == project_id
+                ):
+                    return service
+
         return None
 
     def get_project(self, project_id: str) -> Optional[Project]:
@@ -780,8 +793,12 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def archive_service(self, service_id: str):
         self._archived_services[service_id] = self._services.pop(service_id)
 
-    def is_archived_service(self, service_id: str) -> bool:
-        return service_id in self._archived_services
+    @property
+    def deleted_services(self) -> Dict[str, ServiceModel]:
+        return self._deleted_services
+
+    def delete_service(self, service_id: str):
+        self._deleted_services[service_id] = self._services.pop(service_id)
 
     @property
     def submitted_job(self) -> Optional[CreateInternalProductionJob]:

anyscale/_private/docgen/__main__.py

@@ -245,6 +245,8 @@ ALL_MODULES = [
             service_commands.wait,
             service_commands.rollback,
             service_commands.terminate,
+            service_commands.archive,
+            service_commands.delete,
         ],
         sdk_prefix="anyscale.service",
         sdk_commands=[
@@ -253,6 +255,8 @@ ALL_MODULES = [
             anyscale.service.wait,
             anyscale.service.rollback,
             anyscale.service.terminate,
+            anyscale.service.archive,
+            anyscale.service.delete,
         ],
         models=[
             ServiceConfig,