lightning-sdk 0.1.55__py3-none-any.whl → 0.1.56__py3-none-any.whl

lightning_sdk/cli/list.py

@@ -40,8 +40,7 @@ class _List(_TeamspacesMenu):
                 str(studio.cloud_account),
             )
 
-        console = Console()
-        console.print(table)
+        Console().print(table)
 
     def jobs(self, teamspace: Optional[str] = None) -> None:
         """List jobs for a given teamspace.
@@ -81,8 +80,7 @@ class _List(_TeamspacesMenu):
                 f"{j.total_cost:.3f}",
             )
 
-        console = Console()
-        console.print(table)
+        Console().print(table)
 
     def mmts(self, teamspace: Optional[str] = None) -> None:
         """List multi-machine jobs for a given teamspace.
@@ -122,8 +120,7 @@ class _List(_TeamspacesMenu):
                 str(j.total_cost),
             )
 
-        console = Console()
-        console.print(table)
+        Console().print(table)
 
     def containers(self, teamspace: Optional[str] = None) -> None:
         """Display the list of available containers.
@@ -141,8 +138,7 @@ class _List(_TeamspacesMenu):
         table.add_column("CREATED")
         for repo in result:
             table.add_row(repo["REPOSITORY"], repo["IMAGE ID"], repo["CREATED"])
-        console = Console()
-        console.print(table)
+        Console().print(table)
 
     def machines(self) -> None:
         """Display the list of available machines."""
@@ -156,5 +152,4 @@ class _List(_TeamspacesMenu):
         for name in sorted(machine_types):
             table.add_row(name)
 
-        console = Console()
-        console.print(table)
+        Console().print(table)

lightning_sdk/cli/mmts_menu.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.cli.exceptions import StudioCliError
@@ -20,7 +21,7 @@ class _MMTsMenu:
             if j.name == mmt:
                 return j
 
-        print("Could not find Multi-Machine Job {mmt}, please select it from the list:")
+        Console().print("Could not find Multi-Machine Job {mmt}, please select it from the list:")
         return self._get_mmt_from_interactive_menu(possible_mmts)
 
     @staticmethod

lightning_sdk/cli/run.py

@@ -8,7 +8,7 @@ from lightning_sdk.teamspace import Teamspace
 if TYPE_CHECKING:
     from lightning_sdk.cli.legacy import _LegacyLightningCLI
 
-_MACHINE_VALUES = tuple([machine.value for machine in Machine])
+_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
 
 class _Run:
@@ -156,7 +156,7 @@ class _Run:
             machine = "CPU"
         machine_enum: Union[str, Machine]
         try:
-            machine_enum = Machine[machine.upper()]
+            machine_enum = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             machine_enum = machine
 
@@ -222,7 +222,7 @@ class _Run:
             machine = "CPU"
         machine_enum: Union[str, Machine]
         try:
-            machine_enum = Machine[machine.upper()]
+            machine_enum = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             machine_enum = machine
 

lightning_sdk/cli/serve.py

@@ -167,7 +167,6 @@ class _Docker:
         import litserve as ls
         from litserve import docker_builder
 
-        console = Console()
         requirements = ""
         if os.path.exists("requirements.txt"):
             requirements = "-r requirements.txt"
@@ -210,5 +209,5 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
 [bold]To push the container to a registry:[/bold]
 > [underline]docker push {tag}[/underline]
 """
-        console.print(success_msg)
+        Console().print(success_msg)
         return os.path.abspath("Dockerfile")

lightning_sdk/cli/start.py

@@ -7,7 +7,7 @@ class _Start:
     """Start resources on the Lightning AI platform."""
 
     def __init__(self) -> None:
