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

lightning_sdk/cli/legacy/download.py

@@ -12,6 +12,7 @@ from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
+from lightning_sdk.exceptions import DeprecatedCommand, DeprecatedError
 from lightning_sdk.models import download_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user
@@ -63,134 +64,64 @@ def model(name: str, download_dir: str = ".") -> None:
     )
 
 
-@download.command(name="folder")
-@click.argument("path")
+@download.command(
+    name="folder",
+    cls=DeprecatedCommand,
+    message="Studio downloads via 'lightning download folder' are deprecated. Use 'lightning studio cp -r' instead.",
+)
+@click.argument("path", required=False, nargs=-1)
 @click.option(
     "--studio",
     default=None,
-    help=(
-        "The name of the studio to download from. "
-        "Will show a menu with user's owned studios for selection if not specified. "
-        "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
-        "The teamspace and studio names can be regular expressions to match, "
-        "a menu filtered studios will be shown for final selection."
-    ),
+    hidden=True,
 )
 @click.option(
     "--teamspace",
     default=None,
-    help="The teamspace the drive folder is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.",
+    hidden=True,
 )
 @click.option(
     "--local-path",
     "--local_path",
-    default=".",
-    type=click.Path(file_okay=False, dir_okay=True),
-    help="The path to the directory you want to download the folder to.",
+    default=None,
+    hidden=True,
 )
 def folder(
     path: str = "", studio: Optional[str] = None, teamspace: Optional[str] = None, local_path: str = "."
 ) -> None:
-    """Download a folder from a Studio or a Teamspace drive folder.
-
-    Example:
-      lightning download folder PATH
-
-    PATH: The relative path within the Studio or drive folder you want to download.
-    Defaults to the entire Studio or drive folder.
-    """
-    local_path = Path(local_path)
-    if not local_path.is_dir():
-        raise NotADirectoryError(f"'{local_path}' is not a directory")
-
-    if studio and teamspace:
-        raise ValueError("Either --studio or --teamspace must be provided, not both")
-
-    if studio:
-        path = _expand_remote_path(path)
-        resolved_downloader = _resolve_studio(studio)
-    elif teamspace:
-        menu = TeamspacesMenu()
-        resolved_downloader = menu(teamspace)
-    else:
-        raise ValueError("Either --studio or --teamspace must be provided")
-
-    if not path:
-        local_path /= resolved_downloader.name
-        path = ""
-
-    try:
-        if not path and teamspace:
-            raise FileNotFoundError()
-        resolved_downloader.download_folder(remote_path=path, target_path=str(local_path))
-    except Exception as e:
-        raise StudioCliError(
-            f"Could not download the folder from the given Studio {studio} or Teamspace {teamspace}. "
-            "Please contact Lightning AI directly to resolve this issue."
-        ) from e
+    """[DEPRECATED] Use 'lightning studio cp -r' instead."""
+    raise DeprecatedError(
+        "Studio downloads via 'lightning download folder' are deprecated. Use 'lightning studio cp -r' instead."
+    )
 
 
-@download.command(name="file")
-@click.argument("path")
+@download.command(
+    name="file",
+    cls=DeprecatedCommand,
+    message="Studio downloads via 'lightning download file' are deprecated. Use 'lightning studio cp' instead.",
+)
+@click.argument("path", required=False, nargs=-1)
 @click.option(
     "--studio",
     default=None,
-    help=(
-        "The name of the studio to download from. "
-        "Will show a menu with user's owned studios for selection if not specified. "
-        "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
-        "The teamspace and studio names can be regular expressions to match, "
-        "a menu filtered studios will be shown for final selection."
-    ),
+    hidden=True,
 )
 @click.option(
     "--teamspace",
     default=None,
-    help="The teamspace the file is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.",
+    hidden=True,
 )
 @click.option(
     "--local-path",
     "--local_path",
-    default=".",
-    type=click.Path(file_okay=False, dir_okay=True),
-    help="The path to the directory you want to download the file to.",
+    default=None,
+    hidden=True,
 )
 def file(path: str = "", studio: Optional[str] = None, teamspace: Optional[str] = None, local_path: str = ".") -> None:
