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

lightning_sdk/job/base.py

@@ -90,6 +90,7 @@ class _BaseJob(ABC):
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -134,6 +135,8 @@ class _BaseJob(ABC):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         from lightning_sdk.lightning_cloud.openapi.rest import ApiException
         from lightning_sdk.studio import Studio
@@ -231,6 +234,7 @@ class _BaseJob(ABC):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
 
     @abstractmethod
@@ -251,6 +255,7 @@ class _BaseJob(ABC):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Submit a new job to the Lightning AI platform.
 
@@ -290,6 +295,8 @@ class _BaseJob(ABC):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
 
     @abstractmethod

lightning_sdk/job/job.py

@@ -97,6 +97,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -140,6 +141,8 @@ class Job(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+        reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         ret_val = super().run(
             name=name,
@@ -162,6 +165,7 @@ class Job(_BaseJob):
             path_mappings=path_mappings,
             max_runtime=max_runtime,
             cluster=cluster,
+            reuse_snapshot=reuse_snapshot,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -186,6 +190,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Submit a new job to the Lightning AI platform.
 
@@ -224,6 +229,8 @@ class Job(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         self._job = self._internal_job._submit(
             machine=machine,
@@ -241,6 +248,7 @@ class Job(_BaseJob):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         return self
 

lightning_sdk/job/v1.py

@@ -54,6 +54,7 @@ class _JobV1(_BaseJob):
         cloud_provider: Optional[str] = None,
         interruptible: bool = False,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Start a new async workload from your studio.
 
@@ -89,6 +90,7 @@ class _JobV1(_BaseJob):
             cluster=cluster,
             path_mappings=None,
             max_runtime=None,
+            reuse_snapshot=reuse_snapshot,
         )
 
     def _submit(
@@ -108,6 +110,7 @@ class _JobV1(_BaseJob):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_JobV1":
         """Submit a job to run on a machine.
 

lightning_sdk/job/v2.py

@@ -53,6 +53,7 @@ class _JobV2(_BaseJob):
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
+        reuse_snapshot: bool = True,
     ) -> "_JobV2":
         """Submit a new job to the Lightning AI platform.
 
@@ -87,6 +88,8 @@ class _JobV2(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -129,6 +132,7 @@ class _JobV2(_BaseJob):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         self._job = submitted
         self._name = submitted.name

lightning_sdk/lightning_cloud/login.py

@@ -5,7 +5,7 @@ import logging
 import os
 import pathlib
 from dataclasses import dataclass
-from typing import Optional
+from typing import Optional, Literal
 from urllib.parse import urlencode
 
 import webbrowser
@@ -22,14 +22,21 @@ from lightning_sdk.lightning_cloud.openapi.api import \
     AuthServiceApi
 from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_request import \
     V1GuestLoginRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_login_request import \
+    V1TokenLoginRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import \
+    V1RefreshRequest
 
 logger = logging.getLogger(__name__)
 
+# Authentication override types
+AuthOverride = Literal["auth_token", "api_key", "guest"]
 
 class Keys(Enum):
     # USERNAME = "LIGHTNING_USERNAME"
     USER_ID = "LIGHTNING_USER_ID"
     API_KEY = "LIGHTNING_API_KEY"
+    AUTH_TOKEN = "LIGHTNING_AUTH_TOKEN"
 
     @property
     def suffix(self):
@@ -41,15 +48,18 @@ class Auth:
     # username: Optional[str] = None
     user_id: Optional[str] = None
     api_key: Optional[str] = None
+    auth_token: Optional[str] = None
 
     secrets_file = pathlib.Path(env.LIGHTNING_CREDENTIAL_PATH)
 
     def __post_init__(self):
         for key in Keys:
-            setattr(self, key.suffix, os.environ.get(key.value, None))
+            # Only set from environment if not already set
+            if getattr(self, key.suffix) is None:
+                setattr(self, key.suffix, os.environ.get(key.value, None))
 
         # used by authenticate method
-        self._with_env_var = bool(self.api_key)
+        self._with_env_var = bool(self.api_key or self.auth_token)
 
     def load(self) -> bool:
         """Load credentials from disk and update properties with credentials.
@@ -71,6 +81,7 @@ class Auth:
              token: str = "",
              user_id: str = "",
              api_key: str = "",
+             auth_token: str = "",
              # username: str = "",
         ) -> None:
         """save credentials to disk."""
@@ -81,6 +92,7 @@ class Auth:
                     # f"{Keys.USERNAME.suffix}": username,
                     f"{Keys.USER_ID.suffix}": user_id,
                     f"{Keys.API_KEY.suffix}": api_key,
+                    f"{Keys.AUTH_TOKEN.suffix}": auth_token,
                 },
                 f,
             )
@@ -88,6 +100,7 @@ class Auth:
         # self.username = username
         self.user_id = user_id
         self.api_key = api_key
+        self.auth_token = auth_token
         logger.debug("credentials saved successfully")
 
     @classmethod
@@ -99,15 +112,78 @@ class Auth:
             os.environ.pop(key.value, None)
         logger.debug("credentials removed successfully")
 
-    @property
-    def auth_header(self) -> Optional[str]:
-        """authentication header used by lightning-cloud client."""
-        if self.api_key:
+    def get_auth_header(self, override: Optional[AuthOverride] = None) -> Optional[str]:
+        """Get authentication header with optional override.
+        
+        By default, uses the automatic priority selection (auth_token > api_key).
+        You can override this for specific cases where you need a different auth method.
+        
+        Parameters
+        ----------
+        override : AuthOverride, optional
+            Override the default authentication method:
+            - "auth_token": Force use of JWT auth token (Bearer)
+            - "api_key": Force use of API key (Basic)
+            - "guest": Force use of guest credentials (Basic)
+            - None: Use automatic selection (default)
+            
+        Returns
+        -------
+        Optional[str]
+            The authorization header for the specified method.
+            
+        Raises
+        ------
+        ValueError
+            If the specified override is not available or invalid.
+        """
+        if override == "auth_token":
+            if not self.auth_token:
+                raise ValueError(
+                    "Auth token override requested but no JWT token available. "
+                    "Please use token_login() method first."
+                )
+            return f"Bearer {self.auth_token}"
+        
+        elif override == "api_key":
+            if not self.api_key or not self.user_id:
+                raise ValueError(
+                    "API key override requested but no API key or user ID available. "
+                    "Please set LIGHTNING_API_KEY and LIGHTNING_USER_ID environment variables "
+                    "or use authenticate() method."
+                )
+            token = f"{self.user_id}:{self.api_key}"
+            return f"Basic {base64.b64encode(token.encode('ascii')).decode('ascii')}"  # noqa E501
+        
+        elif override == "guest":
+            if not self.api_key or not self.user_id:
+                raise ValueError(
+                    "Guest override requested but no guest credentials available. "
+                    "Please call guest_login() method first."
+                )
             token = f"{self.user_id}:{self.api_key}"
             return f"Basic {base64.b64encode(token.encode('ascii')).decode('ascii')}"  # noqa E501
-        raise AttributeError(
-            "Authentication Failed, no authentication header available. "
-            "This is most likely a bug in the LightningCloud Framework")
+        
+        elif override is None:
+            # Use the original automatic selection logic (default behavior)
+            if self.auth_token:
+                return f"Bearer {self.auth_token}"
+            elif self.api_key:
+                token = f"{self.user_id}:{self.api_key}"
+                return f"Basic {base64.b64encode(token.encode('ascii')).decode('ascii')}"  # noqa E501
+            else:
+                raise ValueError(
+                    "No authentication credentials available. Please authenticate first using "
+                    "token_login(), guest_login(), or authenticate() methods."
+                )
+        
+        else:
+            raise ValueError(f"Invalid authentication override: {override}")
+
+    @property
+    def auth_header(self) -> Optional[str]:
+        """authentication header used by lightning-cloud client (automatic selection)."""
+        return self.get_auth_header()
 
     def _run_server(self) -> None:
         """start a server to complete authentication."""
@@ -130,6 +206,8 @@ class Auth:
             self._run_server()
             return self.auth_header
 
+        elif self.auth_token:
+            return self.auth_header
         elif self.user_id and self.api_key:
             return self.auth_header
 
@@ -192,6 +270,178 @@ class Auth:
 
         return self.auth_header
 
+    def token_login(self, token_key: str, save_token: bool = True) -> Optional[str]:
+        """Performs token-based authentication.
+        
+        This method sends a request to the token login endpoint to authenticate
+        using an auth token key, optionally saves the returned JWT token, and 
+        returns the authorization header.
+        
+        Parameters
+        ----------
+        token_key : str
+            The auth token key to use for authentication.
+        save_token : bool, optional
+            Whether to save the JWT token for future use. Defaults to True.
+            If False, the token is only used for the current session.
+            
+        Returns
+        -------
+        Optional[str]
+            The authorization header to use for subsequent requests.
+            
+        Raises
+        ------
+        RuntimeError
+            If the token login request fails.
+        ValueError
+            If the response from the server is invalid.
+        """
+        config = Configuration()
+        config.host = env.LIGHTNING_CLOUD_URL
+        api_client = ApiClient(configuration=config)
+        auth_api = AuthServiceApi(api_client)
+
+        logger.debug(f"Attempting token login to {config.host}")
+
+        try:
+            body = V1TokenLoginRequest(token_key=token_key)
+            response = auth_api.auth_service_token_login(body)
+
+        except requests.RequestException as e:
+            logger.error(f"Token login request failed: {e}")
+            raise RuntimeError(
+                "Failed to connect to the token login endpoint. "
+                "Please check your network connection and the server status."
+            ) from e
+
+        # Extract the JWT token from the response
+        jwt_token = getattr(response, "token", None)
+
+        if not jwt_token:
+            logger.error(
+                f"No token received from token login response: {response}"
+            )
+            raise ValueError(
+                "The token login response did not contain a valid JWT token."
+            )
+
+        # Set the JWT token in memory
+        self.auth_token = jwt_token
+        
+        # Optionally save the JWT token to disk
+        if save_token:
+            self.save(auth_token=jwt_token)
+            logger.info("Successfully authenticated using auth token and saved to disk.")
+        else:
+            logger.info("Successfully authenticated using auth token (not saved to disk).")
+
+        return self.auth_header
+
+    def refresh_token(self, duration: int = 43200) -> Optional[str]:
+        """Refreshes the current JWT token.
+        
+        This method sends a request to the refresh endpoint to get a new JWT token
+        with the specified duration, saves the new token, and returns the updated
+        authorization header.
+        
+        Parameters
+        ----------
+        duration : int, optional
+            Duration in seconds for the new token. Can range from 900 seconds (15 minutes)
+            up to a maximum of 129,600 seconds (36 hours), with a default of 43,200 seconds (12 hours).
+            
+        Returns
+        -------
+        Optional[str]
+            The updated authorization header with the new JWT token.
+            
+        Raises
+        ------
+        RuntimeError
+            If the refresh request fails.
+        ValueError
+            If no valid JWT token is available to refresh, or if the response is invalid.
+        """
+        if not self.auth_token:
+            raise ValueError(
+                "No JWT token available to refresh. Please authenticate first using "
+                "token_login() or authenticate() methods."
+            )
+        
+        config = Configuration()
+        config.host = env.LIGHTNING_CLOUD_URL
+        # Set the current auth token as the authorization header
+        config.api_key_prefix['Authorization'] = 'Bearer'
+        config.api_key['Authorization'] = self.auth_token
+        
+        api_client = ApiClient(configuration=config)
+        auth_api = AuthServiceApi(api_client)
+
+        logger.debug(f"Attempting to refresh JWT token with duration {duration} seconds")
+
+        try:
+            body = V1RefreshRequest(duration=str(duration))
+            response = auth_api.auth_service_refresh(body)
+
+        except requests.RequestException as e:
+            logger.error(f"Token refresh request failed: {e}")
+            raise RuntimeError(
+                "Failed to connect to the refresh endpoint. "
+                "Please check your network connection and the server status."
+            ) from e
+
+        # Extract the new JWT token from the response
+        new_jwt_token = getattr(response, "token", None)
+
+        if not new_jwt_token:
+            logger.error(
+                f"No token received from refresh response: {response}"
+            )
+            raise ValueError(
+                "The refresh response did not contain a valid JWT token."
+            )
+
+        # Save the new JWT token
+        self.save(auth_token=new_jwt_token)
+        logger.info("Successfully refreshed JWT token.")
+
+        return self.auth_header
+
+    def create_api_client(self, override: Optional[AuthOverride] = None) -> 'ApiClient':
+        """Create an API client with optional authentication override.
+        
+        This is a convenience method for creating API clients that use a specific
+        authentication method instead of the default automatic selection.
+        
+        Parameters
+        ----------
+        override : AuthOverride, optional
+            Override the default authentication method for this API client.
+            See get_auth_header() for available options.
+            
+        Returns
+        -------
+        ApiClient
+            Configured API client with the specified authentication method.
+        """
+        from lightning_sdk.lightning_cloud.openapi import ApiClient, Configuration
+        
+        config = Configuration()
+        config.host = env.LIGHTNING_CLOUD_URL
+        
+        # Get the auth header for the specified override
+        auth_header = self.get_auth_header(override)
+        
+        # Create the API client
+        client = ApiClient(configuration=config)
+        
+        # Set the Authorization header directly in default_headers
+        if auth_header:
+            client.set_default_header('Authorization', auth_header)
+        
+        return client
+
 
 class AuthServer:
 

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -166,6 +166,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBod
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
 from lightning_sdk.lightning_cloud.openapi.models.kubernetestemplates_id_body import KubernetestemplatesIdBody
+from lightning_sdk.lightning_cloud.openapi.models.license_key_validate_body import LicenseKeyValidateBody
 from lightning_sdk.lightning_cloud.openapi.models.litdatasets_dataset_id_body import LitdatasetsDatasetIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
@@ -394,6 +395,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request i
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_git_credentials_request import V1CreateGitCredentialsRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V1CreateJobRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_license_request import V1CreateLicenseRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
@@ -452,6 +454,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_incident_response im
 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_license_response import V1DeleteLicenseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_response import V1DeleteLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
@@ -546,6 +549,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space impo
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_job import V1FilesystemJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_mmt import V1FilesystemMMT
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_metric import V1FilesystemMetric
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_metrics import V1FilesystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import V1FilesystemSlurmJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
@@ -630,6 +634,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_ids_logger_metrics import V
 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_detail import V1IncidentDetail
 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
@@ -662,6 +667,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template import V1KubernetesTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_license import V1License
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_basic_auth import V1LightningBasicAuth
@@ -730,6 +736,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_space
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_metrics_response import V1ListFilesystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
@@ -745,6 +752,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_response import V
 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_license_response import V1ListLicenseResponse
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_events_response import V1ListLightningappInstanceEventsResponse
@@ -774,8 +782,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_roles_response imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_encryption_keys_response import V1ListOrganizationClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organizations_response import V1ListOrganizationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_pipelines_response import V1ListPipelinesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_platform_notifications_response import V1ListPlatformNotificationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_pod_metrics_response import V1ListPodMetricsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_product_licenses_response import V1ListProductLicensesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_profiler_captures_response import V1ListProfilerCapturesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_artifacts_response import V1ListProjectArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_cluster_accelerators_response import V1ListProjectClusterAcceleratorsResponse
@@ -882,13 +890,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1PipelineTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
+from lightning_sdk.lightning_cloud.openapi.models.v1_platform_notification import V1PlatformNotification
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_pod_metrics import V1PodMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_presigned_url import V1PresignedUrl
-from lightning_sdk.lightning_cloud.openapi.models.v1_product_license import V1ProductLicense
-from lightning_sdk.lightning_cloud.openapi.models.v1_product_license_check_response import V1ProductLicenseCheckResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_capture import V1ProfilerCapture
 from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_enabled_response import V1ProfilerEnabledResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_project import V1Project
@@ -938,6 +945,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_required_balance_reason import V1RequiredBalanceReason
 from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
+from lightning_sdk.lightning_cloud.openapi.models.v1_reset_api_key_request import V1ResetAPIKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_reset_api_key_response import V1ResetAPIKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
@@ -1010,6 +1019,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_login_request import V1TokenLoginRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_login_response import V1TokenLoginResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_owner_type import V1TokenOwnerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
 from lightning_sdk.lightning_cloud.openapi.models.v1_tool import V1Tool
 from lightning_sdk.lightning_cloud.openapi.models.v1_tool_call import V1ToolCall
@@ -1074,6 +1086,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_validate_license_response import V1ValidateLicenseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_request import V1ValidateManagedEndpointRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_response import V1ValidateManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse