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

lightning_sdk/cli/list.py

@@ -1,27 +1,58 @@
-from typing import Optional
+from contextlib import suppress
+from typing import Callable, Optional
 
 from rich.console import Console
 from rich.table import Table
+from typing_extensions import Literal
 
-from lightning_sdk import Machine
+from lightning_sdk import Job, Machine, Studio, Teamspace
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
+from lightning_sdk.utils.resolve import _get_authed_user
 
 
 class _List(_TeamspacesMenu):
     """List resources on the Lightning AI platform."""
 
-    def studios(self, teamspace: Optional[str] = None) -> None:
+    @staticmethod
+    def _sort_studios_key(sort_by: str) -> Callable[[Studio], str]:
+        """Return a key function to sort studios by a given attribute."""
+        sort_key_map = {
+            "name": lambda s: str(s.name or ""),
+            "teamspace": lambda s: str(s.teamspace.name or ""),
+            "status": lambda s: str(s.status or ""),
+            "machine": lambda s: str(s.machine or ""),
+            "cloud-account": lambda s: str(s.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda s: s.name)
+
+    def studios(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "status", "machine", "cloud-account"]] = None,
+    ) -> None:
         """List studios for a given teamspace.
 
         Args:
             teamspace: the teamspace to list studios from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all studios in all teamspaces.
+            sort_by: the attribute to sort the studios by.
+                Can be one of "name", "teamspace", "status", "machine", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        studios = resolved_teamspace.studios
+        studios = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                studios.extend(teamspace.studios)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            studios = resolved_teamspace.studios
 
         table = Table(
             pad_edge=True,
@@ -31,32 +62,59 @@ class _List(_TeamspacesMenu):
         table.add_column("Status")
         table.add_column("Machine")
         table.add_column("Cloud account")
-        for studio in studios:
+        for studio in sorted(studios, key=self._sort_studios_key(sort_by)):
             table.add_row(
                 studio.name,
                 f"{studio.teamspace.owner.name}/{studio.teamspace.name}",
                 str(studio.status),
-                str(studio.machine) if studio.machine is not None else None,
+                str(studio.machine) if studio.machine is not None else None,  # when None the cell is empty
                 str(studio.cloud_account),
             )
 
         Console().print(table)
 
-    def jobs(self, teamspace: Optional[str] = None) -> None:
+    @staticmethod
+    def _sort_jobs_key(sort_by: str) -> Callable[[Job], str]:
+        """Return a key function to sort studios by a given attribute."""
+        sort_key_map = {
+            "name": lambda j: str(j.name or ""),
+            "teamspace": lambda j: str(j.teamspace.name or ""),
+            "status": lambda j: str(j.status or ""),
+            "machine": lambda j: str(j.machine or ""),
+            "studio": lambda j: str(j.studio or ""),
+            "image": lambda j: str(j.image or ""),
+            "cloud-account": lambda j: str(j.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda j: j.name)
+
+    def jobs(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "status", "studio", "machine", "image", "cloud-account"]] = None,
+    ) -> None:
         """List jobs for a given teamspace.
 
         Args:
             teamspace: the teamspace to list jobs from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all jobs in all teamspaces.
+            sort_by: the attribute to sort the jobs by.
+                Can be one of "name", "teamspace", "status", "studio", "machine", "image", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        jobs = resolved_teamspace.jobs
+        jobs = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                jobs.extend(teamspace.jobs)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            jobs = resolved_teamspace.jobs
 
-        table = Table(
-            pad_edge=True,
-        )
+        table = Table(pad_edge=True)
         table.add_column("Name")
         table.add_column("Teamspace")
         table.add_column("Studio")
@@ -64,35 +122,64 @@ class _List(_TeamspacesMenu):
         table.add_column("Status")
         table.add_column("Machine")
         table.add_column("Total Cost")
-        for j in jobs:
+        for j in sorted(jobs, key=self._sort_jobs_key(sort_by)):
             # we know we just fetched these, so no need to refetch
             j._prevent_refetch_latest = True
             j._internal_job._prevent_refetch_latest = True
 
             studio = j.studio
-            table.add_row(
-                j.name,
-                f"{j.teamspace.owner.name}/{j.teamspace.name}",
-                studio.name if studio else None,
-                j.image,
-                str(j.status),
-                str(j.machine),
-                f"{j.total_cost:.3f}",
-            )
+            with suppress(RuntimeError):
+                table.add_row(
+                    j.name,
+                    f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                    studio.name if studio else None,
+                    j.image,
+                    str(j.status) if j.status is not None else None,
+                    str(j.machine),
+                    f"{j.total_cost:.3f}",
+                )
 
         Console().print(table)
 
-    def mmts(self, teamspace: Optional[str] = None) -> None:
+    def _sort_mmts_key(self, sort_by: str) -> Callable[[V1MultiMachineJob], str]:
+        """Return a key function to sort multi-machine jobs by a given attribute."""
+        sort_key_map = {
+            "name": lambda j: str(j.name or ""),
+            "teamspace": lambda j: str(j.teamspace.name or ""),
+            "studio": lambda j: str(j.studio.name or ""),
+            "image": lambda j: str(j.image or ""),
+            "status": lambda j: str(j.status or ""),
+            "machine": lambda j: str(j.machine or ""),
+            "cloud-account": lambda j: str(j.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda j: j.name)
+
+    def mmts(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "studio", "image", "status", "machine", "cloud-account"]] = None,
+    ) -> None:
         """List multi-machine jobs for a given teamspace.
 
         Args:
             teamspace: the teamspace to list jobs from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all multi-machine jobs in all teamspaces.
+            sort_by: the attribute to sort the multi-machine jobs by.
+                Can be one of "name", "teamspace", "studio", "image", "status", "machine", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        jobs = resolved_teamspace.multi_machine_jobs
+        jobs = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                jobs.extend(teamspace.multi_machine_jobs)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            jobs = resolved_teamspace.multi_machine_jobs
 
         table = Table(pad_edge=True)
         table.add_column("Name")
@@ -103,22 +190,24 @@ class _List(_TeamspacesMenu):
         table.add_column("Machine")
         table.add_column("Num Machines")
         table.add_column("Total Cost")
-        for j in jobs:
+        for j in sorted(jobs, key=self._sort_mmts_key(sort_by)):
             # we know we just fetched these, so no need to refetch
             j._prevent_refetch_latest = True
-            j._internal_job._prevent_refetch_latest = True
+            with suppress(AttributeError):
+                j._internal_job._prevent_refetch_latest = True
 
             studio = j.studio
-            table.add_row(
-                j.name,
-                f"{j.teamspace.owner.name}/{j.teamspace.name}",
-                studio.name if studio else None,
-                j.image,
-                str(j.status),
-                str(j.machine),
-                str(j.num_machines),
-                str(j.total_cost),
-            )
+            with suppress(RuntimeError):
+                table.add_row(
+                    j.name,
+                    f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                    studio.name if studio else None,
+                    j.image,
+                    str(j.status),
+                    str(j.machine),
+                    str(j.num_machines),
+                    str(j.total_cost),
+                )
 
         Console().print(table)
 

lightning_sdk/cli/run.py

@@ -162,10 +162,6 @@ class _Run:
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
-        if cloud_account is None:
-            cloud_account = resolved_teamspace.default_cloud_account
-        machine_enum = Machine(machine.upper())
-
         path_mappings_dict = self._resolve_path_mapping(path_mappings=path_mappings)
 
         Job.run(
@@ -227,8 +223,6 @@ class _Run:
             machine_enum = machine
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
-        if cloud_account is None:
-            cloud_account = resolved_teamspace.default_cloud_account
 
         if image is None:
             raise RuntimeError("Image needs to be specified to run a multi-machine job")

lightning_sdk/cli/teamspace_menu.py

@@ -27,7 +27,7 @@ class _TeamspacesMenu:
             if ts["name"] == name and (ts["user"] == owner or ts["org"] == owner):
                 return ts
 
-        Console().print("Could not find Teamspace {teamspace}, please select it from the list:")
+        Console().print(f"Could not find Teamspace {teamspace}, please select it from the list:")
         return self._get_teamspace_from_interactive_menu(possible_teamspaces)
 
     @staticmethod

lightning_sdk/helpers.py

@@ -1,8 +1,13 @@
 import functools
+import importlib
+import os
+import sys
 import warnings
 from typing import Optional
 
 import requests
+import tqdm
+import tqdm.std
 from packaging import version as packaging_version
 
 __package_name__ = "lightning-sdk"
@@ -47,3 +52,18 @@ def _check_version_and_prompt_upgrade(curr_version: str) -> None:
             UserWarning,
         )
     return
+
+
+def _set_tqdm_envvars_noninteractive() -> None:
+    # note: stderr is the default stream tqdm writes progressbars to
+    # so we check that one.
+    if os.isatty(sys.stderr.fileno()):
+        os.unsetenv("TQDM_POSITION")
+        os.unsetenv("TQDM_MININTERVAL")
+    else:
+        # makes use of https://github.com/tqdm/tqdm/blob/master/tqdm/utils.py#L34 to set defaults
+        os.environ.update({"TQDM_POSITION": "-1", "TQDM_MININTERVAL": "1"})
+
+    # reload to make sure env vars are parsed again
+    importlib.reload(tqdm.std)
+    importlib.reload(tqdm)

lightning_sdk/job/job.py

@@ -235,7 +235,7 @@ class Job(_BaseJob):
         return self._internal_job.delete()
 
     @property
-    def status(self) -> "Status":
+    def status(self) -> Optional["Status"]:
         """The current status of the job."""
         return self._internal_job.status
 

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -130,6 +130,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStora
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
 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.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
@@ -182,6 +183,7 @@ from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import
 from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
 from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
 from lightning_sdk.lightning_cloud.openapi.models.serviceexecution_id_body import ServiceexecutionIdBody
+from lightning_sdk.lightning_cloud.openapi.models.setup import Setup
 from lightning_sdk.lightning_cloud.openapi.models.slurm_jobs_body import SlurmJobsBody
 from lightning_sdk.lightning_cloud.openapi.models.snowflake_export_body import SnowflakeExportBody
 from lightning_sdk.lightning_cloud.openapi.models.snowflake_query_body import SnowflakeQueryBody
@@ -440,6 +442,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_component import V1GalleryComponent
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection import V1GcpDataConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection_setup import V1GcpDataConnectionSetup
 from lightning_sdk.lightning_cloud.openapi.models.v1_generate_ssh_key_pair_request import V1GenerateSSHKeyPairRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response import V1GetAffiliateLinkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
@@ -737,6 +740,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1S
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_health import V1ServiceHealth
 from lightning_sdk.lightning_cloud.openapi.models.v1_setup_confirmed_ssh_public_key_response import V1SetupConfirmedSSHPublicKeyResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_response import V1SetupDataConnectionResponse
 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_slurm_node import V1SlurmNode
@@ -773,6 +777,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publicat
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse

lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py

@@ -920,6 +920,111 @@ class DataConnectionServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def data_connection_service_setup_data_connection(self, body: 'Setup', project_id: 'str', **kwargs) -> 'V1SetupDataConnectionResponse':  # noqa: E501
+        """data_connection_service_setup_data_connection  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.data_connection_service_setup_data_connection(body, project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param Setup body: (required)
+        :param str project_id: (required)
+        :return: V1SetupDataConnectionResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.data_connection_service_setup_data_connection_with_http_info(body, project_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.data_connection_service_setup_data_connection_with_http_info(body, project_id, **kwargs)  # noqa: E501
+            return data
+
+    def data_connection_service_setup_data_connection_with_http_info(self, body: 'Setup', project_id: 'str', **kwargs) -> 'V1SetupDataConnectionResponse':  # noqa: E501
+        """data_connection_service_setup_data_connection  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.data_connection_service_setup_data_connection_with_http_info(body, project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param Setup body: (required)
+        :param str project_id: (required)
+        :return: V1SetupDataConnectionResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_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 data_connection_service_setup_data_connection" % 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 `data_connection_service_setup_data_connection`")  # 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 `data_connection_service_setup_data_connection`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_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}/data-connections/setup', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1SetupDataConnectionResponse',  # 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 data_connection_service_update_data_connection(self, body: 'Update', project_id: 'str', id: 'str', **kwargs) -> 'V1DataConnection':  # noqa: E501
         """data_connection_service_update_data_connection  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py

@@ -3503,6 +3503,119 @@ class JobsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def jobs_service_update_deployment_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateDeploymentVisibilityResponse':  # noqa: E501
+        """UpdateDeploymentVisibility updates the deployment visibility, mainly switch between public and private  # 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_update_deployment_visibility(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateDeploymentVisibilityResponse
+                 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_update_deployment_visibility_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.jobs_service_update_deployment_visibility_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def jobs_service_update_deployment_visibility_with_http_info(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateDeploymentVisibilityResponse':  # noqa: E501
+        """UpdateDeploymentVisibility updates the deployment visibility, mainly switch between public and private  # 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_update_deployment_visibility_with_http_info(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateDeploymentVisibilityResponse
+                 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_update_deployment_visibility" % 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_update_deployment_visibility`")  # 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_update_deployment_visibility`")  # 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_update_deployment_visibility`")  # 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}/deployments/{id}/visibility', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateDeploymentVisibilityResponse',  # 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_update_job(self, body: 'JobsIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Job':  # noqa: E501
         """jobs_service_update_job  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py

@@ -1534,7 +1534,7 @@ class LitLoggerServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
+    def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
         """lit_logger_service_update_metrics_stream_visibility  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1543,7 +1543,7 @@ class LitLoggerServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdVisibilityBody body: (required)
+        :param IdVisibilityBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1UpdateMetricsStreamVisibilityResponse
@@ -1557,7 +1557,7 @@ class LitLoggerServiceApi(object):
             (data) = self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
             return data
 
-    def lit_logger_service_update_metrics_stream_visibility_with_http_info(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
+    def lit_logger_service_update_metrics_stream_visibility_with_http_info(self, body: 'IdVisibilityBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse':  # noqa: E501
         """lit_logger_service_update_metrics_stream_visibility  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1566,7 +1566,7 @@ class LitLoggerServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdVisibilityBody body: (required)
+        :param IdVisibilityBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1UpdateMetricsStreamVisibilityResponse