lightning-sdk 2025.10.8__py3-none-any.whl → 2025.10.22__py3-none-any.whl

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

@@ -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,18 +158,22 @@ 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
 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
 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 +200,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
@@ -389,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
@@ -400,6 +407,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,9 +449,12 @@ 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
+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
@@ -567,6 +579,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 +622,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 +632,10 @@ 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_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
 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 +658,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
@@ -648,6 +666,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
@@ -721,12 +740,17 @@ 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_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
@@ -757,7 +781,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_e
 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_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
@@ -869,8 +892,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response impor
 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 +959,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
@@ -990,6 +1013,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
@@ -1054,6 +1080,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

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

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -1570,43 +1570,45 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def assistants_service_get_published_managed_endpoint_by_model_id(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
-        """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata  # noqa: E501
+    def assistants_service_get_published_managed_endpoint(self, **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
+        """GetPublishedManagedEndpoint returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.assistants_service_get_published_managed_endpoint_by_model_id(id, async_req=True)
+        >>> thread = api.assistants_service_get_published_managed_endpoint(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
-        :param str id: (required)
+        :param str id:
+        :param str name:
         :return: V1ManagedEndpoint
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
         if kwargs.get('async_req'):
-            return self.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, **kwargs)  # noqa: E501
+            return self.assistants_service_get_published_managed_endpoint_with_http_info(**kwargs)  # noqa: E501
         else:
-            (data) = self.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, **kwargs)  # noqa: E501
+            (data) = self.assistants_service_get_published_managed_endpoint_with_http_info(**kwargs)  # noqa: E501
             return data
 
-    def assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
-        """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata  # noqa: E501
+    def assistants_service_get_published_managed_endpoint_with_http_info(self, **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
+        """GetPublishedManagedEndpoint returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, async_req=True)
+        >>> thread = api.assistants_service_get_published_managed_endpoint_with_http_info(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
-        :param str id: (required)
+        :param str id:
+        :param str name:
         :return: V1ManagedEndpoint
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['id']  # noqa: E501
+        all_params = ['id', 'name']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1617,22 +1619,20 @@ class AssistantsServiceApi(object):
             if key not in all_params:
                 raise TypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method assistants_service_get_published_managed_endpoint_by_model_id" % key
+                    " to method assistants_service_get_published_managed_endpoint" % key
                 )
             params[key] = val
         del params['kwargs']
-        # verify the required parameter 'id' is set
-        if ('id' not in params or
-                params['id'] is None):
-            raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_published_managed_endpoint_by_model_id`")  # noqa: E501
 
         collection_formats = {}
 
         path_params = {}
-        if 'id' in params:
-            path_params['id'] = params['id']  # noqa: E501
 
         query_params = []
+        if 'id' in params:
+            query_params.append(('id', params['id']))  # noqa: E501
+        if 'name' in params:
+            query_params.append(('name', params['name']))  # noqa: E501
 
         header_params = {}
 
@@ -1648,7 +1648,7 @@ class AssistantsServiceApi(object):
         auth_settings = []  # noqa: E501
 
         return self.api_client.call_api(
-            '/v1/agent-published-managed-model/{id}', 'GET',
+            '/v1/agent-published-managed-model', 'GET',
             path_params,
             query_params,
             header_params,