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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (50) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +1 -0
  3. lightning_sdk/api/lit_container_api.py +19 -2
  4. lightning_sdk/api/teamspace_api.py +20 -16
  5. lightning_sdk/api/utils.py +1 -1
  6. lightning_sdk/cli/entrypoint.py +2 -2
  7. lightning_sdk/cli/serve.py +141 -16
  8. lightning_sdk/cli/upload.py +4 -1
  9. lightning_sdk/deployment/deployment.py +5 -2
  10. lightning_sdk/lightning_cloud/openapi/__init__.py +5 -0
  11. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +98 -1
  12. lightning_sdk/lightning_cloud/openapi/models/__init__.py +5 -0
  13. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +55 -3
  14. lightning_sdk/lightning_cloud/openapi/models/create.py +53 -1
  15. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +55 -3
  16. lightning_sdk/lightning_cloud/openapi/models/update.py +27 -1
  17. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  18. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics_stats.py +357 -0
  19. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  20. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +227 -0
  21. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  22. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +55 -3
  23. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +53 -1
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  26. lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_capacity_reservation_response.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_direct_vpc.py +149 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_cold_start_metrics_stats_response.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +17 -17
  32. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +55 -3
  33. lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py +253 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +50 -24
  37. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_weka_data_connection.py +29 -55
  39. lightning_sdk/lightning_cloud/openapi/models/validate.py +27 -1
  40. lightning_sdk/lit_container.py +12 -2
  41. lightning_sdk/models.py +22 -9
  42. lightning_sdk/serve.py +3 -0
  43. lightning_sdk/teamspace.py +2 -0
  44. lightning_sdk/utils/resolve.py +11 -4
  45. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/METADATA +1 -1
  46. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/RECORD +50 -45
  47. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/LICENSE +0 -0
  48. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/WHEEL +0 -0
  49. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/entry_points.txt +0 -0
  50. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,253 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1R2DataConnection(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'account_id': 'str',
45
+ 'credential_id': 'str',
46
+ 'name': 'str',
47
+ 'secret_access_key': 'str',
48
+ 'source': 'str',
49
+ 'token_value': 'str'
50
+ }
51
+
52
+ attribute_map = {
53
+ 'account_id': 'accountId',
54
+ 'credential_id': 'credentialId',
55
+ 'name': 'name',
56
+ 'secret_access_key': 'secretAccessKey',
57
+ 'source': 'source',
58
+ 'token_value': 'tokenValue'
59
+ }
60
+
61
+ def __init__(self, account_id: 'str' =None, credential_id: 'str' =None, name: 'str' =None, secret_access_key: 'str' =None, source: 'str' =None, token_value: 'str' =None): # noqa: E501
62
+ """V1R2DataConnection - a model defined in Swagger""" # noqa: E501
63
+ self._account_id = None
64
+ self._credential_id = None
65
+ self._name = None
66
+ self._secret_access_key = None
67
+ self._source = None
68
+ self._token_value = None
69
+ self.discriminator = None
70
+ if account_id is not None:
71
+ self.account_id = account_id
72
+ if credential_id is not None:
73
+ self.credential_id = credential_id
74
+ if name is not None:
75
+ self.name = name
76
+ if secret_access_key is not None:
77
+ self.secret_access_key = secret_access_key
78
+ if source is not None:
79
+ self.source = source
80
+ if token_value is not None:
81
+ self.token_value = token_value
82
+
83
+ @property
84
+ def account_id(self) -> 'str':
85
+ """Gets the account_id of this V1R2DataConnection. # noqa: E501
86
+
87
+
88
+ :return: The account_id of this V1R2DataConnection. # noqa: E501
89
+ :rtype: str
90
+ """
91
+ return self._account_id
92
+
93
+ @account_id.setter
94
+ def account_id(self, account_id: 'str'):
95
+ """Sets the account_id of this V1R2DataConnection.
96
+
97
+
98
+ :param account_id: The account_id of this V1R2DataConnection. # noqa: E501
99
+ :type: str
100
+ """
101
+
102
+ self._account_id = account_id
103
+
104
+ @property
105
+ def credential_id(self) -> 'str':
106
+ """Gets the credential_id of this V1R2DataConnection. # noqa: E501
107
+
108
+
109
+ :return: The credential_id of this V1R2DataConnection. # noqa: E501
110
+ :rtype: str
111
+ """
112
+ return self._credential_id
113
+
114
+ @credential_id.setter
115
+ def credential_id(self, credential_id: 'str'):
116
+ """Sets the credential_id of this V1R2DataConnection.
117
+
118
+
119
+ :param credential_id: The credential_id of this V1R2DataConnection. # noqa: E501
120
+ :type: str
121
+ """
122
+
123
+ self._credential_id = credential_id
124
+
125
+ @property
126
+ def name(self) -> 'str':
127
+ """Gets the name of this V1R2DataConnection. # noqa: E501
128
+
129
+
130
+ :return: The name of this V1R2DataConnection. # noqa: E501
131
+ :rtype: str
132
+ """
133
+ return self._name
134
+
135
+ @name.setter
136
+ def name(self, name: 'str'):
137
+ """Sets the name of this V1R2DataConnection.
138
+
139
+
140
+ :param name: The name of this V1R2DataConnection. # noqa: E501
141
+ :type: str
142
+ """
143
+
144
+ self._name = name
145
+
146
+ @property
147
+ def secret_access_key(self) -> 'str':
148
+ """Gets the secret_access_key of this V1R2DataConnection. # noqa: E501
149
+
150
+
151
+ :return: The secret_access_key of this V1R2DataConnection. # noqa: E501
152
+ :rtype: str
153
+ """
154
+ return self._secret_access_key
155
+
156
+ @secret_access_key.setter
157
+ def secret_access_key(self, secret_access_key: 'str'):
158
+ """Sets the secret_access_key of this V1R2DataConnection.
159
+
160
+
161
+ :param secret_access_key: The secret_access_key of this V1R2DataConnection. # noqa: E501
162
+ :type: str
163
+ """
164
+
165
+ self._secret_access_key = secret_access_key
166
+
167
+ @property
168
+ def source(self) -> 'str':
169
+ """Gets the source of this V1R2DataConnection. # noqa: E501
170
+
171
+
172
+ :return: The source of this V1R2DataConnection. # noqa: E501
173
+ :rtype: str
174
+ """
175
+ return self._source
176
+
177
+ @source.setter
178
+ def source(self, source: 'str'):
179
+ """Sets the source of this V1R2DataConnection.
180
+
181
+
182
+ :param source: The source of this V1R2DataConnection. # noqa: E501
183
+ :type: str
184
+ """
185
+
186
+ self._source = source
187
+
188
+ @property
189
+ def token_value(self) -> 'str':
190
+ """Gets the token_value of this V1R2DataConnection. # noqa: E501
191
+
192
+
193
+ :return: The token_value of this V1R2DataConnection. # noqa: E501
194
+ :rtype: str
195
+ """
196
+ return self._token_value
197
+
198
+ @token_value.setter
199
+ def token_value(self, token_value: 'str'):
200
+ """Sets the token_value of this V1R2DataConnection.
201
+
202
+
203
+ :param token_value: The token_value of this V1R2DataConnection. # noqa: E501
204
+ :type: str
205
+ """
206
+
207
+ self._token_value = token_value
208
+
209
+ def to_dict(self) -> dict:
210
+ """Returns the model properties as a dict"""
211
+ result = {}
212
+
213
+ for attr, _ in six.iteritems(self.swagger_types):
214
+ value = getattr(self, attr)
215
+ if isinstance(value, list):
216
+ result[attr] = list(map(
217
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
218
+ value
219
+ ))
220
+ elif hasattr(value, "to_dict"):
221
+ result[attr] = value.to_dict()
222
+ elif isinstance(value, dict):
223
+ result[attr] = dict(map(
224
+ lambda item: (item[0], item[1].to_dict())
225
+ if hasattr(item[1], "to_dict") else item,
226
+ value.items()
227
+ ))
228
+ else:
229
+ result[attr] = value
230
+ if issubclass(V1R2DataConnection, dict):
231
+ for key, value in self.items():
232
+ result[key] = value
233
+
234
+ return result
235
+
236
+ def to_str(self) -> str:
237
+ """Returns the string representation of the model"""
238
+ return pprint.pformat(self.to_dict())
239
+
240
+ def __repr__(self) -> str:
241
+ """For `print` and `pprint`"""
242
+ return self.to_str()
243
+
244
+ def __eq__(self, other: 'V1R2DataConnection') -> bool:
245
+ """Returns true if both objects are equal"""
246
+ if not isinstance(other, V1R2DataConnection):
247
+ return False
248
+
249
+ return self.__dict__ == other.__dict__
250
+
251
+ def __ne__(self, other: 'V1R2DataConnection') -> bool:
252
+ """Returns true if both objects are not equal"""
253
+ return not self == other
@@ -50,6 +50,7 @@ class V1UpdateUserRequest(object):
50
50
  'experimentation_id': 'str',