-    """Download a file from a Studio or Teamspace drive file.
-
-    Example:
-      lightning download file PATH
-
-    PATH: The relative path to the file within the Studio or Teamspace drive file you want to download.
-    """
-    local_path = Path(local_path)
-    if not local_path.is_dir():
-        raise NotADirectoryError(f"'{local_path}' is not a directory")
-
-    if studio and teamspace:
-        raise ValueError("Either --studio or --teamspace must be provided, not both")
-
-    if studio:
-        resolved_downloader = _resolve_studio(studio)
-    elif teamspace:
-        menu = TeamspacesMenu()
-        resolved_downloader = menu(teamspace)
-    else:
-        raise ValueError("Either --studio or --teamspace must be provided")
-
-    if not path:
-        local_path /= resolved_downloader.name
-        path = ""
-
-    try:
-        if not path:
-            raise FileNotFoundError()
-        resolved_downloader.download_file(remote_path=path, file_path=str(local_path / os.path.basename(path)))
-    except Exception as e:
-        raise StudioCliError(
-            f"Could not download the file from the given Studio {studio} or Teamspace {teamspace}. "
-            "Please contact Lightning AI directly to resolve this issue."
-        ) from e
+    """[DEPRECATED] Use 'lightning studio cp' instead."""
+    raise DeprecatedError(
+        "Studio downloads via 'lightning download file' are deprecated. Use 'lightning studio cp' instead."
+    )
 
 
 @download.command(name="container")

lightning_sdk/cli/legacy/upload.py

@@ -18,6 +18,7 @@ from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.constants import _LIGHTNING_DEBUG
+from lightning_sdk.exceptions import DeprecatedCommand, DeprecatedError
 from lightning_sdk.models import upload_model as _upload_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user
@@ -54,56 +55,48 @@ def model(name: str, path: str = ".", cloud_account: Optional[str] = None) -> No
     _upload_model(name, path, cloud_account=cloud_account)
 
 
-@upload.command("folder")
-@click.argument("path", type=click.Path(exists=True))
-@click.option(
-    "--studio",
-    default=None,
-    help=(
-        "The name of the studio to upload to. "
-        "Will show a menu for selection if not specified. "
-        "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME>"
-    ),
+@upload.command(
+    name="folder",
+    cls=DeprecatedCommand,
+    message="Studio uploads via 'lightning upload folder' are deprecated. Use 'lightning studio cp -r' instead.",
 )
+@click.argument("path", type=click.Path(), required=False, nargs=-1)
+@click.option("--studio", default=None, hidden=True)
 @click.option(
     "--remote-path",
     "--remote_path",
     default=None,
-    help=(
-        "The path where the uploaded file should appear on your Studio. "
-        "Has to be within your Studio's home directory and will be relative to that. "
-        "If not specified, will use the name of the folder you want to upload and place it in your home directory."
-    ),
+    hidden=True,
 )
 def folder(path: str, studio: Optional[str], remote_path: Optional[str]) -> None:
-    """Upload a folder to a Studio."""
-    _folder(path=path, studio=studio, remote_path=remote_path)
+    """[DEPRECATED] Use 'lightning studio cp -r' instead."""
+    raise DeprecatedError(
+        "Studio uploads via 'lightning upload folder' are deprecated. Use 'lightning studio cp -r' instead."
+    )
 
 
-@upload.command("file")
-@click.argument("path", type=click.Path(exists=True))
+@upload.command(
+    name="file",
+    cls=DeprecatedCommand,
+    message="Studio uploads via 'lightning upload file' are deprecated. Use 'lightning studio cp' instead.",
+)
+@click.argument("path", type=click.Path(), required=False, nargs=-1)
 @click.option(
     "--studio",
     default=None,
-    help=(
-        "The name of the studio to upload to. "
-        "Will show a menu for selection if not specified. "
-        "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME>"
-    ),
+    hidden=True,
 )
 @click.option(
     "--remote-path",
     "--remote_path",
     default=None,
-    help=(
-        "The path where the uploaded file should appear on your Studio. "
-        "Has to be within your Studio's home directory and will be relative to that. "
-        "If not specified, will use the name of the file you want to upload and place it in your home directory."
-    ),
+    hidden=True,
 )
 def file(path: str, studio: Optional[str] = None, remote_path: Optional[str] = None) -> None:
-    """Upload a file to a Studio."""
-    _file(path=path, studio=studio, remote_path=remote_path)
+    """[DEPRECATED] Use 'lightning studio cp' instead."""
+    raise DeprecatedError(
+        "Studio uploads via 'lightning upload file' are deprecated. Use 'lightning studio cp' instead."
+    )
 
 
 @upload.command("container")

lightning_sdk/cli/studio/cp.py

@@ -8,7 +8,7 @@ import click
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+from lightning_sdk.cli.utils.filesystem import parse_studio_path, resolve_studio
 
 
 @click.command("cp")
@@ -111,10 +111,13 @@ def cp_download(
                 f"'{studio_path_result['destination']}' is a directory. Use -r flag to copy directories recursively."
             )
         folder_name = os.path.basename(studio_path_result["destination"].rstrip("/"))
-        if folder_name == "" and local_path in ("./", "."):
-            # handle root directory case (e.g. lit://lightning-ai/gpt-oss/studios/manual-lime-ylu2/)
-            folder_name = selected_studio.name
-        target_path = os.path.join(local_path, folder_name)
+        if local_path in ("./", "."):
+            if folder_name == "":
+                # handle root directory case (e.g. lit://lightning-ai/gpt-oss/studios/manual-lime-ylu2/)
+                folder_name = selected_studio.name
+            target_path = os.path.join(local_path, folder_name)
+        else:
+            target_path = local_path
 
         selected_studio.download_folder(studio_path_result["destination"], target_path)
         console.print(f"See your folder at {target_path}")

lightning_sdk/cli/studio/ls.py

@@ -2,7 +2,7 @@
 
 import click
 
-from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+from lightning_sdk.cli.utils.filesystem import parse_studio_path, resolve_studio
 
 
 @click.command("ls")

lightning_sdk/cli/studio/rm.py

@@ -3,7 +3,7 @@
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+from lightning_sdk.cli.utils.filesystem import parse_studio_path, resolve_studio
 from lightning_sdk.studio import Studio
 
 

lightning_sdk/cli/utils/{studio_filesystem.py → filesystem.py}

@@ -4,21 +4,22 @@ 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 StudioPathResult(TypedDict):
+class PathResult(TypedDict):
     owner: Optional[str]
     teamspace: Optional[str]
     studio: Optional[str]
     destination: Optional[str]
 
 
-def parse_studio_path(studio_path: str) -> StudioPathResult:
+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: StudioPathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
+    result: PathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
 
     if "/studios/" in path_string:
         prefix_part, suffix_part = path_string.split("/studios/", 1)
@@ -54,12 +55,49 @@ def parse_studio_path(studio_path: str) -> StudioPathResult:
     return result
 
 
-def resolve_studio(studio_name: Optional[str], teamspace: Optional[str], owner: Optional[str]) -> Studio:
+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)
-    resolved_teamspace = teamspace_menu(teamspace=teamspace)
+    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/exceptions.py

@@ -1,6 +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/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
@@ -133,7 +134,6 @@ 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
@@ -143,14 +143,16 @@ 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_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
@@ -178,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
@@ -242,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
@@ -303,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
@@ -422,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
@@ -443,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
@@ -529,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
@@ -578,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
@@ -588,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
@@ -811,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
@@ -928,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
@@ -1178,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

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

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3074,6 +3074,7 @@ class CloudSpaceServiceApi(object):
         :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
+        :param bool only_direct_children: if true, only include direct children of the prefix
         :return: V1GetArtifactsPageResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -3102,12 +3103,13 @@ class CloudSpaceServiceApi(object):
         :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
+        :param bool only_direct_children: if true, only include direct children of the prefix
         :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', 'include_folder_index']  # noqa: E501
+        all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index', 'only_direct_children']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -3152,6 +3154,8 @@ class CloudSpaceServiceApi(object):
             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
+        if 'only_direct_children' in params:
+            query_params.append(('onlyDirectChildren', params['only_direct_children']))  # noqa: E501
 
         header_params = {}