anyscale 0.24.54__tar.gz → 0.24.56__tar.gz

{anyscale-0.24.54 → anyscale-0.24.56}/PKG-INFO

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

{anyscale-0.24.54 → anyscale-0.24.56}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -38,7 +38,10 @@ from anyscale.client.openapi_client.models import (
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
     ExperimentalWorkspace,
+    FineTunedModel,
     InternalProductionJob,
+    SessionSshKey,
+    SessionState,
     StartSessionOptions,
     StopSessionOptions,
 )
@@ -47,6 +50,7 @@ from anyscale.client.openapi_client.models.decorated_schedule import DecoratedSc
 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,
@@ -60,7 +64,6 @@ from anyscale.sdk.anyscale_client.models import (
     ClusterEnvironmentsQuery,
     CreateClusterEnvironment,
     CreateClusterEnvironmentBuild,
-    FineTunedModel,
     Job as APIJobRun,
     Project,
     RollbackServiceModel,
@@ -1189,7 +1192,9 @@ class AnyscaleClient(AnyscaleClientInterface):
         return (cloud_id, project_id)  # type: ignore
 
     def retrieve_finetuned_model(self, model_id: str) -> FineTunedModel:
-        return self._external_api_client.get_model(model_id)
+        return self._internal_api_client.get_model_api_v2_llm_models_model_id_get(
+            model_id
+        )
 
     @handle_api_exceptions
     def create_workspace(self, model: CreateExperimentalWorkspace) -> str:
@@ -1287,3 +1292,41 @@ class AnyscaleClient(AnyscaleClientInterface):
         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.54 → anyscale-0.24.56}/anyscale/_private/anyscale_client/common.py

@@ -8,20 +8,22 @@ from anyscale.client.openapi_client.models import (
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
+    FineTunedModel,
     InternalProductionJob,
     Project,
 )
 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,
     ClusterCompute,
     ClusterEnvironment,
-    FineTunedModel,
     Job as APIJobRun,
     ServiceModel,
+    SessionState,
 )
 from anyscale.sdk.anyscale_client.models.cluster_environment_build import (
     ClusterEnvironmentBuild,
@@ -473,3 +475,23 @@ class AnyscaleClientInterface(ABC):
     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.54 → anyscale-0.24.56}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -16,7 +16,10 @@ from anyscale.client.openapi_client.models import (
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
+    EndpointsFineTunedModelStatus,
     ExperimentalWorkspace,
+    FineTunedModel,
+    FineTuneType,
     HaJobGoalStates,
     HaJobStates,
     InternalProductionJob,
@@ -26,6 +29,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 (
@@ -36,14 +40,12 @@ from anyscale.sdk.anyscale_client.models import (
     ClusterEnvironmentBuild,
     ClusterEnvironmentBuildStatus,
     ComputeNodeType,
-    EndpointsFineTunedModelStatus,
-    FineTunedModel,
-    FineTuneType,
     Job as APIJobRun,
     ProductionServiceV2VersionModel,
     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
@@ -140,6 +142,8 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         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] = {
@@ -862,7 +866,9 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             creator_email=self.DEFAULT_USER_EMAIL,
             organization_id=self.DEFAULT_ORGANIZATION_ID,
             cluster_id=self.DEFAULT_CLOUD_ID,
-            state="Running" if not model.skip_start else "Terminated",
+            state=SessionState.RUNNING
+            if not model.skip_start
+            else SessionState.TERMINATED,
         )
         self._workspaces[workspace.id] = workspace
         return workspace.id
@@ -903,13 +909,13 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         workspace = self._workspaces.get(workspace_id, None)
         if workspace is None:
             raise ValueError(f"Workspace '{workspace_id}' not found.")
-        workspace.state = "Running"
+        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 = "Terminated"
+        workspace.state = SessionState.TERMINATED
 
     def update_workspace_dependencies_offline_only(
         self, workspace_id: str, requirements: List[str]
@@ -920,3 +926,21 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         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.54 → anyscale-0.24.56}/anyscale/_private/docgen/__main__.py

@@ -17,6 +17,7 @@ from anyscale.commands import (
     schedule_commands,
     service_account_commands,
     service_commands,
+    workspace_commands_v2,
 )
 from anyscale.compute_config.models import (
     ComputeConfig,
@@ -47,6 +48,7 @@ from anyscale.service.models import (
     ServiceVersionStatus,
     TracingConfig,
 )
+from anyscale.workspace.models import WorkspaceConfig
 
 
 # Defines all modules to be documented.
@@ -213,6 +215,31 @@ ALL_MODULES = [
         sdk_commands=[],
         models=[],
     ),
+    Module(
+        title="Workspaces",
+        filename="workspaces.md",
+        cli_prefix="anyscale workspace_v2",
+        cli_commands=[
+            workspace_commands_v2.create,
+            workspace_commands_v2.start,
+            workspace_commands_v2.terminate,
+            workspace_commands_v2.status,
+            workspace_commands_v2.wait,
+            workspace_commands_v2.ssh,
+            workspace_commands_v2.run_command,
+        ],
+        sdk_prefix="anyscale.workspace",
+        sdk_commands=[
+            anyscale.workspace.create,
+            anyscale.workspace.start,
+            anyscale.workspace.terminate,
+            anyscale.workspace.status,
+            anyscale.workspace.wait,
+            anyscale.workspace.generate_ssh_config_file,
+            anyscale.workspace.run_command,
+        ],
+        models=[WorkspaceConfig],
+    ),
 ]
 
 

{anyscale-0.24.54 → anyscale-0.24.56}/anyscale/_private/docgen/generator.py

@@ -19,7 +19,7 @@ CUSTOMER_HOSTED_ANCHOR = "#customer-hosted-only"
 CUSTOMER_HOSTED_HEADER = """\
 #### Customer-hosted cloud features {#customer-hosted-only}
 :::note
-Some features are only available on customer-hosted clouds. Reach out to preview-help@anyscale.com for info.
+Some features are only available on customer-hosted clouds. Reach out to support@anyscale.com for info.
 :::"""
 
 CUSTOMER_HOSTED_QUALIFIER = (
@@ -120,7 +120,7 @@ class MarkdownGenerator:
         filename = self._model_type_to_filename[t]
         return f"{filename}#{t.__name__.lower()}"
 
-    def _type_container_to_string(self, t: typing.Type) -> str:
+    def _type_container_to_string(self, t: typing.Type) -> str:  # noqa: PLR0911
         """Return a str representation of a type hint."""
         origin, args = typing.get_origin(t), typing.get_args(t)
         assert origin is not None and args is not None
@@ -142,6 +142,13 @@ class MarkdownGenerator:
             else:
                 return "List"
 
+        if origin is tuple:
+            arg_str = ", ".join([self._model_type_to_string(arg) for arg in args])
+            if arg_str:
+                return f"Tuple[{arg_str}]"
+            else:
+                return "Tuple"
+
         raise NotImplementedError(f"Unhandled type: {t}")
 
     def _model_type_to_string(self, t: Type):  # noqa: PLR0911
@@ -361,7 +368,9 @@ class MarkdownGenerator:
                 if name.startswith("_"):
                     continue
 
-                assert param.annotation is not inspect.Parameter.empty
+                assert (
+                    param.annotation is not inspect.Parameter.empty
+                ), f"SDK command '{sdk_prefix}.{c.__name__}' is missing a type hint for argument '{name}'"
                 type_str = "(" + self._model_type_to_string(param.annotation) + ")"
                 if param.default != inspect.Parameter.empty:
                     type_str += f" = {param.default!s}"

{anyscale-0.24.54 → anyscale-0.24.56}/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
@@ -125,6 +126,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_organization_collaborator_api_v2_projects_project_id_collaborators_organization_post**](docs/DefaultApi.md#create_organization_collaborator_api_v2_projects_project_id_collaborators_organization_post) | **POST** /api/v2/projects/{project_id}/collaborators/organization | Create Organization Collaborator
 *DefaultApi* | [**create_project_api_v2_projects_post**](docs/DefaultApi.md#create_project_api_v2_projects_post) | **POST** /api/v2/projects/ | Create Project
 *DefaultApi* | [**create_project_collaborator_api_v2_projects_project_id_collaborators_users_post**](docs/DefaultApi.md#create_project_collaborator_api_v2_projects_project_id_collaborators_users_post) | **POST** /api/v2/projects/{project_id}/collaborators/users | Create Project Collaborator
+*DefaultApi* | [**create_resource_quota_api_v2_resource_quotas_post**](docs/DefaultApi.md#create_resource_quota_api_v2_resource_quotas_post) | **POST** /api/v2/resource_quotas/ | Create Resource Quota
 *DefaultApi* | [**create_service_account_api_v2_users_service_accounts_post**](docs/DefaultApi.md#create_service_account_api_v2_users_service_accounts_post) | **POST** /api/v2/users/service_accounts | Create Service Account
 *DefaultApi* | [**create_session_command_api_v2_sessions_session_id_create_session_command_post**](docs/DefaultApi.md#create_session_command_api_v2_sessions_session_id_create_session_command_post) | **POST** /api/v2/sessions/{session_id}/create_session_command | Create Session Command
 *DefaultApi* | [**create_web_terminal_api_v2_sessions_session_id_web_terminals_post**](docs/DefaultApi.md#create_web_terminal_api_v2_sessions_session_id_web_terminals_post) | **POST** /api/v2/sessions/{session_id}/web_terminals | Create Web Terminal
@@ -136,8 +138,10 @@ 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_model_api_v2_llm_models_model_id_delete**](docs/DefaultApi.md#delete_model_api_v2_llm_models_model_id_delete) | **DELETE** /api/v2/llm/models/{model_id} | Delete Model
 *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
 *DefaultApi* | [**delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete**](docs/DefaultApi.md#delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete) | **DELETE** /api/v2/projects/{project_id}/collaborators/{role_or_identity_id} | Delete Project Collaborator
 *DefaultApi* | [**delete_session_api_v2_sessions_session_id_delete**](docs/DefaultApi.md#delete_session_api_v2_sessions_session_id_delete) | **DELETE** /api/v2/sessions/{session_id} | Delete Session
@@ -199,6 +203,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
@@ -214,6 +219,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_manage_billing_url_api_v2_organization_billing_manage_billing_url_get**](docs/DefaultApi.md#get_manage_billing_url_api_v2_organization_billing_manage_billing_url_get) | **GET** /api/v2/organization_billing/manage_billing_url | Get Manage Billing Url
 *DefaultApi* | [**get_metronome_embedded_usage_dashboard_api_v2_organization_billing_metronome_embedded_dashboard_url_dashboard_type_get**](docs/DefaultApi.md#get_metronome_embedded_usage_dashboard_api_v2_organization_billing_metronome_embedded_dashboard_url_dashboard_type_get) | **GET** /api/v2/organization_billing/metronome_embedded_dashboard_url/{dashboard_type} | Get Metronome Embedded Usage Dashboard
 *DefaultApi* | [**get_metronome_embedded_usage_dashboard_by_organization_api_v2_organization_billing_organization_id_metronome_embedded_dashboard_url_dashboard_type_get**](docs/DefaultApi.md#get_metronome_embedded_usage_dashboard_by_organization_api_v2_organization_billing_organization_id_metronome_embedded_dashboard_url_dashboard_type_get) | **GET** /api/v2/organization_billing/{organization_id}/metronome_embedded_dashboard_url/{dashboard_type} | Get Metronome Embedded Usage Dashboard By Organization
+*DefaultApi* | [**get_model_api_v2_llm_models_model_id_get**](docs/DefaultApi.md#get_model_api_v2_llm_models_model_id_get) | **GET** /api/v2/llm/models/{model_id} | Get Model
 *DefaultApi* | [**get_model_configuration_api_v2_aica_endpoints_model_config_get**](docs/DefaultApi.md#get_model_configuration_api_v2_aica_endpoints_model_config_get) | **GET** /api/v2/aica_endpoints/model_config | Get Model Configuration
 *DefaultApi* | [**get_model_schema_api_v2_aica_endpoints_model_schema_get**](docs/DefaultApi.md#get_model_schema_api_v2_aica_endpoints_model_schema_get) | **GET** /api/v2/aica_endpoints/model_schema | Get Model Schema
 *DefaultApi* | [**get_monitor_logs_api_v2_sessions_session_id_monitor_logs_get**](docs/DefaultApi.md#get_monitor_logs_api_v2_sessions_session_id_monitor_logs_get) | **GET** /api/v2/sessions/{session_id}/monitor_logs | Get Monitor Logs
@@ -290,6 +296,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
@@ -297,10 +304,12 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_machine_pools_api_v2_machine_pools_get**](docs/DefaultApi.md#list_machine_pools_api_v2_machine_pools_get) | **GET** /api/v2/machine_pools/ | List Machine Pools
 *DefaultApi* | [**list_machines_api_v2_machines_get**](docs/DefaultApi.md#list_machines_api_v2_machines_get) | **GET** /api/v2/machines/ | List Machines
 *DefaultApi* | [**list_models_api_v2_aica_endpoints_models_get**](docs/DefaultApi.md#list_models_api_v2_aica_endpoints_models_get) | **GET** /api/v2/aica_endpoints/models | List Models
+*DefaultApi* | [**list_models_api_v2_llm_models_get**](docs/DefaultApi.md#list_models_api_v2_llm_models_get) | **GET** /api/v2/llm/models | List Models
 *DefaultApi* | [**list_organization_collaborators_api_v2_organization_collaborators_get**](docs/DefaultApi.md#list_organization_collaborators_api_v2_organization_collaborators_get) | **GET** /api/v2/organization_collaborators/ | List Organization Collaborators
 *DefaultApi* | [**list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get**](docs/DefaultApi.md#list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get) | **GET** /api/v2/projects/{project_id}/collaborators/organizations | List Organization Project Collaborators
 *DefaultApi* | [**list_project_collaborators_api_v2_projects_project_id_collaborators_users_get**](docs/DefaultApi.md#list_project_collaborators_api_v2_projects_project_id_collaborators_users_get) | **GET** /api/v2/projects/{project_id}/collaborators/users | List Project Collaborators
 *DefaultApi* | [**list_projects_api_v2_projects_get**](docs/DefaultApi.md#list_projects_api_v2_projects_get) | **GET** /api/v2/projects/ | List Projects
+*DefaultApi* | [**list_resource_quotas_api_v2_resource_quotas_get**](docs/DefaultApi.md#list_resource_quotas_api_v2_resource_quotas_get) | **GET** /api/v2/resource_quotas/ | List Resource Quotas
 *DefaultApi* | [**list_services_api_v2_services_v2_get**](docs/DefaultApi.md#list_services_api_v2_services_v2_get) | **GET** /api/v2/services-v2/ | List Services
 *DefaultApi* | [**list_sessions_api_v2_sessions_get**](docs/DefaultApi.md#list_sessions_api_v2_sessions_get) | **GET** /api/v2/sessions/ | List Sessions
 *DefaultApi* | [**list_versions_api_v2_aica_endpoints_versions_get**](docs/DefaultApi.md#list_versions_api_v2_aica_endpoints_versions_get) | **GET** /api/v2/aica_endpoints/versions | List Versions
@@ -360,6 +369,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
@@ -373,6 +383,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
@@ -549,6 +561,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)
@@ -560,6 +573,7 @@ Class | Method | HTTP request | Description
  - [CreateOrganizationConfiguration](docs/CreateOrganizationConfiguration.md)
  - [CreateOrganizationInvitation](docs/CreateOrganizationInvitation.md)
  - [CreateProductionJobConfig](docs/CreateProductionJobConfig.md)
+ - [CreateResourceQuota](docs/CreateResourceQuota.md)
  - [CreateSchedule](docs/CreateSchedule.md)
  - [CreateSessionFromSnapshotOptions](docs/CreateSessionFromSnapshotOptions.md)
  - [CreateSessionInDb](docs/CreateSessionInDb.md)
@@ -633,6 +647,8 @@ Class | Method | HTTP request | Description
  - [DecoratedunifiedjobResponse](docs/DecoratedunifiedjobResponse.md)
  - [DeleteMachinePoolRequest](docs/DeleteMachinePoolRequest.md)
  - [DeleteMachineRequest](docs/DeleteMachineRequest.md)
+ - [DeletedPlatformFineTunedModel](docs/DeletedPlatformFineTunedModel.md)
+ - [DeletedplatformfinetunedmodelResponse](docs/DeletedplatformfinetunedmodelResponse.md)
  - [DeletemachinepoolresponseResponse](docs/DeletemachinepoolresponseResponse.md)
  - [DetachMachinePoolFromCloudRequest](docs/DetachMachinePoolFromCloudRequest.md)
  - [DetachmachinepoolfromcloudresponseResponse](docs/DetachmachinepoolfromcloudresponseResponse.md)
@@ -640,6 +656,7 @@ Class | Method | HTTP request | Description
  - [EbsBlockDevice](docs/EbsBlockDevice.md)
  - [EditableCloudResource](docs/EditableCloudResource.md)
  - [EditableCloudResourceGCP](docs/EditableCloudResourceGCP.md)
+ - [EndpointsFineTunedModelStatus](docs/EndpointsFineTunedModelStatus.md)
  - [Error](docs/Error.md)
  - [EventLevel](docs/EventLevel.md)
  - [ExecuteCommandResponse](docs/ExecuteCommandResponse.md)
@@ -658,7 +675,10 @@ Class | Method | HTTP request | Description
  - [FeatureflagresponseResponse](docs/FeatureflagresponseResponse.md)
  - [FileModel](docs/FileModel.md)
  - [FilemodelResponse](docs/FilemodelResponse.md)
+ - [FineTuneType](docs/FineTuneType.md)
+ - [FineTunedModel](docs/FineTunedModel.md)
  - [FineTuningJobStatus](docs/FineTuningJobStatus.md)
+ - [FinetunedmodelListResponse](docs/FinetunedmodelListResponse.md)
  - [FinishFTJobRequest](docs/FinishFTJobRequest.md)
  - [FinishFTJobRequestV2](docs/FinishFTJobRequestV2.md)
  - [GCPFileStoreConfig](docs/GCPFileStoreConfig.md)
@@ -681,6 +701,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)
@@ -713,6 +737,7 @@ Class | Method | HTTP request | Description
  - [LbresourceResponse](docs/LbresourceResponse.md)
  - [ListMachinePoolsResponse](docs/ListMachinePoolsResponse.md)
  - [ListMachinesResponse](docs/ListMachinesResponse.md)
+ - [ListResourceQuotasQuery](docs/ListResourceQuotasQuery.md)
  - [ListResponseMetadata](docs/ListResponseMetadata.md)
  - [ListmachinepoolsresponseResponse](docs/ListmachinepoolsresponseResponse.md)
  - [ListmachinesresponseResponse](docs/ListmachinesresponseResponse.md)
@@ -831,6 +856,7 @@ Class | Method | HTTP request | Description
  - [ProviderMetadata](docs/ProviderMetadata.md)
  - [ProvidermetadataResponse](docs/ProvidermetadataResponse.md)
  - [PythonModules](docs/PythonModules.md)
+ - [Quota](docs/Quota.md)
  - [RayGCSExternalStorageConfig](docs/RayGCSExternalStorageConfig.md)
  - [RayRuntimeEnvConfig](docs/RayRuntimeEnvConfig.md)
  - [ReadBillingVersion](docs/ReadBillingVersion.md)
@@ -843,6 +869,9 @@ Class | Method | HTTP request | Description
  - [RequestemailmagiclinkresponseResponse](docs/RequestemailmagiclinkresponseResponse.md)
  - [RequestotpreturnapimodelResponse](docs/RequestotpreturnapimodelResponse.md)
  - [ResetPasswordParams](docs/ResetPasswordParams.md)
+ - [ResourceQuota](docs/ResourceQuota.md)
+ - [ResourcequotaListResponse](docs/ResourcequotaListResponse.md)
+ - [ResourcequotaResponse](docs/ResourcequotaResponse.md)
  - [Resources](docs/Resources.md)
  - [ResubmitFTJobRequest](docs/ResubmitFTJobRequest.md)
  - [RollbackServiceModel](docs/RollbackServiceModel.md)

{anyscale-0.24.54 → anyscale-0.24.56}/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
@@ -199,6 +200,7 @@ from openapi_client.models.create_otp_return_api_model import CreateOTPReturnApi
 from openapi_client.models.create_organization_configuration import CreateOrganizationConfiguration
 from openapi_client.models.create_organization_invitation import CreateOrganizationInvitation
 from openapi_client.models.create_production_job_config import CreateProductionJobConfig
+from openapi_client.models.create_resource_quota import CreateResourceQuota
 from openapi_client.models.create_schedule import CreateSchedule
 from openapi_client.models.create_session_from_snapshot_options import CreateSessionFromSnapshotOptions
 from openapi_client.models.create_session_in_db import CreateSessionInDb
@@ -272,6 +274,8 @@ from openapi_client.models.decoratedunifiedjob_list_response import Decorateduni
 from openapi_client.models.decoratedunifiedjob_response import DecoratedunifiedjobResponse
 from openapi_client.models.delete_machine_pool_request import DeleteMachinePoolRequest
 from openapi_client.models.delete_machine_request import DeleteMachineRequest
+from openapi_client.models.deleted_platform_fine_tuned_model import DeletedPlatformFineTunedModel
+from openapi_client.models.deletedplatformfinetunedmodel_response import DeletedplatformfinetunedmodelResponse
 from openapi_client.models.deletemachinepoolresponse_response import DeletemachinepoolresponseResponse
 from openapi_client.models.detach_machine_pool_from_cloud_request import DetachMachinePoolFromCloudRequest
 from openapi_client.models.detachmachinepoolfromcloudresponse_response import DetachmachinepoolfromcloudresponseResponse
@@ -279,6 +283,7 @@ from openapi_client.models.dismissal_type import DismissalType
 from openapi_client.models.ebs_block_device import EbsBlockDevice
 from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.editable_cloud_resource_gcp import EditableCloudResourceGCP
+from openapi_client.models.endpoints_fine_tuned_model_status import EndpointsFineTunedModelStatus
 from openapi_client.models.error import Error
 from openapi_client.models.event_level import EventLevel
 from openapi_client.models.execute_command_response import ExecuteCommandResponse
@@ -297,7 +302,10 @@ from openapi_client.models.feature_flag_response import FeatureFlagResponse
 from openapi_client.models.featureflagresponse_response import FeatureflagresponseResponse
 from openapi_client.models.file_model import FileModel
 from openapi_client.models.filemodel_response import FilemodelResponse
+from openapi_client.models.fine_tune_type import FineTuneType
+from openapi_client.models.fine_tuned_model import FineTunedModel
 from openapi_client.models.fine_tuning_job_status import FineTuningJobStatus
+from openapi_client.models.finetunedmodel_list_response import FinetunedmodelListResponse
 from openapi_client.models.finish_ft_job_request import FinishFTJobRequest
 from openapi_client.models.finish_ft_job_request_v2 import FinishFTJobRequestV2
 from openapi_client.models.gcp_file_store_config import GCPFileStoreConfig
@@ -320,6 +328,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
@@ -352,6 +364,7 @@ from openapi_client.models.lb_resource import LBResource
 from openapi_client.models.lbresource_response import LbresourceResponse
 from openapi_client.models.list_machine_pools_response import ListMachinePoolsResponse
 from openapi_client.models.list_machines_response import ListMachinesResponse
+from openapi_client.models.list_resource_quotas_query import ListResourceQuotasQuery
 from openapi_client.models.list_response_metadata import ListResponseMetadata
 from openapi_client.models.listmachinepoolsresponse_response import ListmachinepoolsresponseResponse
 from openapi_client.models.listmachinesresponse_response import ListmachinesresponseResponse
@@ -470,6 +483,7 @@ from openapi_client.models.protocols import Protocols
 from openapi_client.models.provider_metadata import ProviderMetadata
 from openapi_client.models.providermetadata_response import ProvidermetadataResponse
 from openapi_client.models.python_modules import PythonModules
+from openapi_client.models.quota import Quota
 from openapi_client.models.ray_gcs_external_storage_config import RayGCSExternalStorageConfig
 from openapi_client.models.ray_runtime_env_config import RayRuntimeEnvConfig
 from openapi_client.models.read_billing_version import ReadBillingVersion
@@ -482,6 +496,9 @@ from openapi_client.models.request_password_reset_params import RequestPasswordR
 from openapi_client.models.requestemailmagiclinkresponse_response import RequestemailmagiclinkresponseResponse
 from openapi_client.models.requestotpreturnapimodel_response import RequestotpreturnapimodelResponse
 from openapi_client.models.reset_password_params import ResetPasswordParams
+from openapi_client.models.resource_quota import ResourceQuota
+from openapi_client.models.resourcequota_list_response import ResourcequotaListResponse
+from openapi_client.models.resourcequota_response import ResourcequotaResponse
 from openapi_client.models.resources import Resources
 from openapi_client.models.resubmit_ft_job_request import ResubmitFTJobRequest
 from openapi_client.models.rollback_service_model import RollbackServiceModel