lightning-sdk 2025.12.17__py3-none-any.whl → 2026.1.27__py3-none-any.whl

lightning_sdk/cli/utils/filesystem.py

@@ -0,0 +1,103 @@
+from typing import Optional, TypedDict
+
+from lightning_sdk.cli.utils.owner_selection import OwnerMenu
+from lightning_sdk.cli.utils.studio_selection import StudiosMenu
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
+from lightning_sdk.studio import Studio
+from lightning_sdk.teamspace import Teamspace
+
+
+class PathResult(TypedDict):
+    owner: Optional[str]
+    teamspace: Optional[str]
+    studio: Optional[str]
+    destination: Optional[str]
+
+
+def parse_studio_path(studio_path: str) -> PathResult:
+    path_string = studio_path.removeprefix("lit://")
+    if not path_string:
+        raise ValueError("Studio path cannot be empty after prefix")
+
+    result: PathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
+
+    if "/studios/" in path_string:
+        prefix_part, suffix_part = path_string.split("/studios/", 1)
+
+        # org and teamspace
+        if prefix_part:
+            org_ts_components = prefix_part.split("/")
+            if len(org_ts_components) == 2:
+                result["owner"], result["teamspace"] = org_ts_components
+            elif len(org_ts_components) == 1:
+                result["teamspace"] = org_ts_components[0]
+            else:
+                raise ValueError(f"Invalid format: '{prefix_part}'")
+
+        # studio and destination
+        path_parts = suffix_part.split("/")
+
+    else:
+        # studio and destination
+        path_parts = path_string.split("/")
+
+    if not path_parts:
+        raise ValueError("Invalid: Missing studio name.")
+
+    if len(path_parts) == 1:
+        raise ValueError(
+            "Invalid: Invalid studio path. To refer to the studio root, add a trailing '/' (e.g., 'lit://<owner>/<my-teamspace>/studios/<my-studio>/')"
+        )
+
+    result["studio"] = path_parts[0]
+    result["destination"] = "/".join(path_parts[1:])
+
+    return result
+
+
+def parse_teamspace_uploads_path(teamspace_path: str) -> PathResult:
+    path_string = teamspace_path.removeprefix("lit://")
+    if not path_string:
+        raise ValueError("Teamspace path cannot be empty after prefix")
+
+    result: PathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
+
+    if "/uploads/" in path_string:
+        prefix_part, suffix_part = path_string.split("/uploads/", 1)
+
+        # org and teamspace
+        if prefix_part:
+            org_ts_components = prefix_part.split("/")
+            if len(org_ts_components) == 2:
+                result["owner"], result["teamspace"] = org_ts_components
+            elif len(org_ts_components) == 1:
+                result["teamspace"] = org_ts_components[0]
+            else:
+                raise ValueError(f"Invalid format: '{prefix_part}'")
+
+        # studio and destination
+        path_parts = suffix_part.split("/")
+
+    else:
+        raise ValueError("Invalid teamspace uploads path: missing '/uploads/' segment")
+
+    if not path_parts:
+        raise ValueError("Invalid: Missing teamspace name.")
+
+    result["destination"] = suffix_part
+
+    return result
+
+
+def resolve_teamspace(teamspace: Optional[str], owner: Optional[str]) -> Teamspace:
+    owner_menu = OwnerMenu()
+    resolved_owner = owner_menu(owner=owner)
+
+    teamspace_menu = TeamspacesMenu(resolved_owner)
+    return teamspace_menu(teamspace=teamspace)
+
+
+def resolve_studio(studio_name: Optional[str], teamspace: Optional[str], owner: Optional[str]) -> Studio:
+    resolved_teamspace = resolve_teamspace(teamspace, owner)
+    studio_menu = StudiosMenu(resolved_teamspace)
+    return studio_menu(studio=studio_name)

lightning_sdk/cli/utils/logging.py