-        _machine_values = tuple([machine.value for machine in Machine])
+        _machine_values = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
         docstr_studio = f"""Start a studio on a given machine.
 
@@ -36,7 +36,7 @@ class _Start:
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         try:
-            resolved_machine = Machine[machine.upper()]
+            resolved_machine = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             resolved_machine = machine
 

lightning_sdk/cli/stop.py

@@ -1,5 +1,7 @@
 from typing import Optional
 
+from rich.console import Console
+
 from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
 from lightning_sdk.studio import Studio
 
@@ -20,7 +22,7 @@ class _Stop(_JobAndMMTAction):
         job = super().job(name=name, teamspace=teamspace)
 
         job.stop()
-        print(f"Successfully stopped {job.name}!")
+        Console().print(f"Successfully stopped {job.name}!")
 
     def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Stop a multi-machine job.
@@ -35,7 +37,7 @@ class _Stop(_JobAndMMTAction):
         mmt = super().mmt(name=name, teamspace=teamspace)
 
         mmt.stop()
-        print(f"Successfully stopped {mmt.name}!")
+        Console().print(f"Successfully stopped {mmt.name}!")
 
     def studio(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Stop a running studio.
@@ -60,4 +62,4 @@ class _Stop(_JobAndMMTAction):
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         studio.stop()
-        print("Studio successfully stopped")
+        Console().print("Studio successfully stopped")

lightning_sdk/cli/studios_menu.py

@@ -1,7 +1,9 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
+from lightning_sdk import Studio
 from lightning_sdk.api import OrgApi, TeamspaceApi
 from lightning_sdk.user import User
 
@@ -18,7 +20,7 @@ class _StudiosMenu:
             if st["teamspace"] == teamspace and name == st["name"]:
                 return st
 
-        print("Could not find Studio {studio}, please select it from the list:")
+        Console().print("Could not find Studio {studio}, please select it from the list:")
         return self._get_studio_from_interactive_menu(possible_studios)
 
     @staticmethod
@@ -76,3 +78,24 @@ class _StudiosMenu:
                 possible_studios.append({"name": st.name, **teamspaces[teamspace_id]})
 
         return possible_studios
+
+    def _get_studio(self, name: str, teamspace: str) -> Studio:
+        """Get studio object from name and teamspace.
+
+        Args:
+            name: Name of the studio
+            teamspace: Name of the teamspace
+        """
+        if teamspace:
+            ts_splits = teamspace.split("/")
+            if len(ts_splits) != 2:
+                raise ValueError(f"Teamspace should be of format <OWNER>/<TEAMSPACE_NAME> but got {teamspace}")
+            owner, teamspace = ts_splits
+        else:
+            owner, teamspace = None, None
+
+        try:
+            studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False)
+        except (RuntimeError, ValueError):
+            studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
+        return studio

lightning_sdk/cli/switch.py

@@ -7,7 +7,7 @@ class _Switch:
     """Switch machines for resources on the Lightning AI platform."""
 
     def __init__(self) -> None:
-        _machine_values = tuple([machine.value for machine in Machine])
+        _machine_values = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
         docstr_studio = f"""Switch a studio to a given machine.
 
@@ -36,7 +36,7 @@ class _Switch:
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         try:
-            resolved_machine = Machine[machine.upper()]
+            resolved_machine = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             resolved_machine = machine
 

lightning_sdk/cli/teamspace_menu.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.api import OrgApi
@@ -26,7 +27,7 @@ class _TeamspacesMenu:
             if ts["name"] == name and (ts["user"] == owner or ts["org"] == owner):
                 return ts
 
-        print("Could not find Teamspace {teamspace}, please select it from the list:")
+        Console().print("Could not find Teamspace {teamspace}, please select it from the list:")
         return self._get_teamspace_from_interactive_menu(possible_teamspaces)
 
     @staticmethod

lightning_sdk/cli/upload.py

@@ -71,6 +71,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             If not specified, will use the file or directory name of the path you want to upload
             and place it in your home directory.
         """
+        console = Console()
         if remote_path is None:
             remote_path = os.path.basename(path)
 
@@ -81,7 +82,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
 
         selected_studio = self._resolve_studio(studio)
 
-        print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
+        console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
         pairs = {}
         for root, _, files in os.walk(path):
@@ -112,7 +113,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             + "/studios/"
             + selected_studio.name
         )
-        print(f"See your files at {studio_url}")
+        console.print(f"See your files at {studio_url}")
 
     def file(self, path: str, studio: Optional[str] = None, remote_path: Optional[str] = None) -> None:
         """Upload a file to a Studio.
