anyscale 0.24.53__tar.gz → 0.24.55__tar.gz

{anyscale-0.24.53 → anyscale-0.24.55}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.53
+Version: 0.24.55
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/__init__.py

@@ -33,6 +33,7 @@ from anyscale.llm import LLMSDK
 from anyscale.schedule import ScheduleSDK
 from anyscale.sdk.anyscale_client.sdk import AnyscaleSDK
 from anyscale.service import ServiceSDK
+from anyscale.workspace import WorkspaceSDK
 
 
 # Note: indentation here matches that of connect.py::ClientBuilder.
@@ -94,6 +95,7 @@ class Anyscale:
         self._schedule_sdk = ScheduleSDK(client=self._anyscale_client)
         self._image_sdk = ImageSDK(client=self._anyscale_client)
         self._llm_sdk = LLMSDK(client=self._anyscale_client)
+        self._workspace_sdk = WorkspaceSDK(client=self._anyscale_client)
 
     @classmethod
     def _init_private(
@@ -114,7 +116,9 @@ class Anyscale:
             client=client, logger=logger, timer=timer,
         )
         obj._image_sdk = ImageSDK(client=client, logger=logger)  # noqa: SLF001
-        obj._llm_sdk = LLMSDK(client=client, logger=logger, timer=timer)  # noqa: SLF001
+        obj._workspace_sdk = WorkspaceSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer,
+        )
         return obj
 
     @property
@@ -141,6 +145,10 @@ class Anyscale:
     def llm(self) -> LLMSDK:  # noqa: F811
         return self._llm_sdk
 
+    @property
+    def workspace(self) -> WorkspaceSDK:  # noqa: F811
+        return self._workspace_sdk
+
 
 def _validate_headers(headers: Optional[Dict[str, str]]):
     if not headers:

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -1,5 +1,7 @@
 from abc import ABC, abstractmethod
+from functools import wraps
 import io
+import json
 import logging
 import os
 import pathlib
@@ -32,16 +34,22 @@ from anyscale.client.openapi_client.models import (
     ComputeTemplateConfig,
     ComputeTemplateQuery,
     CreateComputeTemplate,
+    CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
     ExperimentalWorkspace,
     InternalProductionJob,
+    SessionSshKey,
+    SessionState,
+    StartSessionOptions,
+    StopSessionOptions,
 )
 from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
 from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
 from anyscale.client.openapi_client.models.production_job import ProductionJob
 from anyscale.client.openapi_client.rest import ApiException as InternalApiException
 from anyscale.cluster_compute import parse_cluster_compute_name_version
