lightning-sdk 2025.9.30__py3-none-any.whl → 2025.10.14__py3-none-any.whl

lightning_sdk/__init__.py

@@ -35,6 +35,6 @@ __all__ = [
     "VM",
 ]
 
-__version__ = "2025.09.30"
+__version__ = "2025.10.14"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cloud_account_api.py

@@ -173,8 +173,6 @@ class CloudAccountApi:
                 return CloudProvider.GCP
             if cloud_account.spec.lambda_labs_v1:
                 return CloudProvider.LAMBDA_LABS
-            if cloud_account.spec.vultr_v1:
-                return CloudProvider.VULTR
             if cloud_account.spec.voltage_park_v1:
                 return CloudProvider.VOLTAGE_PARK
             if cloud_account.spec.nebius_v1:

lightning_sdk/api/studio_api.py

@@ -438,6 +438,14 @@ class StudioApi:
 
         return response.compute_config.spot
 
+    def get_public_ip(self, studio_id: str, teamspace_id: str) -> Optional[str]:
+        """Get the public IP address of the Studio."""
+        internal_status = self.get_studio_status(studio_id=studio_id, teamspace_id=teamspace_id).in_use
+        if internal_status is None:
+            return None
+
+        return internal_status.public_ip_address
+
     def _get_machines_for_cloud_account(
         self, teamspace_id: str, cloud_account_id: str, org_id: str
     ) -> List[V1ClusterAccelerator]:

lightning_sdk/base_studio.py

@@ -16,6 +16,8 @@ class BaseStudioInfo:
     name: str
     managed_id: str
     description: str
+    creator: str
+    enabled: bool
 
 
 class BaseStudio:
@@ -80,20 +82,29 @@ class BaseStudio:
             disabled=disabled,
         )
 
-    def list(self, managed: bool = True) -> List[BaseStudioInfo]:
+    def list(self, managed: bool = True, include_disabled: bool = False) -> List[BaseStudioInfo]:
         """List all base studios in the organization.
 
+        Args:
+            managed: Whether to filter for managed base studios.
+            include_disabled: Whether to include disabled base studios in the results.
+
         Returns:
-            List[V1CloudSpaceEnvironmentTemplate]: A list of base studio templates.
+            List[BaseStudioInfo]: A list of base studio templates.
         """
-        result = []
-        for template in self._base_studio_api.get_all_base_studios(self._org.id, managed).templates:
-            result.append(
-                BaseStudioInfo(
-                    id=template.id,
-                    name=template.name,
-                    managed_id=template.managed_id,
-                    description=template.description,
-                ),
+        templates = self._base_studio_api.get_all_base_studios(self._org.id, managed).templates
+
+        return [
+            BaseStudioInfo(
+                id=template.id,
+                name=template.name,
+                managed_id=template.managed_id,
+                description=template.description,
+                creator="⚡ Lightning AI"
+                if template.managed_id
+                else UserApi()._get_user_by_id(template.user_id).username,
+                enabled=not template.disabled,
             )
-        return result
+            for template in templates
+            if include_disabled or not template.disabled
+        ]

lightning_sdk/cli/base_studio/__init__.py

@@ -0,0 +1,10 @@
+"""Base Studio CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register base studio commands with the given group."""
+    from lightning_sdk.cli.base_studio.list import list_base_studios
+
+    group.add_command(list_base_studios)

lightning_sdk/cli/base_studio/list.py

@@ -0,0 +1,45 @@
+"""Base Studio list command."""
+
+import click
+from rich.table import Table
+
+from lightning_sdk.base_studio import BaseStudio
+from lightning_sdk.cli.utils.richt_print import rich_to_str
+
+
+@click.command("list")
+@click.option("--include-disabled", help="Include disabled Base Studios in the list.", is_flag=True)
+def list_base_studios(include_disabled: bool) -> None:
+    """List Base Studios in an org.
+
+    Example:
+        lightning base-studio list
+
+    """
+    return list_impl(include_disabled=include_disabled)
+
+
+def list_impl(include_disabled: bool) -> None:
+    base_studio_cls = BaseStudio()
+    base_studios = base_studio_cls.list(include_disabled=include_disabled) + base_studio_cls.list(
+        managed=False, include_disabled=include_disabled
+    )
+
+    table = Table(
+        pad_edge=True,
+    )
+
+    table.add_column("Name")
+    table.add_column("Description")
+    table.add_column("Creator")
+    table.add_column("Enabled")
+
+    for base_studio in base_studios:
+        table.add_row(
+            base_studio.name.lower().replace(" ", "-"),
+            base_studio.description or "",
+            base_studio.creator,
+            "Yes" if base_studio.enabled else "No",
+        )
+
+    click.echo(rich_to_str(table), color=True)

lightning_sdk/cli/entrypoint.py

@@ -17,6 +17,7 @@ from lightning_sdk.api.studio_api import _cloud_url
 
 # Import legacy groups directly from groups.py
 from lightning_sdk.cli.groups import (
+    base_studio,
     config,
     # job,
     # mmt,
@@ -85,6 +86,7 @@ main_cli.add_command(config)
 # main_cli.add_command(mmt)
 main_cli.add_command(studio)
 main_cli.add_command(vm)
+main_cli.add_command(base_studio)
 if os.environ.get("LIGHTNING_EXPERIMENTAL_CLI_ONLY", "0") != "1":
     #### LEGACY COMMANDS ####
     # these commands are currently supported for backwards compatibility, but will potentially be removed in the future.

lightning_sdk/cli/groups.py

@@ -2,6 +2,7 @@
 
 import click
 
+from lightning_sdk.cli.base_studio import register_commands as register_base_studio_commands
 from lightning_sdk.cli.config import register_commands as register_config_commands
 from lightning_sdk.cli.job import register_commands as register_job_commands
 from lightning_sdk.cli.mmt import register_commands as register_mmt_commands
@@ -34,9 +35,15 @@ def vm() -> None:
     """Manage Lightning AI VMs."""
 
 
+@click.group(name="base-studio")
+def base_studio() -> None:
+    """Manage Lightning AI Base Studios."""
+
+
 # Register config commands with the main config group
 register_job_commands(job)
 register_mmt_commands(mmt)
 register_studio_commands(studio)
 register_config_commands(config)
 register_vm_commands(vm)
+register_base_studio_commands(base_studio)

lightning_sdk/cli/studio/connect.py

@@ -2,20 +2,96 @@
 
 import subprocess
 import sys
-from typing import Optional
+from typing import Dict, Optional, Set
 
 import click
 
+from lightning_sdk.base_studio import BaseStudio
 from lightning_sdk.cli.utils.richt_print import studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_studio_to_config, save_teamspace_to_config
-from lightning_sdk.cli.utils.ssh_connection import download_ssh_keys
+from lightning_sdk.cli.utils.ssh_connection import configure_ssh_internal
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
-from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
-from lightning_sdk.machine import CloudProvider
+from lightning_sdk.machine import CloudProvider, Machine
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.names import random_unique_name
 
+DEFAULT_MACHINE = "CPU"
+
+
+def _split_gpus_spec(gpus: str) -> tuple[str, int]:
+    machine_name, machine_val = gpus.split(":", 1)
+    machine_name = machine_name.strip()
+    machine_val = machine_val.strip()
+
+    if not machine_val.isdigit() or int(machine_val) <= 0:
+        raise ValueError(f"Invalid GPU count '{machine_val}'. Must be a positive integer.")
+
+    machine_num = int(machine_val)
+    return machine_name, machine_num
+
+
+def _construct_available_gpus(machine_options: Dict[str, str]) -> Set[str]:
+    # returns available gpus:count
+    available_gpus = set()
+    for v in machine_options.values():
+        if "_X_" in v:
+            gpu_type_num = v.replace("_X_", ":")
+            available_gpus.add(gpu_type_num)
+        else:
+            available_gpus.add(v)
+    return available_gpus
+
+
+def _get_machine_from_gpus(gpus: str) -> Machine:
+    machine_name = gpus
+    machine_num = 1
+
+    if ":" in gpus:
+        machine_name, machine_num = _split_gpus_spec(gpus)
+
+    machine_options = {
+        m.name.lower(): m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli
+    }
+
+    if machine_num == 1:
+        # e.g. gpus=L4 or gpus=L4:1
+        gpu_key = machine_name.lower()
+        try:
+            return machine_options[gpu_key]
+        except KeyError:
+            available = ", ".join(_construct_available_gpus(machine_options))
+            raise ValueError(f"Invalid GPU type '{machine_name}'. Available options: {available}") from None
+
+    # Else: e.g. gpus=L4:4
+    gpu_key = f"{machine_name.lower()}_x_{machine_num}"
+    try:
+        return machine_options[gpu_key]
+    except KeyError:
+        available = ", ".join(_construct_available_gpus(machine_options))
+        raise ValueError(f"Invalid GPU configuration '{gpus}'. Available options: {available}") from None
+
+
+def _get_base_studio_id(studio_type: Optional[str]) -> Optional[str]:
+    base_studios = BaseStudio()
+    base_studios = base_studios.list()
+    template_id = None
+
+    if base_studios and len(base_studios):
+        # if not specified by user, use the first existing template studio
+        template_id = base_studios[0].id
+        # else, try to match the provided studio_type to base studio name
+        if studio_type:
+            normalized_studio_type = studio_type.lower().replace(" ", "-")
+            match = next(
+                (s for s in base_studios if s.name.lower().replace(" ", "-") == normalized_studio_type),
+                None,
+            )
+            if match:
+                template_id = match.id
+
+    return template_id
+
 
 @click.command("connect")
 @click.argument("name", required=False)
@@ -30,13 +106,32 @@ from lightning_sdk.utils.names import random_unique_name
     help="The cloud account to create the studio on. Defaults to teamspace default.",
     type=click.STRING,
 )
-@click.option("--gpus", help="The number of GPUs to start the studio on. ", type=click.INT)
+@click.option(
+    "--machine",
+    help="The machine type to start the studio on. Defaults to CPU-4",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli),
+)
+@click.option(
+    "--gpus",
+    help="The number and type of GPUs to start the studio on (format: TYPE:COUNT, e.g. L4:4)",
+    type=click.STRING,
+)
+@click.option(
+    "--studio-type",
+    help="The base studio template name to use for creating the studio. "
+    "Must be lowercase and hyphenated (use '-' instead of spaces). "
+    "Run 'lightning base-studio list' to see all available templates. "
+    "Defaults to the first available template.",
+    type=click.STRING,
+)
 def connect_studio(
     name: Optional[str] = None,
     teamspace: Optional[str] = None,
     cloud_provider: Optional[str] = None,
     cloud_account: Optional[str] = None,
-    gpus: Optional[int] = None,
+    machine: Optional[str] = None,
+    gpus: Optional[str] = None,
+    studio_type: Optional[str] = None,
 ) -> None:
     """Connect to a Studio.
 
@@ -53,6 +148,9 @@ def connect_studio(
 
     name = name or random_unique_name()
 
+    # check for available base studios
+    template_id = _get_base_studio_id(studio_type)
+
     try:
         studio = Studio(
             name=name,
@@ -60,35 +158,32 @@ def connect_studio(
             create_ok=True,
             cloud_provider=cloud_provider,
             cloud_account=cloud_account,
+            studio_type=template_id,
         )
     except (RuntimeError, ValueError, ApiException):
         raise ValueError(f"Could not create Studio: '{name}'") from None
 
     click.echo(f"Connecting to Studio '{studio_name_link(studio)}' ...")
 
-    machine = "CPU"
     Studio.show_progress = True
-    if gpus:
-        # TODO: handle something like gpus=4:L4
-        pass
+
+    if machine and gpus:
+        raise click.UsageError("Options --machine and --gpu are mutually exclusive. Provide only one.")
+    elif gpus:
+        machine = _get_machine_from_gpus(gpus.strip())
+    elif not machine:
+        machine = DEFAULT_MACHINE
 
     save_studio_to_config(studio)
-    studio.start(machine, interruptible=True)
+    # by default, interruptible is False
+    studio.start(machine=machine, interruptible=False)
 
-    ssh_private_key_path = _configure_ssh_internal()
+    ssh_private_key_path = configure_ssh_internal()
 
+    ssh_option = "-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o LogLevel=ERROR"
     try:
-        ssh_command = (
-            f"ssh -i {ssh_private_key_path} -o UserKnownHostsFile=/dev/null s_{studio._studio.id}@ssh.lightning.ai"
-        )
+        ssh_command = f"ssh -i {ssh_private_key_path} {ssh_option} s_{studio._studio.id}@ssh.lightning.ai"
         subprocess.run(ssh_command.split())
     except Exception as ex:
         print(f"Failed to establish SSH connection: {ex}")
         sys.exit(1)
-
-
-def _configure_ssh_internal() -> str:
-    """Internal function to configure SSH without Click decorators."""
-    auth = Auth()
-    auth.authenticate()
-    return download_ssh_keys(auth.api_key, force_download=False)

lightning_sdk/cli/studio/ssh.py

@@ -6,10 +6,9 @@ from typing import List, Optional
 import click
 
 from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
-from lightning_sdk.cli.utils.ssh_connection import download_ssh_keys
+from lightning_sdk.cli.utils.ssh_connection import configure_ssh_internal
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
-from lightning_sdk.lightning_cloud.login import Auth
 
 
 @click.command("ssh")
@@ -39,9 +38,7 @@ def ssh_studio(name: Optional[str] = None, teamspace: Optional[str] = None, opti
 
 
 def ssh_impl(name: Optional[str], teamspace: Optional[str], option: Optional[List[str]], vm: bool) -> None:
-    auth = Auth()
-    auth.authenticate()
-    ssh_private_key_path = download_ssh_keys(auth.api_key, force_download=False)
+    ssh_private_key_path = configure_ssh_internal()
 
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
@@ -59,7 +56,7 @@ def ssh_impl(name: Optional[str], teamspace: Optional[str], option: Optional[Lis
         subprocess.run(ssh_command.split())
     except Exception:
         # redownload the keys to be sure they are up to date
-        download_ssh_keys(auth.api_key, force_download=True)
+        ssh_private_key_path = configure_ssh_internal(force_download=True)
         try:
             subprocess.run(ssh_command.split())
         except Exception:

lightning_sdk/cli/utils/ssh_connection.py

@@ -4,9 +4,17 @@ import uuid
 from pathlib import Path
 from typing import Optional
 
+from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH
 
 
+def configure_ssh_internal(force_download: bool = False) -> str:
+    """Internal function to configure SSH without Click decorators."""
+    auth = Auth()
+    auth.authenticate()
+    return download_ssh_keys(auth.api_key, force_download=force_download)
+
+
 def download_ssh_keys(
     api_key: str | None,
     force_download: bool = False,

lightning_sdk/job/job.py

@@ -268,6 +268,11 @@ class Job(_BaseJob):
         """The machine type the job is running on."""
         return self._internal_job.machine
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """The public IP address of the machine the job is running on."""
+        return self._internal_job.public_ip
+
     @property
     def artifact_path(self) -> Optional[str]:
         """Path to the artifacts created by the job within the distributed teamspace filesystem."""

lightning_sdk/job/v1.py

@@ -181,6 +181,14 @@ class _JobV1(_BaseJob):
         """Get the machine the job is running on."""
         return self.work.machine
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Get the public IP of the machine the job is running on."""
+        try:
+            return self._job.status.ip_address
+        except AttributeError:
+            return None
+
     @property
     def name(self) -> str:
         """The name of the job."""

lightning_sdk/job/v2.py

@@ -173,6 +173,14 @@ class _JobV2(_BaseJob):
             _get_org_id(self.teamspace),
         )
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Get the public IP of the machine the job is running on."""
+        try:
+            return self._job.public_ip_address
+        except AttributeError:
+            return None
+
     @property
     def artifact_path(self) -> Optional[str]:
         """The path to the artifacts of the job within the distributed teamspace filesystem."""

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -56,6 +56,7 @@ from lightning_sdk.lightning_cloud.openapi.api.product_license_service_api impor
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.sdk_command_history_service_api import SDKCommandHistoryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.ssh_public_key_service_api import SSHPublicKeyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.schedules_service_api import SchedulesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.secret_service_api import SecretServiceApi
@@ -157,6 +158,8 @@ from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploa
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdVisibilityBody2
+from lightning_sdk.lightning_cloud.openapi.models.incident_id_messages_body import IncidentIdMessagesBody
+from lightning_sdk.lightning_cloud.openapi.models.incidents_id_body import IncidentsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.job_id_reportroutingtelemetry_body import JobIdReportroutingtelemetryBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
@@ -169,6 +172,7 @@ from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import Litpag
 from lightning_sdk.lightning_cloud.openapi.models.litregistry_lit_repo_name_body import LitregistryLitRepoNameBody
 from lightning_sdk.lightning_cloud.openapi.models.loggermetrics_id_body import LoggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.messages_id_body import MessagesIdBody
+from lightning_sdk.lightning_cloud.openapi.models.messages_message_id_body import MessagesMessageIdBody
 from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifacts_body import MetricsStreamIdLoggerartifactsBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
@@ -195,6 +199,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_clusters_body impor
 from lightning_sdk.lightning_cloud.openapi.models.project_id_datasets_body import ProjectIdDatasetsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_endpoints_body import ProjectIdEndpointsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_incidents_body import ProjectIdIncidentsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_kubernetestemplates_body import ProjectIdKubernetestemplatesBody
@@ -400,6 +405,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
 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_sdk_command_history_request import V1CreateSDKCommandHistoryRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_sdk_command_history_response import V1CreateSDKCommandHistoryResponse
 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
@@ -440,6 +447,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_release_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_response import V1DeleteDeploymentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response import V1DeleteEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_git_credentials_response import V1DeleteGitCredentialsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_incident_message_response import V1DeleteIncidentMessageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_incident_response import V1DeleteIncidentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_template_response import V1DeleteKubernetesTemplateResponse
@@ -567,6 +576,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_st
 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_required_balance_status_response import V1GetCloudSpaceRequiredBalanceStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_transfer_estimate_response import V1GetCloudSpaceTransferEstimateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_health_response import V1GetClusterHealthResponse
@@ -609,6 +619,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1Gi
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
 from lightning_sdk.lightning_cloud.openapi.models.v1_group_node_metrics import V1GroupNodeMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_group_pod_metrics import V1GroupPodMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_request import V1GuestLoginRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_response import V1GuestLoginResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_guest_user import V1GuestUser
@@ -618,7 +629,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_ids_logger_metrics import V1IdsLoggerMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_state import V1ImageState
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident import V1Incident
 from lightning_sdk.lightning_cloud.openapi.models.v1_incident_event import V1IncidentEvent
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_message import V1IncidentMessage
 from lightning_sdk.lightning_cloud.openapi.models.v1_incident_severity import V1IncidentSeverity
 from lightning_sdk.lightning_cloud.openapi.models.v1_incident_type import V1IncidentType
 from lightning_sdk.lightning_cloud.openapi.models.v1_index import V1Index
@@ -641,6 +654,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTimi
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
 from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
 from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_indexes import V1K8sIncidentIndexes
+from lightning_sdk.lightning_cloud.openapi.models.v1_kai_scheduler_queue_metrics import V1KaiSchedulerQueueMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
@@ -721,11 +735,15 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_lightningapps_response import V1ListGalleryLightningappsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_git_credentials_response import V1ListGitCredentialsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_group_pod_metrics_response import V1ListGroupPodMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_incident_events_response import V1ListIncidentEventsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_incident_messages_response import V1ListIncidentMessagesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_incidents_response import V1ListIncidentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_files_response import V1ListJobFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_resources_response import V1ListJobResourcesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_response import V1ListJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations_response import V1ListJoinableOrganizationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kai_scheduler_queues_metrics_response import V1ListKaiSchedulerQueuesMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_artifacts_response import V1ListLightningappInstanceArtifactsResponse
@@ -938,6 +956,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_rule_condition import V1Rul
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_effect import V1RuleEffect
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_resource import V1RuleResource
 from lightning_sdk.lightning_cloud.openapi.models.v1_s3_folder_data_connection import V1S3FolderDataConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_sdk_command_history_severity import V1SDKCommandHistorySeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_sdk_command_history_type import V1SDKCommandHistoryType
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -37,6 +37,7 @@ from lightning_sdk.lightning_cloud.openapi.api.product_license_service_api impor
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.sdk_command_history_service_api import SDKCommandHistoryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.ssh_public_key_service_api import SSHPublicKeyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.schedules_service_api import SchedulesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.secret_service_api import SecretServiceApi