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

lightning_sdk/__init__.py

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

lightning_sdk/api/ai_hub_api.py

@@ -165,6 +165,7 @@ class AIHubApi:
                 name=name,
                 replicas=1,
                 spec=template.spec_v2.job,
+                parent_template_id=template_id,
             ),
         )
 

lightning_sdk/api/lit_container_api.py

@@ -1,12 +1,12 @@
 import inspect
 import time
-from typing import Any, Callable, Dict, Generator, Iterator, List
+from typing import Any, Callable, Dict, Generator, Iterator, List, Optional
 
 import docker
 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
@@ -21,6 +21,21 @@ class DockerPushError(Exception):
     pass
 
 
+class DockerNotRunningError(Exception):
+    def __init__(self, message: str = "Failed to connect to Docker") -> None:
+        self.message = message
+        super().__init__(self.message)
+
+    def print_help(self) -> None:
+        console = Console()
+        console.print("[bold red]Docker Error: Failed to connect to Docker. Is it running?[/bold red]")
+        console.print("[yellow]Troubleshooting:[/yellow]")
+        console.print("1. Check if Docker daemon is running")
+        console.print("2. Verify your user has proper permissions")
+        console.print("3. Try restarting Docker service")
+        console.print("4. Read more here: https://docs.docker.com/engine/daemon/start/")
+
+
 def retry_on_lcr_auth_failure(func: Callable) -> Callable:
     def generator_wrapper(self: "LitContainerApi", *args: Any, **kwargs: Any) -> Callable:
         try:
@@ -54,9 +69,7 @@ class LitContainerApi:
             self._docker_client.ping()
             self._docker_auth_config = {}
         except docker.errors.DockerException:
-            raise RuntimeError(
-                "Failed to connect to Docker, follow these steps to start it: https://docs.docker.com/engine/daemon/start/"
-            ) from None
+            raise DockerNotRunningError() from None
 
     def authenticate(self, reauth: bool = False) -> bool:
         resp = None
@@ -85,45 +98,73 @@ class LitContainerApi:
             print(f"Authentication error: {e} resp: {resp}")
             return False
 
-    def list_containers(self, project_id: str) -> List:
-        project = self._client.lit_registry_service_get_lit_project_registry(project_id)
+    def list_containers(self, project_id: str, cloud_account: Optional[str] = None) -> List:
+        """Lists containers of the project ID.
+
+        :param project_id: The non-human readable project ID used internally to identify projects.
+        :param cloud_account: The cluster ID of the cloud account. If None, will use the default cluster.
+        :return:
+        """
+        project = self._client.lit_registry_service_get_lit_project_registry(
+            project_id, cluster_id="" if cloud_account is None else cloud_account
+        )  # cloud account on the CLI is cluster_id
         return project.repositories
 
     def delete_container(self, project_id: str, container: str) -> V1DeleteLitRepositoryResponse:
+        """Deletes the container from LitCR. Garbage collection will come in and do the final sweep of the blob.
+
+        :param project_id: The non-human readable project ID used internally to identify projects.
+        :param container: The name of the docker container a user wants to push up, ie, nginx, vllm, etc
+        :return:
+        """
         try:
             return self._client.lit_registry_service_delete_lit_repository(project_id, container)
         except Exception as e:
             raise ValueError(f"Could not delete container {container} from project {project_id}: {e!s}") from e
 
     @retry_on_lcr_auth_failure
-    def upload_container(self, container: str, teamspace: Teamspace, tag: str) -> Generator[dict, None, None]:
+    def upload_container(
+        self, container: str, teamspace: Teamspace, tag: str, cloud_account: str, platform: str
+    ) -> Generator[dict, None, None]:
+        """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
+        location defined by teamspace + cloud_account. Will retry pushes if not authenticated or if push errors happen
+
+        :param container: The name of the docker container a user wants to push up, ie, nginx, vllm, etc
+        :param teamspace: The teamspace he container is going to appear in. This is <OWNER_ID>/<TEAMSPACE_NAME>
+        :param tag: The container tag, default will latest.
+        :param cloud_account: If empty will be pushed to Lightning SaaS storage. Otherwise, this will be cluster_id.
+            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: Generator[dict, None, dict]
+        """
         try:
-            self._docker_client.images.get(container)
+            self._docker_client.images.get(f"{container}:{tag}")
         except docker.errors.ImageNotFound:
             try:
-                self._docker_client.images.pull(container, tag)
-                self._docker_client.images.get(container)
+                self._docker_client.images.pull(repository=container, tag=tag, platform=platform)
+                self._docker_client.images.get(f"{container}:{tag}")
+            except docker.errors.ImageNotFound as e:
+                raise ValueError(f"Container {container}:{tag} does not exist") from e
             except docker.errors.APIError as e:
                 raise ValueError(f"Could not pull container {container}") from e
-            except docker.errors.ImageNotFound as e:
-                raise ValueError(f"Container {container} does not exist") from e
             except Exception as e:
-                raise ValueError(f"Unable to upload {container}") from e
+                raise ValueError(f"Unable to upload {container}:{tag}") from e
 
         registry_url = _get_registry_url()
         container_basename = container.split("/")[-1]
-        repository = f"{registry_url}/lit-container/{teamspace.owner.name}/{teamspace.name}/{container_basename}"
-        tagged = self._docker_client.api.tag(container, repository, tag)
+        repository = (
+            f"{registry_url}/lit-container{f'-{cloud_account}' if cloud_account is not None else ''}/"
+            f"{teamspace.owner.name}/{teamspace.name}/{container_basename}"
+        )
+        tagged = self._docker_client.api.tag(f"{container}:{tag}", repository, tag)
         if not tagged:
-            raise ValueError(f"Could not tag container {container} with {repository}:{tag}")
-        yield from self._push_with_retry(repository)
-        yield {
-            "finish": True,
-            "url": f"{LIGHTNING_CLOUD_URL}/{teamspace.owner.name}/{teamspace.name}/containers/{container_basename}",
-            "repository": repository,
-        }
-
-    def _push_with_retry(self, repository: str, max_retries: int = 3) -> Iterator[Dict[str, Any]]:
+            raise ValueError(f"Could not tag container {container}:{tag} with {repository}:{tag}")
+        yield from self._push_with_retry(repository, tag=tag)
+
+    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"]
             return any(err in error_msg.lower() for err in auth_errors)
@@ -141,7 +182,7 @@ class LitContainerApi:
                     time.sleep(2)
 
                 lines = self._docker_client.api.push(
-                    repository, stream=True, decode=True, auth_config=self._docker_auth_config
+                    repository, tag=tag, stream=True, decode=True, auth_config=self._docker_auth_config
                 )
                 for line in lines:
                     if isinstance(line, dict) and "error" in line:
@@ -149,7 +190,7 @@ class LitContainerApi:
                         if is_auth_error(error) or is_timeout_error(error):
                             if attempt < max_retries - 1:
                                 break
-                            raise DockerPushError(f"Max retries reached: {error}")
+                            raise DockerPushError(f"Max retries reached while pushing: {error}")
                         raise DockerPushError(f"Push error: {error}")
                     yield line
                 else:
@@ -158,14 +199,26 @@ class LitContainerApi:
             except docker.errors.APIError as e:
                 if (is_auth_error(str(e)) or is_timeout_error(str(e))) and attempt < max_retries - 1:
                     continue
-                raise DockerPushError(f"Push failed: {e}") from e
+                raise DockerPushError("Pushing the container failed") from e
 
         raise DockerPushError("Max push retries reached")
 
     @retry_on_lcr_auth_failure
-    def download_container(self, container: str, teamspace: Teamspace, tag: str) -> Generator[str, None, None]:
+    def download_container(
+        self, container: str, teamspace: Teamspace, tag: str, cloud_account: Optional[str] = None
+    ) -> Generator[str, None, None]:
+        """Will download container from LitCR. Optionally from a BYOC cluster.
+
+        :param container: The name of the container to download
+        :param teamspace: The teamspace containing the container
+        :param tag: The tag of the container to download
+        :return: Generator yielding download status
+        """
         registry_url = _get_registry_url()
-        repository = f"{registry_url}/lit-container/{teamspace.owner.name}/{teamspace.name}/{container}"
+        repository = (
+            f"{registry_url}/lit-container{f'-{cloud_account}' if cloud_account is not None else ''}/"
+            f"{teamspace.owner.name}/{teamspace.name}/{container}"
+        )
         try:
             self._docker_client.images.pull(repository, tag=tag, auth_config=self._docker_auth_config)
         except requests.exceptions.HTTPError as e:

lightning_sdk/api/teamspace_api.py

@@ -215,16 +215,15 @@ class TeamspaceApi:
         version: str,
         local_path: Path,
         remote_path: str,
-        cloud_account: str,
         teamspace_id: str,
         progress_bar: bool = True,
     ) -> None:
+        """Upload a file to the model store."""
         uploader = _ModelFileUploader(
             client=self._client,
             model_id=model_id,
             version=version,
             teamspace_id=teamspace_id,
-            cloud_account=cloud_account,
             file_path=str(local_path),
             remote_path=str(remote_path),
             progress_bar=progress_bar,
@@ -235,20 +234,20 @@ class TeamspaceApi:
         self,
         model_id: str,
         version: str,
-        root_path: Path,
-        filepaths: List[Path],
-        cloud_account: str,
+        file_paths: List[Path],
+        remote_paths: List[str],
         teamspace_id: str,
         progress_bar: bool = True,
     ) -> None:
-        main_pbar = tqdm(total=len(filepaths), desc="Uploading files...", position=0) if progress_bar else None
-        for filepath in filepaths:
+        """Upload files to the model store."""
+        main_pbar = tqdm(total=len(file_paths), desc="Uploading files...", position=0) if progress_bar else None
+        assert len(file_paths) == len(remote_paths), "File paths and remote paths must have the same length"
+        for filepath, remote_path in zip(file_paths, remote_paths):
             self.upload_model_file(
                 model_id=model_id,
                 version=version,
                 local_path=filepath,
-                remote_path=str(filepath.relative_to(root_path)),
-                cloud_account=cloud_account,
+                remote_path=remote_path,
                 teamspace_id=teamspace_id,
                 progress_bar=progress_bar,  # TODO: Global progress bar
             )

lightning_sdk/api/utils.py

@@ -202,7 +202,6 @@ class _ModelFileUploader:
         model_id: str,
         version: str,
         teamspace_id: str,
-        cloud_account: str,
         file_path: str,
         remote_path: str,
         progress_bar: bool,

lightning_sdk/cli/docker.py

