lightning-sdk 0.1.47__py3-none-any.whl → 0.1.48__py3-none-any.whl

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.47"
+__version__ = "0.1.48"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/job_api.py

@@ -182,6 +182,7 @@ class JobApiV1:
 
 
 class JobApiV2:
+    # these are stages the job can be in.
     v2_job_state_pending = "pending"
     v2_job_state_running = "running"
     v2_job_state_stopped = "stopped"
@@ -189,6 +190,9 @@ class JobApiV2:
     v2_job_state_failed = "failed"
     v2_job_state_stopping = "stopping"
 
+    # this is the user action to stop the job.
+    v2_job_state_stop = "stop"
+
     def __init__(self) -> None:
         self._cloud_url = _cloud_url()
         self._client = LightningClient(max_tries=7)
@@ -262,7 +266,7 @@ class JobApiV2:
             return
 
         if current_state != Status.Stopping:
-            update_body = JobsIdBody1(cloudspace_id=current_job.spec.cloudspace_id, state=self.v2_job_state_stopped)
+            update_body = JobsIdBody1(cloudspace_id=current_job.spec.cloudspace_id, state=self.v2_job_state_stop)
             self._client.jobs_service_update_job(body=update_body, project_id=teamspace_id, id=job_id)
 
         while True:
@@ -302,6 +306,13 @@ class JobApiV2:
     def get_command(self, job: V1Job) -> str:
         return job.spec.command
 
+    def get_mmt_name(self, job: V1Job) -> str:
+        if job.multi_machine_job_id:
+            splits = job.name.rsplit("-", 1)
+            if len(splits) == 2:
+                return splits[0]
+        return ""
+
     def _job_state_to_external(self, state: str) -> "Status":
         from lightning_sdk.status import Status
 

lightning_sdk/api/lit_container_api.py

@@ -0,0 +1,19 @@
+from typing import List
+
+from lightning_sdk.lightning_cloud.openapi.models import V1DeleteLitRepositoryResponse
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
+
+
+class LitContainerApi:
+    def __init__(self) -> None:
+        self._client = LightningClient(max_tries=3)
+
+    def list_containers(self, project_id: str) -> List:
+        project = self._client.lit_registry_service_get_lit_project_registry(project_id)
+        return project.repositories
+
+    def delete_container(self, project_id: str, container: str) -> V1DeleteLitRepositoryResponse:
+        try:
+            return self._client.lit_registry_service_delete_lit_repository(project_id, container)
+        except Exception as ex:
+            raise ValueError(f"Could not delete container {container} from project {project_id}") from ex

lightning_sdk/api/utils.py

@@ -353,7 +353,7 @@ def _machine_to_compute_name(machine: Union[Machine, str]) -> str:
 
 _COMPUTE_NAME_TO_MACHINE: Dict[str, Machine] = {v: k for k, v in _MACHINE_TO_COMPUTE_NAME.items()}
 
-_DEFAULT_CLOUD_URL = "https://lightning.ai:443"
+_DEFAULT_CLOUD_URL = "https://lightning.ai"
 
 
 def _get_cloud_url() -> str:

lightning_sdk/cli/delete.py

@@ -0,0 +1,58 @@
+from typing import Optional
+
+from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
+from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.lit_container import LitContainer
+
+
+class _Delete(_JobAndMMTAction, _TeamspacesMenu):
+    """Delete resources on the Lightning AI platform."""
+
+    def container(self, container: str, teamspace: Optional[str] = None) -> None:
+        """Delete a docker container.
+
+        Args:
+            container: The name of the container to delete.
+            teamspace: The teamspace to delete the container from. Should be specified as {owner}/{name}
+                If not provided, can be selected in an interactive menu.
+        """
+        api = LitContainer()
+        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+        try:
+            api.delete_container(container, resolved_teamspace.name, resolved_teamspace.owner.name)
+            print(f"Container {container} deleted successfully.")
+        except Exception as e:
+            raise StudioCliError(
+                f"Could not delete container {container} from project {resolved_teamspace.name}: {e}"
+            ) from None
+
+    def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Delete a job.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        job = super().job(name=name, teamspace=teamspace)
+
+        job.delete()
+        print(f"Successfully deleted {job.name}!")
+
+    def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Delete a multi-machine job.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        mmt = super().mmt(name=name, teamspace=teamspace)
+
+        mmt.delete()
+        print(f"Successfully deleted {mmt.name}!")

lightning_sdk/cli/entrypoint.py