@@ -126,6 +127,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             If not specified, will use the name of the file you want to upload
             and place it in your home directory.
         """
+        console = Console()
         if remote_path is None:
             remote_path = os.path.basename(path)
 
@@ -136,7 +138,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
 
         selected_studio = self._resolve_studio(studio)
 
-        print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
+        console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
         self._single_file_upload(selected_studio, path, remote_path, True)
 
@@ -149,7 +151,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             + "/studios/"
             + selected_studio.name
         )
-        print(f"See your file at {studio_url}")
+        console.print(f"See your file at {studio_url}")
 
     def container(self, container: str, tag: str = "latest", teamspace: Optional[str] = None) -> None:
         teamspace = self._resolve_teamspace(teamspace)

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -124,6 +124,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_publications_body import Id
 from lightning_sdk.lightning_cloud.openapi.models.id_publications_body1 import IdPublicationsBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdReleaseBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
+from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
 from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
@@ -446,6 +447,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
@@ -701,6 +703,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
@@ -710,6 +713,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resource
 from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
 from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
 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_role import V1Role
 from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1529,6 +1529,111 @@ class ClusterServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cluster_service_get_cluster_accelerator_demand(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterAcceleratorDemandResponse':  # noqa: E501
+        """cluster_service_get_cluster_accelerator_demand  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_get_cluster_accelerator_demand(cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str cluster_id: (required)
+        :param str id: (required)
+        :param bool spot:
+        :return: V1GetClusterAcceleratorDemandResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def cluster_service_get_cluster_accelerator_demand_with_http_info(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterAcceleratorDemandResponse':  # noqa: E501
+        """cluster_service_get_cluster_accelerator_demand  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str cluster_id: (required)
+        :param str id: (required)
+        :param bool spot:
+        :return: V1GetClusterAcceleratorDemandResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['cluster_id', 'id', 'spot']  # 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 cluster_service_get_cluster_accelerator_demand" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'cluster_id' is set
+        if ('cluster_id' not in params or
+                params['cluster_id'] is None):
+            raise ValueError("Missing the required parameter `cluster_id` when calling `cluster_service_get_cluster_accelerator_demand`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `cluster_service_get_cluster_accelerator_demand`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cluster_id' in params:
+            path_params['clusterId'] = params['cluster_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+        if 'spot' in params:
+            query_params.append(('spot', params['spot']))  # noqa: E501
+
+        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/core/clusters/{clusterId}/accelerator/{id}/demand', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetClusterAcceleratorDemandResponse',  # 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 cluster_service_get_cluster_availability(self, **kwargs) -> 'V1ClusterAvailability':  # noqa: E501
         """cluster_service_get_cluster_availability  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py

@@ -3035,6 +3035,119 @@ class JobsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def jobs_service_report_restart_timings(self, body: 'IdReportrestarttimingsBody', project_id: 'str', id: 'str', **kwargs) -> 'V1ReportRestartTimingsResponse':  # noqa: E501
+        """The tired proxy collects the time at which the user logs started and inform the CP  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.jobs_service_report_restart_timings(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdReportrestarttimingsBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1ReportRestartTimingsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.jobs_service_report_restart_timings_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.jobs_service_report_restart_timings_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def jobs_service_report_restart_timings_with_http_info(self, body: 'IdReportrestarttimingsBody', project_id: 'str', id: 'str', **kwargs) -> 'V1ReportRestartTimingsResponse':  # noqa: E501
+        """The tired proxy collects the time at which the user logs started and inform the CP  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.jobs_service_report_restart_timings_with_http_info(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdReportrestarttimingsBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1ReportRestartTimingsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', '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 jobs_service_report_restart_timings" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `jobs_service_report_restart_timings`")  # noqa: E501
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `jobs_service_report_restart_timings`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `jobs_service_report_restart_timings`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/jobs/{id}/report-restart-timings', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ReportRestartTimingsResponse',  # 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 jobs_service_restore_deployment_release(self, body: 'object', project_id: 'str', deployment_id: 'str', id: 'str', **kwargs) -> 'V1RestoreDeploymentReleaseResponse':  # noqa: E501
         """jobs_service_restore_deployment_release  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py

@@ -53,7 +53,7 @@ class LitRegistryServiceApi(object):
 
         :param async_req bool
         :param ProjectIdLitregistryBody body: (required)
-        :param str project_id: (required)
+        :param str project_id: The teamspace project_id (required)
         :return: V1LitRegistryProject
                  If the method is called asynchronously,
                  returns the request thread.
@@ -75,7 +75,7 @@ class LitRegistryServiceApi(object):
 
         :param async_req bool
         :param ProjectIdLitregistryBody body: (required)
-        :param str project_id: (required)
+        :param str project_id: The teamspace project_id (required)
         :return: V1LitRegistryProject
                  If the method is called asynchronously,
                  returns the request thread.
@@ -563,6 +563,7 @@ class LitRegistryServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str lit_repo_name: (required)
+        :param bool latest_only:
         :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -585,12 +586,13 @@ class LitRegistryServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str lit_repo_name: (required)
+        :param bool latest_only:
         :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'lit_repo_name']  # noqa: E501
+        all_params = ['project_id', 'lit_repo_name', 'latest_only']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -623,6 +625,8 @@ class LitRegistryServiceApi(object):
             path_params['litRepoName'] = params['lit_repo_name']  # noqa: E501
 
         query_params = []
+        if 'latest_only' in params:
+            query_params.append(('latestOnly', params['latest_only']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py

@@ -578,7 +578,6 @@ class ProjectsServiceApi(object):
 
         :param async_req bool
         :param str id: (required)
-        :param bool skip_balance_check:
         :param str balance_destination_id:
         :param bool is_org:
         :return: V1DeleteProjectResponse
@@ -602,7 +601,6 @@ class ProjectsServiceApi(object):
 
         :param async_req bool
         :param str id: (required)
-        :param bool skip_balance_check:
         :param str balance_destination_id:
         :param bool is_org:
         :return: V1DeleteProjectResponse
@@ -610,7 +608,7 @@ class ProjectsServiceApi(object):
                  returns the request thread.
         """
 
-        all_params = ['id', 'skip_balance_check', 'balance_destination_id', 'is_org']  # noqa: E501
+        all_params = ['id', 'balance_destination_id', 'is_org']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -637,8 +635,6 @@ class ProjectsServiceApi(object):
             path_params['id'] = params['id']  # noqa: E501
 
         query_params = []
-        if 'skip_balance_check' in params:
-            query_params.append(('skipBalanceCheck', params['skip_balance_check']))  # noqa: E501
         if 'balance_destination_id' in params:
             query_params.append(('balanceDestinationId', params['balance_destination_id']))  # noqa: E501
         if 'is_org' in params: