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

lightning_sdk/__init__.py

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

lightning_sdk/api/llm_api.py

@@ -16,6 +16,7 @@ from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi.models import (
     StreamResultOfV1ConversationResponseChunk,
     V1ConversationResponseChunk,
+    V1ManagedModel,
     V1ResponseChoice,
     V1ResponseChoiceDelta,
 )
@@ -67,13 +68,23 @@ def authenticate(model: str, shall_confirm: bool = True) -> None:
 class LLMApi:
     def __init__(self) -> None:
         self._client = LightningClient(retry=False, max_tries=0)
+        self._assistant = None
+        self._model = None
 
     def get_assistant(self, model_provider: str, model_name: str, user_name: str, org_name: str) -> str:
         result = self._client.assistants_service_get_managed_model_assistant(
             model_provider=model_provider, model_name=model_name, user_name=user_name, org_name=org_name
         )
+        self._assistant = result
         return result.id
 
+    def get_model_metadata(self, teamspace_id: str, model_name: str) -> V1ManagedModel:
+        if self._assistant and not self._model:
+            self._model = self._client.assistants_service_get_managed_model_by_name(
+                teamspace_id, self._assistant.managed_endpoint_id, model_name
+            )
+        return self._model
+
     def _parse_stream_line(self, decoded_line: str) -> Optional[V1ConversationResponseChunk]:
         try:
             payload = json.loads(decoded_line)
@@ -132,6 +143,7 @@ class LLMApi:
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
         tools: Optional[List[Dict[str, Any]]] = None,
+        reasoning_effort: Optional[str] = None,
         **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
@@ -161,6 +173,9 @@ class LLMApi:
         if max_completion_tokens is not None:
             body["max_completion_tokens"] = max_completion_tokens
 
+        if reasoning_effort:
+            body["reasoning_effort"] = reasoning_effort
+
         if images:
             for image in images:
                 url = image
@@ -191,6 +206,7 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        reasoning_effort: Optional[str] = None,
         **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, AsyncGenerator[V1ConversationResponseChunk, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
@@ -220,6 +236,9 @@ class LLMApi:
         if max_completion_tokens is not None:
             body["max_completion_tokens"] = max_completion_tokens
 
+        if reasoning_effort:
+            body["reasoning_effort"] = reasoning_effort
+
         if images:
             for image in images:
                 url = image

lightning_sdk/api/studio_api.py

@@ -261,12 +261,19 @@ class StudioApi:
         return getattr(getattr(studio.code_status, "in_use", None), "phase", None)
 
     def _request_switch(
-        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool,
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type."""
         compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))
+        if cloud_account:
+            body.compute_config.cluster_override = cloud_account
         self._client.cloud_space_service_update_cloud_space_instance_config(
             id=studio_id,
             project_id=teamspace_id,
@@ -274,11 +281,20 @@ class StudioApi:
         )
 
     def switch_studio_machine(
-        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool,
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type."""
         self._request_switch(
-            studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible
+            studio_id=studio_id,
+            teamspace_id=teamspace_id,
+            machine=machine,
+            interruptible=interruptible,
+            cloud_account=cloud_account,
         )
 
         # Wait until it's time to switch
@@ -324,12 +340,17 @@ class StudioApi:
         machine: Union[Machine, str],
         interruptible: bool,
         progress: Any,  # StudioProgressTracker - avoid circular import
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type with progress tracking."""
         progress.update_progress(10, "Requesting machine switch...")
 
         self._request_switch(
-            studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible
+            studio_id=studio_id,
+            teamspace_id=teamspace_id,
+            machine=machine,
+            interruptible=interruptible,
+            cloud_account=cloud_account,
         )
 
         progress.update_progress(20, "Waiting for machine allocation...")

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -146,6 +146,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_output_body import IdOutput
 from lightning_sdk.lightning_cloud.openapi.models.id_publications_body import IdPublicationsBody
 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_render_body import IdRenderBody
 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_sleepconfig_body import IdSleepconfigBody
@@ -159,6 +160,7 @@ 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
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
+from lightning_sdk.lightning_cloud.openapi.models.kubernetestemplates_id_body import KubernetestemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litdatasets_dataset_id_body import LitdatasetsDatasetIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
@@ -173,6 +175,7 @@ from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import
 from lightning_sdk.lightning_cloud.openapi.models.model_id_visibility_body import ModelIdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.models_id_body import ModelsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
+from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body1 import ModelsModelIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -192,6 +195,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_endpoints_body impo
 from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_kubernetestemplates_body import ProjectIdKubernetestemplatesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_litdatasets_body import ProjectIdLitdatasetsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_litregistry_body import ProjectIdLitregistryBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
@@ -384,6 +388,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_pa
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_model_metrics_response import V1CreateModelMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
@@ -431,6 +436,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response im
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_git_credentials_response import V1DeleteGitCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_template_response import V1DeleteKubernetesTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_response import V1DeleteLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
@@ -529,6 +535,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_firewall_rule import V1FirewallRule
 from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
 from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_function_call import V1FunctionCall
@@ -584,6 +591,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_storage_metadata_response import V1GetProjectStorageMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_settings_response import V1GetSettingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_snowflake_query_response import V1GetSnowflakeQueryResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_temp_bucket_credentials_response import V1GetTempBucketCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_usage_details_response import V1GetUsageDetailsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response import V1GetUserBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
@@ -593,6 +601,7 @@ 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_group_node_metrics import V1GroupNodeMetrics
 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
@@ -602,6 +611,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_ids_logger_metrics import V1IdsLoggerMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_state import V1ImageState
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_event import V1IncidentEvent
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_severity import V1IncidentSeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_type import V1IncidentType
 from lightning_sdk.lightning_cloud.openapi.models.v1_index import V1Index
 from lightning_sdk.lightning_cloud.openapi.models.v1_input import V1Input
 from lightning_sdk.lightning_cloud.openapi.models.v1_instance_overprovisioning_spec import V1InstanceOverprovisioningSpec
@@ -621,10 +633,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_spec import V1JobSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTiming
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
 from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
+from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_indexes import V1K8sIncidentIndexes
 from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template import V1KubernetesTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -698,10 +713,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_lightningapps_response import V1ListGalleryLightningappsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_git_credentials_response import V1ListGitCredentialsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_incident_events_response import V1ListIncidentEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_files_response import V1ListJobFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_resources_response import V1ListJobResourcesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_response import V1ListJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations_response import V1ListJoinableOrganizationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_artifacts_response import V1ListLightningappInstanceArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_events_response import V1ListLightningappInstanceEventsResponse
@@ -875,6 +892,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1Re
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1RefreshResponse
 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_render_kubernetes_template_response import V1RenderKubernetesTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
@@ -884,6 +902,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cloud_space_access_
 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_required_balance_reason import V1RequiredBalanceReason
 from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -358,6 +358,111 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def assistants_service_create_model_metrics(self, body: 'ModelsModelIdBody', model_id: 'str', **kwargs) -> 'V1CreateModelMetricsResponse':  # noqa: E501
+        """assistants_service_create_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_create_model_metrics(body, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelsModelIdBody body: (required)
+        :param str model_id: (required)
+        :return: V1CreateModelMetricsResponse
+                 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_create_model_metrics_with_http_info(body, model_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.assistants_service_create_model_metrics_with_http_info(body, model_id, **kwargs)  # noqa: E501
+            return data
+
+    def assistants_service_create_model_metrics_with_http_info(self, body: 'ModelsModelIdBody', model_id: 'str', **kwargs) -> 'V1CreateModelMetricsResponse':  # noqa: E501
+        """assistants_service_create_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_create_model_metrics_with_http_info(body, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelsModelIdBody body: (required)
+        :param str model_id: (required)
+        :return: V1CreateModelMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', '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_create_model_metrics" % 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 `assistants_service_create_model_metrics`")  # noqa: E501
+        # 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_create_model_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
+        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/agents/metrics/models/{modelId}', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1CreateModelMetricsResponse',  # 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_delete_assistant(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteAssistantResponse':  # noqa: E501
         """assistants_service_delete_assistant  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py

@@ -584,6 +584,107 @@ class DataConnectionServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def data_connection_service_get_temp_bucket_credentials(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetTempBucketCredentialsResponse':  # noqa: E501
+        """data_connection_service_get_temp_bucket_credentials  # 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_get_temp_bucket_credentials(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1GetTempBucketCredentialsResponse
+                 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_get_temp_bucket_credentials_with_http_info(project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.data_connection_service_get_temp_bucket_credentials_with_http_info(project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def data_connection_service_get_temp_bucket_credentials_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetTempBucketCredentialsResponse':  # noqa: E501
+        """data_connection_service_get_temp_bucket_credentials  # 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_get_temp_bucket_credentials_with_http_info(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1GetTempBucketCredentialsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['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 data_connection_service_get_temp_bucket_credentials" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # 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_get_temp_bucket_credentials`")  # 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 `data_connection_service_get_temp_bucket_credentials`")  # 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
+        # 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/projects/{projectId}/data-connections/{id}/temp-bucket-credentials', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetTempBucketCredentialsResponse',  # 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_list_data_connection_artifacts(self, project_id: 'str', id: 'str', **kwargs) -> 'V1ListDataConnectionArtifactsResponse':  # noqa: E501
         """data_connection_service_list_data_connection_artifacts  # noqa: E501