@@ -30,10 +30,11 @@ def _log_command(message: str = "", duration: int = 0, error: Optional[str] = No
     body = V1CreateSDKCommandHistoryRequest(
         command=original_command,
         duration=duration,
-        message=f"VERSION: {__version__} | {message}",
+        message=f"{message}",
         project_id=None,
         severity=V1SDKCommandHistorySeverity.INFO,
         type=V1SDKCommandHistoryType.CLI,
+        version=__version__,
     )
 
     if error:

lightning_sdk/cli/utils/teamspace_selection.py

@@ -51,6 +51,11 @@ class TeamspacesMenu:
         return TerminalMenu(possible_teamspaces, title=title, clear_menu_on_exit=True)
 
     def _get_possible_teamspaces(self, user: User) -> Dict[str, str]:
+        if self._owner is None:
+            from lightning_sdk.cli.utils.owner_selection import OwnerMenu
+
+            menu = OwnerMenu()
+            self._owner = menu()
         user_api = user._user_api
 
         memberships = user_api._get_all_teamspace_memberships(

lightning_sdk/exceptions.py

@@ -1,2 +1,33 @@
+from typing import Any
+
+import click
+
+
 class OutOfCapacityError(RuntimeError):
     """Raised when the requested machine is not available in the selected cloud account."""
+
+
+class NotSupportedError(RuntimeError):
+    """Raised when the requested machine is not supported in the selected cloud account."""
+
+
+class DeprecatedError(RuntimeError):
+    """Raised when a deprecated feature is used."""
+
+
+class DeprecatedCommand(click.Command):
+    """Custom exception for deprecated commands."""
+
+    def __init__(self, *args: Any, message: str, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self.deprecated_message = message
+
+    def get_help(self, ctx: click.Context) -> str:
+        if self.deprecated_message:
+            raise DeprecatedError(self.deprecated_message)
+        return super().get_help(ctx)
+
+    def invoke(self, ctx: click.Context) -> Any:
+        if self.deprecated_message:
+            raise DeprecatedError(self.deprecated_message)
+        return super().invoke(ctx)

lightning_sdk/job/base.py

@@ -201,7 +201,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 raise RuntimeError(
                     "image and studio are mutually exclusive as both define the environment to run the job in"
                 )
-            if cloud_account is None and in_studio():
+            if cloud_account is None and cloud_provider is None and in_studio():
                 try:
                     with skip_studio_setup():
                         resolve_studio = Studio(teamspace=teamspace, user=user, org=org)

lightning_sdk/k8s_cluster.py

@@ -1,8 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-from typing import List, Optional
-
-import pandas as pd
+from typing import Any, Dict, List, Optional
 
 from lightning_sdk.api.k8s_api import K8sClusterApi
 from lightning_sdk.api.utils import to_iso_z
@@ -34,25 +32,26 @@ class K8sCluster:
         self._cloud_account = cloud_account
         self._k8s_cluster = K8sClusterApi(cloud_account=cloud_account)
 
-    def _convert_to_k8s_usage_response(self, df: pd.DataFrame) -> K8sUsageResponse:
-        """Converts a DataFrame to K8sUsageResponse.
+    def _convert_to_k8s_usage_response(self, data: List[Dict[str, Any]]) -> K8sUsageResponse:
+        """Converts a list of usage data to K8sUsageResponse.
 
         Args:
-            df (pd.DataFrame): The DataFrame containing GPU usage data.
+            data (List[Dict[str, Any]]): The list of dictionaries containing GPU usage data.
 
         Returns:
             K8sUsageResponse: The converted response containing hourly usage and total usage.
         """
-        if df.empty:
+        if not data:
             return K8sUsageResponse(hours=[], total_usage=0.0)
-        # Convert each row of the DataFrame to HourlyUsage
+
+        # Convert each row to HourlyUsage
         hourly_usage_list: List[HourlyUsage] = [
             HourlyUsage(time=row["hour"], available_gpus=row["num_gpus"], billed_gpus=row["billed_gpus"])
-            for _, row in df.iterrows()
+            for row in data
         ]
 
         # Calculate total usage (sum of billed GPUs)
-        total_usage = df["billed_gpus"].sum()
+        total_usage = sum(row["billed_gpus"] for row in data)
 
         # Create and return the K8sUsageResponse
         return K8sUsageResponse(hours=hourly_usage_list, total_usage=total_usage)

lightning_sdk/lightning_cloud/__version__.py

@@ -1 +1 @@
-__version__ = "0.5.70"
+__version__ = "0.6.0"

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -32,6 +32,7 @@ from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloudy_service_api import CloudyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.container_registry_service_api import ContainerRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.deployment_templates_service_api import DeploymentTemplatesServiceApi
@@ -42,7 +43,6 @@ from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api impor
 from lightning_sdk.lightning_cloud.openapi.api.incidents_service_api import IncidentsServiceApi
 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.kubernetes_virtual_machine_service_api import KubernetesVirtualMachineServiceApi
 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
@@ -68,6 +68,7 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.virtual_machine_service_api import VirtualMachineServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi
 # import ApiClient
 from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
@@ -133,22 +134,25 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_clo
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_version_publication_body import CloudSpaceServiceUpdateCloudSpaceVersionPublicationBody
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_visibility_body import CloudSpaceServiceUpdateCloudSpaceVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudy_service_update_user_cloudy_settings_body import CloudyServiceUpdateUserCloudySettingsBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_add_container_registry_body import ClusterServiceAddContainerRegistryBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_capacity_reservation_body import ClusterServiceCreateClusterCapacityReservationBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_proxy_body import ClusterServiceCreateClusterProxyBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_usage_restriction_body import ClusterServiceCreateClusterUsageRestrictionBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_machine_body import ClusterServiceCreateMachineBody
+from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_org_cluster_capacity_reservation_body import ClusterServiceCreateOrgClusterCapacityReservationBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project_cluster_body import ClusterServiceCreateProjectClusterBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_server_alert_body import ClusterServiceCreateServerAlertBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_interrupt_server_body import ClusterServiceInterruptServerBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_purchase_capacity_block_body import ClusterServicePurchaseCapacityBlockBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_refresh_container_registry_credentials_body import ClusterServiceRefreshContainerRegistryCredentialsBody
+from lightning_sdk.lightning_cloud.openapi.models.cluster_service_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_server_check_in_body import ClusterServiceServerCheckInBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_sleep_server_body import ClusterServiceSleepServerBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_body import ClusterServiceUpdateClusterBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_usage_restriction_body import ClusterServiceUpdateClusterUsageRestrictionBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_project_cluster_body import ClusterServiceUpdateProjectClusterBody
-from lightning_sdk.lightning_cloud.openapi.models.cluster_service_validate_container_registry_body import ClusterServiceValidateContainerRegistryBody
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_config_ecr import ContainerRegistryConfigECR
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_provider import ContainerRegistryProvider
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_create_container_registry_body import ContainerRegistryServiceCreateContainerRegistryBody
+from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_refresh_container_registry_credentials_body import ContainerRegistryServiceRefreshContainerRegistryCredentialsBody
 from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_service_create_data_connection_body import DataConnectionServiceCreateDataConnectionBody
@@ -176,6 +180,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment_body import JobsServiceCreateDeploymentBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_job_body import JobsServiceCreateJobBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_multi_machine_job_body import JobsServiceCreateMultiMachineJobBody
+from lightning_sdk.lightning_cloud.openapi.models.jobs_service_duplicate_deployment_body import JobsServiceDuplicateDeploymentBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_deployment_routing_telemetry_body import JobsServiceReportDeploymentRoutingTelemetryBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_job_system_metrics_body import JobsServiceReportJobSystemMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_logs_activity_body import JobsServiceReportLogsActivityBody
@@ -195,8 +200,6 @@ from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_create_ku
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_render_kubernetes_template_body import K8SClusterServiceRenderKubernetesTemplateBody
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_report_k8_s_cluster_metrics_body import K8SClusterServiceReportK8SClusterMetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.k8_s_cluster_service_update_kubernetes_template_body import K8SClusterServiceUpdateKubernetesTemplateBody
-from lightning_sdk.lightning_cloud.openapi.models.kubernetes_virtual_machine_service_create_kubernetes_virtual_machine_body import KubernetesVirtualMachineServiceCreateKubernetesVirtualMachineBody
-from lightning_sdk.lightning_cloud.openapi.models.kubernetes_virtual_machine_service_update_kubernetes_virtual_machine_body import KubernetesVirtualMachineServiceUpdateKubernetesVirtualMachineBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_update_lightningapp_instance_body import LightningappInstanceServiceUpdateLightningappInstanceBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_update_lightningapp_instance_release_body import LightningappInstanceServiceUpdateLightningappInstanceReleaseBody
 from lightning_sdk.lightning_cloud.openapi.models.lightningapp_instance_service_upload_lightningapp_instance_artifact_body import LightningappInstanceServiceUploadLightningappInstanceArtifactBody
@@ -216,9 +219,11 @@ from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit
 from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit_dataset_version_body import LitDatasetServiceUpdateLitDatasetVersionBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit_dataset_visibility_body import LitDatasetServiceUpdateLitDatasetVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_append_logger_metrics_body import LitLoggerServiceAppendLoggerMetricsBody
+from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_lit_logger_media_body import LitLoggerServiceCreateLitLoggerMediaBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_logger_artifact_body import LitLoggerServiceCreateLoggerArtifactBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_metrics_stream_body import LitLoggerServiceCreateMetricsStreamBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_delete_metrics_stream_body import LitLoggerServiceDeleteMetricsStreamBody
+from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_lit_logger_media_body import LitLoggerServiceUpdateLitLoggerMediaBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_metrics_stream_body import LitLoggerServiceUpdateMetricsStreamBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_metrics_stream_visibility_body import LitLoggerServiceUpdateMetricsStreamVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_shared_metrics_stream_body import LitLoggerServiceUpdateSharedMetricsStreamBody
@@ -240,6 +245,7 @@ from lightning_sdk.lightning_cloud.openapi.models.organizations_service_auto_joi
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_body import OrganizationsServiceCreateOrgMembershipBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_role_binding_body import OrganizationsServiceCreateOrgMembershipRoleBindingBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_role_body import OrganizationsServiceCreateOrgRoleBody
+from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_org_role_body import OrganizationsServiceUpdateOrgRoleBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_body import OrganizationsServiceUpdateOrganizationBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_credits_auto_replenish_body import OrganizationsServiceUpdateOrganizationCreditsAutoReplenishBody
 from lightning_sdk.lightning_cloud.openapi.models.organizations_service_validate_auto_join_domain_body import OrganizationsServiceValidateAutoJoinDomainBody
@@ -301,7 +307,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import
 from lightning_sdk.lightning_cloud.openapi.models.v1_abort_storage_transfer_response import V1AbortStorageTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_accelerator_quota_info import V1AcceleratorQuotaInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_ack_user_storage_violation_response import V1AckUserStorageViolationResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_add_container_registry_response import V1AddContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_add_job_timing_response import V1AddJobTimingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_affiliate_link import V1AffiliateLink
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_complete_part_response import V1AgentCompletePartResponse
@@ -310,7 +315,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_aggregated_pod_metrics import V1AggregatedPodMetrics
-from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
 from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
@@ -343,6 +347,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_running_cloud_space_instance_transfer_response import V1CancelRunningCloudSpaceInstanceTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
 from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
+from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_reservation_used_by import V1CapacityReservationUsedBy
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_response import V1CheckClusterNameAvailabilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_external_service_status_response import V1CheckExternalServiceStatusResponse
@@ -420,9 +425,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_rea
 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_container_registry import V1ContainerRegistry
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_info import V1ContainerRegistryInfo
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_integration import V1ContainerRegistryIntegration
-from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_status import V1ContainerRegistryStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_config import V1ContainerRegistryConfig
+from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_scopes import V1ContainerRegistryScopes
 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
@@ -441,6 +445,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_k
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_response import V1CreateClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_request import V1CreateClusterRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response import V1CreateClusterResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_container_registry_response import V1CreateContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
 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
@@ -448,6 +453,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_incident_request imp
 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_logger_media_response import V1CreateLitLoggerMediaResponse
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_machine_response import V1CreateMachineResponse
@@ -455,6 +461,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_model_metrics_response import V1CreateModelMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_org_cluster_capacity_reservation_response import V1CreateOrgClusterCapacityReservationResponse
 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
@@ -468,6 +475,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_request import V1CreateSubscriptionCheckoutSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_response import V1CreateSubscriptionCheckoutSessionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_user_secret_request import V1CreateUserSecretRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_cudo_direct_v1 import V1CudoDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_daily_model_metrics import V1DailyModelMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_daily_usage import V1DailyUsage
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1DataConnection
@@ -506,7 +514,6 @@ 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_kubernetes_virtual_machine_response import V1DeleteKubernetesVirtualMachineResponse
 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
@@ -514,6 +521,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instanc
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_response import V1DeleteLitDatasetResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_version_response import V1DeleteLitDatasetVersionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_logger_media_response import V1DeleteLitLoggerMediaResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_registry_repository_image_artifact_version_by_digest_response import V1DeleteLitRegistryRepositoryImageArtifactVersionByDigestResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
@@ -524,6 +532,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_multi_machine_job_response import V1DeleteMultiMachineJobResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_cluster_capacity_reservation_response import V1DeleteOrgClusterCapacityReservationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_response import V1DeleteOrgMembershipResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
@@ -543,6 +552,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response impo
 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_virtual_machine_response import V1DeleteVirtualMachineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state import V1DependencyCacheState
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
@@ -572,6 +582,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_paramet
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_type import V1DeploymentTemplateParameterType
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_summary import V1DeploymentTemplateSummary
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_type import V1DeploymentTemplateType
+from lightning_sdk.lightning_cloud.openapi.models.v1_describe_org_cluster_capacity_reservation_response import V1DescribeOrgClusterCapacityReservationResponse
 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_drive import V1Drive
@@ -582,9 +593,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config import V1ECRRegistryConfig
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config_input import V1ECRRegistryConfigInput
-from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_details import V1ECRRegistryDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
@@ -648,6 +656,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_tele
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_response import V1GetJobSystemMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_kubernetes_pod_logs_response import V1GetKubernetesPodLogsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_kubernetes_pod_response import V1GetKubernetesPodResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_latest_model_metrics_response import V1GetLatestModelMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
@@ -657,7 +667,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_files_url_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_page_response import V1GetLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_logger_metrics_response import V1GetLoggerMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in_cloud_space_response import V1GetLongRunningCommandInCloudSpaceResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_get_machine_response import V1GetMachineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_market_pricing_response import V1GetMarketPricingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
@@ -718,6 +727,8 @@ 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_k8s_incident_setting import V1K8sIncidentSetting
+from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_type import V1K8sIncidentType
 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
@@ -726,13 +737,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_settings_
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod import V1KubernetesPod
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod_logs_page import V1KubernetesPodLogsPage
 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_kubernetes_virtual_machine import V1KubernetesVirtualMachine
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_config import V1KubevirtConfig
-from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_provider_configuration import V1KubevirtProviderConfiguration
-from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_vm_configuration import V1KubevirtVmConfiguration
-from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_vm_resources import V1KubevirtVmResources
 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
@@ -805,7 +813,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_respon
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_response import V1ListFilesystemMMTsResponse
+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
@@ -824,8 +832,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_sort_order import
 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_pods_response import V1ListKubernetesPodsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_pods_sort_order import V1ListKubernetesPodsSortOrder
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_virtual_machines_response import V1ListKubernetesVirtualMachinesResponse
 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
@@ -835,6 +843,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_dataset_versions_response import V1ListLitDatasetVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response import V1ListLitDatasetsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_logger_media_response import V1ListLitLoggerMediaResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
@@ -879,9 +888,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_storage_transfers_response import V1ListStorageTransfersResponse
 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_list_virtual_machines_response import V1ListVirtualMachinesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_file import V1LitDatasetFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_version_archive import V1LitDatasetVersionArchive
+from lightning_sdk.lightning_cloud.openapi.models.v1_lit_logger_media import V1LitLoggerMedia
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
@@ -906,6 +917,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1M
 from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model import V1ManagedModel
 from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model_abilities import V1ManagedModelAbilities
 from lightning_sdk.lightning_cloud.openapi.models.v1_market_price import V1MarketPrice
+from lightning_sdk.lightning_cloud.openapi.models.v1_media_type import V1MediaType
 from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Membership
 from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
 from lightning_sdk.lightning_cloud.openapi.models.v1_message_action import V1MessageAction
@@ -918,6 +930,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics import V1Metrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_stream import V1MetricsStream
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1MetricsTags
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
+from lightning_sdk.lightning_cloud.openapi.models.v1_mithril_direct_v1 import V1MithrilDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_metrics import V1ModelMetrics
@@ -1012,12 +1025,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
 from lightning_sdk.lightning_cloud.openapi.models.v1_render_kubernetes_template_response import V1RenderKubernetesTemplateResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_idle_state_response import V1ReportCloudSpaceInstanceIdleStateResponse
 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_deployment_routing_telemetry_response import V1ReportDeploymentRoutingTelemetryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_job_system_metrics_response import V1ReportJobSystemMetricsResponse
 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_machine_system_metrics_response import V1ReportMachineSystemMetricsResponse
 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
@@ -1098,6 +1113,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1Sys
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
 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_tenant_credentials import V1TenantCredentials
 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
@@ -1165,7 +1181,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_requested_flow_compute
 from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_response import V1UserSLURMJobActionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_validate_container_registry_response import V1ValidateContainerRegistryResponse
 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
@@ -1175,11 +1190,16 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_storage_transfer_response import V1ValidateStorageTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_response import V1VerifyVerificationResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_virtual_machine import V1VirtualMachine
+from lightning_sdk.lightning_cloud.openapi.models.v1_vm_configuration import V1VmConfiguration
+from lightning_sdk.lightning_cloud.openapi.models.v1_vm_provider_configuration import V1VmProviderConfiguration
 from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
 from lightning_sdk.lightning_cloud.openapi.models.v1_volume_state import V1VolumeState
 from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
+from lightning_sdk.lightning_cloud.openapi.models.virtual_machine_service_create_virtual_machine_body import VirtualMachineServiceCreateVirtualMachineBody
+from lightning_sdk.lightning_cloud.openapi.models.virtual_machine_service_update_virtual_machine_body import VirtualMachineServiceUpdateVirtualMachineBody
 from lightning_sdk.lightning_cloud.openapi.models.volume_service_update_volume_body import VolumeServiceUpdateVolumeBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec as Gridv1ImageSpec

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -13,6 +13,7 @@ from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloudy_service_api import CloudyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.container_registry_service_api import ContainerRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.deployment_templates_service_api import DeploymentTemplatesServiceApi
@@ -23,7 +24,6 @@ from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api impor
 from lightning_sdk.lightning_cloud.openapi.api.incidents_service_api import IncidentsServiceApi
 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.kubernetes_virtual_machine_service_api import KubernetesVirtualMachineServiceApi
 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
@@ -49,4 +49,5 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.virtual_machine_service_api import VirtualMachineServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi