lightning-sdk 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.5"
+__version__ = "0.2.6"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/cli/serve.py

@@ -1,5 +1,6 @@
 import os
 import subprocess
+from datetime import datetime
 from pathlib import Path
 from typing import Optional, Union
 
@@ -7,11 +8,11 @@ import click
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm
+from rich.syntax import Syntax
 
 from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.deployment_api import DeploymentApi
 from lightning_sdk.api.lit_container_api import LitContainerApi
-from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.serve import _LitServeDeployer
 
@@ -109,7 +110,7 @@ def api(
     script_path: str,
     easy: bool,
     cloud: bool,
-    name: str,
+    name: Optional[str],
     non_interactive: bool,
     machine: str,
     interruptible: bool,
@@ -137,6 +138,7 @@ def api(
         user=user,
         cloud_account=cloud_account,
         port=port,
+        replicas=replicas,
         min_replica=min_replica,
         max_replica=max_replica,
         include_credentials=not no_credentials,
@@ -172,6 +174,10 @@ def api_impl(
 
     _LitServeDeployer.generate_client() if easy else None
 
+    if not repository:
+        timestr = datetime.now().strftime("%b-%d-%H_%M")
+        repository = f"litserve-{timestr}".lower()
+
     if cloud:
         repository = repository or "litserve-model"
         machine = Machine.from_str(machine)
@@ -223,12 +229,30 @@ def _handle_cloud(
     replicas: Optional[int] = 1,
     include_credentials: Optional[bool] = True,
 ) -> None:
+    deployment_name = os.path.basename(repository)
+    tag = tag if tag else "latest"
     if teamspace is None:
         menu = _TeamspacesMenu()
         resolved_teamspace = menu._resolve_teamspace(teamspace)
     else:
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
+    if DeploymentApi().get_deployment_by_name(deployment_name, resolved_teamspace.id):
+        syntax = Syntax(
+            "from lightning_sdk import Deployment\n\n"
+            f"deployment = Deployment('{deployment_name}', teamspace={resolved_teamspace.name})\n"
+            "deployment.update(...)",
+            "python",
+            line_numbers=True,
+        )
+
+        console.print(
+            f"\nDeployment with name [b]{deployment_name}[/b] already running. "
+            "To update the deployment, use the Deployment API:\n",
+            syntax,
+        )
+        return
+
     port = port or 8000
     ls_deployer = _LitServeDeployer()
     path = ls_deployer.dockerize_api(script_path, port=port, gpu=not machine.is_cpu(), tag=tag, print_success=False)
@@ -242,14 +266,10 @@ def _handle_cloud(
         console.print("Please fix the Dockerfile and try again.", style="red")
         return
 
-    tag = tag if tag else "latest"
-
-    lit_cr = LitContainerApi()
-    deployment_name = os.path.basename(repository)
-
     ls_deployer.authenticate()
-    if DeploymentApi().get_deployment_by_name(deployment_name, resolved_teamspace.id):
-        raise StudioCliError(f"Deployment {deployment_name} already exists. Please choose a different name.") from None
+    # list containers to create the project if it doesn't exist
+    lit_cr = LitContainerApi()
+    lit_cr.list_containers(resolved_teamspace.id)
 
     with Progress(
         SpinnerColumn(),

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -196,6 +196,7 @@ from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
 from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import SecretsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
+from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
 from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
 from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
 from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
@@ -273,6 +274,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version im
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_collab_status import V1CloudSpaceInstanceCollabStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_config import V1CloudSpaceInstanceConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_startup_status import V1CloudSpaceInstanceStartupStatus
@@ -346,6 +348,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_pipeline_template_request import V1CreatePipelineTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_ssh_public_key_request import V1CreateSSHPublicKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_server_alert_response import V1CreateServerAlertResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_request import V1CreateSharedMetricsStreamRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_response import V1CreateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection_response import V1CreateSnowflakeConnectionResponse
@@ -770,6 +773,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
@@ -797,6 +801,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret import V1Secret
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret_type import V1SecretType
 from lightning_sdk.lightning_cloud.openapi.models.v1_select import V1Select
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert import V1ServerAlert
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
@@ -821,6 +828,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
+from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics

lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py

@@ -334,12 +334,12 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def cloud_space_environment_template_service_list_get_cloud_space_environment_templates(self, **kwargs) -> 'V1ListCloudSpaceEnvironmentTemplatesResponse':  # noqa: E501
-        """cloud_space_environment_template_service_list_get_cloud_space_environment_templates  # noqa: E501
+    def cloud_space_environment_template_service_list_cloud_space_environment_templates(self, **kwargs) -> 'V1ListCloudSpaceEnvironmentTemplatesResponse':  # noqa: E501
+        """cloud_space_environment_template_service_list_cloud_space_environment_templates  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.cloud_space_environment_template_service_list_get_cloud_space_environment_templates(async_req=True)
+        >>> thread = api.cloud_space_environment_template_service_list_cloud_space_environment_templates(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
@@ -352,17 +352,17 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
         """
         kwargs['_return_http_data_only'] = True
         if kwargs.get('async_req'):
-            return self.cloud_space_environment_template_service_list_get_cloud_space_environment_templates_with_http_info(**kwargs)  # noqa: E501
+            return self.cloud_space_environment_template_service_list_cloud_space_environment_templates_with_http_info(**kwargs)  # noqa: E501
         else:
-            (data) = self.cloud_space_environment_template_service_list_get_cloud_space_environment_templates_with_http_info(**kwargs)  # noqa: E501
+            (data) = self.cloud_space_environment_template_service_list_cloud_space_environment_templates_with_http_info(**kwargs)  # noqa: E501
             return data
 
-    def cloud_space_environment_template_service_list_get_cloud_space_environment_templates_with_http_info(self, **kwargs) -> 'V1ListCloudSpaceEnvironmentTemplatesResponse':  # noqa: E501
-        """cloud_space_environment_template_service_list_get_cloud_space_environment_templates  # noqa: E501
+    def cloud_space_environment_template_service_list_cloud_space_environment_templates_with_http_info(self, **kwargs) -> 'V1ListCloudSpaceEnvironmentTemplatesResponse':  # noqa: E501
+        """cloud_space_environment_template_service_list_cloud_space_environment_templates  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.cloud_space_environment_template_service_list_get_cloud_space_environment_templates_with_http_info(async_req=True)
+        >>> thread = api.cloud_space_environment_template_service_list_cloud_space_environment_templates_with_http_info(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
@@ -385,7 +385,7 @@ class CloudSpaceEnvironmentTemplateServiceApi(object):
             if key not in all_params:
                 raise TypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method cloud_space_environment_template_service_list_get_cloud_space_environment_templates" % key
+                    " to method cloud_space_environment_template_service_list_cloud_space_environment_templates" % key
                 )
             params[key] = val
         del params['kwargs']

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -665,6 +665,111 @@ class ClusterServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cluster_service_create_server_alert(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse':  # noqa: E501
+        """cluster_service_create_server_alert  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_create_server_alert(body, server_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ServerIdAlertsBody body: (required)
+        :param str server_id: (required)
+        :return: V1CreateServerAlertResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs)  # noqa: E501
+            return data
+
+    def cluster_service_create_server_alert_with_http_info(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse':  # noqa: E501
+        """cluster_service_create_server_alert  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_create_server_alert_with_http_info(body, server_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ServerIdAlertsBody body: (required)
+        :param str server_id: (required)
+        :return: V1CreateServerAlertResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'server_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method cluster_service_create_server_alert" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `cluster_service_create_server_alert`")  # noqa: E501
+        # verify the required parameter 'server_id' is set
+        if ('server_id' not in params or
+                params['server_id'] is None):
+            raise ValueError("Missing the required parameter `server_id` when calling `cluster_service_create_server_alert`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'server_id' in params:
+            path_params['serverId'] = params['server_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/core/servers/{serverId}/alerts', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1CreateServerAlertResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def cluster_service_delete_cluster(self, id: 'str', **kwargs) -> 'V1DeleteClusterResponse':  # noqa: E501
         """cluster_service_delete_cluster  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py

@@ -159,6 +159,7 @@ class LitRegistryServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str lit_repo_name: (required)
+        :param str cluster_id:
         :return: V1DeleteLitRepositoryResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -181,12 +182,13 @@ class LitRegistryServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str lit_repo_name: (required)
+        :param str cluster_id:
         :return: V1DeleteLitRepositoryResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'lit_repo_name']  # noqa: E501
+        all_params = ['project_id', 'lit_repo_name', 'cluster_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -219,6 +221,8 @@ class LitRegistryServiceApi(object):
             path_params['litRepoName'] = params['lit_repo_name']  # noqa: E501
 
         query_params = []
+        if 'cluster_id' in params:
+            query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
 
         header_params = {}
 
@@ -475,6 +479,7 @@ class LitRegistryServiceApi(object):
         :param str project_id: (required)
         :param str lit_repo_name: (required)
         :param bool latest_only:
+        :param str cluster_id:
         :return: V1LitRepository
                  If the method is called asynchronously,
                  returns the request thread.
@@ -498,12 +503,13 @@ class LitRegistryServiceApi(object):
         :param str project_id: (required)
         :param str lit_repo_name: (required)
         :param bool latest_only:
+        :param str cluster_id:
         :return: V1LitRepository
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'lit_repo_name', 'latest_only']  # noqa: E501
+        all_params = ['project_id', 'lit_repo_name', 'latest_only', 'cluster_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -538,6 +544,8 @@ class LitRegistryServiceApi(object):
         query_params = []
         if 'latest_only' in params:
             query_params.append(('latestOnly', params['latest_only']))  # noqa: E501
+        if 'cluster_id' in params:
+            query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
 
         header_params = {}
 
@@ -580,6 +588,7 @@ class LitRegistryServiceApi(object):
         :param str project_id: (required)
         :param str lit_repo_name: (required)
         :param bool latest_only:
+        :param str cluster_id:
         :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -603,12 +612,13 @@ class LitRegistryServiceApi(object):
         :param str project_id: (required)
         :param str lit_repo_name: (required)
         :param bool latest_only:
+        :param str cluster_id:
         :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'lit_repo_name', 'latest_only']  # noqa: E501
+        all_params = ['project_id', 'lit_repo_name', 'latest_only', 'cluster_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -643,6 +653,8 @@ class LitRegistryServiceApi(object):
         query_params = []
         if 'latest_only' in params:
             query_params.append(('latestOnly', params['latest_only']))  # noqa: E501
+        if 'cluster_id' in params:
+            query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -154,6 +154,7 @@ from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
 from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import SecretsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
+from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
 from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
 from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
 from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
@@ -231,6 +232,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version im
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_collab_status import V1CloudSpaceInstanceCollabStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_config import V1CloudSpaceInstanceConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_startup_status import V1CloudSpaceInstanceStartupStatus
@@ -304,6 +306,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_pipeline_template_request import V1CreatePipelineTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_ssh_public_key_request import V1CreateSSHPublicKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_server_alert_response import V1CreateServerAlertResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_request import V1CreateSharedMetricsStreamRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_response import V1CreateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection_response import V1CreateSnowflakeConnectionResponse
@@ -728,6 +731,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
@@ -755,6 +759,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret import V1Secret
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret_type import V1SecretType
 from lightning_sdk.lightning_cloud.openapi.models.v1_select import V1Select
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert import V1ServerAlert
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
@@ -779,6 +786,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
+from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics

lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py

@@ -41,6 +41,7 @@ class EnvironmenttemplatesIdBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'config': 'V1CloudSpaceEnvironmentTemplateConfig',
         'created_at': 'datetime',
         'name': 'str',
         'org_id': 'str',
@@ -49,6 +50,7 @@ class EnvironmenttemplatesIdBody(object):
     }
 
     attribute_map = {
+        'config': 'config',
         'created_at': 'createdAt',
         'name': 'name',
         'org_id': 'orgId',
@@ -56,14 +58,17 @@ class EnvironmenttemplatesIdBody(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, created_at: 'datetime' =None, name: 'str' =None, org_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, config: 'V1CloudSpaceEnvironmentTemplateConfig' =None, created_at: 'datetime' =None, name: 'str' =None, org_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """EnvironmenttemplatesIdBody - a model defined in Swagger"""  # noqa: E501
+        self._config = None
         self._created_at = None
         self._name = None
         self._org_id = None
         self._updated_at = None
         self._user_id = None
         self.discriminator = None
+        if config is not None:
+            self.config = config
         if created_at is not None:
             self.created_at = created_at
         if name is not None:
@@ -75,6 +80,27 @@ class EnvironmenttemplatesIdBody(object):
         if user_id is not None:
             self.user_id = user_id
 
+    @property
+    def config(self) -> 'V1CloudSpaceEnvironmentTemplateConfig':
+        """Gets the config of this EnvironmenttemplatesIdBody.  # noqa: E501
+
+
+        :return: The config of this EnvironmenttemplatesIdBody.  # noqa: E501
+        :rtype: V1CloudSpaceEnvironmentTemplateConfig
+        """
+        return self._config
+
+    @config.setter
+    def config(self, config: 'V1CloudSpaceEnvironmentTemplateConfig'):
+        """Sets the config of this EnvironmenttemplatesIdBody.
+
+
+        :param config: The config of this EnvironmenttemplatesIdBody.  # noqa: E501
+        :type: V1CloudSpaceEnvironmentTemplateConfig
+        """
+
+        self._config = config
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this EnvironmenttemplatesIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py

@@ -41,6 +41,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'alerts': 'list[V1ServerAlert]',
         'app_url': 'str',
         'bytes_to_sync': 'str',
         'cloud_space_id': 'str',
@@ -73,6 +74,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
     }
 
     attribute_map = {
+        'alerts': 'alerts',
         'app_url': 'appUrl',
         'bytes_to_sync': 'bytesToSync',
         'cloud_space_id': 'cloudSpaceId',
@@ -104,8 +106,9 @@ class Externalv1CloudSpaceInstanceStatus(object):
         'vscode_url': 'vscodeUrl'
     }
 
-    def __init__(self, app_url: 'str' =None, bytes_to_sync: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, files_to_sync: 'str' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_last_updated: 'datetime' =None, termination_time: 'datetime' =None, vscode_url: 'str' =None):  # noqa: E501
+    def __init__(self, alerts: 'list[V1ServerAlert]' =None, app_url: 'str' =None, bytes_to_sync: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, files_to_sync: 'str' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_last_updated: 'datetime' =None, termination_time: 'datetime' =None, vscode_url: 'str' =None):  # noqa: E501
         """Externalv1CloudSpaceInstanceStatus - a model defined in Swagger"""  # noqa: E501
+        self._alerts = None
         self._app_url = None
         self._bytes_to_sync = None
         self._cloud_space_id = None
@@ -136,6 +139,8 @@ class Externalv1CloudSpaceInstanceStatus(object):
         self._termination_time = None
         self._vscode_url = None
         self.discriminator = None
+        if alerts is not None:
+            self.alerts = alerts
         if app_url is not None:
             self.app_url = app_url
         if bytes_to_sync is not None:
@@ -195,6 +200,27 @@ class Externalv1CloudSpaceInstanceStatus(object):
         if vscode_url is not None:
             self.vscode_url = vscode_url
 
+    @property
+    def alerts(self) -> 'list[V1ServerAlert]':
+        """Gets the alerts of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+
+
+        :return: The alerts of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+        :rtype: list[V1ServerAlert]
+        """
+        return self._alerts
+
+    @alerts.setter
+    def alerts(self, alerts: 'list[V1ServerAlert]'):
+        """Sets the alerts of this Externalv1CloudSpaceInstanceStatus.
+
+
+        :param alerts: The alerts of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+        :type: list[V1ServerAlert]
+        """
+
+        self._alerts = alerts
+
     @property
     def app_url(self) -> 'str':
         """Gets the app_url of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/externalv1_cluster.py

@@ -44,6 +44,7 @@ class Externalv1Cluster(object):
         'created_at': 'datetime',
         'id': 'str',
         'name': 'str',
+        'protected': 'bool',
         'spec': 'V1ClusterSpec',
         'status': 'V1ClusterStatus'
     }
@@ -52,15 +53,17 @@ class Externalv1Cluster(object):
         'created_at': 'createdAt',
         'id': 'id',
         'name': 'name',
+        'protected': 'protected',
         'spec': 'spec',
         'status': 'status'
     }
 
-    def __init__(self, created_at: 'datetime' =None, id: 'str' =None, name: 'str' =None, spec: 'V1ClusterSpec' =None, status: 'V1ClusterStatus' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, id: 'str' =None, name: 'str' =None, protected: 'bool' =None, spec: 'V1ClusterSpec' =None, status: 'V1ClusterStatus' =None):  # noqa: E501
         """Externalv1Cluster - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
         self._id = None
         self._name = None
+        self._protected = None
         self._spec = None
         self._status = None
         self.discriminator = None
@@ -70,6 +73,8 @@ class Externalv1Cluster(object):
             self.id = id
         if name is not None:
             self.name = name
+        if protected is not None:
+            self.protected = protected
         if spec is not None:
             self.spec = spec
         if status is not None:
@@ -138,6 +143,27 @@ class Externalv1Cluster(object):
 
         self._name = name
 
+    @property
+    def protected(self) -> 'bool':
+        """Gets the protected of this Externalv1Cluster.  # noqa: E501
+
+
+        :return: The protected of this Externalv1Cluster.  # noqa: E501
+        :rtype: bool
+        """
+        return self._protected
+
+    @protected.setter
+    def protected(self, protected: 'bool'):
+        """Sets the protected of this Externalv1Cluster.
+
+
+        :param protected: The protected of this Externalv1Cluster.  # noqa: E501
+        :type: bool
+        """
+
+        self._protected = protected
+
     @property
     def spec(self) -> 'V1ClusterSpec':
         """Gets the spec of this Externalv1Cluster.  # noqa: E501