+from anyscale.feature_flags import FLAG_DEFAULT_WORKING_DIR_FOR_PROJ
 from anyscale.sdk.anyscale_client.api.default_api import DefaultApi as ExternalApi
 from anyscale.sdk.anyscale_client.models import (
     ApplyServiceModel,
@@ -98,6 +106,21 @@ OVERWRITE_EXISTING_CLOUD_STORAGE_FILES = (
 # internal_logger is used for logging internal errors or debug messages that we do not expect users to see.
 internal_logger = logging.getLogger(__name__)
 
+# A decorator to handle ApiException and raise ValueError with the error message.
+def handle_api_exceptions(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except ApiException as e:
+            if e.status >= 400 and e.status < 500:
+                body_dict = json.loads(e.body)
+                msg = body_dict["error"]["detail"]
+                raise ValueError(msg) from None
+            raise e from None
+
+    return wrapper
+
 
 class AWSS3ClientInterface(ABC):
     @abstractmethod
@@ -1025,22 +1048,11 @@ class AnyscaleClient(AnyscaleClientInterface):
         logs = self._read_log_lines(all_log_chunk_urls, head, bearer_token, max_lines)
         return logs
 
+    @handle_api_exceptions
     def apply_schedule(self, model: CreateSchedule) -> DecoratedSchedule:
-        try:
-            return self._internal_api_client.create_or_update_job_api_v2_experimental_cron_jobs_put(
-                model
-            ).result
-        except ApiException as e:
-            if e.status == 400 and (
-                "Please specify a valid cron expression" in e.body
-                or "Please specify a valid tz database timezone" in e.body
-            ):
-                import json
-
-                body_dict = json.loads(e.body)
-                msg = body_dict["error"]["detail"]
-                raise ValueError(msg) from None
-            raise e from None
+        return self._internal_api_client.create_or_update_job_api_v2_experimental_cron_jobs_put(
+            model
+        ).result
 
     def get_schedule(
         self,
@@ -1181,3 +1193,138 @@ class AnyscaleClient(AnyscaleClientInterface):
 
     def retrieve_finetuned_model(self, model_id: str) -> FineTunedModel:
         return self._external_api_client.get_model(model_id)
+
+    @handle_api_exceptions
+    def create_workspace(self, model: CreateExperimentalWorkspace) -> str:
+        return self._internal_api_client.create_workspace_api_v2_experimental_workspaces_post(
+            create_experimental_workspace=model,
+        ).result.id
+
+    def get_workspace(
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+    ) -> Optional[ExperimentalWorkspace]:
+        """Get a workspace by either name or id. Filter by cloud and project.
+
+        Returns None if not found.
+        """
+        if id is not None:
+            try:
+                return self._internal_api_client.get_workspace_api_v2_experimental_workspaces_workspace_id_get(
+                    id
+                ).result
+            except ExternalApiException as e:
+                if e.status == 404:
+                    return None
+                raise e from None
+        else:
+            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)
+            resp = self._internal_api_client.list_workspaces_api_v2_experimental_workspaces_get(
+                project_id=project_id,
+                name=name,
+                count=self.LIST_ENDPOINT_COUNT,
+                paging_token=paging_token,
+            )
+
+            if len(resp.results) == 0:
+                return None
+
+            workspace = resp.results[0]
+            return workspace
+
+    @handle_api_exceptions
+    def update_workspace_dependencies_offline_only(
+        self, workspace_id: str, requirements: List[str]
+    ):
+        return self._internal_api_client.put_workspace_proxied_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_proxied_dataplane_artifacts_put(
+            workspace_id=workspace_id,
+            workspace_dataplane_proxied_artifacts={
+                "requirements": "\n".join(requirements),
+            },
+        )
+
+    @handle_api_exceptions
+    def update_workspace_env_vars_offline_only(
+        self, workspace_id: str, env_vars: Dict[str, str]
+    ):
+        return self._internal_api_client.put_workspace_proxied_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_proxied_dataplane_artifacts_put(
+            workspace_id=workspace_id,
+            workspace_dataplane_proxied_artifacts={
+                "environment_variables": [
+                    f"{key}={value}" for key, value in env_vars.items()
+                ],
+            },
+        )
+
+    @handle_api_exceptions
+    def start_workspace(self, workspace_id: str):
+        """Start a workspace."""
+        workspace_model = self.get_workspace(id=workspace_id)
+        if workspace_model is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+
+        return self._internal_api_client.start_session_api_v2_sessions_session_id_start_post(
+            session_id=workspace_model.cluster_id,
+            start_session_options=StartSessionOptions(),
+        )
+
+    @handle_api_exceptions
+    def terminate_workspace(self, workspace_id: str):
+        """Terminate a workspace."""
+        workspace_model = self.get_workspace(id=workspace_id)
+        if workspace_model is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+
+        options = StopSessionOptions(
+            terminate=True,
+            workers_only=False,
+            keep_min_workers=False,
+            take_snapshot=False,
+        )
+        return self._internal_api_client.stop_session_api_v2_sessions_session_id_stop_post(
+            session_id=workspace_model.cluster_id, stop_session_options=options,
+        )
+
+    @handle_api_exceptions
+    def get_workspace_status(self, workspace_id: str) -> SessionState:
+        workspace_model = self.get_workspace(id=workspace_id)
+        if workspace_model is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+
+        result = self._internal_api_client.get_session_api_v2_sessions_session_id_get(
+            session_id=workspace_model.cluster_id
+        )
+        return result.result.state
+
+    @handle_api_exceptions
+    def get_cluster_head_node_ip(self, cluster_id: str) -> str:
+        head_ip = self._internal_api_client.get_session_head_ip_api_v2_sessions_session_id_head_ip_get(
+            cluster_id
+        ).result.head_ip
+        return head_ip
+
+    @handle_api_exceptions
+    def get_cluster_ssh_key(self, cluster_id: str) -> SessionSshKey:
+        return self._internal_api_client.get_session_ssh_key_api_v2_sessions_session_id_ssh_key_get(
+            cluster_id
+        ).result
+
+    @handle_api_exceptions
+    def get_workspace_default_dir_name(self, workspace_id) -> str:
+        workspace = self.get_workspace(id=workspace_id)
+        assert workspace, f"Workspace '{workspace_id}' not found."
+        project = self._internal_api_client.get_project_api_v2_projects_project_id_get(
+            workspace.project_id
+        ).result
+        if self._internal_api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
+            FLAG_DEFAULT_WORKING_DIR_FOR_PROJ
+        ).result.is_on:
+            return project.directory_name
+        else:
+            return project.name

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/_private/anyscale_client/common.py

@@ -5,6 +5,7 @@ from anyscale._private.models.image_uri import ImageURI
 from anyscale.client.openapi_client.models import (
     Cloud,
     ComputeTemplateConfig,
+    CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
     InternalProductionJob,
@@ -13,6 +14,7 @@ from anyscale.client.openapi_client.models import (
 from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
 from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
 from anyscale.client.openapi_client.models.production_job import ProductionJob
+from anyscale.client.openapi_client.models.session_ssh_key import SessionSshKey
 from anyscale.sdk.anyscale_client.models import (
     ApplyServiceModel,
     Cluster,
@@ -21,6 +23,7 @@ from anyscale.sdk.anyscale_client.models import (
     FineTunedModel,
     Job as APIJobRun,
     ServiceModel,
+    SessionState,
 )
 from anyscale.sdk.anyscale_client.models.cluster_environment_build import (
     ClusterEnvironmentBuild,
@@ -420,9 +423,75 @@ class AnyscaleClientInterface(ABC):
         self, dataset_id_or_name: str, version: Optional[int]
     ) -> bytes:
         """See docstring for `anyscale.llm.dataset.download`."""
+
+    @abstractmethod
+    def create_workspace(self, model: CreateExperimentalWorkspace) -> str:
+        """Creates a workspace
+
+        Returns the id of the workspace created.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_workspace(
+        self,
+        *,
+        name: Optional[str],
+        id: Optional[str],  # noqa: A002
+        cloud: Optional[str],
+        project: Optional[str],
+    ) -> Optional[DecoratedSchedule]:
+        """Get a workspace by either name or id. Filter by cloud and project.
+
+        Returns None if not found.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def start_workspace(self, workspace_id: str):
+        """Start a workspace."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def terminate_workspace(self, workspace_id: str):
+        """Terminate a workspace."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def update_workspace_dependencies_offline_only(
+        self, workspace_id: str, requirements: List[str]
+    ):
+        """Updates the dynamic dependencies of a workspace while the workspace is offline"""
+        raise NotImplementedError
+
+    @abstractmethod
+    def update_workspace_env_vars_offline_only(
+        self, workspace_id: str, env_vars: Dict[str, str]
+    ):
+        """Updates the dynamic dependencies of a workspace"""
         raise NotImplementedError
 
     @abstractmethod
     def retrieve_finetuned_model(self, id: str) -> FineTunedModel:  # noqa: A002
         """Returns LLM model information for the given model ID"""
         raise NotImplementedError
+
+    @abstractmethod
+    def get_workspace_status(self, workspace_id: str) -> SessionState:
+        """Get the status of a workspace."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_cluster_head_node_ip(self, cluster_id: str) -> Optional[str]:
+        """Get the head node IP of the cluster."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_cluster_ssh_key(self, cluster_id: str) -> SessionSshKey:
+        """Get the SSH key for the cluster."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_workspace_default_dir_name(self, workspace_id) -> str:
+        """Get the default directory name for a workspace."""
+        raise NotImplementedError

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -13,8 +13,10 @@ from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.models import (
     Cloud,
     ComputeTemplateConfig,
+    CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
+    ExperimentalWorkspace,
     HaJobGoalStates,
     HaJobStates,
     InternalProductionJob,
@@ -24,6 +26,7 @@ from anyscale.client.openapi_client.models import (
 )
 from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
 from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSchedule
+from anyscale.client.openapi_client.models.session_ssh_key import SessionSshKey
 from anyscale.cluster_compute import parse_cluster_compute_name_version
 from anyscale.sdk.anyscale_client.configuration import Configuration
 from anyscale.sdk.anyscale_client.models import (
@@ -42,6 +45,7 @@ from anyscale.sdk.anyscale_client.models import (
     ServiceEventCurrentState,
     ServiceModel,
     ServiceVersionState,
+    SessionState,
 )
 from anyscale.sdk.anyscale_client.models.cluster_environment import ClusterEnvironment
 from anyscale.shared_anyscale_utils.latest_ray_version import LATEST_RAY_VERSION
@@ -65,6 +69,9 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     DEFAULT_CLUSTER_COMPUTE_NAME = "fake-default-cluster-compute"
     DEFAULT_CLUSTER_COMPUTE_ID = "fake-default-cluster-compute-id"
     DEFAULT_CLUSTER_ENV_BUILD_ID = "fake-default-cluster-env-build-id"
+    DEFAULT_USER_ID = "fake-user-id"
+    DEFAULT_USER_EMAIL = "user@email.com"
+    DEFAULT_ORGANIZATION_ID = "fake-org-id"
 
     WORKSPACE_ID = "fake-workspace-id"
     WORKSPACE_CLOUD_ID = "fake-workspace-cloud-id"
@@ -132,6 +139,11 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._job_run_logs: Dict[str, str] = {}
         self._schedules: Dict[str, DecoratedSchedule] = {}
         self._schedule_trigger_counts: Dict[str, int] = defaultdict(int)
+        self._workspaces: Dict[str, ExperimentalWorkspace] = {}
+        self._workspaces_dependencies: Dict[str, List[str]] = {}
+        self._workspaces_env_vars: Dict[str, Dict[str, str]] = {}
+        self._clusters_headnode_ip: Dict[str, str] = {}
+        self._clusters_ssh_key: Dict[str, SessionSshKey] = {}
 
         # Cloud ID -> Cloud.
         self._clouds: Dict[str, Cloud] = {
@@ -833,3 +845,102 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             created_at=datetime.utcnow(),
             storage_uri="s3://fake_bucket/fake_folder/",
         )
+
+    def create_workspace(self, model: CreateExperimentalWorkspace) -> str:
+        workspace_id = uuid.uuid4()
+
+        # this usually happens on the backend
+        compute_config = self.get_compute_config(model.compute_config_id)
+        assert compute_config is not None
+        compute_config.idle_timeout_minutes = model.idle_timeout_minutes
+
+        workspace = ExperimentalWorkspace(
+            id=f"workspace-id-{workspace_id!s}",
+            name=model.name,
+            project_id=model.project_id or self.get_project_id(),
+            compute_config_id=model.compute_config_id,
+            environment_id=model.cluster_environment_build_id,
+            cloud_id=model.cloud_id or self.get_cloud_id(),
+            created_at=datetime.now(),
+            creator_id=self.DEFAULT_USER_ID,
+            creator_email=self.DEFAULT_USER_EMAIL,
+            organization_id=self.DEFAULT_ORGANIZATION_ID,
+            cluster_id=self.DEFAULT_CLOUD_ID,
+            state=SessionState.RUNNING
+            if not model.skip_start
+            else SessionState.TERMINATED,
+        )
+        self._workspaces[workspace.id] = workspace
+        return workspace.id
+
+    def get_workspace(
+        self,
+        *,
+        id: Optional[str] = None,  # noqa: A002
+        name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        project: Optional[str] = None,
+    ) -> Optional[ExperimentalWorkspace]:
+        if id is not None:
+            return self._workspaces.get(id, None)
+        else:
+            cloud_id = self.get_cloud_id(cloud_name=cloud)
+            cloud_project_dict = self._project_to_id.get(cloud_id, None)
+            project_id = (
+                cloud_project_dict.get(project, None) if cloud_project_dict else None
+            )
+            result: ExperimentalWorkspace = None
+            for workspace in self._workspaces.values():
+                if (
+                    workspace is not None
+                    and workspace.name == name
+                    and (project_id is None or workspace.project_id == project_id)
+                ):
+                    result = workspace
+                    break
+
+        return result
+
+    @property
+    def workspaces(self) -> Dict[str, ExperimentalWorkspace]:
+        return self._workspaces
+
+    def start_workspace(self, workspace_id: str):
+        workspace = self._workspaces.get(workspace_id, None)
+        if workspace is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+        workspace.state = SessionState.RUNNING
+
+    def terminate_workspace(self, workspace_id: str):
+        workspace = self._workspaces.get(workspace_id, None)
+        if workspace is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+        workspace.state = SessionState.TERMINATED
+
+    def update_workspace_dependencies_offline_only(
+        self, workspace_id: str, requirements: List[str]
+    ):
+        self._workspaces_dependencies[workspace_id] = requirements
+
+    def update_workspace_env_vars_offline_only(
+        self, workspace_id: str, env_vars: Dict[str, str]
+    ):
+        self._workspaces_env_vars[workspace_id] = env_vars
+
+    def get_workspace_status(self, workspace_id: str) -> SessionState:
+        workspace = self._workspaces.get(workspace_id, None)
+        if workspace is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+        return workspace.state
+
+    def get_cluster_head_node_ip(self, cluster_id: str) -> str:
+        return self._clusters_headnode_ip.get(cluster_id, "")
+
+    def get_cluster_ssh_key(self, cluster_id: str) -> SessionSshKey:
+        return self._clusters_ssh_key.get(cluster_id, None)
+
+    def get_workspace_default_dir_name(self, workspace_id) -> str:
+        workspace = self._workspaces.get(workspace_id, None)
+        if workspace is None:
+            raise ValueError(f"Workspace '{workspace_id}' not found.")
+        return "/home/ray/default"

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/client/README.md

@@ -115,6 +115,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_connect_session_api_v2_sessions_create_connect_session_post**](docs/DefaultApi.md#create_connect_session_api_v2_sessions_create_connect_session_post) | **POST** /api/v2/sessions/create_connect_session | Create Connect Session
 *DefaultApi* | [**create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post**](docs/DefaultApi.md#create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post) | **POST** /api/v2/organizations/{organization_id}/create_endpoint_user | Create Endpoint User For Org
 *DefaultApi* | [**create_fine_tuning_job_api_v2_fine_tuning_jobs_create_post**](docs/DefaultApi.md#create_fine_tuning_job_api_v2_fine_tuning_jobs_create_post) | **POST** /api/v2/fine_tuning/jobs/create | Create Fine Tuning Job
+*DefaultApi* | [**create_instance_usage_budget_api_v2_instance_usage_budgets_post**](docs/DefaultApi.md#create_instance_usage_budget_api_v2_instance_usage_budgets_post) | **POST** /api/v2/instance_usage_budgets/ | Create Instance Usage Budget
 *DefaultApi* | [**create_invitation_api_v2_organization_invitations_post**](docs/DefaultApi.md#create_invitation_api_v2_organization_invitations_post) | **POST** /api/v2/organization_invitations/ | Create Invitation
 *DefaultApi* | [**create_job_api_v2_decorated_ha_jobs_create_post**](docs/DefaultApi.md#create_job_api_v2_decorated_ha_jobs_create_post) | **POST** /api/v2/decorated_ha_jobs/create | Create Job
 *DefaultApi* | [**create_machine_api_v2_machines_create_post**](docs/DefaultApi.md#create_machine_api_v2_machines_create_post) | **POST** /api/v2/machines/create | Create Machine
@@ -136,6 +137,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete**](docs/DefaultApi.md#delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete) | **DELETE** /api/v2/clouds/{cloud_id}/collaborators/{identity_id} | Delete Cloud Collaborator
 *DefaultApi* | [**delete_compute_template_api_v2_compute_templates_template_id_delete**](docs/DefaultApi.md#delete_compute_template_api_v2_compute_templates_template_id_delete) | **DELETE** /api/v2/compute_templates/{template_id} | Delete Compute Template
 *DefaultApi* | [**delete_hosted_cloud_api_v2_clouds_hosted_cloud_id_delete**](docs/DefaultApi.md#delete_hosted_cloud_api_v2_clouds_hosted_cloud_id_delete) | **DELETE** /api/v2/clouds/hosted/{cloud_id} | Delete Hosted Cloud
+*DefaultApi* | [**delete_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_delete**](docs/DefaultApi.md#delete_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_delete) | **DELETE** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Delete Instance Usage Budget
 *DefaultApi* | [**delete_machine_api_v2_machines_delete_post**](docs/DefaultApi.md#delete_machine_api_v2_machines_delete_post) | **POST** /api/v2/machines/delete | Delete Machine
 *DefaultApi* | [**delete_machine_pool_api_v2_machine_pools_delete_post**](docs/DefaultApi.md#delete_machine_pool_api_v2_machine_pools_delete_post) | **POST** /api/v2/machine_pools/delete | Delete Machine Pool
 *DefaultApi* | [**delete_project_api_v2_projects_project_id_delete**](docs/DefaultApi.md#delete_project_api_v2_projects_project_id_delete) | **DELETE** /api/v2/projects/{project_id} | Delete Project
@@ -175,6 +177,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get**](docs/DefaultApi.md#get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get) | **GET** /api/v2/logs/cluster_product_autoscaler_flag/{session_id} | Get Cluster Product Autoscaler Flag
 *DefaultApi* | [**get_compute_template_api_v2_compute_templates_template_id_get**](docs/DefaultApi.md#get_compute_template_api_v2_compute_templates_template_id_get) | **GET** /api/v2/compute_templates/{template_id} | Get Compute Template
 *DefaultApi* | [**get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get**](docs/DefaultApi.md#get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get) | **GET** /api/v2/experimental_cron_jobs/{cron_job_id} | Get Cron Job
+*DefaultApi* | [**get_dataset_download_url_api_v2_datasets_download_get**](docs/DefaultApi.md#get_dataset_download_url_api_v2_datasets_download_get) | **GET** /api/v2/datasets/download | Get Dataset Download Url
 *DefaultApi* | [**get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get**](docs/DefaultApi.md#get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get) | **GET** /api/v2/decorated_sessions/{cluster_id} | Get Decorated Cluster
 *DefaultApi* | [**get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get**](docs/DefaultApi.md#get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get) | **GET** /api/v2/decorated_interactive_sessions/{interactive_session_id} | Get Decorated Interactive Session
 *DefaultApi* | [**get_decorated_interactive_session_logs_api_v2_decorated_interactive_sessions_interactive_session_id_logs_get**](docs/DefaultApi.md#get_decorated_interactive_session_logs_api_v2_decorated_interactive_sessions_interactive_session_id_logs_get) | **GET** /api/v2/decorated_interactive_sessions/{interactive_session_id}/logs | Get Decorated Interactive Session Logs
@@ -198,6 +201,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_file_api_v2_files_file_id_get**](docs/DefaultApi.md#get_file_api_v2_files_file_id_get) | **GET** /api/v2/files/{file_id} | Get File
 *DefaultApi* | [**get_file_upload_url_api_v2_files_upload_url_post**](docs/DefaultApi.md#get_file_upload_url_api_v2_files_upload_url_post) | **POST** /api/v2/files/upload_url | Get File Upload Url
 *DefaultApi* | [**get_fine_tuning_job_api_v2_fine_tuning_jobs_fine_tuning_job_id_get**](docs/DefaultApi.md#get_fine_tuning_job_api_v2_fine_tuning_jobs_fine_tuning_job_id_get) | **GET** /api/v2/fine_tuning/jobs/{fine_tuning_job_id} | Get Fine Tuning Job
+*DefaultApi* | [**get_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_get**](docs/DefaultApi.md#get_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_get) | **GET** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Get Instance Usage Budget
 *DefaultApi* | [**get_invitation_api_v2_organization_invitations_invitation_id_get**](docs/DefaultApi.md#get_invitation_api_v2_organization_invitations_invitation_id_get) | **GET** /api/v2/organization_invitations/{invitation_id} | Get Invitation
 *DefaultApi* | [**get_job_api_v2_decorated_ha_jobs_production_job_id_get**](docs/DefaultApi.md#get_job_api_v2_decorated_ha_jobs_production_job_id_get) | **GET** /api/v2/decorated_ha_jobs/{production_job_id} | Get Job
 *DefaultApi* | [**get_job_api_v2_decorated_unified_jobs_job_id_get**](docs/DefaultApi.md#get_job_api_v2_decorated_unified_jobs_job_id_get) | **GET** /api/v2/decorated_unified_jobs/{job_id} | Get Job
@@ -289,6 +293,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_decorated_runtime_envs_api_v2_decorated_runtime_envs_get**](docs/DefaultApi.md#list_decorated_runtime_envs_api_v2_decorated_runtime_envs_get) | **GET** /api/v2/decorated_runtime_envs/ | List Decorated Runtime Envs
 *DefaultApi* | [**list_decorated_serve_deployments_api_v2_decorated_serve_deployments_get**](docs/DefaultApi.md#list_decorated_serve_deployments_api_v2_decorated_serve_deployments_get) | **GET** /api/v2/decorated_serve_deployments/ | List Decorated Serve Deployments
 *DefaultApi* | [**list_fine_tuning_jobs_api_v2_fine_tuning_jobs_get**](docs/DefaultApi.md#list_fine_tuning_jobs_api_v2_fine_tuning_jobs_get) | **GET** /api/v2/fine_tuning/jobs | List Fine Tuning Jobs
+*DefaultApi* | [**list_instance_usage_budgets_api_v2_instance_usage_budgets_get**](docs/DefaultApi.md#list_instance_usage_budgets_api_v2_instance_usage_budgets_get) | **GET** /api/v2/instance_usage_budgets/ | List Instance Usage Budgets
 *DefaultApi* | [**list_invitations_api_v2_organization_invitations_get**](docs/DefaultApi.md#list_invitations_api_v2_organization_invitations_get) | **GET** /api/v2/organization_invitations/ | List Invitations
 *DefaultApi* | [**list_job_queues_api_v2_job_queues_post**](docs/DefaultApi.md#list_job_queues_api_v2_job_queues_post) | **POST** /api/v2/job_queues/ | List Job Queues
 *DefaultApi* | [**list_job_states_api_v2_decorated_ha_job_states_get**](docs/DefaultApi.md#list_job_states_api_v2_decorated_ha_job_states_get) | **GET** /api/v2/decorated_ha_job_states/ | List Job States
@@ -359,6 +364,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post**](docs/DefaultApi.md#terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/terminate | Terminate Job
 *DefaultApi* | [**terminate_service_api_v2_services_v2_service_id_terminate_post**](docs/DefaultApi.md#terminate_service_api_v2_services_v2_service_id_terminate_post) | **POST** /api/v2/services-v2/{service_id}/terminate | Terminate Service
 *DefaultApi* | [**test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post**](docs/DefaultApi.md#test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post) | **POST** /api/v2/organizations/{organization_id}/test_saml_acs | Test Saml Acs
+*DefaultApi* | [**toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post**](docs/DefaultApi.md#toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post) | **POST** /api/v2/instance_usage_budgets/{instance_usage_budget_id}/toggle_is_enabled | Toggle Instance Usage Budget Is Enabled
 *DefaultApi* | [**trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post**](docs/DefaultApi.md#trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post) | **POST** /api/v2/experimental_cron_jobs/{cron_job_id}/trigger | Trigger Cron Job
 *DefaultApi* | [**try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post**](docs/DefaultApi.md#try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post) | **POST** /api/v2/aioa_cloud_waitlist/claim_cloud | Try Claim Cloud
 *DefaultApi* | [**try_login_api_v2_users_try_login_post**](docs/DefaultApi.md#try_login_api_v2_users_try_login_post) | **POST** /api/v2/users/try-login | Try Login
@@ -372,6 +378,8 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**update_customer_aggregated_logs_config_api_v2_clouds_cloud_id_update_customer_aggregated_logs_config_put**](docs/DefaultApi.md#update_customer_aggregated_logs_config_api_v2_clouds_cloud_id_update_customer_aggregated_logs_config_put) | **PUT** /api/v2/clouds/{cloud_id}/update_customer_aggregated_logs_config | Update Customer Aggregated Logs Config
 *DefaultApi* | [**update_default_cloud_api_v2_organizations_update_default_cloud_post**](docs/DefaultApi.md#update_default_cloud_api_v2_organizations_update_default_cloud_post) | **POST** /api/v2/organizations/update_default_cloud | Update Default Cloud
 *DefaultApi* | [**update_dns_api_v2_dns_post**](docs/DefaultApi.md#update_dns_api_v2_dns_post) | **POST** /api/v2/dns/ | Update Dns
+*DefaultApi* | [**update_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_put**](docs/DefaultApi.md#update_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_put) | **PUT** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Update Instance Usage Budget
+*DefaultApi* | [**update_job_queue_config_api_v2_decorated_ha_jobs_production_job_id_update_job_queue_config_put**](docs/DefaultApi.md#update_job_queue_config_api_v2_decorated_ha_jobs_production_job_id_update_job_queue_config_put) | **PUT** /api/v2/decorated_ha_jobs/{production_job_id}/update_job_queue_config | Update Job Queue Config
 *DefaultApi* | [**upload_file_api_v2_files_post**](docs/DefaultApi.md#upload_file_api_v2_files_post) | **POST** /api/v2/files | Upload File
 *DefaultApi* | [**upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post**](docs/DefaultApi.md#upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post) | **POST** /api/v2/session_commands/{session_command_id}/upload_logs | Upload Session Command Logs
 *DefaultApi* | [**upsert_billing_version_api_v2_organization_billing_billing_versions_put**](docs/DefaultApi.md#upsert_billing_version_api_v2_organization_billing_billing_versions_put) | **PUT** /api/v2/organization_billing/billing_versions | Upsert Billing Version
@@ -548,6 +556,7 @@ Class | Method | HTTP request | Description
  - [CreateExperimentalWorkspaceFromJob](docs/CreateExperimentalWorkspaceFromJob.md)
  - [CreateFineTuningHyperparameters](docs/CreateFineTuningHyperparameters.md)
  - [CreateFineTuningJobProductRequest](docs/CreateFineTuningJobProductRequest.md)
+ - [CreateInstanceUsageBudget](docs/CreateInstanceUsageBudget.md)
  - [CreateInternalProductionJob](docs/CreateInternalProductionJob.md)
  - [CreateJobQueueConfig](docs/CreateJobQueueConfig.md)
  - [CreateMachinePoolRequest](docs/CreateMachinePoolRequest.md)
@@ -680,6 +689,10 @@ Class | Method | HTTP request | Description
  - [IamInstanceProfileSpecification](docs/IamInstanceProfileSpecification.md)
  - [IdleTerminationStatus](docs/IdleTerminationStatus.md)
  - [ImportAicaModel](docs/ImportAicaModel.md)
+ - [InstanceUsageBudget](docs/InstanceUsageBudget.md)
+ - [InstanceUsageBudgetEvaluationPeriod](docs/InstanceUsageBudgetEvaluationPeriod.md)
+ - [InstanceusagebudgetListResponse](docs/InstanceusagebudgetListResponse.md)
+ - [InstanceusagebudgetResponse](docs/InstanceusagebudgetResponse.md)
  - [IntegrationDetails](docs/IntegrationDetails.md)
  - [InteractiveSessionLogs](docs/InteractiveSessionLogs.md)
  - [InteractivesessionlogsResponse](docs/InteractivesessionlogsResponse.md)

{anyscale-0.24.53 → anyscale-0.24.55}/anyscale/client/openapi_client/__init__.py

@@ -188,6 +188,7 @@ from openapi_client.models.create_experimental_workspace import CreateExperiment
 from openapi_client.models.create_experimental_workspace_from_job import CreateExperimentalWorkspaceFromJob
 from openapi_client.models.create_fine_tuning_hyperparameters import CreateFineTuningHyperparameters
 from openapi_client.models.create_fine_tuning_job_product_request import CreateFineTuningJobProductRequest
+from openapi_client.models.create_instance_usage_budget import CreateInstanceUsageBudget
 from openapi_client.models.create_internal_production_job import CreateInternalProductionJob
 from openapi_client.models.create_job_queue_config import CreateJobQueueConfig
 from openapi_client.models.create_machine_pool_request import CreateMachinePoolRequest
@@ -320,6 +321,10 @@ from openapi_client.models.http_protocol_config import HttpProtocolConfig
 from openapi_client.models.iam_instance_profile_specification import IamInstanceProfileSpecification
 from openapi_client.models.idle_termination_status import IdleTerminationStatus
 from openapi_client.models.import_aica_model import ImportAicaModel
+from openapi_client.models.instance_usage_budget import InstanceUsageBudget
+from openapi_client.models.instance_usage_budget_evaluation_period import InstanceUsageBudgetEvaluationPeriod
+from openapi_client.models.instanceusagebudget_list_response import InstanceusagebudgetListResponse
+from openapi_client.models.instanceusagebudget_response import InstanceusagebudgetResponse
 from openapi_client.models.integration_details import IntegrationDetails
 from openapi_client.models.interactive_session_logs import InteractiveSessionLogs
 from openapi_client.models.interactivesessionlogs_response import InteractivesessionlogsResponse