@@ -3,11 +3,14 @@ from lightning_utilities.core.imports import RequirementCache
 
 from lightning_sdk.api.studio_api import _cloud_url
 from lightning_sdk.cli.ai_hub import _AIHub
+from lightning_sdk.cli.delete import _Delete
 from lightning_sdk.cli.download import _Downloads
+from lightning_sdk.cli.inspect import _Inspect
 from lightning_sdk.cli.legacy import _LegacyLightningCLI
 from lightning_sdk.cli.list import _List
 from lightning_sdk.cli.run import _Run
 from lightning_sdk.cli.serve import _Docker, _LitServe
+from lightning_sdk.cli.stop import _Stop
 from lightning_sdk.cli.upload import _Uploads
 from lightning_sdk.lightning_cloud.login import Auth
 
@@ -25,6 +28,9 @@ class StudioCLI:
         self.serve = _LitServe()
         self.dockerize = _Docker()
         self.list = _List()
+        self.delete = _Delete()
+        self.inspect = _Inspect()
+        self.stop = _Stop()
 
     def login(self) -> None:
         """Login to Lightning AI Studios."""

lightning_sdk/cli/inspect.py

@@ -0,0 +1,31 @@
+from typing import Optional
+
+from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
+
+
+class _Inspect(_JobAndMMTAction):
+    """Inspect resources of the Lightning AI platform to get additional details as JSON."""
+
+    def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Inspect a job for further details as JSON.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        print(super().job(name=name, teamspace=teamspace).json())
+
+    def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Inspect a multi-machine job for further details as JSON.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        print(super().mmt(name=name, teamspace=teamspace).json())

lightning_sdk/cli/job_and_mmt_action.py

@@ -0,0 +1,37 @@
+from typing import Optional
+
+from lightning_sdk.cli.jobs_menu import _JobsMenu
+from lightning_sdk.cli.mmts_menu import _MMTsMenu
+from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.job import Job
+from lightning_sdk.mmt import MMT
+
+
+class _JobAndMMTAction(_TeamspacesMenu, _JobsMenu, _MMTsMenu):
+    """Inspect resources of the Lightning AI platform to get additional details as JSON."""
+
+    def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> Job:
+        """Fetch a job for further processing.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        resolved_teamspace = self._resolve_teamspace(teamspace)
+        return self._resolve_job(name, teamspace=resolved_teamspace)
+
+    def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> MMT:
+        """Fetch a multi-machine job for further processing.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        resolved_teamspace = self._resolve_teamspace(teamspace)
+        return self._resolve_mmt(name, teamspace=resolved_teamspace)

lightning_sdk/cli/jobs_menu.py

@@ -0,0 +1,57 @@
+from typing import Dict, List, Optional
+
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.job import Job
+from lightning_sdk.teamspace import Teamspace
+
+
+class _JobsMenu:
+    def _get_job_from_interactive_menu(self, possible_jobs: Dict[str, Job]) -> Job:
+        job_ids = sorted(possible_jobs.keys())
+        terminal_menu = self._prepare_terminal_menu_jobs([possible_jobs[k] for k in job_ids])
+        terminal_menu.show()
+
+        return possible_jobs[terminal_menu.chosen_menu_entry]
+
+    def _get_job_from_name(self, job: str, possible_jobs: Dict[str, Job]) -> Job:
+        for _, j in possible_jobs.items():
+            if j.name == job:
+                return j
+
+        print("Could not find Job {job}, please select it from the list:")
+        return self._get_job_from_interactive_menu(possible_jobs)
+
+    @staticmethod
+    def _prepare_terminal_menu_jobs(possible_jobs: List[Job], title: Optional[str] = None) -> TerminalMenu:
+        if title is None:
+            title = "Please select a Job of the following:"
+
+        return TerminalMenu([j.name for j in possible_jobs], title=title, clear_menu_on_exit=True)
+
+    @staticmethod
+    def _get_possible_jobs(teamspace: Teamspace) -> Dict[str, Job]:
+        jobs = {}
+        for j in teamspace.jobs:
+            jobs[j.name] = j
+
+        return jobs
+
+    def _resolve_job(self, job: Optional[str], teamspace: Teamspace) -> Job:
+        try:
+            possible_jobs = self._get_possible_jobs(teamspace)
+            if job is None:
+                resolved_job = self._get_job_from_interactive_menu(possible_jobs)
+            else:
+                resolved_job = self._get_job_from_name(job=job, possible_jobs=possible_jobs)
+
+            return resolved_job
+        except KeyboardInterrupt:
+            raise KeyboardInterrupt from None
+
+        except Exception as e:
+            raise StudioCliError(
+                f"Could not find the given Job {job} in Teamspace {teamspace.name}. "
+                "Please contact Lightning AI directly to resolve this issue."
+            ) from e