@@ -19,4 +19,4 @@ def api(server_filename: str, port: int = 8000, gpu: bool = False, tag: str = "l
 
 
 def _api(server_filename: str, port: int = 8000, gpu: bool = False, tag: str = "litserve-model") -> str:
-    return _LitServeDeployer().dockerize_api(server_filename=server_filename, port=port, gpu=gpu, tag=tag)
+    return _LitServeDeployer(None, None).dockerize_api(server_filename=server_filename, port=port, gpu=gpu, tag=tag)

lightning_sdk/cli/download.py

@@ -143,7 +143,15 @@ def file(path: str = "", studio: Optional[str] = None, local_path: str = ".") ->
 @click.argument("container")
 @click.option("--teamspace", default=None, help="The name of the teamspace to download the container from")
 @click.option("--tag", default="latest", show_default=True, help="The tag of the container to download.")
-def download_container(container: str, teamspace: Optional[str] = None, tag: str = "latest") -> None:
+@click.option(
+    "--cloud-account",
+    "--cloud_account",  # The UI will present the above variant, using this as a secondary to be consistent w/ models
+    default=None,
+    help="The name of the cloud account to download the Container from.",
+)
+def download_container(
+    container: str, teamspace: Optional[str] = None, tag: str = "latest", cloud_account: Optional[str] = None
+) -> None:
     """Download a docker container from a teamspace.
 
     Example:
@@ -156,7 +164,7 @@ def download_container(container: str, teamspace: Optional[str] = None, tag: str
     resolved_teamspace = menu._resolve_teamspace(teamspace)
     with console.status("Downloading container..."):
         api = LitContainerApi()
-        api.download_container(container, resolved_teamspace, tag)
+        api.download_container(container, resolved_teamspace, tag, cloud_account)
         console.print("Container downloaded successfully", style="green")
 
 

lightning_sdk/cli/serve.py

@@ -8,28 +8,35 @@ 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.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.serve import _LitServeDeployer
+from lightning_sdk.serve import _LitServeDeployer, authenticate
 
 _MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
 
-@click.group("serve")
+class _ServeGroup(click.Group):
+    def parse_args(self, ctx: click.Context, args: list) -> click.Group:
+        # Check if first arg is a file path and not a command name
+        if args and os.path.exists(args[0]) and args[0] not in self.commands:
+            # Insert the 'api' command before the file path
+            args.insert(0, "api")
+        return super().parse_args(ctx, args)
+
+
+@click.group("serve", cls=_ServeGroup)
 def serve() -> None:
     """Serve a LitServe model.
 
     Example:
-        lightning serve api server.py  # serve locally
+        lightning serve server.py  # serve locally
 
     Example:
-        lightning serve api server.py --cloud --name litserve-api  # deploy to the cloud
+        lightning serve server.py --cloud  # deploy to the cloud
 
-    You can deploy the API to the cloud by running `lightning serve api server.py --cloud`.
+    You can deploy the API to the cloud by running `lightning serve server.py --cloud`.
     This will build a docker container for the server.py script and deploy it to the Lightning AI platform.
     """
 
@@ -44,11 +51,11 @@ def serve() -> None:
     help="Generate a client for the model",
 )
 @click.option(
-    "--cloud",
+    "--local",
     is_flag=True,
     default=False,
     flag_value=True,
-    help="Deploy the model to the Lightning AI platform",
+    help="Run the model locally",
 )
 @click.option("--name", default=None, help="Name of the deployed API (e.g., 'classification-api', 'Llama-api')")
 @click.option(
@@ -109,7 +116,7 @@ def serve() -> None:
 def api(
     script_path: str,
     easy: bool,
-    cloud: bool,
+    local: bool,
     name: Optional[str],
     non_interactive: bool,
     machine: str,
@@ -128,7 +135,7 @@ def api(
     return api_impl(
         script_path=script_path,
         easy=easy,
-        cloud=cloud,
+        local=local,
         repository=name,
         non_interactive=non_interactive,
         machine=machine,
@@ -148,7 +155,7 @@ def api(
 def api_impl(
     script_path: Union[str, Path],
     easy: bool = False,
-    cloud: bool = False,
+    local: bool = False,
     repository: [str] = None,
     tag: Optional[str] = None,
     non_interactive: bool = False,
@@ -178,7 +185,7 @@ def api_impl(
         timestr = datetime.now().strftime("%b-%d-%H_%M")
         repository = f"litserve-{timestr}".lower()
 
-    if cloud:
+    if not local:
         repository = repository or "litserve-model"
         machine = Machine.from_str(machine)
         return _handle_cloud(
@@ -231,34 +238,21 @@ def _handle_cloud(
 ) -> None:
     deployment_name = os.path.basename(repository)
     tag = tag if tag else "latest"
+
+    if non_interactive:
+        console.print("[italic]non-interactive[/italic] mode enabled, skipping confirmation prompts", style="blue")
+
+    # Authenticate with LitServe affiliate
+    authenticate()
     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()
+    ls_deployer = _LitServeDeployer(name=deployment_name, teamspace=resolved_teamspace)
     path = ls_deployer.dockerize_api(script_path, port=port, gpu=not machine.is_cpu(), tag=tag, print_success=False)
-    console.clear()
-    if non_interactive:
-        console.print("[italic]non-interactive[/italic] mode enabled, skipping confirmation prompts", style="blue")
 
     console.print(f"\nPlease review the Dockerfile at [u]{path}[/u] and make sure it is correct.", style="bold")
     correct_dockerfile = True if non_interactive else Confirm.ask("Is the Dockerfile correct?", default=True)
@@ -266,10 +260,9 @@ def _handle_cloud(
         console.print("Please fix the Dockerfile and try again.", style="red")
         return
 
-    ls_deployer.authenticate()
     # list containers to create the project if it doesn't exist
     lit_cr = LitContainerApi()
-    lit_cr.list_containers(resolved_teamspace.id)
+    lit_cr.list_containers(resolved_teamspace.id, cloud_account=cloud_account)
 
     with Progress(
         SpinnerColumn(),
@@ -279,18 +272,35 @@ def _handle_cloud(
         transient=True,
     ) as progress:
         try:
-            ls_deployer.build_container(path, repository, tag, console, progress)
-            push_status = ls_deployer.push_container(repository, tag, resolved_teamspace, lit_cr, progress)
+            # Build the container
+            build_task = progress.add_task("Building Docker image", total=None)
+            for line in ls_deployer.build_container(path, repository, tag):
+                console.print(line.strip())
+                progress.update(build_task, advance=1)
+            progress.update(build_task, description="[green]Build completed![/green]", completed=1.0)
+            progress.remove_task(build_task)
+
+            # Push the container to the registry
+            console.print("\nPushing image to registry. It may take a while...", style="bold")
+            push_task = progress.add_task("Pushing to registry", total=None)
+            push_status = {}
+            for line in ls_deployer.push_container(
+                repository, tag, resolved_teamspace, lit_cr, cloud_account=cloud_account
+            ):
+                push_status = line
+                progress.update(push_task, advance=1)
+                if not ("Pushing" in line["status"] or "Waiting" in line["status"]):
+                    console.print(line["status"])
+            progress.update(push_task, description="[green]Push completed![/green]")
         except Exception as e:
             console.print(f"❌ Deployment failed: {e}", style="red")
             return
     console.print(f"\n✅ Image pushed to {repository}:{tag}")
-    console.print(f"🔗 You can access the container at: [i]{push_status.get('url')}[/i]")
-    repository = push_status.get("repository")
+    image = push_status.get("image")
 
-    deployment_status = ls_deployer._run_on_cloud(
+    deployment_status = ls_deployer.run_on_cloud(
         deployment_name=deployment_name,
-        image=repository,
+        image=image,
         teamspace=resolved_teamspace,
         metric=None,
         machine=machine,

lightning_sdk/cli/upload.py

@@ -11,11 +11,12 @@ from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from simple_term_menu import TerminalMenu
 from tqdm import tqdm
 
-from lightning_sdk.api.lit_container_api import LCRAuthFailedError, LitContainerApi
+from lightning_sdk.api.lit_container_api import DockerNotRunningError, LCRAuthFailedError, LitContainerApi
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.studios_menu import _StudiosMenu
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.models import upload_model as _upload_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
@@ -116,11 +117,28 @@ def file(path: str, studio: Optional[str] = None, remote_path: Optional[str] = N
         "If not specified, tries to infer from the environment (e.g. when run from within a Studio.)"
     ),
 )
-def upload_container(container: str, tag: str = "latest", teamspace: Optional[str] = None) -> None:
+@click.option(
+    "--cloud-account",
+    "--cloud_account",  # The UI will present the above variant, using this as a secondary to be consistent w/ models
+    default=None,
+    help="The name of the cloud account to store the Container in.",
+)
+@click.option(
+    "--platform",
+    "--platform",
+    default="linux/amd64",
+    help="This is the platform the container pulled and push to Lightning AI will run on.",
+)
+def upload_container(
+    container: str,
+    tag: str = "latest",
+    teamspace: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+    platform: Optional[str] = "linux/amd64",
+) -> None:
     """Upload a container to Lightning AI's container registry."""
     menu = _TeamspacesMenu()
     teamspace = menu._resolve_teamspace(teamspace)
-    api = LitContainerApi()
     console = Console()
     with Progress(
         SpinnerColumn(),
@@ -129,8 +147,13 @@ def upload_container(container: str, tag: str = "latest", teamspace: Optional[st
         console=console,
         transient=False,
     ) as progress:
-        push_task = progress.add_task("Pushing Docker image", total=None)
         try:
+            if platform != "linux/amd64":
+                console.print(
+                    "[yellow]Warning: The platform you selected (" + platform + ") may not deploy successfully[/yellow]"
+                )
+            api = LitContainerApi()
+            push_task = progress.add_task("Pushing Docker image", total=None)
             console.print("Authenticating with Lightning Container Registry...")
             try:
                 api.authenticate()
@@ -139,15 +162,27 @@ def upload_container(container: str, tag: str = "latest", teamspace: Optional[st
                 # let the push with retry take control of auth moving forward
                 pass
 
-            lines = api.upload_container(container, teamspace, tag)
+            lines = api.upload_container(container, teamspace, tag, cloud_account, platform)
             _print_docker_push(lines, console, progress, push_task)
+        except DockerNotRunningError as e:
+            e.print_help()
+            return
         except LCRAuthFailedError:
             console.print("Re-authenticating with Lightning Container Registry...")
             if not api.authenticate(reauth=True):
                 raise StudioCliError("Failed to authenticate with Lightning Container Registry") from None
             console.print("Authenticated with Lightning Container Registry", style="green")
-            lines = api.upload_container(container, teamspace, tag)
+            lines = api.upload_container(container, teamspace, tag, cloud_account, platform)
             _print_docker_push(lines, console, progress, push_task)
+        except Exception as e:
+            if _LIGHTNING_DEBUG:
+                print(e)
+                if e.__cause__:
+                    print(e.__cause__)
+
+            progress.update(push_task, description=f"[bold red]Error: {e!s}[/]")
+            progress.stop()
+            return
         progress.update(push_task, description="[green]Container pushed![/green]")
 
 

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -66,6 +66,8 @@ from lightning_sdk.lightning_cloud.openapi.configuration import Configuration
 from lightning_sdk.lightning_cloud.openapi.models.affiliatelinks_id_body import AffiliatelinksIdBody
 from lightning_sdk.lightning_cloud.openapi.models.agentmanagedendpoints_id_body import AgentmanagedendpointsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.agents_id_body import AgentsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.alerts_config_billing import AlertsConfigBilling
+from lightning_sdk.lightning_cloud.openapi.models.alerts_config_studios import AlertsConfigStudios
 from lightning_sdk.lightning_cloud.openapi.models.app_id_works_body import AppIdWorksBody
 from lightning_sdk.lightning_cloud.openapi.models.appinstances_id_body import AppinstancesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.approveautojoindomain_domain_body import ApproveautojoindomainDomainBody
@@ -233,6 +235,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
+from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_app_type import V1AppType
 from lightning_sdk.lightning_cloud.openapi.models.v1_append_logger_metrics_response import V1AppendLoggerMetricsResponse
@@ -272,6 +276,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_type import V1CloudSpaceArtifactEventType
 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_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
@@ -574,6 +579,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_cold_start_metrics_response import V1ListCloudSpaceColdStartMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_environment_templates_response import V1ListCloudSpaceEnvironmentTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_instances_response import V1ListCloudSpaceInstancesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_publications_response import V1ListCloudSpacePublicationsResponse
@@ -768,6 +774,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1Re
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1RefreshResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -802,6 +809,7 @@ 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_phase import V1ServerAlertPhase
 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