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

lightning_sdk/__version__.py

@@ -1,3 +1,3 @@
 """Version information for lightning_sdk."""
 
-__version__ = "2025.12.16"
+__version__ = "2025.12.17"

lightning_sdk/api/org_api.py

@@ -1,4 +1,5 @@
 from lightning_sdk.lightning_cloud.openapi import (
+    V1CreateProjectRequest,
     V1Organization,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
@@ -20,3 +21,9 @@ class OrgApi:
     def _get_org_by_id(self, org_id: str) -> V1Organization:
         """Gets the organization from the given ID."""
         return self._client.organizations_service_get_organization(id=org_id)
+
+    def create_teamspace(self, name: str, organization_id: str) -> None:
+        """Creates a new teamspace."""
+        self._client.projects_service_create_project(
+            body=V1CreateProjectRequest(name=name, organization_id=organization_id, display_name=name)
+        )

lightning_sdk/api/studio_api.py

@@ -962,9 +962,9 @@ class StudioApi:
             interruptible=interruptible,
         )
 
-    def start_new_port(self, teamspace_id: str, studio_id: str, name: str, port: int, auto_start: bool = False) -> str:
+    def add_port(self, teamspace_id: str, studio_id: str, name: str, port: int, auto_start: bool = False) -> V1Endpoint:
         """Starts a new port to the given Studio."""
-        endpoint = self._client.endpoint_service_create_endpoint(
+        return self._client.endpoint_service_create_endpoint(
             project_id=teamspace_id,
             body=EndpointServiceCreateEndpointBody(
                 name=name,
@@ -974,7 +974,32 @@ class StudioApi:
                 ),
             ),
         )
-        return endpoint.urls[0]
+
+    def get_port_url(
+        self, teamspace_id: str, studio_id: str, port: Optional[int] = None, name: Optional[str] = None
+    ) -> str:
+        if port is None and name is None:
+            raise ValueError("Either 'port' or 'name' must be provided")
+
+        endpoints = self.list_ports(teamspace_id=teamspace_id, studio_id=studio_id)
+
+        for endpoint in endpoints:
+            if port is not None and port in endpoint.ports:
+                idx = endpoint.ports.index(port)
+                return endpoint.urls[idx]
+
+            if name is not None and endpoint.name == name:
+                return endpoint.urls[0] if endpoint.urls else None
+
+        identifier = f"port {port}" if port else f"name '{name}'"
+        raise ValueError(f"Endpoint with {identifier} not found")
+
+    def list_ports(self, teamspace_id: str, studio_id: str) -> List[V1Endpoint]:
+        """List ports that are exposed in the Studio."""
+        return self._client.endpoint_service_list_endpoints(
+            project_id=teamspace_id,
+            cloudspace_id=studio_id,
+        ).endpoints
 
     def create_assistant(self, studio_id: str, teamspace_id: str, port: int, assistant_name: str) -> V1Assistant:
         target_teamspace = self._client.projects_service_get_project(teamspace_id)

lightning_sdk/api/user_api.py

@@ -5,6 +5,7 @@ from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
     SecretServiceUpdateUserSecretBody,
     V1CloudSpace,
+    V1CreateProjectRequest,
     V1CreateUserSecretRequest,
     V1GetUserResponse,
     V1ListCloudSpacesResponse,