lightning_sdk/cli/list.py

@@ -4,7 +4,7 @@ from rich.console import Console
 from rich.table import Table
 
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lit_registry import LitRegistry
+from lightning_sdk.lit_container import LitContainer
 
 
 class _List(_TeamspacesMenu):
@@ -41,7 +41,7 @@ class _List(_TeamspacesMenu):
             teamspace: The teamspace to list containers from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
         """
-        api = LitRegistry()
+        api = LitContainer()
         resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
         result = api.list_containers(teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name)
         table = Table(pad_edge=True, box=None)

lightning_sdk/cli/mmts_menu.py

@@ -0,0 +1,57 @@
+from typing import Dict, List, Optional
+
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.mmt import MMT
+from lightning_sdk.teamspace import Teamspace
+
+
+class _MMTsMenu:
+    def _get_mmt_from_interactive_menu(self, possible_mmts: Dict[str, MMT]) -> MMT:
+        job_ids = sorted(possible_mmts.keys())
+        terminal_menu = self._prepare_terminal_menu_mmts([possible_mmts[k] for k in job_ids])
+        terminal_menu.show()
+
+        return possible_mmts[terminal_menu.chosen_menu_entry]
+
+    def _get_mmt_from_name(self, mmt: str, possible_mmts: Dict[str, MMT]) -> MMT:
+        for _, j in possible_mmts.items():
+            if j.name == mmt:
+                return j
+
+        print("Could not find Multi-Machine Job {mmt}, please select it from the list:")
+        return self._get_mmt_from_interactive_menu(possible_mmts)
+
+    @staticmethod
+    def _prepare_terminal_menu_mmts(possible_mmts: List[MMT], title: Optional[str] = None) -> TerminalMenu:
+        if title is None:
+            title = "Please select a Multi-Machine Job of the following:"
+
+        return TerminalMenu([m.name for m in possible_mmts], title=title, clear_menu_on_exit=True)
+
+    @staticmethod
+    def _get_possible_mmts(teamspace: Teamspace) -> Dict[str, MMT]:
+        jobs = {}
+        for j in teamspace.multi_machine_jobs:
+            jobs[j.name] = j
+
+        return jobs
+
+    def _resolve_mmt(self, mmt: Optional[str], teamspace: Teamspace) -> MMT:
+        try:
+            possible_mmts = self._get_possible_mmts(teamspace)
+            if mmt is None:
+                resolved_mmt = self._get_mmt_from_interactive_menu(possible_mmts)
+            else:
+                resolved_mmt = self._get_mmt_from_name(mmt=mmt, possible_mmts=possible_mmts)
+
+            return resolved_mmt
+        except KeyboardInterrupt:
+            raise KeyboardInterrupt from None
+
+        except Exception as e:
+            raise StudioCliError(
+                f"Could not find the given Multi-Machine-Job {mmt} in Teamspace {teamspace.name}. "
+                "Please contact Lightning AI directly to resolve this issue."
+            ) from e

lightning_sdk/cli/run.py

@@ -53,6 +53,10 @@ class _Run:
                 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` 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.
+                Only applicable when submitting docker jobs.
         """
         # TODO: the docstrings from artifacts_local and artifacts_remote don't show up completely,
         # might need to switch to explicit cli definition
@@ -93,6 +97,10 @@ class _Run:
                 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` 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.
+                Only applicable when submitting docker jobs.
         """
         # TODO: the docstrings from artifacts_local and artifacts_remote don't show up completely,
         # might need to switch to explicit cli definition
@@ -118,6 +126,7 @@ class _Run:
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
     ) -> None:
         if machine is None:
             # TODO: infer from studio
@@ -145,6 +154,7 @@ class _Run:
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )
 
     # TODO: sadly, fire displays both Optional[type] and Union[type, None] as Optional[Optional]
@@ -167,6 +177,7 @@ class _Run:
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
     ) -> None:
         if name is None:
             from datetime import datetime
@@ -203,4 +214,5 @@ class _Run:
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )

lightning_sdk/cli/stop.py

@@ -0,0 +1,37 @@
+from typing import Optional
+
+from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
+
+
+class _Stop(_JobAndMMTAction):
+    """Stop resources on the Lightning AI platform."""
+
+    def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Stop a job.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        job = super().job(name=name, teamspace=teamspace)
+
+        job.stop()
+        print(f"Successfully stopped {job.name}!")
+
+    def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+        """Stop a multi-machine job.
+
+        Args:
+            name: the name of the job. If not specified can be selected interactively.
+            teamspace: the name of the teamspace the job lives in.
+                Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace).
+                If not specified can be selected interactively.
+
+        """
+        mmt = super().mmt(name=name, teamspace=teamspace)
+
+        mmt.stop()
+        print(f"Successfully stopped {mmt.name}!")

lightning_sdk/job/base.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any, Dict, Optional, TypedDict, Union
 
+from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace
 
 if TYPE_CHECKING:
@@ -345,7 +346,10 @@ class _BaseJob(ABC):
         studio_name = self._job_api.get_studio_name(self._guaranteed_job)
         if not studio_name:
             raise RuntimeError("Cannot extract studio name from job")
-        return f"https://lightning.ai/{self.teamspace.owner.name}/{self.teamspace.name}/studios/{studio_name}/app?app_id=jobs&job_name={self.name}"
+        return (
+            f"{_get_cloud_url()}/{self.teamspace.owner.name}/{self.teamspace.name}/studios/"
+            f"{studio_name}/app?app_id=jobs&job_name={self.name}"
+        )
 
     @property
     def _guaranteed_job(self) -> Any:

lightning_sdk/job/job.py

@@ -50,15 +50,40 @@ class Job(_BaseJob):
             user: the name of the user owning the :param`teamspace`
                 in case it is owned directly by a user instead of an org.
         """
-        internal_job_cls = _JobV2 if _has_jobs_v2() and not self._force_v1 else _JobV1
-
-        self._internal_job = internal_job_cls(
-            name=name,
-            teamspace=teamspace,
-            org=org,
-            user=user,
-            _fetch_job=_fetch_job,
-        )
+        from lightning_sdk.lightning_cloud.openapi.rest import ApiException
+
+        if _has_jobs_v2() and not self._force_v1:
+            # try with v2 and fall back to v1
+            try:
+                job = _JobV2(
+                    name=name,
+                    teamspace=teamspace,
+                    org=org,
+                    user=user,
+                    _fetch_job=_fetch_job,
+                )
+            except ApiException as e:
+                try:
+                    job = _JobV1(
+                        name=name,
+                        teamspace=teamspace,
+                        org=org,
+                        user=user,
+                        _fetch_job=_fetch_job,
+                    )
+                except ApiException:
+                    raise e from e
+
+        else:
+            job = _JobV1(
+                name=name,
+                teamspace=teamspace,
+                org=org,
+                user=user,
+                _fetch_job=_fetch_job,
+            )
+
+        self._internal_job = job
 
     @classmethod
     def run(

lightning_sdk/job/v2.py

@@ -1,6 +1,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 from lightning_sdk.api.job_api import JobApiV2
+from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.job.base import _BaseJob
 
 if TYPE_CHECKING:
@@ -184,11 +185,18 @@ class _JobV2(_BaseJob):
 
     @property
     def link(self) -> str:
+        mmt_name = self._job_api.get_mmt_name(self._guaranteed_job)
+
         if self._job_api.get_image_name(self._guaranteed_job):
-            return (
-                f"https://lightning.ai/{self.teamspace.owner.name}/{self.teamspace.name}/jobs/{self.name}?app_id=jobs"
-            )
+            if mmt_name:
+                # don't go via the studio unless we use studio env
+                return (
+                    f"{_get_cloud_url()}/{self.teamspace.owner.name}/{self.teamspace.name}/"
+                    f"jobs/{mmt_name}?app_id=mmt&machine_name={self.name}"
+                )
+            return f"{_get_cloud_url()}/{self.teamspace.owner.name}/{self.teamspace.name}/jobs/{self.name}?app_id=jobs"
 
+        # TODO: MMT env with studio
         return super().link
 
     @property

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -327,7 +327,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_capacity_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_key_response import V1DeleteClusterEncryptionKeyResponse
 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_container_response import V1DeleteContainerResponse
 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
@@ -342,6 +341,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instanc
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
+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_managed_endpoint_response import V1DeleteManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
@@ -794,8 +794,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1Vu
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
 from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
 from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody
+from lightning_sdk.lightning_cloud.openapi.models.version_default_body import VersionDefaultBody
 from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body import VersionUploadsBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
 from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec as Gridv1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody as Body