lightning-sdk 2025.9.23__py3-none-any.whl → 2025.9.30__py3-none-any.whl

lightning_sdk/cli/utils/studio_selection.py

@@ -1,12 +1,12 @@
 import os
 from contextlib import suppress
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Union
 
 import click
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.cli.legacy.exceptions import StudioCliError
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -17,54 +17,58 @@ class StudiosMenu:
     It can be used to select a studio from a list of possible studios, or to resolve a studio from a name.
     """
 
-    def __init__(self, teamspace: Teamspace) -> None:
+    def __init__(self, teamspace: Teamspace, vm: bool = False) -> None:
         """Initialize the StudiosMenu with a teamspace.
 
         Args:
             teamspace: The teamspace to list studios from
         """
         self.teamspace = teamspace
+        self.vm = vm
 
-    def _get_studio_from_interactive_menu(self, possible_studios: Dict[str, Studio]) -> Studio:
+    def _get_studio_from_interactive_menu(self, possible_studios: Dict[str, Union[Studio, VM]]) -> Union[Studio, VM]:
         studio_names = sorted(possible_studios.keys())
-        terminal_menu = self._prepare_terminal_menu_studios(studio_names)
+        terminal_menu = self._prepare_terminal_menu_studios(studio_names, vm=self.vm)
         terminal_menu.show()
 
         selected_name = studio_names[terminal_menu.chosen_menu_index]
         return possible_studios[selected_name]
 
-    def _get_studio_from_name(self, studio: str, possible_studios: Dict[str, Studio]) -> Studio:
+    def _get_studio_from_name(self, studio: str, possible_studios: Dict[str, Union[Studio, VM]]) -> Union[Studio, VM]:
         if studio in possible_studios:
             return possible_studios[studio]
 
-        click.echo(f"Could not find Studio {studio}, please select it from the list:")
+        click.echo(f"Could not find {'VM' if self.vm else 'Studio'} {studio}, please select it from the list:")
         return self._get_studio_from_interactive_menu(possible_studios)
 
     @staticmethod
-    def _prepare_terminal_menu_studios(studio_names: List[str], title: Optional[str] = None) -> TerminalMenu:
+    def _prepare_terminal_menu_studios(
+        studio_names: List[str], title: Optional[str] = None, vm: bool = False
+    ) -> TerminalMenu:
         if title is None:
-            title = "Please select a Studio out of the following:"
+            title = f"Please select a {'VM' if vm else 'Studio'} out of the following:"
 
         return TerminalMenu(studio_names, title=title, clear_menu_on_exit=True)
 
-    def _get_possible_studios(self) -> Dict[str, Studio]:
+    def _get_possible_studios(self) -> Dict[str, Union[Studio, VM]]:
         """Get all available studios in the teamspace."""
         studios = {}
 
         user = _get_authed_user()
+        studios = self.teamspace.vms if self.vm else self.teamspace.studios
         for studio in self.teamspace.studios:
             if studio._studio.user_id == user.id:
                 studios[studio.name] = studio
         return studios
 
-    def __call__(self, studio: Optional[str] = None) -> Studio:
+    def __call__(self, studio: Optional[str] = None) -> Union[Studio, VM]:
         """Select a studio from the teamspace.
 
         Args:
             studio: Optional studio name to select. If not provided, will show interactive menu.
 
         Returns:
-            Selected Studio object
+            Selected Studio/VM object
 
         Raises:
             StudioCliError: If studio selection fails
@@ -73,15 +77,18 @@ class StudiosMenu:
             # try to resolve the studio from the name, environment or config
             resolved_studio = None
 
+            selected_cls = VM if self.vm else Studio
+
             with suppress(Exception):
-                resolved_studio = Studio(name=studio, teamspace=self.teamspace, create_ok=False)
+                resolved_studio = selected_cls(name=studio, teamspace=self.teamspace, create_ok=False)
 
             if resolved_studio is not None:
                 return resolved_studio
 
             if os.environ.get("LIGHTNING_NON_INTERACTIVE", "0") == "1" and studio is None:
                 raise ValueError(
-                    "Studio selection is not supported in non-interactive mode. Please provide a studio name."
+                    f"{'VM' if self.vm else 'Studio'} selection is not supported in non-interactive mode. "
+                    "Please provide a studio name."
                 )
 
             click.echo(f"Listing studios in teamspace {self.teamspace.owner.name}/{self.teamspace.name}...")
@@ -101,6 +108,6 @@ class StudiosMenu:
 
         except Exception as e:
             raise StudioCliError(
-                "Could not resolve a Studio. "
+                f"Could not resolve a {'VM' if self.vm else 'Studio'}. "
                 "Please pass it as an argument or contact Lightning AI directly to resolve this issue."
             ) from e

lightning_sdk/cli/vm/__init__.py

@@ -0,0 +1,20 @@
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register studio commands with the given group."""
+    from lightning_sdk.cli.vm.create import create_vm
+    from lightning_sdk.cli.vm.delete import delete_vm
+    from lightning_sdk.cli.vm.list import list_vms
+    from lightning_sdk.cli.vm.ssh import ssh_vm
+    from lightning_sdk.cli.vm.start import start_vm
+    from lightning_sdk.cli.vm.stop import stop_vm
+    from lightning_sdk.cli.vm.switch import switch_vm
+
+    group.add_command(create_vm)
+    group.add_command(delete_vm)
+    group.add_command(list_vms)
+    group.add_command(ssh_vm)
+    group.add_command(start_vm)
+    group.add_command(stop_vm)
+    group.add_command(switch_vm)

lightning_sdk/cli/vm/create.py

@@ -0,0 +1,33 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.create import create_impl
+from lightning_sdk.machine import CloudProvider
+
+
+@click.command("create")
+@click.option("--name", help="The name of the VM to create. If not provided, a random name will be generated.")
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--cloud-provider",
+    help="The cloud provider to start the VM on. Defaults to teamspace default.",
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to create the VM on. Defaults to teamspace default.",
+    type=click.STRING,
+)
+def create_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+) -> None:
+    """Create a new VM.
+
+    Example:
+        lightning vm create
+    """
+    create_impl(name=name, teamspace=teamspace, cloud_provider=cloud_provider, cloud_account=cloud_account, vm=True)

lightning_sdk/cli/vm/delete.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.delete import delete_impl
+
+
+@click.command("delete")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to delete. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+def delete_vm(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete a VM.
+
+    Example:
+      lightning vm delete --name my-vm
+
+    """
+    return delete_impl(name=name, teamspace=teamspace, vm=True)

lightning_sdk/cli/vm/list.py

@@ -0,0 +1,30 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.list import list_impl
+
+
+@click.command("list")
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--all",
+    is_flag=True,
+    flag_value=True,
+    default=False,
+    help="List all VMs, not just the ones belonging to the authed user",
+)
+@click.option(
+    "--sort-by",
+    default=None,
+    type=click.Choice(["name", "teamspace", "status", "machine", "cloud-account"], case_sensitive=False),
+    help="the attribute to sort the VMs by.",
+)
+def list_vms(teamspace: Optional[str] = None, all: bool = False, sort_by: Optional[str] = None) -> None:  # noqa: A002
+    """List VMs in a teamspace.
+
+    Example:
+        lightning vm list --teamspace owner/teamspace
+
+    """
+    return list_impl(teamspace=teamspace, all=all, sort_by=sort_by, vm=True)

lightning_sdk/cli/vm/ssh.py

@@ -0,0 +1,31 @@
+from typing import List, Optional
+
+import click
+
+from lightning_sdk.cli.studio.ssh import ssh_impl
+
+
+@click.command("ssh")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to ssh into. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)", type=click.STRING)
+@click.option(
+    "--option",
+    "-o",
+    help="Additional options to pass to the SSH command. Can be specified multiple times.",
+    multiple=True,
+    type=click.STRING,
+)
+def ssh_vm(name: Optional[str] = None, teamspace: Optional[str] = None, option: Optional[List[str]] = None) -> None:
+    """SSH into a VM.
+
+    Example:
+        lightning vm ssh --name my-vm
+    """
+    return ssh_impl(name=name, teamspace=teamspace, option=option, vm=False)

lightning_sdk/cli/vm/start.py