@@ -123,3 +124,7 @@ class UserApi:
         """
         pattern = r"^[A-Za-z_][A-Za-z0-9_]*$"
         return re.match(pattern, name) is not None
+
+    def create_teamspace(self, name: str) -> None:
+        """Creates a new teamspace."""
+        self._client.projects_service_create_project(body=V1CreateProjectRequest(name=name, display_name=name))

lightning_sdk/cli/legacy/download.py

@@ -242,7 +242,8 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
                     and (re.match(studio_name, st["name"]) or studio_name in st["name"]),
                     possible_studios,
                 )
-                if not len(possible_studios):
+                possible_studios = list(possible_studios)
+                if len(possible_studios) == 0:
                     raise ValueError(
                         f"Could not find Studio like '{studio}', please consider update your filtering pattern."
                     )

lightning_sdk/cli/legacy/studios_menu.py

@@ -31,7 +31,14 @@ class _StudiosMenu:
             title = "Please select a Studio of the following studios:"
 
         return TerminalMenu(
-            [f"{s['teamspace']}/{s['name']}" for s in possible_studios], title=title, clear_menu_on_exit=True
+            [
+                f"{s['org']}/{s['teamspace']}/{s['name']}"
+                if s.get("org") is not None
+                else f"{s['teamspace']}/{s['name']}"
+                for s in possible_studios
+            ],
+            title=title,
+            clear_menu_on_exit=True,
         )
 
     @staticmethod

lightning_sdk/job/base.py

@@ -87,7 +87,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
@@ -193,7 +193,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
                 )
 
-            if entrypoint != "sh -c":
+            if entrypoint is not None:
                 raise ValueError("Specifying the entrypoint has no effect for jobs with Studio envs.")
 
         else:
@@ -219,6 +219,20 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
                 )
 
+            # command specified, so use the default entrypoint of sh -c
+            if command is not None and entrypoint is None:
+                entrypoint = "sh -c"
+
+            # entrypoint specifically set to empty string, so set to None here to fall back to the image entrypoint
+            elif entrypoint == "":  # noqa: SIM114
+                entrypoint = None
+
+            # entrypoint not specified, but also no command specified, so use the image entrypoint
+            elif entrypoint is None:
+                entrypoint = None
+
+            # all other cases, the entrypoint has been specifically set, so use it as is
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
         return inst._submit(
             machine=machine,
@@ -255,7 +269,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
@@ -292,7 +306,12 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 within it.
                 Note that the connection needs to be added to the teamspace already in order for it to be found.
                 Only supported for jobs with a docker image compute environment.
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             max_runtime: the duration (in seconds) for which to allocate the machine.

lightning_sdk/job/job.py

@@ -97,7 +97,7 @@ class Job(_BaseJob):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
@@ -132,8 +132,10 @@ class Job(_BaseJob):
         cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
             Required if the registry is part of a cloud provider (e.g. ECR).
         entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-            just runs the provided command in a standard shell.
-            To use the pre-defined entrypoint of the provided image, set this to an empty string.
+            just runs the provided command in a standard shell if a command is provided.
+            If no command is provided, it will run the pre-defined entrypoint of the provided image.
+            To use the pre-defined entrypoint of the provided image with a specified command,
+            set this to an empty string.
             Only applicable when submitting docker jobs.
         path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
             represents the data-connection name and the path inside that connection.
@@ -193,7 +195,7 @@ class Job(_BaseJob):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
@@ -222,7 +224,12 @@ class Job(_BaseJob):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value

lightning_sdk/job/v1.py

@@ -108,7 +108,7 @@ class _JobV1(_BaseJob):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
@@ -127,7 +127,10 @@ class _JobV1(_BaseJob):
             image_credentials: The image credentials for the job (not supported).
             cloud_account_auth: Whether to use cloud account authentication for the job (not supported).
             entrypoint: The entrypoint of your docker container (not supported).
-                Defaults to `sh -c` which just runs the provided command in a standard shell.
+                Defaults to `sh -c` which just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: The mappings from data connection inside your container (not supported)

lightning_sdk/job/v2.py

@@ -49,7 +49,7 @@ class _JobV2(_BaseJob):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
@@ -74,6 +74,14 @@ class _JobV2(_BaseJob):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
+                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                Only applicable when submitting docker jobs.
             artifacts_local: The path of inside the docker container, you want to persist images from.
                 CAUTION: When setting this to "/", it will effectively erase your container.
                 Only supported for jobs with a docker image compute environment.

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -42,6 +42,7 @@ 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
@@ -132,6 +133,7 @@ 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
@@ -140,11 +142,13 @@ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project
 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_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.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
@@ -191,6 +195,8 @@ 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
@@ -295,6 +301,7 @@ 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
@@ -412,6 +419,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1Com
 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_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_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
@@ -481,6 +492,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_k
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restriction_response import V1DeleteClusterUsageRestrictionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_registry_response import V1DeleteContainerRegistryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
@@ -494,6 +506,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_incident_response im
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_template_response import V1DeleteKubernetesTemplateResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_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
@@ -569,6 +582,9 @@ 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
@@ -712,6 +728,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod import V1KubernetesPod
 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
@@ -764,6 +785,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_respon
 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_container_registries_response import V1ListContainerRegistriesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversation_message_actions_response import V1ListConversationMessageActionsResponse
 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
@@ -783,7 +805,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_mm_ts_response import V1ListFilesystemMMTsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_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
@@ -803,6 +825,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations
 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_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
@@ -981,6 +1004,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_container_registry_credentials_response import V1RefreshContainerRegistryCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -1141,6 +1165,7 @@ 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

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -23,6 +23,7 @@ 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

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3073,6 +3073,7 @@ class CloudSpaceServiceApi(object):
         :param bool include_download_url:
         :param str cluster_id:
         :param bool local_index: if true, fetch from the local index
+        :param bool include_folder_index: if true, include the folder index in the response
         :return: V1GetArtifactsPageResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -3100,12 +3101,13 @@ class CloudSpaceServiceApi(object):
         :param bool include_download_url:
         :param str cluster_id:
         :param bool local_index: if true, fetch from the local index
+        :param bool include_folder_index: if true, include the folder index in the response
         :return: V1GetArtifactsPageResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index']  # noqa: E501
+        all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -3148,6 +3150,8 @@ class CloudSpaceServiceApi(object):
             query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
         if 'local_index' in params:
             query_params.append(('localIndex', params['local_index']))  # noqa: E501
+        if 'include_folder_index' in params:
+            query_params.append(('includeFolderIndex', params['include_folder_index']))  # noqa: E501
 
         header_params = {}