lightning-sdk 2025.8.18.post0__py3-none-any.whl → 2025.8.21__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py

@@ -160,6 +160,119 @@ class K8SClusterServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def k8_s_cluster_service_list_aggregated_pod_metrics(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListAggregatedPodMetricsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_aggregated_pod_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.k8_s_cluster_service_list_aggregated_pod_metrics(project_id, cluster_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str namespace:
+        :param datetime start: Date range.
+        :param datetime end:
+        :return: V1ListAggregatedPodMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.k8_s_cluster_service_list_aggregated_pod_metrics_with_http_info(project_id, cluster_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.k8_s_cluster_service_list_aggregated_pod_metrics_with_http_info(project_id, cluster_id, **kwargs)  # noqa: E501
+            return data
+
+    def k8_s_cluster_service_list_aggregated_pod_metrics_with_http_info(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListAggregatedPodMetricsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_aggregated_pod_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.k8_s_cluster_service_list_aggregated_pod_metrics_with_http_info(project_id, cluster_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str namespace:
+        :param datetime start: Date range.
+        :param datetime end:
+        :return: V1ListAggregatedPodMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cluster_id', 'namespace', 'start', 'end']  # 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 k8_s_cluster_service_list_aggregated_pod_metrics" % 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 `k8_s_cluster_service_list_aggregated_pod_metrics`")  # noqa: E501
+        # verify the required parameter 'cluster_id' is set
+        if ('cluster_id' not in params or
+                params['cluster_id'] is None):
+            raise ValueError("Missing the required parameter `cluster_id` when calling `k8_s_cluster_service_list_aggregated_pod_metrics`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cluster_id' in params:
+            path_params['clusterId'] = params['cluster_id']  # noqa: E501
+
+        query_params = []
+        if 'namespace' in params:
+            query_params.append(('namespace', params['namespace']))  # noqa: E501
+        if 'start' in params:
+            query_params.append(('start', params['start']))  # noqa: E501
+        if 'end' in params:
+            query_params.append(('end', params['end']))  # 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/projects/{projectId}/clusters/{clusterId}/aggregated-metrics/pods', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListAggregatedPodMetricsResponse',  # 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 k8_s_cluster_service_list_cluster_metrics(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListClusterMetricsResponse':  # noqa: E501
         """k8_s_cluster_service_list_cluster_metrics  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -590,6 +590,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_like_status import V1LikeSt
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_affiliate_links_response import V1ListAffiliateLinksResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_response import V1ListAgentJobArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_aggregated_pod_metrics_response import V1ListAggregatedPodMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_blog_posts_response import V1ListBlogPostsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
@@ -804,6 +805,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest import V1Quest
 from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1QuestStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
+from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse

lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py

@@ -46,7 +46,7 @@ class AssistantIdConversationsBody(object):
         'conversation_id': 'str',
         'ephemeral': 'bool',
         'internal_conversation': 'bool',
-        'max_tokens': 'str',
+        'max_completion_tokens': 'str',
         'message': 'V1Message',
         'metadata': 'dict(str, str)',
         'name': 'str',
@@ -66,7 +66,7 @@ class AssistantIdConversationsBody(object):
         'conversation_id': 'conversationId',
         'ephemeral': 'ephemeral',
         'internal_conversation': 'internalConversation',
-        'max_tokens': 'maxTokens',
+        'max_completion_tokens': 'maxCompletionTokens',
         'message': 'message',
         'metadata': 'metadata',
         'name': 'name',
@@ -80,14 +80,14 @@ class AssistantIdConversationsBody(object):
         'tools': 'tools'
     }
 
-    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None, tools: 'list[V1Tool]' =None):  # noqa: E501
+    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_completion_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None, tools: 'list[V1Tool]' =None):  # noqa: E501
         """AssistantIdConversationsBody - a model defined in Swagger"""  # noqa: E501
         self._auto_name = None
         self._billing_project_id = None
         self._conversation_id = None
         self._ephemeral = None
         self._internal_conversation = None
-        self._max_tokens = None
+        self._max_completion_tokens = None
         self._message = None
         self._metadata = None
         self._name = None
@@ -110,8 +110,8 @@ class AssistantIdConversationsBody(object):
             self.ephemeral = ephemeral
         if internal_conversation is not None:
             self.internal_conversation = internal_conversation
-        if max_tokens is not None:
-            self.max_tokens = max_tokens
+        if max_completion_tokens is not None:
+            self.max_completion_tokens = max_completion_tokens
         if message is not None:
             self.message = message
         if metadata is not None:
@@ -241,25 +241,25 @@ class AssistantIdConversationsBody(object):
         self._internal_conversation = internal_conversation
 
     @property
-    def max_tokens(self) -> 'str':
-        """Gets the max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+    def max_completion_tokens(self) -> 'str':
+        """Gets the max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
 
 
-        :return: The max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+        :return: The max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
         :rtype: str
         """
-        return self._max_tokens
+        return self._max_completion_tokens
 
-    @max_tokens.setter
-    def max_tokens(self, max_tokens: 'str'):
-        """Sets the max_tokens of this AssistantIdConversationsBody.
+    @max_completion_tokens.setter
+    def max_completion_tokens(self, max_completion_tokens: 'str'):
+        """Sets the max_completion_tokens of this AssistantIdConversationsBody.
 
 
-        :param max_tokens: The max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+        :param max_completion_tokens: The max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
         :type: str
         """
 
-        self._max_tokens = max_tokens
+        self._max_completion_tokens = max_completion_tokens
 
     @property
     def message(self) -> 'V1Message':

lightning_sdk/lightning_cloud/openapi/models/id_codeconfig_body.py

@@ -43,47 +43,32 @@ class IdCodeconfigBody(object):
     swagger_types = {
         'compute_config': 'V1UserRequestedComputeConfig',
         'data_connection_mounts': 'list[V1DataConnectionMount]',
-        'disable_auto_shutdown': 'bool',
         'ide': 'str',
-        'idle_shutdown_seconds': 'int',
-        'port_plugin_port': 'str',
-        'switch_to_default_machine_on_idle': 'bool'
+        'port_plugin_port': 'str'
     }
 
     attribute_map = {
         'compute_config': 'computeConfig',
         'data_connection_mounts': 'dataConnectionMounts',
-        'disable_auto_shutdown': 'disableAutoShutdown',
         'ide': 'ide',
-        'idle_shutdown_seconds': 'idleShutdownSeconds',
-        'port_plugin_port': 'portPluginPort',
-        'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle'
+        'port_plugin_port': 'portPluginPort'
     }
 
-    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disable_auto_shutdown: 'bool' =None, ide: 'str' =None, idle_shutdown_seconds: 'int' =None, port_plugin_port: 'str' =None, switch_to_default_machine_on_idle: 'bool' =None):  # noqa: E501
+    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, ide: 'str' =None, port_plugin_port: 'str' =None):  # noqa: E501
         """IdCodeconfigBody - a model defined in Swagger"""  # noqa: E501
         self._compute_config = None
         self._data_connection_mounts = None
-        self._disable_auto_shutdown = None
         self._ide = None
-        self._idle_shutdown_seconds = None
         self._port_plugin_port = None
-        self._switch_to_default_machine_on_idle = None
         self.discriminator = None
         if compute_config is not None:
             self.compute_config = compute_config
         if data_connection_mounts is not None:
             self.data_connection_mounts = data_connection_mounts
-        if disable_auto_shutdown is not None:
-            self.disable_auto_shutdown = disable_auto_shutdown
         if ide is not None:
             self.ide = ide
-        if idle_shutdown_seconds is not None:
-            self.idle_shutdown_seconds = idle_shutdown_seconds
         if port_plugin_port is not None:
             self.port_plugin_port = port_plugin_port
-        if switch_to_default_machine_on_idle is not None:
-            self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
 
     @property
     def compute_config(self) -> 'V1UserRequestedComputeConfig':
@@ -127,27 +112,6 @@ class IdCodeconfigBody(object):
 
         self._data_connection_mounts = data_connection_mounts
 
-    @property
-    def disable_auto_shutdown(self) -> 'bool':
-        """Gets the disable_auto_shutdown of this IdCodeconfigBody.  # noqa: E501
-
-
-        :return: The disable_auto_shutdown of this IdCodeconfigBody.  # noqa: E501
-        :rtype: bool
-        """
-        return self._disable_auto_shutdown
-
-    @disable_auto_shutdown.setter
-    def disable_auto_shutdown(self, disable_auto_shutdown: 'bool'):
-        """Sets the disable_auto_shutdown of this IdCodeconfigBody.
-
-
-        :param disable_auto_shutdown: The disable_auto_shutdown of this IdCodeconfigBody.  # noqa: E501
-        :type: bool
-        """
-
-        self._disable_auto_shutdown = disable_auto_shutdown
-
     @property
     def ide(self) -> 'str':
         """Gets the ide of this IdCodeconfigBody.  # noqa: E501
@@ -169,27 +133,6 @@ class IdCodeconfigBody(object):
 
         self._ide = ide
 
-    @property
-    def idle_shutdown_seconds(self) -> 'int':
-        """Gets the idle_shutdown_seconds of this IdCodeconfigBody.  # noqa: E501
-
-
-        :return: The idle_shutdown_seconds of this IdCodeconfigBody.  # noqa: E501
-        :rtype: int
-        """
-        return self._idle_shutdown_seconds
-
-    @idle_shutdown_seconds.setter
-    def idle_shutdown_seconds(self, idle_shutdown_seconds: 'int'):
-        """Sets the idle_shutdown_seconds of this IdCodeconfigBody.
-
-
-        :param idle_shutdown_seconds: The idle_shutdown_seconds of this IdCodeconfigBody.  # noqa: E501
-        :type: int
-        """
-
-        self._idle_shutdown_seconds = idle_shutdown_seconds
-
     @property
     def port_plugin_port(self) -> 'str':
         """Gets the port_plugin_port of this IdCodeconfigBody.  # noqa: E501
@@ -211,27 +154,6 @@ class IdCodeconfigBody(object):
 
         self._port_plugin_port = port_plugin_port
 
-    @property
-    def switch_to_default_machine_on_idle(self) -> 'bool':
-        """Gets the switch_to_default_machine_on_idle of this IdCodeconfigBody.  # noqa: E501
-
-
-        :return: The switch_to_default_machine_on_idle of this IdCodeconfigBody.  # noqa: E501
-        :rtype: bool
-        """
-        return self._switch_to_default_machine_on_idle
-
-    @switch_to_default_machine_on_idle.setter
-    def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
-        """Sets the switch_to_default_machine_on_idle of this IdCodeconfigBody.
-
-
-        :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this IdCodeconfigBody.  # noqa: E501
-        :type: bool
-        """
-
-        self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
-
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py

@@ -73,6 +73,7 @@ class OrgsIdBody(object):
         'preferred_cluster': 'str',
         'preferred_deployment_provider': 'str',
         'preferred_studio_provider': 'str',
+        'show_model_apis_tab': 'bool',
         'start_studios_on_spot_instance': 'bool',
         'switch_to_default_machine_on_idle': 'bool',
         'teamspace_default_credits': 'float',
@@ -113,6 +114,7 @@ class OrgsIdBody(object):
         'preferred_cluster': 'preferredCluster',
         'preferred_deployment_provider': 'preferredDeploymentProvider',
         'preferred_studio_provider': 'preferredStudioProvider',
+        'show_model_apis_tab': 'showModelApisTab',
         'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
         'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle',
         'teamspace_default_credits': 'teamspaceDefaultCredits',
@@ -120,7 +122,7 @@ class OrgsIdBody(object):
         'workload_max_run_duration': 'workloadMaxRunDuration'
     }
 
-    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, default_machine_type: 'str' =None, default_project_id: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_nebius_saas: 'bool' =None, disallow_voltage_park_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
+    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, default_machine_type: 'str' =None, default_project_id: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_nebius_saas: 'bool' =None, disallow_voltage_park_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, show_model_apis_tab: 'bool' =None, start_studios_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
         """OrgsIdBody - a model defined in Swagger"""  # noqa: E501
         self._alerts_config = None
         self._allow_budgeting = None
@@ -154,6 +156,7 @@ class OrgsIdBody(object):
         self._preferred_cluster = None
         self._preferred_deployment_provider = None
         self._preferred_studio_provider = None
+        self._show_model_apis_tab = None
         self._start_studios_on_spot_instance = None
         self._switch_to_default_machine_on_idle = None
         self._teamspace_default_credits = None
@@ -224,6 +227,8 @@ class OrgsIdBody(object):
             self.preferred_deployment_provider = preferred_deployment_provider
         if preferred_studio_provider is not None:
             self.preferred_studio_provider = preferred_studio_provider
+        if show_model_apis_tab is not None:
+            self.show_model_apis_tab = show_model_apis_tab
         if start_studios_on_spot_instance is not None:
             self.start_studios_on_spot_instance = start_studios_on_spot_instance
         if switch_to_default_machine_on_idle is not None:
@@ -907,6 +912,27 @@ class OrgsIdBody(object):
 
         self._preferred_studio_provider = preferred_studio_provider
 
+    @property
+    def show_model_apis_tab(self) -> 'bool':
+        """Gets the show_model_apis_tab of this OrgsIdBody.  # noqa: E501
+
+
+        :return: The show_model_apis_tab of this OrgsIdBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._show_model_apis_tab
+
+    @show_model_apis_tab.setter
+    def show_model_apis_tab(self, show_model_apis_tab: 'bool'):
+        """Sets the show_model_apis_tab of this OrgsIdBody.
+
+
+        :param show_model_apis_tab: The show_model_apis_tab of this OrgsIdBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._show_model_apis_tab = show_model_apis_tab
+
     @property
     def start_studios_on_spot_instance(self) -> 'bool':
         """Gets the start_studios_on_spot_instance of this OrgsIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py

@@ -43,25 +43,30 @@ class ProjectIdStorageBody(object):
     swagger_types = {
         'cluster_id': 'str',
         'count': 'str',
+        'data_connection_id': 'str',
         'filename': 'str'
     }
 
     attribute_map = {
         'cluster_id': 'clusterId',
         'count': 'count',
+        'data_connection_id': 'dataConnectionId',
         'filename': 'filename'
     }
 
-    def __init__(self, cluster_id: 'str' =None, count: 'str' =None, filename: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, count: 'str' =None, data_connection_id: 'str' =None, filename: 'str' =None):  # noqa: E501
         """ProjectIdStorageBody - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._count = None
+        self._data_connection_id = None
         self._filename = None
         self.discriminator = None
         if cluster_id is not None:
             self.cluster_id = cluster_id
         if count is not None:
             self.count = count
+        if data_connection_id is not None:
+            self.data_connection_id = data_connection_id
         if filename is not None:
             self.filename = filename
 
@@ -107,6 +112,27 @@ class ProjectIdStorageBody(object):
 
         self._count = count
 
+    @property
+    def data_connection_id(self) -> 'str':
+        """Gets the data_connection_id of this ProjectIdStorageBody.  # noqa: E501
+
+
+        :return: The data_connection_id of this ProjectIdStorageBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._data_connection_id
+
+    @data_connection_id.setter
+    def data_connection_id(self, data_connection_id: 'str'):
+        """Sets the data_connection_id of this ProjectIdStorageBody.
+
+
+        :param data_connection_id: The data_connection_id of this ProjectIdStorageBody.  # noqa: E501
+        :type: str
+        """
+
+        self._data_connection_id = data_connection_id
+
     @property
     def filename(self) -> 'str':
         """Gets the filename of this ProjectIdStorageBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py

@@ -42,6 +42,7 @@ class StorageCompleteBody(object):
     """
     swagger_types = {
         'cluster_id': 'str',
+        'data_connection_id': 'str',
         'filename': 'str',
         'parts': 'list[V1CompleteUpload]',
         'suppress_reindex': 'bool',
@@ -50,15 +51,17 @@ class StorageCompleteBody(object):
 
     attribute_map = {
         'cluster_id': 'clusterId',
+        'data_connection_id': 'dataConnectionId',
         'filename': 'filename',
         'parts': 'parts',
         'suppress_reindex': 'suppressReindex',
         'upload_id': 'uploadId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, suppress_reindex: 'bool' =None, upload_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, data_connection_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, suppress_reindex: 'bool' =None, upload_id: 'str' =None):  # noqa: E501
         """StorageCompleteBody - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
+        self._data_connection_id = None
         self._filename = None
         self._parts = None
         self._suppress_reindex = None
@@ -66,6 +69,8 @@ class StorageCompleteBody(object):
         self.discriminator = None
         if cluster_id is not None:
             self.cluster_id = cluster_id
+        if data_connection_id is not None:
+            self.data_connection_id = data_connection_id
         if filename is not None:
             self.filename = filename
         if parts is not None:
@@ -96,6 +101,27 @@ class StorageCompleteBody(object):
 
         self._cluster_id = cluster_id
 
+    @property
+    def data_connection_id(self) -> 'str':
+        """Gets the data_connection_id of this StorageCompleteBody.  # noqa: E501
+
+
+        :return: The data_connection_id of this StorageCompleteBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._data_connection_id
+
+    @data_connection_id.setter
+    def data_connection_id(self, data_connection_id: 'str'):
+        """Sets the data_connection_id of this StorageCompleteBody.
+
+
+        :param data_connection_id: The data_connection_id of this StorageCompleteBody.  # noqa: E501
+        :type: str
+        """
+
+        self._data_connection_id = data_connection_id
+
     @property
     def filename(self) -> 'str':
         """Gets the filename of this StorageCompleteBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py

@@ -42,24 +42,29 @@ class UploadsUploadIdBody1(object):
     """
     swagger_types = {
         'cluster_id': 'str',
+        'data_connection_id': 'str',
         'filename': 'str',
         'parts': 'list[str]'
     }
 
     attribute_map = {
         'cluster_id': 'clusterId',
+        'data_connection_id': 'dataConnectionId',
         'filename': 'filename',
         'parts': 'parts'
     }
 
-    def __init__(self, cluster_id: 'str' =None, filename: 'str' =None, parts: 'list[str]' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, data_connection_id: 'str' =None, filename: 'str' =None, parts: 'list[str]' =None):  # noqa: E501
         """UploadsUploadIdBody1 - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
+        self._data_connection_id = None
         self._filename = None
         self._parts = None
         self.discriminator = None
         if cluster_id is not None:
             self.cluster_id = cluster_id
+        if data_connection_id is not None:
+            self.data_connection_id = data_connection_id
         if filename is not None:
             self.filename = filename
         if parts is not None:
@@ -86,6 +91,27 @@ class UploadsUploadIdBody1(object):
 
         self._cluster_id = cluster_id
 
+    @property
+    def data_connection_id(self) -> 'str':
+        """Gets the data_connection_id of this UploadsUploadIdBody1.  # noqa: E501
+
+
+        :return: The data_connection_id of this UploadsUploadIdBody1.  # noqa: E501
+        :rtype: str
+        """
+        return self._data_connection_id
+
+    @data_connection_id.setter
+    def data_connection_id(self, data_connection_id: 'str'):
+        """Sets the data_connection_id of this UploadsUploadIdBody1.
+
+
+        :param data_connection_id: The data_connection_id of this UploadsUploadIdBody1.  # noqa: E501
+        :type: str
+        """
+
+        self._data_connection_id = data_connection_id
+
     @property
     def filename(self) -> 'str':
         """Gets the filename of this UploadsUploadIdBody1.  # noqa: E501