lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc0__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 (70) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +9 -2
  3. lightning_sdk/api/deployment_api.py +9 -9
  4. lightning_sdk/api/llm_api.py +5 -11
  5. lightning_sdk/api/pipeline_api.py +31 -11
  6. lightning_sdk/api/studio_api.py +4 -0
  7. lightning_sdk/base_studio.py +22 -6
  8. lightning_sdk/deployment/deployment.py +17 -7
  9. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  11. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  12. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  13. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  14. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  15. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/models/__init__.py +11 -0
  17. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  20. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  21. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  24. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  40. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  41. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  43. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  46. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +185 -29
  58. lightning_sdk/llm/llm.py +113 -115
  59. lightning_sdk/llm/public_assistants.json +8 -0
  60. lightning_sdk/pipeline/pipeline.py +17 -2
  61. lightning_sdk/pipeline/printer.py +11 -10
  62. lightning_sdk/pipeline/steps.py +4 -1
  63. lightning_sdk/pipeline/utils.py +29 -4
  64. lightning_sdk/studio.py +3 -0
  65. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  66. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +70 -57
  67. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  68. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  69. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  70. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -53,7 +53,6 @@ class V1UserFeatures(object):
53
53
  'capacity_reservation_byoc': 'bool',
54
54
  'capacity_reservation_dry_run': 'bool',
55
55
  'chat_models': 'bool',
56
- 'cloud_space_environment_templates': 'bool',
57
56
  'cloudy_vibe_code': 'bool',
58
57
  'code_tab': 'bool',
59
58
  'collab_screen_sharing': 'bool',
@@ -69,13 +68,16 @@ class V1UserFeatures(object):
69
68
  'enterprise_compute_admin': 'bool',
70
69
  'fair_share': 'bool',
71
70
  'featured_studios_admin': 'bool',
71
+ 'gcs_connections_optimized': 'bool',
72
72
  'gcs_folders': 'bool',
73
73
  'instant_capacity_reservation': 'bool',
74
74
  'job_artifacts_v2': 'bool',
75
+ 'kubernetes_clusters': 'bool',
75
76
  'lambda_labs': 'bool',
76
77
  'lambda_labs_studios': 'bool',
77
78
  'landing_studios': 'bool',
78
79
  'lit_logger': 'bool',
80
+ 'litcr_byoc_gcp': 'bool',
79
81
  'machine_selector_v2': 'bool',
80
82
  'marketplace': 'bool',
81
83
  'mmt_fault_tolerance': 'bool',
@@ -91,6 +93,7 @@ class V1UserFeatures(object):
91
93
  'org_level_member_permissions': 'bool',
92
94
  'org_usage_limits': 'bool',
93
95
  'paygo_free_storage_limit_check': 'bool',
96
+ 'persistent_disk': 'bool',
94
97
  'pipelines': 'bool',
95
98
  'plugin_distributed': 'bool',
96
99
  'plugin_inference': 'bool',
@@ -112,15 +115,18 @@ class V1UserFeatures(object):
112
115
  'show_dev_admin': 'bool',
113
116
  'single_wallet': 'bool',
114
117
  'slurm': 'bool',
118
+ 'specialised_studios': 'bool',
115
119
  'storage_overuse_deletion': 'bool',
116
120
  'studio_config': 'bool',
117
121
  'studio_version_visibility': 'bool',
118
122
  'trainium2': 'bool',
123
+ 'use_internal_data_connection_mounts': 'bool',
119
124
  'use_rclone_mounts_only': 'bool',
120
125
  'voltage_park': 'bool',
121
126
  'voltage_park_studios': 'bool',
122
127
  'vultr': 'bool',
123
- 'weka': 'bool'
128
+ 'weka': 'bool',
129
+ 'writable_s3_connections': 'bool'
124
130
  }
125
131
 
126
132
  attribute_map = {
@@ -136,7 +142,6 @@ class V1UserFeatures(object):
136
142
  'capacity_reservation_byoc': 'capacityReservationByoc',
137
143
  'capacity_reservation_dry_run': 'capacityReservationDryRun',
138
144
  'chat_models': 'chatModels',
139
- 'cloud_space_environment_templates': 'cloudSpaceEnvironmentTemplates',
140
145
  'cloudy_vibe_code': 'cloudyVibeCode',
141
146
  'code_tab': 'codeTab',
142
147
  'collab_screen_sharing': 'collabScreenSharing',
@@ -152,13 +157,16 @@ class V1UserFeatures(object):
152
157
  'enterprise_compute_admin': 'enterpriseComputeAdmin',
153
158
  'fair_share': 'fairShare',
154
159
  'featured_studios_admin': 'featuredStudiosAdmin',
160
+ 'gcs_connections_optimized': 'gcsConnectionsOptimized',
155
161
  'gcs_folders': 'gcsFolders',
156
162
  'instant_capacity_reservation': 'instantCapacityReservation',
157
163
  'job_artifacts_v2': 'jobArtifactsV2',
164
+ 'kubernetes_clusters': 'kubernetesClusters',
158
165
  'lambda_labs': 'lambdaLabs',
159
166
  'lambda_labs_studios': 'lambdaLabsStudios',
160
167
  'landing_studios': 'landingStudios',
161
168
  'lit_logger': 'litLogger',
169
+ 'litcr_byoc_gcp': 'litcrByocGcp',
162
170
  'machine_selector_v2': 'machineSelectorV2',
163
171
  'marketplace': 'marketplace',
164
172
  'mmt_fault_tolerance': 'mmtFaultTolerance',
@@ -174,6 +182,7 @@ class V1UserFeatures(object):
174
182
  'org_level_member_permissions': 'orgLevelMemberPermissions',
175
183
  'org_usage_limits': 'orgUsageLimits',
176
184
  'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
185
+ 'persistent_disk': 'persistentDisk',
177
186
  'pipelines': 'pipelines',
178
187
  'plugin_distributed': 'pluginDistributed',
179
188
  'plugin_inference': 'pluginInference',
@@ -195,18 +204,21 @@ class V1UserFeatures(object):
195
204
  'show_dev_admin': 'showDevAdmin',
196
205
  'single_wallet': 'singleWallet',
197
206
  'slurm': 'slurm',
207
+ 'specialised_studios': 'specialisedStudios',
198
208
  'storage_overuse_deletion': 'storageOveruseDeletion',
199
209
  'studio_config': 'studioConfig',
200
210
  'studio_version_visibility': 'studioVersionVisibility',
201
211
  'trainium2': 'trainium2',
212
+ 'use_internal_data_connection_mounts': 'useInternalDataConnectionMounts',
202
213
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
203
214
  'voltage_park': 'voltagePark',
204
215
  'voltage_park_studios': 'voltageParkStudios',
205
216
  'vultr': 'vultr',
206
- 'weka': 'weka'
217
+ 'weka': 'weka',
218
+ 'writable_s3_connections': 'writableS3Connections'
207
219
  }
208
220
 
209
- def __init__(self, accurate_billing: 'bool' =None, 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, 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, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, data_connection_flushing_v2: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, machine_selector_v2: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: '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, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: '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, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
221
+ def __init__(self, accurate_billing: 'bool' =None, 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, 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, cloudy_vibe_code: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, data_connection_flushing_v2: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_clusters: 'bool' =None, lambda_labs: 'bool' =None, lambda_labs_studios: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, litcr_byoc_gcp: 'bool' =None, machine_selector_v2: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multicloud_folders: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nebius: 'bool' =None, nebius_cpu_studios: 'bool' =None, nebius_gpu_studios: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, persistent_disk: '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, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, reserved_machines_tab: '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, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, voltage_park_studios: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None): # noqa: E501
210
222
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
211
223
  self._accurate_billing = None
212
224
  self._affiliate_links = None
@@ -220,7 +232,6 @@ class V1UserFeatures(object):
220
232
  self._capacity_reservation_byoc = None
221
233
  self._capacity_reservation_dry_run = None
222
234
  self._chat_models = None
223
- self._cloud_space_environment_templates = None
224
235
  self._cloudy_vibe_code = None
225
236
  self._code_tab = None
226
237
  self._collab_screen_sharing = None
@@ -236,13 +247,16 @@ class V1UserFeatures(object):
236
247
  self._enterprise_compute_admin = None
237
248
  self._fair_share = None
238
249
  self._featured_studios_admin = None
250
+ self._gcs_connections_optimized = None
239
251
  self._gcs_folders = None
240
252
  self._instant_capacity_reservation = None
241
253
  self._job_artifacts_v2 = None
254
+ self._kubernetes_clusters = None
242
255
  self._lambda_labs = None
243
256
  self._lambda_labs_studios = None
244
257
  self._landing_studios = None
245
258
  self._lit_logger = None
259
+ self._litcr_byoc_gcp = None
246
260
  self._machine_selector_v2 = None
247
261
  self._marketplace = None
248
262
  self._mmt_fault_tolerance = None
@@ -258,6 +272,7 @@ class V1UserFeatures(object):
258
272
  self._org_level_member_permissions = None
259
273
  self._org_usage_limits = None
260
274
  self._paygo_free_storage_limit_check = None
275
+ self._persistent_disk = None
261
276
  self._pipelines = None
262
277
  self._plugin_distributed = None
263
278
  self._plugin_inference = None
@@ -279,15 +294,18 @@ class V1UserFeatures(object):
279
294
  self._show_dev_admin = None
280
295
  self._single_wallet = None
281
296
  self._slurm = None
297
+ self._specialised_studios = None
282
298
  self._storage_overuse_deletion = None
283
299
  self._studio_config = None
284
300
  self._studio_version_visibility = None
285
301
  self._trainium2 = None
302
+ self._use_internal_data_connection_mounts = None
286
303
  self._use_rclone_mounts_only = None
287
304
  self._voltage_park = None
288
305
  self._voltage_park_studios = None
289
306
  self._vultr = None
290
307
  self._weka = None
308
+ self._writable_s3_connections = None
291
309
  self.discriminator = None
292
310
  if accurate_billing is not None:
293
311
  self.accurate_billing = accurate_billing
@@ -313,8 +331,6 @@ class V1UserFeatures(object):
313
331
  self.capacity_reservation_dry_run = capacity_reservation_dry_run
314
332
  if chat_models is not None:
315
333
  self.chat_models = chat_models
316
- if cloud_space_environment_templates is not None:
317
- self.cloud_space_environment_templates = cloud_space_environment_templates
318
334
  if cloudy_vibe_code is not None:
319
335
  self.cloudy_vibe_code = cloudy_vibe_code
320
336
  if code_tab is not None:
@@ -345,12 +361,16 @@ class V1UserFeatures(object):
345
361
  self.fair_share = fair_share
346
362
  if featured_studios_admin is not None:
347
363
  self.featured_studios_admin = featured_studios_admin
364
+ if gcs_connections_optimized is not None:
365
+ self.gcs_connections_optimized = gcs_connections_optimized
348
366
  if gcs_folders is not None:
349
367
  self.gcs_folders = gcs_folders
350
368
  if instant_capacity_reservation is not None:
351
369
  self.instant_capacity_reservation = instant_capacity_reservation
352
370
  if job_artifacts_v2 is not None:
353
371
  self.job_artifacts_v2 = job_artifacts_v2
372
+ if kubernetes_clusters is not None:
373
+ self.kubernetes_clusters = kubernetes_clusters
354
374
  if lambda_labs is not None:
355
375
  self.lambda_labs = lambda_labs
356
376
  if lambda_labs_studios is not None:
@@ -359,6 +379,8 @@ class V1UserFeatures(object):
359
379
  self.landing_studios = landing_studios
360
380
  if lit_logger is not None:
361
381
  self.lit_logger = lit_logger
382
+ if litcr_byoc_gcp is not None:
383
+ self.litcr_byoc_gcp = litcr_byoc_gcp
362
384
  if machine_selector_v2 is not None:
363
385
  self.machine_selector_v2 = machine_selector_v2
364
386
  if marketplace is not None:
@@ -389,6 +411,8 @@ class V1UserFeatures(object):
389
411
  self.org_usage_limits = org_usage_limits
390
412
  if paygo_free_storage_limit_check is not None:
391
413
  self.paygo_free_storage_limit_check = paygo_free_storage_limit_check
414
+ if persistent_disk is not None:
415
+ self.persistent_disk = persistent_disk
392
416
  if pipelines is not None:
393
417
  self.pipelines = pipelines
394
418
  if plugin_distributed is not None:
@@ -431,6 +455,8 @@ class V1UserFeatures(object):
431
455
  self.single_wallet = single_wallet
432
456
  if slurm is not None:
433
457
  self.slurm = slurm
458
+ if specialised_studios is not None:
459
+ self.specialised_studios = specialised_studios
434
460
  if storage_overuse_deletion is not None:
435
461
  self.storage_overuse_deletion = storage_overuse_deletion
436
462
  if studio_config is not None:
@@ -439,6 +465,8 @@ class V1UserFeatures(object):
439
465
  self.studio_version_visibility = studio_version_visibility
440
466
  if trainium2 is not None:
441
467
  self.trainium2 = trainium2
468
+ if use_internal_data_connection_mounts is not None:
469
+ self.use_internal_data_connection_mounts = use_internal_data_connection_mounts
442
470
  if use_rclone_mounts_only is not None:
443
471
  self.use_rclone_mounts_only = use_rclone_mounts_only
444
472
  if voltage_park is not None:
@@ -449,6 +477,8 @@ class V1UserFeatures(object):
449
477
  self.vultr = vultr
450
478
  if weka is not None:
451
479
  self.weka = weka
480
+ if writable_s3_connections is not None:
481
+ self.writable_s3_connections = writable_s3_connections
452
482
 
453
483
  @property
454
484
  def accurate_billing(self) -> 'bool':
@@ -702,27 +732,6 @@ class V1UserFeatures(object):
702
732
 
703
733
  self._chat_models = chat_models
704
734
 
705
- @property
706
- def cloud_space_environment_templates(self) -> 'bool':
707
- """Gets the cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
708
-
709
-
710
- :return: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
711
- :rtype: bool
712
- """
713
- return self._cloud_space_environment_templates
714
-
715
- @cloud_space_environment_templates.setter
716
- def cloud_space_environment_templates(self, cloud_space_environment_templates: 'bool'):
717
- """Sets the cloud_space_environment_templates of this V1UserFeatures.
718
-
719
-
720
- :param cloud_space_environment_templates: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
721
- :type: bool
722
- """
723
-
724
- self._cloud_space_environment_templates = cloud_space_environment_templates
725
-
726
735
  @property
727
736
  def cloudy_vibe_code(self) -> 'bool':
728
737
  """Gets the cloudy_vibe_code of this V1UserFeatures. # noqa: E501
@@ -1038,6 +1047,27 @@ class V1UserFeatures(object):
1038
1047
 
1039
1048
  self._featured_studios_admin = featured_studios_admin
1040
1049
 
1050
+ @property
1051
+ def gcs_connections_optimized(self) -> 'bool':
1052
+ """Gets the gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1053
+
1054
+
1055
+ :return: The gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1056
+ :rtype: bool
1057
+ """
1058
+ return self._gcs_connections_optimized
1059
+
1060
+ @gcs_connections_optimized.setter
1061
+ def gcs_connections_optimized(self, gcs_connections_optimized: 'bool'):
1062
+ """Sets the gcs_connections_optimized of this V1UserFeatures.
1063
+
1064
+
1065
+ :param gcs_connections_optimized: The gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1066
+ :type: bool
1067
+ """
1068
+
1069
+ self._gcs_connections_optimized = gcs_connections_optimized
1070
+
1041
1071
  @property
1042
1072
  def gcs_folders(self) -> 'bool':
1043
1073
  """Gets the gcs_folders of this V1UserFeatures. # noqa: E501
@@ -1101,6 +1131,27 @@ class V1UserFeatures(object):
1101
1131
 
1102
1132
  self._job_artifacts_v2 = job_artifacts_v2
1103
1133
 
1134
+ @property
1135
+ def kubernetes_clusters(self) -> 'bool':
1136
+ """Gets the kubernetes_clusters of this V1UserFeatures. # noqa: E501
1137
+
1138
+
1139
+ :return: The kubernetes_clusters of this V1UserFeatures. # noqa: E501
1140
+ :rtype: bool
1141
+ """
1142
+ return self._kubernetes_clusters
1143
+
1144
+ @kubernetes_clusters.setter
1145
+ def kubernetes_clusters(self, kubernetes_clusters: 'bool'):
1146
+ """Sets the kubernetes_clusters of this V1UserFeatures.
1147
+
1148
+
1149
+ :param kubernetes_clusters: The kubernetes_clusters of this V1UserFeatures. # noqa: E501
1150
+ :type: bool
1151
+ """
1152
+
1153
+ self._kubernetes_clusters = kubernetes_clusters
1154
+
1104
1155
  @property
1105
1156
  def lambda_labs(self) -> 'bool':
1106
1157
  """Gets the lambda_labs of this V1UserFeatures. # noqa: E501
@@ -1185,6 +1236,27 @@ class V1UserFeatures(object):
1185
1236
 
1186
1237
  self._lit_logger = lit_logger
1187
1238
 
1239
+ @property
1240
+ def litcr_byoc_gcp(self) -> 'bool':
1241
+ """Gets the litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1242
+
1243
+
1244
+ :return: The litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1245
+ :rtype: bool
1246
+ """
1247
+ return self._litcr_byoc_gcp
1248
+
1249
+ @litcr_byoc_gcp.setter
1250
+ def litcr_byoc_gcp(self, litcr_byoc_gcp: 'bool'):
1251
+ """Sets the litcr_byoc_gcp of this V1UserFeatures.
1252
+
1253
+
1254
+ :param litcr_byoc_gcp: The litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1255
+ :type: bool
1256
+ """
1257
+
1258
+ self._litcr_byoc_gcp = litcr_byoc_gcp
1259
+
1188
1260
  @property
1189
1261
  def machine_selector_v2(self) -> 'bool':
1190
1262
  """Gets the machine_selector_v2 of this V1UserFeatures. # noqa: E501
@@ -1500,6 +1572,27 @@ class V1UserFeatures(object):
1500
1572
 
1501
1573
  self._paygo_free_storage_limit_check = paygo_free_storage_limit_check
1502
1574
 
1575
+ @property
1576
+ def persistent_disk(self) -> 'bool':
1577
+ """Gets the persistent_disk of this V1UserFeatures. # noqa: E501
1578
+
1579
+
1580
+ :return: The persistent_disk of this V1UserFeatures. # noqa: E501
1581
+ :rtype: bool
1582
+ """
1583
+ return self._persistent_disk
1584
+
1585
+ @persistent_disk.setter
1586
+ def persistent_disk(self, persistent_disk: 'bool'):
1587
+ """Sets the persistent_disk of this V1UserFeatures.
1588
+
1589
+
1590
+ :param persistent_disk: The persistent_disk of this V1UserFeatures. # noqa: E501
1591
+ :type: bool
1592
+ """
1593
+
1594
+ self._persistent_disk = persistent_disk
1595
+
1503
1596
  @property
1504
1597
  def pipelines(self) -> 'bool':
1505
1598
  """Gets the pipelines of this V1UserFeatures. # noqa: E501
@@ -1941,6 +2034,27 @@ class V1UserFeatures(object):
1941
2034
 
1942
2035
  self._slurm = slurm
1943
2036
 
2037
+ @property
2038
+ def specialised_studios(self) -> 'bool':
2039
+ """Gets the specialised_studios of this V1UserFeatures. # noqa: E501
2040
+
2041
+
2042
+ :return: The specialised_studios of this V1UserFeatures. # noqa: E501
2043
+ :rtype: bool
2044
+ """
2045
+ return self._specialised_studios
2046
+
2047
+ @specialised_studios.setter
2048
+ def specialised_studios(self, specialised_studios: 'bool'):
2049
+ """Sets the specialised_studios of this V1UserFeatures.
2050
+
2051
+
2052
+ :param specialised_studios: The specialised_studios of this V1UserFeatures. # noqa: E501
2053
+ :type: bool
2054
+ """
2055
+
2056
+ self._specialised_studios = specialised_studios
2057
+
1944
2058
  @property
1945
2059
  def storage_overuse_deletion(self) -> 'bool':
1946
2060
  """Gets the storage_overuse_deletion of this V1UserFeatures. # noqa: E501
@@ -2025,6 +2139,27 @@ class V1UserFeatures(object):
2025
2139
 
2026
2140
  self._trainium2 = trainium2
2027
2141
 
2142
+ @property
2143
+ def use_internal_data_connection_mounts(self) -> 'bool':
2144
+ """Gets the use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2145
+
2146
+
2147
+ :return: The use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2148
+ :rtype: bool
2149
+ """
2150
+ return self._use_internal_data_connection_mounts
2151
+
2152
+ @use_internal_data_connection_mounts.setter
2153
+ def use_internal_data_connection_mounts(self, use_internal_data_connection_mounts: 'bool'):
2154
+ """Sets the use_internal_data_connection_mounts of this V1UserFeatures.
2155
+
2156
+
2157
+ :param use_internal_data_connection_mounts: The use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2158
+ :type: bool
2159
+ """
2160
+
2161
+ self._use_internal_data_connection_mounts = use_internal_data_connection_mounts
2162
+
2028
2163
  @property
2029
2164
  def use_rclone_mounts_only(self) -> 'bool':
2030
2165
  """Gets the use_rclone_mounts_only of this V1UserFeatures. # noqa: E501
@@ -2130,6 +2265,27 @@ class V1UserFeatures(object):
2130
2265
 
2131
2266
  self._weka = weka
2132
2267
 
2268
+ @property
2269
+ def writable_s3_connections(self) -> 'bool':
2270
+ """Gets the writable_s3_connections of this V1UserFeatures. # noqa: E501
2271
+
2272
+
2273
+ :return: The writable_s3_connections of this V1UserFeatures. # noqa: E501
2274
+ :rtype: bool
2275
+ """
2276
+ return self._writable_s3_connections
2277
+
2278
+ @writable_s3_connections.setter
2279
+ def writable_s3_connections(self, writable_s3_connections: 'bool'):
2280
+ """Sets the writable_s3_connections of this V1UserFeatures.
2281
+
2282
+
2283
+ :param writable_s3_connections: The writable_s3_connections of this V1UserFeatures. # noqa: E501
2284
+ :type: bool
2285
+ """
2286
+
2287
+ self._writable_s3_connections = writable_s3_connections
2288
+
2133
2289
  def to_dict(self) -> dict:
2134
2290
  """Returns the model properties as a dict"""
2135
2291
  result = {}