lightning-sdk 2025.8.28__py3-none-any.whl → 2025.9.2__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.28"
+__version__ = "2025.09.02"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/teamspace_api.py

@@ -390,6 +390,8 @@ class TeamspaceApi:
             params=query_params,
             stream=True,
         )
+        if r.status_code == 404:
+            raise FileNotFoundError(f"File {path} not found")
         total_length = int(r.headers.get("content-length"))
 
         if progress_bar:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -250,6 +250,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_aggregated_pod_metrics import V1AggregatedPodMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
 from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
@@ -280,6 +281,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_body import V1Body
 from lightning_sdk.lightning_cloud.openapi.models.v1_build_spec import V1BuildSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V1CPUSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_running_cloud_space_instance_transfer_response import V1CancelRunningCloudSpaceInstanceTransferResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
 from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
@@ -395,6 +397,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_request import V1CreateSubscriptionCheckoutSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_response import V1CreateSubscriptionCheckoutSessionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_user_secret_request import V1CreateUserSecretRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_daily_model_metrics import V1DailyModelMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_daily_usage import V1DailyUsage
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1DataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
@@ -548,6 +551,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_open_ports_response import V1GetCloudSpaceInstanceOpenPortsResponse
 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_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_required_balance_status_response import V1GetCloudSpaceRequiredBalanceStatusResponse
 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
@@ -558,6 +562,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_tele
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_response import V1GetJobSystemMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_latest_model_metrics_response import V1GetLatestModelMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_system_metrics_response import V1GetLightningappInstanceSystemMetricsResponse
@@ -570,6 +575,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_metrics_response import V1GetModelMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_total_usage_metrics_response import V1GetModelTotalUsageMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
@@ -586,6 +593,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
+from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_request import V1GuestLoginRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_response import V1GuestLoginResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_guest_user import V1GuestUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_header import V1Header
 from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_exec import V1HealthCheckExec
 from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get import V1HealthCheckHttpGet
@@ -779,6 +789,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1Metri
 from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
 from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
+from lightning_sdk.lightning_cloud.openapi.models.v1_model_metrics import V1ModelMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
 from lightning_sdk.lightning_cloud.openapi.models.v1_modify_filesystem_volume_response import V1ModifyFilesystemVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

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

lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py

@@ -213,6 +213,103 @@ class AuthServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def auth_service_guest_login(self, body: 'V1GuestLoginRequest', **kwargs) -> 'V1GuestLoginResponse':  # noqa: E501
+        """auth_service_guest_login  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_guest_login(body, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param V1GuestLoginRequest body: (required)
+        :return: V1GuestLoginResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.auth_service_guest_login_with_http_info(body, **kwargs)  # noqa: E501
+        else:
+            (data) = self.auth_service_guest_login_with_http_info(body, **kwargs)  # noqa: E501
+            return data
+
+    def auth_service_guest_login_with_http_info(self, body: 'V1GuestLoginRequest', **kwargs) -> 'V1GuestLoginResponse':  # noqa: E501
+        """auth_service_guest_login  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_guest_login_with_http_info(body, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param V1GuestLoginRequest body: (required)
+        :return: V1GuestLoginResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body']  # 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 auth_service_guest_login" % 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 `auth_service_guest_login`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        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/auth/guest-login', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GuestLoginResponse',  # 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 auth_service_login(self, body: 'V1LoginRequest', **kwargs) -> 'V1LoginResponse':  # noqa: E501
         """auth_service_login  # noqa: E501