@@ -0,0 +1,60 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.start import start_impl
+from lightning_sdk.machine import CloudProvider, Machine
+
+
+@click.command("start")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to start. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option("--create", is_flag=True, help="Create the VM if it doesn't exist")
+@click.option(
+    "--machine",
+    help="The machine type to start the VM on. Defaults to CPU-4",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli),
+)
+@click.option("--interruptible", is_flag=True, help="Start the VM on an interruptible instance.")
+@click.option(
+    "--cloud-provider",
+    help=("The cloud provider to start the VM on. Defaults to teamspace default. Only used if --create is specified."),
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to start the VM on. Defaults to teamspace default. Only used if --create is specified.",
+    type=click.STRING,
+)
+def start_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    create: bool = False,
+    machine: str = "CPU",
+    interruptible: bool = False,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+) -> None:
+    """Start a VM.
+
+    Example:
+        lightning vm start --name my-vm
+
+    """
+    return start_impl(
+        name=name,
+        teamspace=teamspace,
+        create=create,
+        machine=machine,
+        interruptible=interruptible,
+        cloud_provider=cloud_provider,
+        cloud_account=cloud_account,
+        vm=True,
+    )

lightning_sdk/cli/vm/stop.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.stop import stop_impl
+
+
+@click.command("stop")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to stop. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+def stop_vm(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Stop a VM.
+
+    Example:
+        lightning vm stop --name my-vm
+
+    """
+    return stop_impl(name=name, teamspace=teamspace, vm=True)

lightning_sdk/cli/vm/switch.py

@@ -0,0 +1,38 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.switch import switch_impl
+from lightning_sdk.machine import Machine
+
+
+@click.command("switch")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to switch to a different machine. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--machine",
+    help="The machine type to switch the studio to.",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine)),
+)
+@click.option("--interruptible", is_flag=True, help="Switch the studio to an interruptible instance.")
+def switch_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    machine: Optional[str] = None,
+    interruptible: bool = False,
+) -> None:
+    """Switch a VM to a different machine type."""
+    return switch_impl(
+        name=name,
+        teamspace=teamspace,
+        machine=machine,
+        interruptible=interruptible,
+        vm=True,
+    )

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -107,6 +107,7 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body1 import
 from lightning_sdk.lightning_cloud.openapi.models.create import Create
 from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
 from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
+from lightning_sdk.lightning_cloud.openapi.models.create_machine_request_represents_the_request_to_create_a_machine import CreateMachineRequestRepresentsTheRequestToCreateAMachine
 from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body import CreditsAutoreplenishBody
 from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body1 import CreditsAutoreplenishBody1
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
@@ -156,6 +157,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploa
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdVisibilityBody2
+from lightning_sdk.lightning_cloud.openapi.models.job_id_reportroutingtelemetry_body import JobIdReportroutingtelemetryBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -207,6 +209,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body impo
 from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import ProjectIdStorageBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_storagetransfers_body import ProjectIdStoragetransfersBody
 from lightning_sdk.lightning_cloud.openapi.models.project_tab_management_messages import ProjectTabManagementMessages
 from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body import ProjectsProjectIdBody
@@ -246,6 +249,7 @@ from lightning_sdk.lightning_cloud.openapi.models.user_user_id_body import UserU
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_cluster_credentials import V1AWSClusterCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1 import V1AWSDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import V1AWSDirectV1Status
+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_job_timing_response import V1AddJobTimingResponse
@@ -388,6 +392,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_machine_response import V1CreateMachineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
 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
@@ -448,6 +453,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response im
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_machine_response import V1DeleteMachineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_endpoint_response import V1DeleteManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
@@ -579,12 +585,12 @@ 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_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_metrics_response import V1GetModelMetricsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_total_usage_metrics_response import V1GetModelTotalUsageMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
@@ -645,6 +651,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_basic_auth import V1LightningBasicAuth
+from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_elastic_cluster_v1 import V1LightningElasticClusterV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_o_auth_auth import V1LightningOAuthAuth
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_run import V1LightningRun
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightningapp_instance_artifact import V1LightningappInstanceArtifact
@@ -731,6 +738,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response
 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_machines_response import V1ListMachinesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
@@ -768,6 +776,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedule_runs_response import V1ListScheduleRunsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_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_lit_dataset import V1LitDataset
@@ -788,6 +797,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_logout_request import V1Log
 from lightning_sdk.lightning_cloud.openapi.models.v1_logout_response import V1LogoutResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_logs_response import V1LogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_lustre_data_connection import V1LustreDataConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_machine import V1Machine
+from lightning_sdk.lightning_cloud.openapi.models.v1_machine_direct_v1 import V1MachineDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_machines_selector import V1MachinesSelector
 from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request import V1MagicLinkLoginRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
@@ -840,6 +851,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1Pa
 from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathMapping
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
+from lightning_sdk.lightning_cloud.openapi.models.v1_pause_storage_transfer_response import V1PauseStorageTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline import V1Pipeline
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter import V1PipelineParameter
@@ -875,6 +887,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_project_tab import V1Projec
 from lightning_sdk.lightning_cloud.openapi.models.v1_prompt_suggestion import V1PromptSuggestion
 from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_response import V1PublishCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_request import V1PurchaseAnnualUpsellRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_response import V1PurchaseAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
@@ -897,6 +910,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer impo
 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_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_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
@@ -914,6 +928,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_restart_cloud_space_instance_response import V1RestartCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1RestartTiming
 from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_resume_storage_transfer_response import V1ResumeStorageTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
 from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_routing_telemetry import V1RoutingTelemetry
@@ -947,6 +962,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
 from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
+from lightning_sdk.lightning_cloud.openapi.models.v1_slack_notifier import V1SlackNotifier
+from lightning_sdk.lightning_cloud.openapi.models.v1_slack_notifier_type import V1SlackNotifierType
 from lightning_sdk.lightning_cloud.openapi.models.v1_sleep_server_response import V1SleepServerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
@@ -961,6 +978,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_stop_cloud_space_instance_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset import V1StorageAsset
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V1StorageAssetType
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_storage_transfer import V1StorageTransfer
+from lightning_sdk.lightning_cloud.openapi.models.v1_storage_transfer_status import V1StorageTransferStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
 from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -1058,7 +1058,7 @@ class AssistantsServiceApi(object):
             collection_formats=collection_formats)
 
     def assistants_service_get_latest_model_metrics(self, **kwargs) -> 'V1GetLatestModelMetricsResponse':  # noqa: E501
-        """assistants_service_get_latest_model_metrics  # noqa: E501
+        """model metrics  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1079,7 +1079,7 @@ class AssistantsServiceApi(object):
             return data
 
     def assistants_service_get_latest_model_metrics_with_http_info(self, **kwargs) -> 'V1GetLatestModelMetricsResponse':  # noqa: E501
-        """assistants_service_get_latest_model_metrics  # noqa: E501
+        """model metrics  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
@@ -1570,99 +1570,6 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def assistants_service_get_model_total_usage_metrics(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse':  # noqa: E501
-        """model metrics  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.assistants_service_get_model_total_usage_metrics(model_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :param str model_id: (required)
-        :return: V1GetModelTotalUsageMetricsResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        if kwargs.get('async_req'):
-            return self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs)  # noqa: E501
-        else:
-            (data) = self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs)  # noqa: E501
-            return data
-
-    def assistants_service_get_model_total_usage_metrics_with_http_info(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse':  # noqa: E501
-        """model metrics  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :param str model_id: (required)
-        :return: V1GetModelTotalUsageMetricsResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        all_params = ['model_id']  # noqa: E501
-        all_params.append('async_req')
-        all_params.append('_return_http_data_only')
-        all_params.append('_preload_content')
-        all_params.append('_request_timeout')
-
-        params = locals()
-        for key, val in six.iteritems(params['kwargs']):
-            if key not in all_params:
-                raise TypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method assistants_service_get_model_total_usage_metrics" % key
-                )
-            params[key] = val
-        del params['kwargs']
-        # verify the required parameter 'model_id' is set
-        if ('model_id' not in params or
-                params['model_id'] is None):
-            raise ValueError("Missing the required parameter `model_id` when calling `assistants_service_get_model_total_usage_metrics`")  # noqa: E501
-
-        collection_formats = {}
-
-        path_params = {}
-        if 'model_id' in params:
-            path_params['modelId'] = params['model_id']  # noqa: E501
-
-        query_params = []
-
-        header_params = {}
-
-        form_params = []
-        local_var_files = {}
-
-        body_params = None
-        # HTTP header `Accept`
-        header_params['Accept'] = self.api_client.select_header_accept(
-            ['application/json'])  # noqa: E501
-
-        # Authentication setting
-        auth_settings = []  # noqa: E501
-
-        return self.api_client.call_api(
-            '/v1/agents/{modelId}/total-usage', 'GET',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='V1GetModelTotalUsageMetricsResponse',  # noqa: E501
-            auth_settings=auth_settings,
-            async_req=params.get('async_req'),
-            _return_http_data_only=params.get('_return_http_data_only'),
-            _preload_content=params.get('_preload_content', True),
-            _request_timeout=params.get('_request_timeout'),
-            collection_formats=collection_formats)
-
     def assistants_service_get_published_managed_endpoint_by_model_id(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
         """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata  # noqa: E501