lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.8.8__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.06rc2"
+__version__ = "2025.08.08"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/cli/clusters_menu.py

@@ -1,12 +1,12 @@
 import sys
-from typing import List
+from typing import List, Optional
 
 from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk import Teamspace
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
-from lightning_sdk.lightning_cloud.openapi import Externalv1Cluster, V1ProjectClusterBinding
+from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1ProjectClusterBinding
 
 
 class _ClustersMenu:
@@ -22,18 +22,21 @@ class _ClustersMenu:
 
         return TerminalMenu(cluster_ids, title=title, clear_menu_on_exit=True)
 
-    def _resolve_cluster(self, teamspace: Teamspace) -> Externalv1Cluster:
+    def _resolve_cluster(self, teamspace: Teamspace) -> Optional[str]:
         selected_cluster_id = None
         console = Console()
         try:
-            selected_cluster_id = self._get_cluster_from_interactive_menu(
+            selected_cluster_id = teamspace.default_cloud_account or self._get_cluster_from_interactive_menu(
                 possible_clusters=teamspace.cloud_account_objs
             )
+
             cloud_account_api = CloudAccountApi()
 
-            return cloud_account_api.get_cloud_account(
-                cluster_id=selected_cluster_id, org_id=teamspace.owner.id, project_id=teamspace.id
+            resolved_cluster_obj = cloud_account_api.get_cloud_account(
+                cloud_account_id=selected_cluster_id, org_id=teamspace.owner.id, teamspace_id=teamspace.id
             )
+
+            return None if resolved_cluster_obj.spec.cluster_type == V1ClusterType.GLOBAL else resolved_cluster_obj.id
         except KeyboardInterrupt:
             console.print("Operation cancelled by user")
             sys.exit(0)

lightning_sdk/cli/deploy/serve.py

@@ -15,6 +15,7 @@ from rich.prompt import Confirm
 from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_registry_url
+from lightning_sdk.cli.clusters_menu import _ClustersMenu
 from lightning_sdk.cli.deploy._auth import (
     _AuthMode,
     _Onboarding,
@@ -377,18 +378,24 @@ def _handle_cloud(
     else:
         resolved_teamspace = select_teamspace(teamspace, org, user)
 
+    lightning_containers_cloud_account = cloud_account
+    if not cloud_account:
+        clusters_menu = _ClustersMenu()
+        lightning_containers_cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
+        cloud_account = resolved_teamspace.default_cloud_account
+
     # list containers to create the project if it doesn't exist
     lit_cr = LitContainerApi()
-    lit_cr.list_containers(resolved_teamspace.id, cloud_account=cloud_account)
+    lit_cr.list_containers(resolved_teamspace.id, cloud_account=lightning_containers_cloud_account)
 
     registry_url = _get_registry_url()
     container_basename = repository.split("/")[-1]
     image = (
-        f"{registry_url}/lit-container{f'-{cloud_account}' if cloud_account is not None else ''}/"
-        f"{resolved_teamspace.owner.name}/{resolved_teamspace.name}/{container_basename}"
+        f"{registry_url}/lit-container"
+        + (f"-{lightning_containers_cloud_account}" if lightning_containers_cloud_account is not None else "")
+        + f"/{resolved_teamspace.owner.name}/{resolved_teamspace.name}/{container_basename}"
     )
 
-    cloud_account = cloud_account or resolved_teamspace.default_cloud_account
     if from_onboarding:
         thread = Thread(
             target=ls_deployer.run_on_cloud,
@@ -411,13 +418,17 @@ def _handle_cloud(
         )
         thread.start()
         console.print("🚀 Deployment started")
-        if not _upload_container(console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, cloud_account):
+        if not _upload_container(
+            console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, lightning_containers_cloud_account
+        ):
             thread.join()
             return
         thread.join()
         return
 
-    if not _upload_container(console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, cloud_account):
+    if not _upload_container(
+        console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, lightning_containers_cloud_account
+    ):
         return
 
     deployment_status = ls_deployer.run_on_cloud(

lightning_sdk/cli/download.py

@@ -136,7 +136,7 @@ def folder(
     "--studio",
     default=None,
     help=(
-        "The name of the studio to upload to. "
+        "The name of the studio to download from. "
         "Will show a menu with user's owned studios for selection if not specified. "
         "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
         "The teamspace and studio names can be regular expressions to match, "
@@ -256,7 +256,7 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
     # give user friendlier error message
     except Exception as e:
         raise StudioCliError(
-            f"Could not find the given Studio {studio} to upload files to. "
+            f"Could not find the given Studio {studio} to download files from. "
             "Please contact Lightning AI directly to resolve this issue."
         ) from e
 

lightning_sdk/cli/list.py

@@ -9,7 +9,7 @@ from typing_extensions import Literal
 from lightning_sdk import Job, Machine, Studio, Teamspace
 from lightning_sdk.cli.clusters_menu import _ClustersMenu
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1MultiMachineJob
+from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -247,8 +247,7 @@ def containers(teamspace: Optional[str] = None, cloud_account: Optional[str] = N
     resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
 
     if not cloud_account:
-        cloud_account_obj = clusters_menu._resolve_cluster(resolved_teamspace)
-        cloud_account = "" if cloud_account_obj.spec.cluster_type == V1ClusterType.GLOBAL else cloud_account_obj.id
+        cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
 
     result = api.list_containers(
         teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name, cloud_account=cloud_account

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -326,6 +326,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_deletion_options import V1ClusterDeletionOptions
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_encryption_key import V1ClusterEncryptionKey
+from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_metrics import V1ClusterMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_names import V1ClusterNames
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_proxy import V1ClusterProxy
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_resource_tag import V1ClusterResourceTag
@@ -730,6 +731,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_managed_endp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_quests_response import V1ListQuestsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_response import V1ListSLURMClusterUsersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
+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_studio_jobs_response import V1ListStudioJobsResponse
@@ -884,6 +886,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSH
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_action_type import V1ScheduleActionType
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_run import V1ScheduleRun
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response import V1SearchUsersResponse

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -965,6 +965,7 @@ class AssistantsServiceApi(object):
         :param str user_name:
         :param str org_name:
         :param str model_provider:
+        :param str model_display_name:
         :return: V1Assistant
                  If the method is called asynchronously,
                  returns the request thread.
@@ -989,12 +990,13 @@ class AssistantsServiceApi(object):
         :param str user_name:
         :param str org_name:
         :param str model_provider:
+        :param str model_display_name:
         :return: V1Assistant
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['model_name', 'user_name', 'org_name', 'model_provider']  # noqa: E501
+        all_params = ['model_name', 'user_name', 'org_name', 'model_provider', 'model_display_name']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1027,6 +1029,8 @@ class AssistantsServiceApi(object):
             query_params.append(('orgName', params['org_name']))  # noqa: E501
         if 'model_provider' in params:
             query_params.append(('modelProvider', params['model_provider']))  # noqa: E501
+        if 'model_display_name' in params:
+            query_params.append(('modelDisplayName', params['model_display_name']))  # noqa: E501
 
         header_params = {}
 
@@ -1057,6 +1061,111 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def assistants_service_get_managed_model_assistant2(self, **kwargs) -> 'V1Assistant':  # noqa: E501
+        """Each managed model has a dedicated assistant for direct interaction. By using user_name, org_name, or model_provider as query parameters,  this endpoint retrieves that specific assistant only—excluding any other assistants that may use the same model.  # 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_managed_model_assistant2(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str user_name:
+        :param str org_name:
+        :param str model_provider:
+        :param str model_name:
+        :param str model_display_name:
+        :return: V1Assistant
+                 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_managed_model_assistant2_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.assistants_service_get_managed_model_assistant2_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def assistants_service_get_managed_model_assistant2_with_http_info(self, **kwargs) -> 'V1Assistant':  # noqa: E501
+        """Each managed model has a dedicated assistant for direct interaction. By using user_name, org_name, or model_provider as query parameters,  this endpoint retrieves that specific assistant only—excluding any other assistants that may use the same model.  # 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_managed_model_assistant2_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str user_name:
+        :param str org_name:
+        :param str model_provider:
+        :param str model_name:
+        :param str model_display_name:
+        :return: V1Assistant
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['user_name', 'org_name', 'model_provider', 'model_name', 'model_display_name']  # 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_managed_model_assistant2" % key
+                )
+            params[key] = val
+        del params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'user_name' in params:
+            query_params.append(('userName', params['user_name']))  # noqa: E501
+        if 'org_name' in params:
+            query_params.append(('orgName', params['org_name']))  # noqa: E501
+        if 'model_provider' in params:
+            query_params.append(('modelProvider', params['model_provider']))  # noqa: E501
+        if 'model_name' in params:
+            query_params.append(('modelName', params['model_name']))  # noqa: E501
+        if 'model_display_name' in params:
+            query_params.append(('modelDisplayName', params['model_display_name']))  # 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/agents/managed-model', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1Assistant',  # 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_managed_model_by_name(self, project_id: 'str', managed_endpoint_id: 'str', name: 'str', **kwargs) -> 'V1ManagedModel':  # noqa: E501
         """assistants_service_get_managed_model_by_name  # noqa: E501