lightning-sdk 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

lightning_sdk/__init__.py

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

lightning_sdk/api/deployment_api.py

@@ -228,6 +228,7 @@ class DeploymentApi:
         return self._client.jobs_service_create_deployment(
             project_id=deployment.project_id,
             body=CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(
+                cloudspace_id=deployment.cloudspace_id,
                 autoscaling=deployment.autoscaling,
                 cluster_id=deployment.spec.cluster_id,
                 endpoint=deployment.endpoint,

lightning_sdk/api/lit_container_api.py

@@ -7,6 +7,7 @@ import requests
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_registry_url
+from lightning_sdk.lightning_cloud.env import LIGHTNING_CLOUD_URL
 from lightning_sdk.lightning_cloud.openapi.models import V1DeleteLitRepositoryResponse
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 from lightning_sdk.teamspace import Teamspace
@@ -126,8 +127,14 @@ class LitContainerApi:
 
     @retry_on_lcr_auth_failure
     def upload_container(
-        self, container: str, teamspace: Teamspace, tag: str, cloud_account: str, platform: str
-    ) -> Generator[dict, None, None]:
+        self,
+        container: str,
+        teamspace: Teamspace,
+        tag: str,
+        cloud_account: str,
+        platform: str,
+        return_final_dict: bool = False,
+    ) -> Generator[dict, None, Dict]:
         """Upload container will push the container to LitCR.
 
         It uses docker push API to interact with docker daemon which will then push the container to a storage
@@ -140,6 +147,7 @@ class LitContainerApi:
             Named cloud-account in the CLI options.
         :param platform: If empty will be linux/amd64. This is important because our entire deployment infra runs on
             linux/amd64. Will show user a warning otherwise.
+        :return_final_dict: Controls whether we respond with the dictionary containing metadata about container upload
         :return: Generator[dict, None, dict]
         """
         try:
@@ -166,6 +174,15 @@ class LitContainerApi:
             raise ValueError(f"Could not tag container {container}:{tag} with {repository}:{tag}")
         yield from self._push_with_retry(repository, tag=tag)
 
+        if return_final_dict:
+            yield {
+                "finish": True,
+                "url": f"{LIGHTNING_CLOUD_URL}/{teamspace.owner.name}/{teamspace.name}/containers/"
+                f"{container_basename}?section=tags"
+                f"{f'?clusterId={cloud_account}' if cloud_account is not None else ''}",
+                "repository": repository,
+            }
+
     def _push_with_retry(self, repository: str, tag: str, max_retries: int = 3) -> Iterator[Dict[str, Any]]:
         def is_auth_error(error_msg: str) -> bool:
             auth_errors = ["unauthorized", "authentication required", "unauth"]

lightning_sdk/api/teamspace_api.py

@@ -34,7 +34,7 @@ class TeamspaceApi:
 
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=7)
-        self._models: Optional[ModelsStoreApi] = None
+        self._models_api: Optional[ModelsStoreApi] = None
 
     def get_teamspace(self, name: str, owner_id: str) -> V1Project:
         """Get the current teamspace from the owner."""
@@ -166,12 +166,12 @@ class TeamspaceApi:
 
     # lazy property which is only created when needed
     @property
-    def models(self) -> ModelsStoreApi:
-        if not self._models:
-            self._models = ModelsStoreApi(self._client.api_client)
-        return self._models
+    def models_api(self) -> ModelsStoreApi:
+        if not self._models_api:
+            self._models_api = ModelsStoreApi(self._client.api_client)
+        return self._models_api
 
-    def get_model_version(self, name: str, version: str, teamspace_id: str) -> V1ModelVersionArchive:
+    def get_model_version(self, name: str, version: Optional[str], teamspace_id: str) -> V1ModelVersionArchive:
         return _get_model_version(client=self._client, name=name, version=version, teamspace_id=teamspace_id)
 
     def create_model(
@@ -184,14 +184,14 @@ class TeamspaceApi:
         cloud_account: str,
     ) -> V1ModelVersionArchive:
         # ask if such model already exists by listing models with specific name
-        models = self.models.models_store_list_models(project_id=teamspace_id, name=name).models
+        models = self.models_api.models_store_list_models(project_id=teamspace_id, name=name).models
         if len(models) == 0:
-            return self.models.models_store_create_model(
+            return self.models_api.models_store_create_model(
                 body=ProjectIdModelsBody(cluster_id=cloud_account, metadata=metadata, name=name, private=private),
                 project_id=teamspace_id,
             )
         assert len(models) == 1, "Multiple models with the same name found"
-        return self.models.models_store_create_model_version(
+        return self.models_api.models_store_create_model_version(
             body=ModelIdVersionsBody(cluster_id=cloud_account, version=version),
             project_id=teamspace_id,
             model_id=models[0].id,
@@ -199,16 +199,18 @@ class TeamspaceApi:
 
     def delete_model(self, name: str, version: Optional[str], teamspace_id: str) -> None:
         """Delete a model or a version from the model store."""
-        models = self.models.models_store_list_models(project_id=teamspace_id, name=name).models
+        models = self.models_api.models_store_list_models(project_id=teamspace_id, name=name).models
         assert len(models) == 1, "Multiple models with the same name found"
-        model_id = models[0].id
+        model = models[0]
         # decide if delete only version of whole model
         if version:
             if version == "default":
-                version = models[0].default_version
-            self.models.models_store_delete_model_version(project_id=teamspace_id, model_id=model_id, version=version)
+                version = model.default_version
+            self.models_api.models_store_delete_model_version(
+                project_id=teamspace_id, model_id=model.id, version=version
+            )
         else:
-            self.models.models_store_delete_model(project_id=teamspace_id, model_id=model_id)
+            self.models_api.models_store_delete_model(project_id=teamspace_id, model_id=model.id)
 
     def upload_model_file(
         self,
@@ -256,7 +258,7 @@ class TeamspaceApi:
                 main_pbar.update(1)
 
     def complete_model_upload(self, model_id: str, version: str, teamspace_id: str) -> None:
-        self.models.models_store_complete_model_upload(
+        self.models_api.models_store_complete_model_upload(
             body=_DummyBody(),
             project_id=teamspace_id,
             model_id=model_id,
@@ -266,12 +268,14 @@ class TeamspaceApi:
     def download_model_files(
         self,
         name: str,
-        version: str,
+        version: Optional[str],
         download_dir: Path,
         teamspace_name: str,
         teamspace_owner_name: str,
         progress_bar: bool = True,
     ) -> List[str]:
+        if version is None:
+            version = "default"
         return _download_model_files(
             client=self._client,
             teamspace_name=teamspace_name,

lightning_sdk/api/utils.py

@@ -493,7 +493,7 @@ def _get_model_version(client: LightningClient, teamspace_id: str, name: str, ve
         raise ValueError(f"Model `{name}` does not exist")
     elif len(models) > 1:
         raise ValueError("Multiple models with the same name found")
-    if version == ("default"):
+    if version is None or version == "default":
         return models[0].default_version
     versions = api.models_store_list_model_versions(project_id=teamspace_id, model_id=models[0].id).versions
     if not versions:

lightning_sdk/cli/entrypoint.py

@@ -21,7 +21,7 @@ from lightning_sdk.cli.inspect import inspect
 from lightning_sdk.cli.list import list_cli
 from lightning_sdk.cli.open import open
 from lightning_sdk.cli.run import run
-from lightning_sdk.cli.serve import serve
+from lightning_sdk.cli.serve import deploy
 from lightning_sdk.cli.start import start
 from lightning_sdk.cli.stop import stop
 from lightning_sdk.cli.switch import switch
@@ -76,7 +76,7 @@ main_cli.add_command(generate)
 main_cli.add_command(inspect)
 main_cli.add_command(list_cli)
 main_cli.add_command(run)
-main_cli.add_command(serve)
+main_cli.add_command(deploy)
 main_cli.add_command(start)
 main_cli.add_command(stop)
 main_cli.add_command(switch)

lightning_sdk/cli/serve.py

@@ -1,10 +1,12 @@
 import os
+import socket
 import subprocess
 import time
 import webbrowser
 from datetime import datetime
+from enum import Enum
 from pathlib import Path
-from typing import Optional, Union
+from typing import List, Optional, TypedDict, Union
 from urllib.parse import urlencode
 
 import click
@@ -18,10 +20,13 @@ from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.login import Auth, AuthServer
+from lightning_sdk.lightning_cloud.openapi import V1CloudSpace
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
 from lightning_sdk.serve import _LitServeDeployer
 from lightning_sdk.utils.resolve import _get_authed_user, _resolve_teamspace
 
 _MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_POLL_TIMEOUT = 600
 
 
 class _ServeGroup(click.Group):
@@ -33,22 +38,22 @@ class _ServeGroup(click.Group):
         return super().parse_args(ctx, args)
 
 
-@click.group("serve", cls=_ServeGroup)
-def serve() -> None:
-    """Serve a LitServe model.
+@click.group("deploy", cls=_ServeGroup)
+def deploy() -> None:
+    """Deploy a LitServe model.
 
     Example:
-        lightning serve server.py  # deploy to the cloud
+        lightning deploy server.py  # deploy to the cloud
 
     Example:
-        lightning serve server.py --local  # serve locally
+        lightning deploy server.py --local  # run locally
 
-    You can deploy the API to the cloud by running `lightning serve server.py`.
+    You can deploy the API to the cloud by running `lightning deploy server.py`.
     This will build a docker container for the server.py script and deploy it to the Lightning AI platform.
     """
 
 
-@serve.command("api")
+@deploy.command("api")
 @click.argument("script-path", type=click.Path(exists=True))
 @click.option(
     "--easy",
@@ -272,20 +277,127 @@ def select_teamspace(teamspace: Optional[str], org: Optional[str], user: Optiona
     return _resolve_teamspace(teamspace=teamspace, org=org, user=user)
 
 
-def poll_verified_status() -> bool:
+class _UserStatus(TypedDict):
+    verified: bool
+    onboarded: bool
+
+
+def poll_verified_status(timeout: int = _POLL_TIMEOUT) -> _UserStatus:
     """Polls the verified status of the user until it is True or a timeout occurs."""
     user_api = UserApi()
     user = _get_authed_user()
     start_time = datetime.now()
-    timeout = 600  # 10 minutes
+    result = {"onboarded": False, "verified": False}
     while True:
         user_resp = user_api.get_user(name=user.name)
+        result["onboarded"] = user_resp.status.completed_project_onboarding
+        result["verified"] = user_resp.status.verified
         if user_resp.status.verified:
-            return True
+            return result
         if (datetime.now() - start_time).total_seconds() > timeout:
             break
         time.sleep(5)
-    return False
+    return result
+
+
+class _OnboardingStatus(Enum):
+    NOT_VERIFIED = "not_verified"
+    ONBOARDING = "onboarding"
+    ONBOARDED = "onboarded"
+
+
+class _Onboarding:
+    def __init__(self, console: Console) -> None:
+        self.console = console
+        self.user = _get_authed_user()
+        self.user_api = UserApi()
+        self.client = LightningClient(max_tries=7)
+
+    @property
+    def verified(self) -> bool:
+        return self.user_api.get_user(name=self.user.name).status.verified
+
+    @property
+    def is_onboarded(self) -> bool:
+        return self.user_api.get_user(name=self.user.name).status.completed_project_onboarding
+
+    @property
+    def can_join_org(self) -> bool:
+        return len(self.client.organizations_service_list_joinable_organizations().joinable_organizations) > 0
+
+    @property
+    def status(self) -> _OnboardingStatus:
+        if not self.verified:
+            return _OnboardingStatus.NOT_VERIFIED
+        if self.is_onboarded:
+            return _OnboardingStatus.ONBOARDED
+        return _OnboardingStatus.ONBOARDING
+
+    def _wait(self, timeout: int = _POLL_TIMEOUT) -> None:
+        """Wait for user onboarding if they can join the teamspace otherwise move to select a teamspace."""
+        status = self.status
+        if status == _OnboardingStatus.ONBOARDED:
+            return
+
+        start_time = datetime.now()
+        while self.status != _OnboardingStatus.ONBOARDED:
+            time.sleep(5)
+            if self.is_onboarded:
+                return
+            if (datetime.now() - start_time).total_seconds() > timeout:
+                break
+
+        raise RuntimeError("Timed out waiting for onboarding status")
+
+    def get_cloudspace_id(self, teamspace: Teamspace) -> Optional[str]:
+        cloudspaces: List[V1CloudSpace] = self.client.cloud_space_service_list_cloud_spaces(teamspace.id).cloudspaces
+        for cloudspace in cloudspaces:
+            if "scratch-studio" in cloudspace.name or "scratch-studio" in cloudspace.display_name:
+                return cloudspace.id
+        return None
+
+    def select_teamspace(self, teamspace: Optional[str], org: Optional[str], user: Optional[str]) -> Teamspace:
+        """Select a teamspace while onboarding.
+
+        If user is being onboarded and can't join any org, the teamspace it will be resolved to the default
+         personal teamspace.
+        If user is being onboarded and can join an org then it will select default teamspace from the org.
+        """
+        if self.is_onboarded:
+            return select_teamspace(teamspace, org, user)
+
+        # Run only when user hasn't completed onboarding yet.
+        menu = _TeamspacesMenu()
+        possible_teamspaces = menu._get_possible_teamspaces(self.user)
+        can_join_org = self.can_join_org
+
+        if len(possible_teamspaces) == 1 and can_join_org:
+            # wait for onboarding to complete so that user can join an org
+            # create deployment in the org default teamspace
+            self.console.print("Waiting for account setup. Visit lightning.ai")
+            self._wait()
+
+        possible_teamspaces = menu._get_possible_teamspaces(self.user)
+        if len(possible_teamspaces) == 1:
+            # User didn't select any org
+            value = next(iter(possible_teamspaces.values()))
+            return Teamspace(name=value["name"], org=value["org"], user=value["user"])
+
+        for _, value in possible_teamspaces.items():
+            # User select an org
+            # Onboarding teamspace will be the default teamspace in the selected org
+            if value["org"]:
+                return Teamspace(name=value["name"], org=value["org"], user=value["user"])
+        raise RuntimeError("Unable to select teamspace. Visit lightning.ai")
+
+
+def is_connected(host: str = "8.8.8.8", port: int = 53, timeout: int = 10) -> bool:
+    try:
+        socket.setdefaulttimeout(timeout)
+        socket.create_connection((host, port))
+        return True
+    except OSError:
+        return False
 
 
 def _handle_cloud(
@@ -306,6 +418,11 @@ def _handle_cloud(
     replicas: Optional[int] = 1,
     include_credentials: Optional[bool] = True,
 ) -> None:
+    if not is_connected():
+        console.print("❌ Internet connection required to deploy to the cloud.", style="red")
+        console.print("To run locally instead, use: `lightning serve [SCRIPT | server.py] --local`")
+        return
+
     deployment_name = os.path.basename(repository)
     tag = tag if tag else "latest"
 
@@ -346,11 +463,17 @@ def _handle_cloud(
     console.print("\nPushing container to registry. It may take a while...", style="bold")
     # Authenticate with LitServe affiliate
     authenticate(shall_confirm=not non_interactive)
-    resolved_teamspace = select_teamspace(teamspace, org, user)
-    verified = poll_verified_status()
-    if not verified:
+    user_status = poll_verified_status()
+    cloudspace_id: Optional[str] = None
+    if not user_status["verified"]:
         console.print("❌ Verify phone number to continue. Visit lightning.ai.", style="red")
         return
+    if not user_status["onboarded"]:
+        onboarding = _Onboarding(console)
+        resolved_teamspace = onboarding.select_teamspace(teamspace, org, user)
+        cloudspace_id = onboarding.get_cloudspace_id(resolved_teamspace)
+    else:
+        resolved_teamspace = select_teamspace(teamspace, org, user)
 
     # list containers to create the project if it doesn't exist
     lit_cr = LitContainerApi()
@@ -393,6 +516,8 @@ def _handle_cloud(
         max_replica=max_replica,
         replicas=replicas,
         include_credentials=include_credentials,
+        cloudspace_id=cloudspace_id,
     )
     console.print(f"🚀 Deployment started, access at [i]{deployment_status.get('url')}[/i]")
-    webbrowser.open(deployment_status.get("url"))
+    if user_status["onboarded"]:
+        webbrowser.open(deployment_status.get("url"))

lightning_sdk/cli/upload.py

@@ -1,6 +1,7 @@
 import concurrent.futures
 import json
 import os
+import webbrowser
 from pathlib import Path
 from typing import Dict, Generator, List, Optional
 
@@ -162,7 +163,7 @@ def upload_container(
                 # let the push with retry take control of auth moving forward
                 pass
 
-            lines = api.upload_container(container, teamspace, tag, cloud_account, platform)
+            lines = api.upload_container(container, teamspace, tag, cloud_account, platform, return_final_dict=True)
             _print_docker_push(lines, console, progress, push_task)
         except DockerNotRunningError as e:
             e.print_help()
@@ -303,6 +304,8 @@ def _print_docker_push(lines: Generator, console: Console, progress: Progress, p
             console.print(f"\n[red]{line}[/red]")
             return
         elif "finish" in line:
+            if "url" in line:
+                webbrowser.open(line["url"])
             console.print(f"Container available at [i]{line['url']}[/i]")
             return
         else:

lightning_sdk/deployment/deployment.py

@@ -117,6 +117,7 @@ class Deployment:
         cloud_account: Optional[str] = None,
         custom_domain: Optional[str] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        cloudspace_id: Optional[str] = None,
         quantity: Optional[int] = None,
         include_credentials: Optional[bool] = None,
     ) -> None:
@@ -141,9 +142,10 @@ class Deployment:
             auth: The auth config to protect your services. Only Basic and Token supported.
             cloud_account: The name of the cloud account, the studio should be created on.
                 Doesn't matter when the studio already exists.
-            custom_domain: Whether your service would be referenced under a custom doamin.
+            custom_domain: Whether your service would be referenced under a custom domain.
+            cloudspace_id: Connect deployment to a Studio.
             quantity: The number of machines per replica to deploy.
-            include_credentials: Whether to include the environement variables for the SDK to authenticate
+            include_credentials: Whether to include the environment variables for the SDK to authenticate
 
         Note:
             Since a teamspace can either be owned by an org or by a user directly,
@@ -166,6 +168,7 @@ class Deployment:
                 name=self._name,
                 project_id=self._teamspace.id,
                 replicas=replicas,
+                cloudspace_id=cloudspace_id,
                 spec=to_spec(
                     cloud_account=cloud_account,
                     command=command,

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -278,6 +278,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
 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_cold_start_metrics import V1CloudSpaceColdStartMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
 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
@@ -293,6 +294,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -484,6 +486,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
 from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
@@ -495,6 +498,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
@@ -770,6 +774,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1Quest
 from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3174,6 +3174,99 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_get_cloud_space_cold_start_metrics_stats(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_cold_start_metrics_stats  # 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_service_get_cloud_space_cold_start_metrics_stats(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceColdStartMetricsStatsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_cold_start_metrics_stats  # 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_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceColdStartMetricsStatsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_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 cloud_space_service_get_cloud_space_cold_start_metrics_stats" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_get_cloud_space_cold_start_metrics_stats`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/cold-start-stats', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetCloudSpaceColdStartMetricsStatsResponse',  # 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 cloud_space_service_get_cloud_space_folder_index(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetFolderIndexResponse':  # noqa: E501
         """cloud_space_service_get_cloud_space_folder_index  # noqa: E501
 
@@ -3499,6 +3592,7 @@ class CloudSpaceServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str cloudspace_id:
         :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -3520,12 +3614,13 @@ class CloudSpaceServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str cloudspace_id:
         :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'cloudspace_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -3552,6 +3647,8 @@ class CloudSpaceServiceApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'cloudspace_id' in params:
+            query_params.append(('cloudspaceId', params['cloudspace_id']))  # noqa: E501
 
         header_params = {}