51
51
  'first_name': 'str',
52
52
  'general_audience_mode': 'bool',
53
+ 'git_credentials_id': 'str',
53
54
  'last_name': 'str',
54
55
  'non_developer_mode': 'bool',
55
56
  'opted_in_marketing_emails': 'bool',
@@ -77,6 +78,7 @@ class V1UpdateUserRequest(object):
77
78
  'experimentation_id': 'experimentationId',
78
79
  'first_name': 'firstName',
79
80
  'general_audience_mode': 'generalAudienceMode',
81
+ 'git_credentials_id': 'gitCredentialsId',
80
82
  'last_name': 'lastName',
81
83
  'non_developer_mode': 'nonDeveloperMode',
82
84
  'opted_in_marketing_emails': 'optedInMarketingEmails',
@@ -94,7 +96,7 @@ class V1UpdateUserRequest(object):
94
96
  'website': 'website'
95
97
  }
96
98
 
97
- def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None): # noqa: E501
99
+ def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, git_credentials_id: 'str' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None): # noqa: E501
98
100
  """V1UpdateUserRequest - a model defined in Swagger""" # noqa: E501
99
101
  self._agree_to_terms_and_conditions = None
100
102
  self._complete_sign_up = None
@@ -105,6 +107,7 @@ class V1UpdateUserRequest(object):
105
107
  self._experimentation_id = None
106
108
  self._first_name = None
107
109
  self._general_audience_mode = None
110
+ self._git_credentials_id = None
108
111
  self._last_name = None
109
112
  self._non_developer_mode = None
110
113
  self._opted_in_marketing_emails = None
@@ -139,6 +142,8 @@ class V1UpdateUserRequest(object):
139
142
  self.first_name = first_name
140
143
  if general_audience_mode is not None:
141
144
  self.general_audience_mode = general_audience_mode
145
+ if git_credentials_id is not None:
146
+ self.git_credentials_id = git_credentials_id
142
147
  if last_name is not None:
143
148
  self.last_name = last_name
144
149
  if non_developer_mode is not None:
@@ -359,6 +364,27 @@ class V1UpdateUserRequest(object):
359
364
 
360
365
  self._general_audience_mode = general_audience_mode
361
366
 
367
+ @property
368
+ def git_credentials_id(self) -> 'str':
369
+ """Gets the git_credentials_id of this V1UpdateUserRequest. # noqa: E501
370
+
371
+
372
+ :return: The git_credentials_id of this V1UpdateUserRequest. # noqa: E501
373
+ :rtype: str
374
+ """
375
+ return self._git_credentials_id
376
+
377
+ @git_credentials_id.setter
378
+ def git_credentials_id(self, git_credentials_id: 'str'):
379
+ """Sets the git_credentials_id of this V1UpdateUserRequest.
380
+
381
+
382
+ :param git_credentials_id: The git_credentials_id of this V1UpdateUserRequest. # noqa: E501
383
+ :type: str
384
+ """
385
+
386
+ self._git_credentials_id = git_credentials_id
387
+
362
388
  @property
