lightning-sdk 2025.8.6rc0__py3-none-any.whl → 2025.8.6rc2__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.06rc0"
+__version__ = "2025.08.06rc2"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/llm_api.py

@@ -4,9 +4,15 @@ import datetime
 import json
 import os
 import threading
+import time
 import warnings
 from typing import Any, AsyncGenerator, Dict, Generator, List, Optional, Union
+from urllib.parse import urlencode
 
+from rich.prompt import Confirm
+
+from lightning_sdk.lightning_cloud import env
+from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi.models import (
     StreamResultOfV1ConversationResponseChunk,
     V1ConversationResponseChunk,
@@ -15,6 +21,48 @@ from lightning_sdk.lightning_cloud.openapi.models import (
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
+LITAI_CODE = os.environ.get("LITAI_CODE", "x334uv8t7v")
+
+
+class _AuthServer(AuthServer):
+    def __init__(self, model: str, *args: Any, **kwargs: Any) -> None:
+        self._model = model
+        super().__init__(*args, **kwargs)
+
+    def get_auth_url(self, port: int) -> str:
+        redirect_uri = f"http://localhost:{port}/login-complete"
+        params = urlencode({"redirectTo": redirect_uri, "okbhrt": LITAI_CODE, "litaimodel": self._model})
+        return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
+
+
+class _AuthLitAI(Auth):
+    def __init__(self, model: str, shall_confirm: bool = False) -> None:
+        super().__init__()
+        self._model = model
+        self._shall_confirm = shall_confirm
+
+    def _run_server(self) -> None:
+        if self._shall_confirm:
+            proceed = Confirm.ask(
+                "[bold yellow]LitAI needs to authenticate with Lightning AI.[/bold yellow]\n"
+                "This will open a browser window for login.\n"
+                "Do you want to continue?",
+                default=True,
+            )
+            if not proceed:
+                raise RuntimeError(
+                    "Login cancelled. Please login at https://lightning.ai/sign-up. Or run `lightning login` to login."
+                )
+        print("Opening browser for authentication...")
+        print("Please come back to the terminal after logging in.")
+        time.sleep(1)
+        _AuthServer(self._model).login_with_browser(self)
+
+
+def authenticate(model: str, shall_confirm: bool = True) -> None:
+    auth = _AuthLitAI(model, shall_confirm)
+    auth.authenticate()
+
 
 class LLMApi:
     def __init__(self) -> None:

lightning_sdk/lightning_cloud/login.py

@@ -43,6 +43,7 @@ class Auth:
         for key in Keys:
             setattr(self, key.suffix, os.environ.get(key.value, None))
 
+        # used by authenticate method
         self._with_env_var = bool(self.api_key)
 
     def load(self) -> bool:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -40,6 +40,7 @@ from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import Ex
 from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api import GitCredentialsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.k8_s_cluster_service_api import K8SClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
@@ -95,12 +96,12 @@ from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_visibility_body
 from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
+from lightning_sdk.lightning_cloud.openapi.models.cluster_id_metrics_body import ClusterIdMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_proxies_body import ClusterIdProxiesBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_slurmusers_body import ClusterIdSlurmusersBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_usagerestrictions_body import ClusterIdUsagerestrictionsBody
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body1 import ClustersIdBody1
-from lightning_sdk.lightning_cloud.openapi.models.command_argument_command_argument_type import CommandArgumentCommandArgumentType
 from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import ConversationsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body1 import ConversationsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.create import Create
@@ -123,13 +124,11 @@ from lightning_sdk.lightning_cloud.openapi.models.externalv1_cluster import Exte
 from lightning_sdk.lightning_cloud.openapi.models.externalv1_lightningapp_instance import Externalv1LightningappInstance
 from lightning_sdk.lightning_cloud.openapi.models.externalv1_lightningwork import Externalv1Lightningwork
 from lightning_sdk.lightning_cloud.openapi.models.externalv1_user_status import Externalv1UserStatus
-from lightning_sdk.lightning_cloud.openapi.models.fileendpoints_id_body import FileendpointsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.get_endpoint_request_endpoint_ref_type import GetEndpointRequestEndpointRefType
 from lightning_sdk.lightning_cloud.openapi.models.id_action_body import IdActionBody
 from lightning_sdk.lightning_cloud.openapi.models.id_artifacts_body import IdArtifactsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCodeconfigBody
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
-from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_contactowner_body import IdContactownerBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
@@ -150,10 +149,8 @@ from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdRelea
 from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
-from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
 from lightning_sdk.lightning_cloud.openapi.models.id_transfer_body import IdTransferBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
-from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
 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
@@ -191,7 +188,6 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_cloudspaces_body im
 from lightning_sdk.lightning_cloud.openapi.models.project_id_clusters_body import ProjectIdClustersBody
 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_fileendpoints_body import ProjectIdFileendpointsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
 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
@@ -204,7 +200,6 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_pipelines_body impo
 from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body import ProjectIdSchedulesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
-from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import ProjectIdStorageBody
 from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
@@ -218,9 +213,7 @@ from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import Secrets
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
 from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
-from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
 from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
-from lightning_sdk.lightning_cloud.openapi.models.serviceexecution_id_body import ServiceexecutionIdBody
 from lightning_sdk.lightning_cloud.openapi.models.setup import Setup
 from lightning_sdk.lightning_cloud.openapi.models.slurm_jobs_body import SlurmJobsBody
 from lightning_sdk.lightning_cloud.openapi.models.snowflake_export_body import SnowflakeExportBody
@@ -322,6 +315,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_specialized_view import V1CloudSpaceSpecializedView
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_transfer_metadata import V1CloudSpaceTransferMetadata
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
@@ -345,21 +339,21 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_upload import V1Clu
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_usage_restriction import V1ClusterUsageRestriction
 from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1CollabAction
 from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
-from lightning_sdk.lightning_cloud.openapi.models.v1_command_argument import V1CommandArgument
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_agent_multipart_upload import V1CompleteAgentMultipartUpload
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_multi_part_upload_response import V1CompleteLitDatasetMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_upload_response import V1CompleteLitDatasetUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_model_upload_response import V1CompleteModelUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_multi_part_upload_response import V1CompleteMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_onboarding_response import V1CompleteOnboardingResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_complete_running_cloud_space_instance_transfer_response import V1CompleteRunningCloudSpaceInstanceTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload import V1CompleteUpload
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_project_artifact_response import V1CompleteUploadProjectArtifactResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_service_execution_artifact_response import V1CompleteUploadServiceExecutionArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_temporary_artifact_request import V1CompleteUploadTemporaryArtifactRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_completed_part import V1CompletedPart
 from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1ComputeConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_container_metrics import V1ContainerMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
 from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
 from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -429,7 +423,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_alerting_
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_release_response import V1DeleteDeploymentReleaseResponse
 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_file_endpoint_response import V1DeleteFileEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_git_credentials_response import V1DeleteGitCredentialsResponse
 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
@@ -464,7 +457,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_role_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_ssh_public_key_response import V1DeleteSSHPublicKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_schedule_response import V1DeleteScheduleResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response import V1DeleteSecretResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_delete_service_execution_response import V1DeleteServiceExecutionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_studio_job_response import V1DeleteStudioJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_response import V1DeleteUserSLURMJobResponse
@@ -499,7 +491,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_summary
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_type import V1DeploymentTemplateType
 from lightning_sdk.lightning_cloud.openapi.models.v1_download_job_logs_response import V1DownloadJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_download_lightningapp_instance_logs_response import V1DownloadLightningappInstanceLogsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_download_service_execution_artifact_response import V1DownloadServiceExecutionArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive import V1Drive
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_spec import V1DriveSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_state import V1DriveState
@@ -522,13 +513,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster import V1ExternalCluster
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster_spec import V1ExternalClusterSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
-from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_filestore_data_connection import V1FilestoreDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_app import V1FilesystemApp
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space import V1FilesystemCloudSpace
 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_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
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
@@ -582,7 +573,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_storage_metadata_response import V1GetProjectStorageMetadataResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_get_service_execution_status_response import V1GetServiceExecutionStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_settings_response import V1GetSettingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_snowflake_query_response import V1GetSnowflakeQueryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_usage_details_response import V1GetUsageDetailsResponse
@@ -666,6 +656,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reser
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_response import V1ListClusterProxiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_metrics_response import V1ListContainerMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversations_response import V1ListConversationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifacts_response import V1ListDataConnectionArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connections_response import V1ListDataConnectionsResponse
@@ -680,7 +671,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_templates_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployments_response import V1ListDeploymentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_endpoints_response import V1ListEndpointsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_experiments_response import V1ListExperimentsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_file_endpoints_response import V1ListFileEndpointsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_response import V1ListFilesystemAppsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
@@ -713,6 +703,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_node_file_system_metrics_response import V1ListNodeFileSystemMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_node_metrics_response import V1ListNodeMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_notification_dialogs_response import V1ListNotificationDialogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
@@ -721,6 +713,7 @@ 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_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
@@ -739,8 +732,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_execution_lightningapp_instances_response import V1ListServiceExecutionLightningappInstancesResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_executions_response import V1ListServiceExecutionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
@@ -793,6 +784,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics im
 from lightning_sdk.lightning_cloud.openapi.models.v1_nebius_direct_v1 import V1NebiusDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_network_config import V1NetworkConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_new_feature import V1NewFeature
+from lightning_sdk.lightning_cloud.openapi.models.v1_node_metrics import V1NodeMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_notification_preference import V1NotificationPreference
 from lightning_sdk.lightning_cloud.openapi.models.v1_notification_preferences_request import V1NotificationPreferencesRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_notification_type import V1NotificationType
@@ -822,6 +814,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
 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
@@ -860,8 +853,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1Regio
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_request_cloud_space_access_response import V1RequestCloudSpaceAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
@@ -900,8 +895,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_phase import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
-from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_health import V1ServiceHealth
 from lightning_sdk.lightning_cloud.openapi.models.v1_setup_confirmed_ssh_public_key_response import V1SetupConfirmedSSHPublicKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_response import V1SetupDataConnectionResponse
@@ -947,6 +940,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_request
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_response import V1UpdateAgentStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_billing_subscription_request import V1UpdateBillingSubscriptionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_collab_response import V1UpdateCloudSpaceCollabResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_instance_config_request import V1UpdateCloudSpaceInstanceConfigRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publication_response import V1UpdateCloudSpacePublicationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_visibility_response import V1UpdateCloudSpaceVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
@@ -976,8 +970,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_project_artifact_parts_response import V1UploadProjectArtifactPartsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_project_artifact_response import V1UploadProjectArtifactResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_parts_response import V1UploadServiceExecutionArtifactPartsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_response import V1UploadServiceExecutionArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_settings_response import V1UploadSettingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_temporary_artifact_request import V1UploadTemporaryArtifactRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_upstream_cloud_space import V1UpstreamCloudSpace

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -21,6 +21,7 @@ from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import Ex
 from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api import GitCredentialsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.k8_s_cluster_service_api import K8SClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -1579,6 +1579,7 @@ class AssistantsServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
+        :param str org_id:
         :return: V1ListPublishedManagedEndpointsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1599,12 +1600,13 @@ class AssistantsServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
+        :param str org_id:
         :return: V1ListPublishedManagedEndpointsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = []  # noqa: E501
+        all_params = ['org_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1625,6 +1627,8 @@ class AssistantsServiceApi(object):
         path_params = {}
 
         query_params = []
+        if 'org_id' in params:
+            query_params.append(('orgId', params['org_id']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -231,6 +231,107 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_complete_running_cloud_space_instance_transfer(self, project_id: 'str', id: 'str', **kwargs) -> 'V1CompleteRunningCloudSpaceInstanceTransferResponse':  # noqa: E501
+        """cloud_space_service_complete_running_cloud_space_instance_transfer  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_complete_running_cloud_space_instance_transfer(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1CompleteRunningCloudSpaceInstanceTransferResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_complete_running_cloud_space_instance_transfer_with_http_info(project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_complete_running_cloud_space_instance_transfer_with_http_info(project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_complete_running_cloud_space_instance_transfer_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1CompleteRunningCloudSpaceInstanceTransferResponse':  # noqa: E501
+        """cloud_space_service_complete_running_cloud_space_instance_transfer  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_complete_running_cloud_space_instance_transfer_with_http_info(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1CompleteRunningCloudSpaceInstanceTransferResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method cloud_space_service_complete_running_cloud_space_instance_transfer" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_complete_running_cloud_space_instance_transfer`")  # noqa: E501
+        # 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 `cloud_space_service_complete_running_cloud_space_instance_transfer`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/{id}/complete-transfer', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1CompleteRunningCloudSpaceInstanceTransferResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def cloud_space_service_create_cloud_space(self, body: 'ProjectIdCloudspacesBody', project_id: 'str', **kwargs) -> 'V1CloudSpace':  # noqa: E501
         """cloud_space_service_create_cloud_space  # noqa: E501
 
@@ -6660,6 +6761,7 @@ class CloudSpaceServiceApi(object):
         :param str org_id:
         :param str tag:
         :param str internal_name:
+        :param list[str] cloud_space_ids:
         :return: V1ListPublishedCloudSpacesResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -6688,12 +6790,13 @@ class CloudSpaceServiceApi(object):
         :param str org_id:
         :param str tag:
         :param str internal_name:
+        :param list[str] cloud_space_ids:
         :return: V1ListPublishedCloudSpacesResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['username', 'search_query', 'page_token', 'limit', 'sort_by', 'org_id', 'tag', 'internal_name']  # noqa: E501
+        all_params = ['username', 'search_query', 'page_token', 'limit', 'sort_by', 'org_id', 'tag', 'internal_name', 'cloud_space_ids']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -6730,6 +6833,9 @@ class CloudSpaceServiceApi(object):
             query_params.append(('tag', params['tag']))  # noqa: E501
         if 'internal_name' in params:
             query_params.append(('internalName', params['internal_name']))  # noqa: E501
+        if 'cloud_space_ids' in params:
+            query_params.append(('cloudSpaceIds', params['cloud_space_ids']))  # noqa: E501
+            collection_formats['cloudSpaceIds'] = 'multi'  # noqa: E501
 
         header_params = {}
 
@@ -7309,6 +7415,127 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_request_cloud_space_access(self, body: 'object', user_name: 'str', project_name: 'str', cloudspace_name: 'str', **kwargs) -> 'V1RequestCloudSpaceAccessResponse':  # noqa: E501
+        """RequestCloudSpaceAccess  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_request_cloud_space_access(body, user_name, project_name, cloudspace_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param object body: (required)
+        :param str user_name: (required)
+        :param str project_name: (required)
+        :param str cloudspace_name: (required)
+        :return: V1RequestCloudSpaceAccessResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_request_cloud_space_access_with_http_info(body, user_name, project_name, cloudspace_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_request_cloud_space_access_with_http_info(body, user_name, project_name, cloudspace_name, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_request_cloud_space_access_with_http_info(self, body: 'object', user_name: 'str', project_name: 'str', cloudspace_name: 'str', **kwargs) -> 'V1RequestCloudSpaceAccessResponse':  # noqa: E501
+        """RequestCloudSpaceAccess  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_request_cloud_space_access_with_http_info(body, user_name, project_name, cloudspace_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param object body: (required)
+        :param str user_name: (required)
+        :param str project_name: (required)
+        :param str cloudspace_name: (required)
+        :return: V1RequestCloudSpaceAccessResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'user_name', 'project_name', 'cloudspace_name']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method cloud_space_service_request_cloud_space_access" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `cloud_space_service_request_cloud_space_access`")  # noqa: E501
+        # verify the required parameter 'user_name' is set
+        if ('user_name' not in params or
+                params['user_name'] is None):
+            raise ValueError("Missing the required parameter `user_name` when calling `cloud_space_service_request_cloud_space_access`")  # noqa: E501
+        # verify the required parameter 'project_name' is set
+        if ('project_name' not in params or
+                params['project_name'] is None):
+            raise ValueError("Missing the required parameter `project_name` when calling `cloud_space_service_request_cloud_space_access`")  # noqa: E501
+        # verify the required parameter 'cloudspace_name' is set
+        if ('cloudspace_name' not in params or
+                params['cloudspace_name'] is None):
+            raise ValueError("Missing the required parameter `cloudspace_name` when calling `cloud_space_service_request_cloud_space_access`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'user_name' in params:
+            path_params['userName'] = params['user_name']  # noqa: E501
+        if 'project_name' in params:
+            path_params['projectName'] = params['project_name']  # noqa: E501
+        if 'cloudspace_name' in params:
+            path_params['cloudspaceName'] = params['cloudspace_name']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/users/{userName}/projects/{projectName}/cloudspaces/{cloudspaceName}/request-access', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1RequestCloudSpaceAccessResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def cloud_space_service_restart_cloud_space_instance(self, project_id: 'str', id: 'str', **kwargs) -> 'V1RestartCloudSpaceInstanceResponse':  # noqa: E501
         """cloud_space_service_restart_cloud_space_instance  # noqa: E501