lightning-sdk 0.2.9__py3-none-any.whl → 0.2.11__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3174,6 +3174,99 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_get_cloud_space_cold_start_metrics_stats(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_cold_start_metrics_stats  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_get_cloud_space_cold_start_metrics_stats(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceColdStartMetricsStatsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_cold_start_metrics_stats  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceColdStartMetricsStatsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method cloud_space_service_get_cloud_space_cold_start_metrics_stats" % 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 `cloud_space_service_get_cloud_space_cold_start_metrics_stats`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # 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}/cloudspaces/cold-start-stats', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetCloudSpaceColdStartMetricsStatsResponse',  # 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 cloud_space_service_get_cloud_space_folder_index(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetFolderIndexResponse':  # noqa: E501
         """cloud_space_service_get_cloud_space_folder_index  # noqa: E501
 
@@ -3499,6 +3592,7 @@ class CloudSpaceServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str cloudspace_id:
         :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -3520,12 +3614,13 @@ class CloudSpaceServiceApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str cloudspace_id:
         :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'cloudspace_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -3552,6 +3647,8 @@ class CloudSpaceServiceApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'cloudspace_id' in params:
+            query_params.append(('cloudspaceId', params['cloudspace_id']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -236,6 +236,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics import V1CloudSpaceColdStartMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
@@ -251,6 +252,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -442,6 +444,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
 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_gcp_direct_vpc import V1GCPDirectVPC
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
@@ -453,6 +456,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
 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_size_response import V1GetCloudSpaceSizeResponse
@@ -728,6 +732,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1Quest
 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_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
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest

lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py

@@ -44,31 +44,37 @@ class ClusterIdCapacityreservationsBody(object):
         'capacity_reservation_type': 'str',
         'cloud_provider_capacity_reservation_id': 'str',
         'end_time': 'datetime',
+        'full_cloud_provider_reservation_string': 'str',
         'instance_type': 'str',
         'match_pattern': 'str',
         'region': 'str',
-        'start_time': 'datetime'
+        'start_time': 'datetime',
+        'zone': 'str'
     }
 
     attribute_map = {
         'capacity_reservation_type': 'capacityReservationType',
         'cloud_provider_capacity_reservation_id': 'cloudProviderCapacityReservationId',
         'end_time': 'endTime',
+        'full_cloud_provider_reservation_string': 'fullCloudProviderReservationString',
         'instance_type': 'instanceType',
         'match_pattern': 'matchPattern',
         'region': 'region',
-        'start_time': 'startTime'
+        'start_time': 'startTime',
+        'zone': 'zone'
     }
 
-    def __init__(self, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, end_time: 'datetime' =None, instance_type: 'str' =None, match_pattern: 'str' =None, region: 'str' =None, start_time: 'datetime' =None):  # noqa: E501
+    def __init__(self, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, zone: 'str' =None):  # noqa: E501
         """ClusterIdCapacityreservationsBody - a model defined in Swagger"""  # noqa: E501
         self._capacity_reservation_type = None
         self._cloud_provider_capacity_reservation_id = None
         self._end_time = None
+        self._full_cloud_provider_reservation_string = None
         self._instance_type = None
         self._match_pattern = None
         self._region = None
         self._start_time = None
+        self._zone = None
         self.discriminator = None
         if capacity_reservation_type is not None:
             self.capacity_reservation_type = capacity_reservation_type
@@ -76,6 +82,8 @@ class ClusterIdCapacityreservationsBody(object):
             self.cloud_provider_capacity_reservation_id = cloud_provider_capacity_reservation_id
         if end_time is not None:
             self.end_time = end_time
+        if full_cloud_provider_reservation_string is not None:
+            self.full_cloud_provider_reservation_string = full_cloud_provider_reservation_string
         if instance_type is not None:
             self.instance_type = instance_type
         if match_pattern is not None:
@@ -84,6 +92,8 @@ class ClusterIdCapacityreservationsBody(object):
             self.region = region
         if start_time is not None:
             self.start_time = start_time
+        if zone is not None:
+            self.zone = zone
 
     @property
     def capacity_reservation_type(self) -> 'str':
@@ -148,6 +158,27 @@ class ClusterIdCapacityreservationsBody(object):
 
         self._end_time = end_time
 
+    @property
+    def full_cloud_provider_reservation_string(self) -> 'str':
+        """Gets the full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody.  # noqa: E501
+
+
+        :return: The full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._full_cloud_provider_reservation_string
+
+    @full_cloud_provider_reservation_string.setter
+    def full_cloud_provider_reservation_string(self, full_cloud_provider_reservation_string: 'str'):
+        """Sets the full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody.
+
+
+        :param full_cloud_provider_reservation_string: The full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._full_cloud_provider_reservation_string = full_cloud_provider_reservation_string
+
     @property
     def instance_type(self) -> 'str':
         """Gets the instance_type of this ClusterIdCapacityreservationsBody.  # noqa: E501
@@ -232,6 +263,27 @@ class ClusterIdCapacityreservationsBody(object):
 
         self._start_time = start_time
 
+    @property
+    def zone(self) -> 'str':
+        """Gets the zone of this ClusterIdCapacityreservationsBody.  # noqa: E501
+
+
+        :return: The zone of this ClusterIdCapacityreservationsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._zone
+
+    @zone.setter
+    def zone(self, zone: 'str'):
+        """Sets the zone of this ClusterIdCapacityreservationsBody.
+
+
+        :param zone: The zone of this ClusterIdCapacityreservationsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._zone = zone
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/create.py

@@ -53,9 +53,11 @@ class Create(object):
         'gcs_folder': 'V1GCSFolderDataConnection',
         'id': 'str',
         'name': 'str',
+        'r2': 'V1R2DataConnection',
         'run_cmds': 'list[str]',
         's3_folder': 'V1S3FolderDataConnection',
         'snowflake': 'V1SnowflakeDataConnection',
+        'weka': 'V1WekaDataConnection',
         'writable': 'bool'
     }
 
@@ -72,13 +74,15 @@ class Create(object):
         'gcs_folder': 'gcsFolder',
         'id': 'id',
         'name': 'name',
+        'r2': 'r2',
         'run_cmds': 'runCmds',
         's3_folder': 's3Folder',
         'snowflake': 'snowflake',
+        'weka': 'weka',
         'writable': 'writable'
     }
 