363
389
  def last_name(self) -> 'str':
364
390
  """Gets the last_name of this V1UpdateUserRequest. # noqa: E501
@@ -63,8 +63,8 @@ class V1UserFeatures(object):
63
63
  'default_one_cluster': 'bool',
64
64
  'deployment_alerts': 'bool',
65
65
  'deployment_persistent_disk': 'bool',
66
- 'deployment_reservations': 'bool',
67
66
  'dgx_cloud': 'bool',
67
+ 'doc_helper_chat': 'bool',
68
68
  'docs_agent': 'bool',
69
69
  'drive_v2': 'bool',
70
70
  'enable_storage_limits': 'bool',
@@ -85,6 +85,7 @@ class V1UserFeatures(object):
85
85
  'multiple_studio_versions': 'bool',
86
86
  'org_admin_alerts': 'bool',
87
87
  'org_level_member_permissions': 'bool',
88
+ 'org_usage_limits': 'bool',
88
89
  'pipelines': 'bool',
89
90
  'plugin_distributed': 'bool',
90
91
  'plugin_inference': 'bool',
@@ -140,8 +141,8 @@ class V1UserFeatures(object):
140
141
  'default_one_cluster': 'defaultOneCluster',
141
142
  'deployment_alerts': 'deploymentAlerts',
142
143
  'deployment_persistent_disk': 'deploymentPersistentDisk',
143
- 'deployment_reservations': 'deploymentReservations',
144
144
  'dgx_cloud': 'dgxCloud',
145
+ 'doc_helper_chat': 'docHelperChat',
145
146
  'docs_agent': 'docsAgent',
146
147
  'drive_v2': 'driveV2',
147
148
  'enable_storage_limits': 'enableStorageLimits',
@@ -162,6 +163,7 @@ class V1UserFeatures(object):
162
163
  'multiple_studio_versions': 'multipleStudioVersions',
163
164
  'org_admin_alerts': 'orgAdminAlerts',
164
165
  'org_level_member_permissions': 'orgLevelMemberPermissions',
166
+ 'org_usage_limits': 'orgUsageLimits',
165
167
  'pipelines': 'pipelines',
166
168
  'plugin_distributed': 'pluginDistributed',
167
169
  'plugin_inference': 'pluginInference',
@@ -194,7 +196,7 @@ class V1UserFeatures(object):
194
196
  'weka': 'weka'
195
197
  }
196
198
 
197
- def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_reservations: 'bool' =None, dgx_cloud: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, inactive_notify_delete: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, org_admin_alerts: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, studios_dashboard: 'bool' =None, studios_dashboard_system_metrics: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
199
+ def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, dgx_cloud: 'bool' =None, doc_helper_chat: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, inactive_notify_delete: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, org_admin_alerts: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, studios_dashboard: 'bool' =None, studios_dashboard_system_metrics: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
198
200
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
199
201
  self._affiliate_links = None
200
202
  self._agents_v2 = None
@@ -218,8 +220,8 @@ class V1UserFeatures(object):
218
220
  self._default_one_cluster = None
219
221
  self._deployment_alerts = None
220
222
  self._deployment_persistent_disk = None
221
- self._deployment_reservations = None
222
223
  self._dgx_cloud = None
224
+ self._doc_helper_chat = None
223
225
  self._docs_agent = None
224
226
  self._drive_v2 = None
225
227
  self._enable_storage_limits = None
@@ -240,6 +242,7 @@ class V1UserFeatures(object):
240
242
  self._multiple_studio_versions = None
241
243
  self._org_admin_alerts = None
242
244
  self._org_level_member_permissions = None
245
+ self._org_usage_limits = None
243
246
  self._pipelines = None
244
247
  self._plugin_distributed = None
245
248
  self._plugin_inference = None
@@ -315,10 +318,10 @@ class V1UserFeatures(object):
315
318
  self.deployment_alerts = deployment_alerts
316
319
  if deployment_persistent_disk is not None:
317
320
  self.deployment_persistent_disk = deployment_persistent_disk
318
- if deployment_reservations is not None:
319
- self.deployment_reservations = deployment_reservations
320
321
  if dgx_cloud is not None:
321
322
  self.dgx_cloud = dgx_cloud
323
+ if doc_helper_chat is not None:
324
+ self.doc_helper_chat = doc_helper_chat
322
325
  if docs_agent is not None:
323
326
  self.docs_agent = docs_agent
324
327
  if drive_v2 is not None:
@@ -359,6 +362,8 @@ class V1UserFeatures(object):
359
362
  self.org_admin_alerts = org_admin_alerts
360
363
  if org_level_member_permissions is not None:
361
364
  self.org_level_member_permissions = org_level_member_permissions
365
+ if org_usage_limits is not None:
366
+ self.org_usage_limits = org_usage_limits
362
367
  if pipelines is not None:
363
368
  self.pipelines = pipelines
364
369
  if plugin_distributed is not None:
@@ -883,46 +888,46 @@ class V1UserFeatures(object):
883
888
  self._deployment_persistent_disk = deployment_persistent_disk
884
889
 
885
890
  @property
886
- def deployment_reservations(self) -> 'bool':
887
- """Gets the deployment_reservations of this V1UserFeatures. # noqa: E501
891
+ def dgx_cloud(self) -> 'bool':
892
+ """Gets the dgx_cloud of this V1UserFeatures. # noqa: E501
888
893
 
889
894
 
890
- :return: The deployment_reservations of this V1UserFeatures. # noqa: E501
895
+ :return: The dgx_cloud of this V1UserFeatures. # noqa: E501
891
896
  :rtype: bool
892
897
  """
893
- return self._deployment_reservations
898
+ return self._dgx_cloud
894
899
 
895
- @deployment_reservations.setter
896
- def deployment_reservations(self, deployment_reservations: 'bool'):
897
- """Sets the deployment_reservations of this V1UserFeatures.
900
+ @dgx_cloud.setter
901
+ def dgx_cloud(self, dgx_cloud: 'bool'):
902
+ """Sets the dgx_cloud of this V1UserFeatures.
898
903
 
899
904
 
900
- :param deployment_reservations: The deployment_reservations of this V1UserFeatures. # noqa: E501
905
+ :param dgx_cloud: The dgx_cloud of this V1UserFeatures. # noqa: E501
901
906
  :type: bool
902
907
  """
903
908
 
904
- self._deployment_reservations = deployment_reservations
909
+ self._dgx_cloud = dgx_cloud
905
910
 
906
911
  @property
907
- def dgx_cloud(self) -> 'bool':
908
- """Gets the dgx_cloud of this V1UserFeatures. # noqa: E501
912
+ def doc_helper_chat(self) -> 'bool':
913
+ """Gets the doc_helper_chat of this V1UserFeatures. # noqa: E501
909
914
 
910
915
 
911
- :return: The dgx_cloud of this V1UserFeatures. # noqa: E501
916
+ :return: The doc_helper_chat of this V1UserFeatures. # noqa: E501
912
917
  :rtype: bool
913
918
  """
914
- return self._dgx_cloud
919
+ return self._doc_helper_chat
915
920
 
916
- @dgx_cloud.setter
917
- def dgx_cloud(self, dgx_cloud: 'bool'):
918
- """Sets the dgx_cloud of this V1UserFeatures.
921
+ @doc_helper_chat.setter
922
+ def doc_helper_chat(self, doc_helper_chat: 'bool'):
923
+ """Sets the doc_helper_chat of this V1UserFeatures.
919
924
 
920
925
 
921
- :param dgx_cloud: The dgx_cloud of this V1UserFeatures. # noqa: E501
926
+ :param doc_helper_chat: The doc_helper_chat of this V1UserFeatures. # noqa: E501
922
927
  :type: bool
923
928
  """
924
929
 
925
- self._dgx_cloud = dgx_cloud
930
+ self._doc_helper_chat = doc_helper_chat
926
931
 
927
932
  @property
928
933
  def docs_agent(self) -> 'bool':
@@ -1344,6 +1349,27 @@ class V1UserFeatures(object):
1344
1349
 
1345
1350
  self._org_level_member_permissions = org_level_member_permissions
1346
1351
 
1352
+ @property
1353
+ def org_usage_limits(self) -> 'bool':
1354
+ """Gets the org_usage_limits of this V1UserFeatures. # noqa: E501
1355
+
1356
+
1357
+ :return: The org_usage_limits of this V1UserFeatures. # noqa: E501
1358
+ :rtype: bool
1359
+ """
1360
+ return self._org_usage_limits
1361
+
1362
+ @org_usage_limits.setter
1363
+ def org_usage_limits(self, org_usage_limits: 'bool'):
1364
+ """Sets the org_usage_limits of this V1UserFeatures.
1365
+
1366
+
1367
+ :param org_usage_limits: The org_usage_limits of this V1UserFeatures. # noqa: E501
1368
+ :type: bool
1369
+ """
1370
+
1371
+ self._org_usage_limits = org_usage_limits
1372
+
1347
1373
  @property
1348
1374
  def pipelines(self) -> 'bool':
1349
1375
  """Gets the pipelines of this V1UserFeatures. # noqa: E501
@@ -45,6 +45,7 @@ class V1ValidateDataConnectionResponse(object):
45
45
  'efs': 'V1EfsConfig',
46
46
  'gcp': 'V1GcpDataConnection',
47
47
  'gcs_folder': 'V1GCSFolderDataConnection',
48
+ 'r2': 'V1R2DataConnection',
48
49
  's3_folder': 'V1S3FolderDataConnection'
49
50
  }
50
51
 
@@ -53,15 +54,17 @@ class V1ValidateDataConnectionResponse(object):
53
54
  'efs': 'efs',
54
55
  'gcp': 'gcp',
55
56
  'gcs_folder': 'gcsFolder',
57
+ 'r2': 'r2',
56
58
  's3_folder': 's3Folder'
57
59
  }
58
60
 
59
- def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None): # noqa: E501
61
+ def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, r2: 'V1R2DataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None): # noqa: E501
60
62
  """V1ValidateDataConnectionResponse - a model defined in Swagger""" # noqa: E501
61
63
  self._aws = None
62
64
  self._efs = None
63
65
  self._gcp = None
64
66
  self._gcs_folder = None
67
+ self._r2 = None
65
68
  self._s3_folder = None
66
69
  self.discriminator = None
67
70
  if aws is not None:
@@ -72,6 +75,8 @@ class V1ValidateDataConnectionResponse(object):
72
75
  self.gcp = gcp
73
76
  if gcs_folder is not None:
74
77
  self.gcs_folder = gcs_folder
78
+ if r2 is not None:
79
+ self.r2 = r2
75
80
  if s3_folder is not None:
76
81
  self.s3_folder = s3_folder
77
82
 
@@ -159,6 +164,27 @@ class V1ValidateDataConnectionResponse(object):
159
164
 
160
165
  self._gcs_folder = gcs_folder
161
166
 
167
+ @property
168
+ def r2(self) -> 'V1R2DataConnection':
169
+ """Gets the r2 of this V1ValidateDataConnectionResponse. # noqa: E501
170
+
171
+
172
+ :return: The r2 of this V1ValidateDataConnectionResponse. # noqa: E501
173
+ :rtype: V1R2DataConnection
174
+ """
175
+ return self._r2
176
+
177
+ @r2.setter
178
+ def r2(self, r2: 'V1R2DataConnection'):
179
+ """Sets the r2 of this V1ValidateDataConnectionResponse.
180
+
181
+
182
+ :param r2: The r2 of this V1ValidateDataConnectionResponse. # noqa: E501
183
+ :type: V1R2DataConnection
184
+ """
185
+
186
+ self._r2 = r2
187
+
162
188
  @property
163
189
  def s3_folder(self) -> 'V1S3FolderDataConnection':
164
190
  """Gets the s3_folder of this V1ValidateDataConnectionResponse. # noqa: E501