-    def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, r2: 'V1R2DataConnection' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, weka: 'V1WekaDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """Create - a model defined in Swagger"""  # noqa: E501
         self._access_cluster_ids = None
         self._aws = None
@@ -92,9 +96,11 @@ class Create(object):
         self._gcs_folder = None
         self._id = None
         self._name = None
+        self._r2 = None
         self._run_cmds = None
         self._s3_folder = None
         self._snowflake = None
+        self._weka = None
         self._writable = None
         self.discriminator = None
         if access_cluster_ids is not None:
@@ -121,12 +127,16 @@ class Create(object):
             self.id = id
         if name is not None:
             self.name = name
+        if r2 is not None:
+            self.r2 = r2
         if run_cmds is not None:
             self.run_cmds = run_cmds
         if s3_folder is not None:
             self.s3_folder = s3_folder
         if snowflake is not None:
             self.snowflake = snowflake
+        if weka is not None:
+            self.weka = weka
         if writable is not None:
             self.writable = writable
 
@@ -382,6 +392,27 @@ class Create(object):
 
         self._name = name
 
+    @property
+    def r2(self) -> 'V1R2DataConnection':
+        """Gets the r2 of this Create.  # noqa: E501
+
+
+        :return: The r2 of this Create.  # noqa: E501
+        :rtype: V1R2DataConnection
+        """
+        return self._r2
+
+    @r2.setter
+    def r2(self, r2: 'V1R2DataConnection'):
+        """Sets the r2 of this Create.
+
+
+        :param r2: The r2 of this Create.  # noqa: E501
+        :type: V1R2DataConnection
+        """
+
+        self._r2 = r2
+
     @property
     def run_cmds(self) -> 'list[str]':
         """Gets the run_cmds of this Create.  # noqa: E501
@@ -445,6 +476,27 @@ class Create(object):
 
         self._snowflake = snowflake
 
+    @property
+    def weka(self) -> 'V1WekaDataConnection':
+        """Gets the weka of this Create.  # noqa: E501
+
+
+        :return: The weka of this Create.  # noqa: E501
+        :rtype: V1WekaDataConnection
+        """
+        return self._weka
+
+    @weka.setter
+    def weka(self, weka: 'V1WekaDataConnection'):
+        """Sets the weka of this Create.
+
+
+        :param weka: The weka of this Create.  # noqa: E501
+        :type: V1WekaDataConnection
+        """
+
+        self._weka = weka
+
     @property
     def writable(self) -> 'bool':
         """Gets the writable of this Create.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py

@@ -43,6 +43,7 @@ class OrgsIdBody(object):
     swagger_types = {
         'alerts_config': 'V1AlertsConfig',
         'allow_aws_saas': 'bool',
+        'allow_budgeting': 'bool',
         'allow_gcp_saas': 'bool',
         'allow_guest': 'bool',
         'allow_lambda_saas': 'bool',
@@ -64,12 +65,14 @@ class OrgsIdBody(object):
         'preferred_studio_provider': 'str',
         'start_studios_on_spot_instance': 'bool',
         'teamspace_default_credits': 'float',
-        'twitter_username': 'str'
+        'twitter_username': 'str',
+        'workload_max_run_duration': 'str'
     }
 
     attribute_map = {
         'alerts_config': 'alertsConfig',
         'allow_aws_saas': 'allowAwsSaas',
+        'allow_budgeting': 'allowBudgeting',
         'allow_gcp_saas': 'allowGcpSaas',
         'allow_guest': 'allowGuest',
         'allow_lambda_saas': 'allowLambdaSaas',
@@ -91,13 +94,15 @@ class OrgsIdBody(object):
         'preferred_studio_provider': 'preferredStudioProvider',
         'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
         'teamspace_default_credits': 'teamspaceDefaultCredits',
-        'twitter_username': 'twitterUsername'
+        'twitter_username': 'twitterUsername',
+        'workload_max_run_duration': 'workloadMaxRunDuration'
     }
 
-    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, default_machine_type: 'str' =None, description: 'str' =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, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None):  # noqa: E501
+    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, default_machine_type: 'str' =None, description: 'str' =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, 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_aws_saas = None
+        self._allow_budgeting = None
         self._allow_gcp_saas = None
         self._allow_guest = None
         self._allow_lambda_saas = None
@@ -120,11 +125,14 @@ class OrgsIdBody(object):
         self._start_studios_on_spot_instance = None
         self._teamspace_default_credits = None
         self._twitter_username = None
+        self._workload_max_run_duration = None
         self.discriminator = None
         if alerts_config is not None:
             self.alerts_config = alerts_config
         if allow_aws_saas is not None:
             self.allow_aws_saas = allow_aws_saas
+        if allow_budgeting is not None:
+            self.allow_budgeting = allow_budgeting
         if allow_gcp_saas is not None:
             self.allow_gcp_saas = allow_gcp_saas
         if allow_guest is not None:
@@ -169,6 +177,8 @@ class OrgsIdBody(object):
             self.teamspace_default_credits = teamspace_default_credits
         if twitter_username is not None:
             self.twitter_username = twitter_username
+        if workload_max_run_duration is not None:
+            self.workload_max_run_duration = workload_max_run_duration
 
     @property
     def alerts_config(self) -> 'V1AlertsConfig':
@@ -212,6 +222,27 @@ class OrgsIdBody(object):
 
         self._allow_aws_saas = allow_aws_saas
 
+    @property
+    def allow_budgeting(self) -> 'bool':
+        """Gets the allow_budgeting of this OrgsIdBody.  # noqa: E501
+
+
+        :return: The allow_budgeting of this OrgsIdBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_budgeting
+
+    @allow_budgeting.setter
+    def allow_budgeting(self, allow_budgeting: 'bool'):
+        """Sets the allow_budgeting of this OrgsIdBody.
+
+
+        :param allow_budgeting: The allow_budgeting of this OrgsIdBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_budgeting = allow_budgeting
+
     @property
     def allow_gcp_saas(self) -> 'bool':
         """Gets the allow_gcp_saas of this OrgsIdBody.  # noqa: E501
@@ -674,6 +705,27 @@ class OrgsIdBody(object):
 
         self._twitter_username = twitter_username
 
+    @property
+    def workload_max_run_duration(self) -> 'str':
+        """Gets the workload_max_run_duration of this OrgsIdBody.  # noqa: E501
+
+
+        :return: The workload_max_run_duration of this OrgsIdBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_max_run_duration
+
+    @workload_max_run_duration.setter
+    def workload_max_run_duration(self, workload_max_run_duration: 'str'):
+        """Sets the workload_max_run_duration of this OrgsIdBody.
+
+
+        :param workload_max_run_duration: The workload_max_run_duration of this OrgsIdBody.  # noqa: E501
+        :type: str
+        """
+
+        self._workload_max_run_duration = workload_max_run_duration
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/update.py

@@ -46,6 +46,7 @@ class Update(object):
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         'name': 'str',
+        'r2': 'V1R2DataConnection',
         'run_cmds': 'list[str]',
         's3_folder': 'V1S3FolderDataConnection',
         'snowflake': 'V1SnowflakeDataConnection',
@@ -58,19 +59,21 @@ class Update(object):
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         'name': 'name',
+        'r2': 'r2',
         'run_cmds': 'runCmds',
         's3_folder': 's3Folder',
         'snowflake': 'snowflake',
         'writable': 'writable'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, r2: 'V1R2DataConnection' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """Update - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._efs = None
         self._gcp = None
         self._gcs_folder = None
         self._name = None
+        self._r2 = None
         self._run_cmds = None
         self._s3_folder = None
         self._snowflake = None
@@ -86,6 +89,8 @@ class Update(object):
             self.gcs_folder = gcs_folder
         if name is not None:
             self.name = name
+        if r2 is not None:
+            self.r2 = r2
         if run_cmds is not None:
             self.run_cmds = run_cmds
         if s3_folder is not None:
@@ -200,6 +205,27 @@ class Update(object):
 
         self._name = name
 
+    @property
+    def r2(self) -> 'V1R2DataConnection':
+        """Gets the r2 of this Update.  # noqa: E501
+
+
+        :return: The r2 of this Update.  # noqa: E501
+        :rtype: V1R2DataConnection
+        """
+        return self._r2
+
+    @r2.setter
+    def r2(self, r2: 'V1R2DataConnection'):
+        """Sets the r2 of this Update.
+
+
+        :param r2: The r2 of this Update.  # noqa: E501
+        :type: V1R2DataConnection
+        """
+
+        self._r2 = r2
+
     @property
     def run_cmds(self) -> 'list[str]':
         """Gets the run_cmds of this Update.  # noqa: E501