lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc1__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 (83) 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 +23 -13
  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 +18 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +2 -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/cluster_service_api.py +9 -1
  14. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +13 -1
  15. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  17. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  18. lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py +258 -0
  19. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  20. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  23. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +53 -1
  25. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  27. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py +103 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +29 -3
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_volume_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  56. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +149 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py +104 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +1 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_update_volume_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +317 -31
  70. lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.py +123 -0
  71. lightning_sdk/llm/llm.py +118 -115
  72. lightning_sdk/llm/public_assistants.json +8 -0
  73. lightning_sdk/pipeline/pipeline.py +17 -2
  74. lightning_sdk/pipeline/printer.py +11 -10
  75. lightning_sdk/pipeline/steps.py +4 -1
  76. lightning_sdk/pipeline/utils.py +29 -4
  77. lightning_sdk/studio.py +3 -0
  78. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/METADATA +1 -1
  79. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/RECORD +83 -64
  80. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/LICENSE +0 -0
  81. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/WHEEL +0 -0
  82. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/entry_points.txt +0 -0
  83. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/top_level.txt +0 -0
@@ -41,7 +41,9 @@ class V1UserFeatures(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'academic_tier': 'bool',
44
45
  'accurate_billing': 'bool',
46
+ 'add_data_v2': 'bool',
45
47
  'affiliate_links': 'bool',
46
48
  'agents_v2': 'bool',
47
49
  'ai_hub_monetization': 'bool',
@@ -53,7 +55,7 @@ class V1UserFeatures(object):
53
55
  'capacity_reservation_byoc': 'bool',
54
56
  'capacity_reservation_dry_run': 'bool',
55
57
  'chat_models': 'bool',
56
- 'cloud_space_environment_templates': 'bool',
58
+ 'cloudspace_schedules': 'bool',
57
59
  'cloudy_vibe_code': 'bool',
58
60
  'code_tab': 'bool',
59
61
  'collab_screen_sharing': 'bool',
@@ -69,13 +71,17 @@ class V1UserFeatures(object):
69
71
  'enterprise_compute_admin': 'bool',
70
72
  'fair_share': 'bool',
71
73
  'featured_studios_admin': 'bool',
74
+ 'gcs_connections_optimized': 'bool',
72
75
  'gcs_folders': 'bool',
76
+ 'gcs_fuse': 'bool',
73
77
  'instant_capacity_reservation': 'bool',
74
78
  'job_artifacts_v2': 'bool',
79
+ 'kubernetes_clusters': 'bool',
75
80
  'lambda_labs': 'bool',
76
81
  'lambda_labs_studios': 'bool',
77
82
  'landing_studios': 'bool',
78
83
  'lit_logger': 'bool',
84
+ 'litcr_byoc_gcp': 'bool',
79
85
  'machine_selector_v2': 'bool',
80
86
  'marketplace': 'bool',
81
87
  'mmt_fault_tolerance': 'bool',
@@ -88,10 +94,11 @@ class V1UserFeatures(object):
88
94
  'nebius_cpu_studios': 'bool',
89
95
  'nebius_gpu_studios': 'bool',
90
96
  'nerf_fs_nonpaying': 'bool',
97
+ 'onboarding_v2': 'bool',
91
98
  'org_level_member_permissions': 'bool',
92
99
  'org_usage_limits': 'bool',
93
100
  'paygo_free_storage_limit_check': 'bool',
94
- 'pipelines': 'bool',
101
+ 'persistent_disk': 'bool',
95
102
  'plugin_distributed': 'bool',
96
103
  'plugin_inference': 'bool',
97
104
  'plugin_label_studio': 'bool',
@@ -112,19 +119,25 @@ class V1UserFeatures(object):
112
119
  'show_dev_admin': 'bool',
113
120
  'single_wallet': 'bool',
114
121
  'slurm': 'bool',
122
+ 'specialised_studios': 'bool',
115
123
  'storage_overuse_deletion': 'bool',
116
124
  'studio_config': 'bool',
125
+ 'studio_sharing_v2': 'bool',
117
126
  'studio_version_visibility': 'bool',
118
127
  'trainium2': 'bool',
128
+ 'use_internal_data_connection_mounts': 'bool',
119
129
  'use_rclone_mounts_only': 'bool',
120
130
  'voltage_park': 'bool',
121
131
  'voltage_park_studios': 'bool',
122
132
  'vultr': 'bool',
123
- 'weka': 'bool'
133
+ 'weka': 'bool',
134
+ 'writable_s3_connections': 'bool'
124
135
  }
125
136
 
126
137
  attribute_map = {
138
+ 'academic_tier': 'academicTier',
127
139
  'accurate_billing': 'accurateBilling',
140
+ 'add_data_v2': 'addDataV2',
128
141
  'affiliate_links': 'affiliateLinks',
129
142
  'agents_v2': 'agentsV2',
130
143
  'ai_hub_monetization': 'aiHubMonetization',
@@ -136,7 +149,7 @@ class V1UserFeatures(object):
136
149
  'capacity_reservation_byoc': 'capacityReservationByoc',
137
150
  'capacity_reservation_dry_run': 'capacityReservationDryRun',
138
151
  'chat_models': 'chatModels',
139
- 'cloud_space_environment_templates': 'cloudSpaceEnvironmentTemplates',
152
+ 'cloudspace_schedules': 'cloudspaceSchedules',
140
153
  'cloudy_vibe_code': 'cloudyVibeCode',
141
154
  'code_tab': 'codeTab',
142
155
  'collab_screen_sharing': 'collabScreenSharing',
@@ -152,13 +165,17 @@ class V1UserFeatures(object):
152
165
  'enterprise_compute_admin': 'enterpriseComputeAdmin',
153
166
  'fair_share': 'fairShare',
154
167
  'featured_studios_admin': 'featuredStudiosAdmin',
168
+ 'gcs_connections_optimized': 'gcsConnectionsOptimized',
155
169
  'gcs_folders': 'gcsFolders',
170
+ 'gcs_fuse': 'gcsFuse',
156
171
  'instant_capacity_reservation': 'instantCapacityReservation',
157
172
  'job_artifacts_v2': 'jobArtifactsV2',
173
+ 'kubernetes_clusters': 'kubernetesClusters',
158
174
  'lambda_labs': 'lambdaLabs',
159
175
  'lambda_labs_studios': 'lambdaLabsStudios',
160
176
  'landing_studios': 'landingStudios',
161
177
  'lit_logger': 'litLogger',
178
+ 'litcr_byoc_gcp': 'litcrByocGcp',
162
179
  'machine_selector_v2': 'machineSelectorV2',
163
180
  'marketplace': 'marketplace',
164
181
  'mmt_fault_tolerance': 'mmtFaultTolerance',
@@ -171,10 +188,11 @@ class V1UserFeatures(object):
171
188
  'nebius_cpu_studios': 'nebiusCpuStudios',
172
189
  'nebius_gpu_studios': 'nebiusGpuStudios',
173
190
  'nerf_fs_nonpaying': 'nerfFsNonpaying',
191
+ 'onboarding_v2': 'onboardingV2',
174
192
  'org_level_member_permissions': 'orgLevelMemberPermissions',
175
193
  'org_usage_limits': 'orgUsageLimits',
176
194
  'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
177
- 'pipelines': 'pipelines',
195
+ 'persistent_disk': 'persistentDisk',
178
196
  'plugin_distributed': 'pluginDistributed',
179
197
  'plugin_inference': 'pluginInference',
180
198
  'plugin_label_studio': 'pluginLabelStudio',
@@ -195,20 +213,26 @@ class V1UserFeatures(object):
195
213
  'show_dev_admin': 'showDevAdmin',
196
214
  'single_wallet': 'singleWallet',
197
215
  'slurm': 'slurm',
216
+ 'specialised_studios': 'specialisedStudios',
198
217
  'storage_overuse_deletion': 'storageOveruseDeletion',
199
218
  'studio_config': 'studioConfig',
219
+ 'studio_sharing_v2': 'studioSharingV2',
200
220
  'studio_version_visibility': 'studioVersionVisibility',
201
221
  'trainium2': 'trainium2',
222
+ 'use_internal_data_connection_mounts': 'useInternalDataConnectionMounts',
202
223
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
203
224
  'voltage_park': 'voltagePark',
204
225
  'voltage_park_studios': 'voltageParkStudios',
205
226
  'vultr': 'vultr',
206
- 'weka': 'weka'
227
+ 'weka': 'weka',
228
+ 'writable_s3_connections': 'writableS3Connections'
207
229
  }
208
230
 
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
231
+ def __init__(self, academic_tier: 'bool' =None, accurate_billing: 'bool' =None, add_data_v2: '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, cloudspace_schedules: '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, gcs_fuse: '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, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, paygo_free_storage_limit_check: 'bool' =None, persistent_disk: '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_sharing_v2: '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
232
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
233
+ self._academic_tier = None
211
234
  self._accurate_billing = None
235
+ self._add_data_v2 = None
212
236
  self._affiliate_links = None
213
237
  self._agents_v2 = None
214
238
  self._ai_hub_monetization = None
@@ -220,7 +244,7 @@ class V1UserFeatures(object):
220
244
  self._capacity_reservation_byoc = None
221
245
  self._capacity_reservation_dry_run = None
222
246
  self._chat_models = None
223
- self._cloud_space_environment_templates = None
247
+ self._cloudspace_schedules = None
224
248
  self._cloudy_vibe_code = None
225
249
  self._code_tab = None
226
250
  self._collab_screen_sharing = None
@@ -236,13 +260,17 @@ class V1UserFeatures(object):
236
260
  self._enterprise_compute_admin = None
237
261
  self._fair_share = None
238
262
  self._featured_studios_admin = None
263
+ self._gcs_connections_optimized = None
239
264
  self._gcs_folders = None
265
+ self._gcs_fuse = None
240
266
  self._instant_capacity_reservation = None
241
267
  self._job_artifacts_v2 = None
268
+ self._kubernetes_clusters = None
242
269
  self._lambda_labs = None
243
270
  self._lambda_labs_studios = None
244
271
  self._landing_studios = None
245
272
  self._lit_logger = None
273
+ self._litcr_byoc_gcp = None
246
274
  self._machine_selector_v2 = None
247
275
  self._marketplace = None
248
276
  self._mmt_fault_tolerance = None
@@ -255,10 +283,11 @@ class V1UserFeatures(object):
255
283
  self._nebius_cpu_studios = None
256
284
  self._nebius_gpu_studios = None
257
285
  self._nerf_fs_nonpaying = None
286
+ self._onboarding_v2 = None
258
287
  self._org_level_member_permissions = None
259
288
  self._org_usage_limits = None
260
289
  self._paygo_free_storage_limit_check = None
261
- self._pipelines = None
290
+ self._persistent_disk = None
262
291
  self._plugin_distributed = None
263
292
  self._plugin_inference = None
264
293
  self._plugin_label_studio = None
@@ -279,18 +308,26 @@ class V1UserFeatures(object):
279
308
  self._show_dev_admin = None
280
309
  self._single_wallet = None
281
310
  self._slurm = None
311
+ self._specialised_studios = None
282
312
  self._storage_overuse_deletion = None
283
313
  self._studio_config = None
314
+ self._studio_sharing_v2 = None
284
315
  self._studio_version_visibility = None
285
316
  self._trainium2 = None
317
+ self._use_internal_data_connection_mounts = None
286
318
  self._use_rclone_mounts_only = None
287
319
  self._voltage_park = None
288
320
  self._voltage_park_studios = None
289
321
  self._vultr = None
290
322
  self._weka = None
323
+ self._writable_s3_connections = None
291
324
  self.discriminator = None
325
+ if academic_tier is not None:
326
+ self.academic_tier = academic_tier
292
327
  if accurate_billing is not None:
293
328
  self.accurate_billing = accurate_billing
329
+ if add_data_v2 is not None:
330
+ self.add_data_v2 = add_data_v2
294
331
  if affiliate_links is not None:
295
332
  self.affiliate_links = affiliate_links
296
333
  if agents_v2 is not None:
@@ -313,8 +350,8 @@ class V1UserFeatures(object):
313
350
  self.capacity_reservation_dry_run = capacity_reservation_dry_run
314
351
  if chat_models is not None:
315
352
  self.chat_models = chat_models
316
- if cloud_space_environment_templates is not None:
317
- self.cloud_space_environment_templates = cloud_space_environment_templates
353
+ if cloudspace_schedules is not None:
354
+ self.cloudspace_schedules = cloudspace_schedules
318
355
  if cloudy_vibe_code is not None:
319
356
  self.cloudy_vibe_code = cloudy_vibe_code
320
357
  if code_tab is not None:
@@ -345,12 +382,18 @@ class V1UserFeatures(object):
345
382
  self.fair_share = fair_share
346
383
  if featured_studios_admin is not None:
347
384
  self.featured_studios_admin = featured_studios_admin
385
+ if gcs_connections_optimized is not None:
386
+ self.gcs_connections_optimized = gcs_connections_optimized
348
387
  if gcs_folders is not None:
349
388
  self.gcs_folders = gcs_folders
389
+ if gcs_fuse is not None:
390
+ self.gcs_fuse = gcs_fuse
350
391
  if instant_capacity_reservation is not None:
351
392
  self.instant_capacity_reservation = instant_capacity_reservation
352
393
  if job_artifacts_v2 is not None:
353
394
  self.job_artifacts_v2 = job_artifacts_v2
395
+ if kubernetes_clusters is not None:
396
+ self.kubernetes_clusters = kubernetes_clusters
354
397
  if lambda_labs is not None:
355
398
  self.lambda_labs = lambda_labs
356
399
  if lambda_labs_studios is not None:
@@ -359,6 +402,8 @@ class V1UserFeatures(object):
359
402
  self.landing_studios = landing_studios
360
403
  if lit_logger is not None:
361
404
  self.lit_logger = lit_logger
405
+ if litcr_byoc_gcp is not None:
406
+ self.litcr_byoc_gcp = litcr_byoc_gcp
362
407
  if machine_selector_v2 is not None:
363
408
  self.machine_selector_v2 = machine_selector_v2
364
409
  if marketplace is not None:
@@ -383,14 +428,16 @@ class V1UserFeatures(object):
383
428
  self.nebius_gpu_studios = nebius_gpu_studios
384
429
  if nerf_fs_nonpaying is not None:
385
430
  self.nerf_fs_nonpaying = nerf_fs_nonpaying
431
+ if onboarding_v2 is not None:
432
+ self.onboarding_v2 = onboarding_v2
386
433
  if org_level_member_permissions is not None:
387
434
  self.org_level_member_permissions = org_level_member_permissions
388
435
  if org_usage_limits is not None:
389
436
  self.org_usage_limits = org_usage_limits
390
437
  if paygo_free_storage_limit_check is not None:
391
438
  self.paygo_free_storage_limit_check = paygo_free_storage_limit_check
392
- if pipelines is not None:
393
- self.pipelines = pipelines
439
+ if persistent_disk is not None:
440
+ self.persistent_disk = persistent_disk
394
441
  if plugin_distributed is not None:
395
442
  self.plugin_distributed = plugin_distributed
396
443
  if plugin_inference is not None:
@@ -431,14 +478,20 @@ class V1UserFeatures(object):
431
478
  self.single_wallet = single_wallet
432
479
  if slurm is not None:
433
480
  self.slurm = slurm
481
+ if specialised_studios is not None:
482
+ self.specialised_studios = specialised_studios
434
483
  if storage_overuse_deletion is not None:
435
484
  self.storage_overuse_deletion = storage_overuse_deletion
436
485
  if studio_config is not None:
437
486
  self.studio_config = studio_config
487
+ if studio_sharing_v2 is not None:
488
+ self.studio_sharing_v2 = studio_sharing_v2
438
489
  if studio_version_visibility is not None:
439
490
  self.studio_version_visibility = studio_version_visibility
440
491
  if trainium2 is not None:
441
492
  self.trainium2 = trainium2
493
+ if use_internal_data_connection_mounts is not None:
494
+ self.use_internal_data_connection_mounts = use_internal_data_connection_mounts
442
495
  if use_rclone_mounts_only is not None:
443
496
  self.use_rclone_mounts_only = use_rclone_mounts_only
444
497
  if voltage_park is not None:
@@ -449,6 +502,29 @@ class V1UserFeatures(object):
449
502
  self.vultr = vultr
450
503
  if weka is not None:
451
504
  self.weka = weka
505
+ if writable_s3_connections is not None:
506
+ self.writable_s3_connections = writable_s3_connections
507
+
508
+ @property
509
+ def academic_tier(self) -> 'bool':
510
+ """Gets the academic_tier of this V1UserFeatures. # noqa: E501
511
+
512
+
513
+ :return: The academic_tier of this V1UserFeatures. # noqa: E501
514
+ :rtype: bool
515
+ """
516
+ return self._academic_tier
517
+
518
+ @academic_tier.setter
519
+ def academic_tier(self, academic_tier: 'bool'):
520
+ """Sets the academic_tier of this V1UserFeatures.
521
+
522
+
523
+ :param academic_tier: The academic_tier of this V1UserFeatures. # noqa: E501
524
+ :type: bool
525
+ """
526
+
527
+ self._academic_tier = academic_tier
452
528
 
453
529
  @property
454
530
  def accurate_billing(self) -> 'bool':
@@ -471,6 +547,27 @@ class V1UserFeatures(object):
471
547
 
472
548
  self._accurate_billing = accurate_billing
473
549
 
550
+ @property
551
+ def add_data_v2(self) -> 'bool':
552
+ """Gets the add_data_v2 of this V1UserFeatures. # noqa: E501
553
+
554
+
555
+ :return: The add_data_v2 of this V1UserFeatures. # noqa: E501
556
+ :rtype: bool
557
+ """
558
+ return self._add_data_v2
559
+
560
+ @add_data_v2.setter
561
+ def add_data_v2(self, add_data_v2: 'bool'):
562
+ """Sets the add_data_v2 of this V1UserFeatures.
563
+
564
+
565
+ :param add_data_v2: The add_data_v2 of this V1UserFeatures. # noqa: E501
566
+ :type: bool
567
+ """
568
+
569
+ self._add_data_v2 = add_data_v2
570
+
474
571
  @property
475
572
  def affiliate_links(self) -> 'bool':
476
573
  """Gets the affiliate_links of this V1UserFeatures. # noqa: E501
@@ -703,25 +800,25 @@ class V1UserFeatures(object):
703
800
  self._chat_models = chat_models
704
801
 
705
802
  @property
706
- def cloud_space_environment_templates(self) -> 'bool':
707
- """Gets the cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
803
+ def cloudspace_schedules(self) -> 'bool':
804
+ """Gets the cloudspace_schedules of this V1UserFeatures. # noqa: E501
708
805
 
709
806
 
710
- :return: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
807
+ :return: The cloudspace_schedules of this V1UserFeatures. # noqa: E501
711
808
  :rtype: bool
712
809
  """
713
- return self._cloud_space_environment_templates
810
+ return self._cloudspace_schedules
714
811
 
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.
812
+ @cloudspace_schedules.setter
813
+ def cloudspace_schedules(self, cloudspace_schedules: 'bool'):
814
+ """Sets the cloudspace_schedules of this V1UserFeatures.
718
815
 
719
816
 
720
- :param cloud_space_environment_templates: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
817
+ :param cloudspace_schedules: The cloudspace_schedules of this V1UserFeatures. # noqa: E501
721
818
  :type: bool
722
819
  """
723
820
 
724
- self._cloud_space_environment_templates = cloud_space_environment_templates
821
+ self._cloudspace_schedules = cloudspace_schedules
725
822
 
726
823
  @property
727
824
  def cloudy_vibe_code(self) -> 'bool':
@@ -1038,6 +1135,27 @@ class V1UserFeatures(object):
1038
1135
 
1039
1136
  self._featured_studios_admin = featured_studios_admin
1040
1137
 
1138
+ @property
1139
+ def gcs_connections_optimized(self) -> 'bool':
1140
+ """Gets the gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1141
+
1142
+
1143
+ :return: The gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1144
+ :rtype: bool
1145
+ """
1146
+ return self._gcs_connections_optimized
1147
+
1148
+ @gcs_connections_optimized.setter
1149
+ def gcs_connections_optimized(self, gcs_connections_optimized: 'bool'):
1150
+ """Sets the gcs_connections_optimized of this V1UserFeatures.
1151
+
1152
+
1153
+ :param gcs_connections_optimized: The gcs_connections_optimized of this V1UserFeatures. # noqa: E501
1154
+ :type: bool
1155
+ """
1156
+
1157
+ self._gcs_connections_optimized = gcs_connections_optimized
1158
+
1041
1159
  @property
1042
1160
  def gcs_folders(self) -> 'bool':
1043
1161
  """Gets the gcs_folders of this V1UserFeatures. # noqa: E501
@@ -1059,6 +1177,27 @@ class V1UserFeatures(object):
1059
1177
 
1060
1178
  self._gcs_folders = gcs_folders
1061
1179
 
1180
+ @property
1181
+ def gcs_fuse(self) -> 'bool':
1182
+ """Gets the gcs_fuse of this V1UserFeatures. # noqa: E501
1183
+
1184
+
1185
+ :return: The gcs_fuse of this V1UserFeatures. # noqa: E501
1186
+ :rtype: bool
1187
+ """
1188
+ return self._gcs_fuse
1189
+
1190
+ @gcs_fuse.setter
1191
+ def gcs_fuse(self, gcs_fuse: 'bool'):
1192
+ """Sets the gcs_fuse of this V1UserFeatures.
1193
+
1194
+
1195
+ :param gcs_fuse: The gcs_fuse of this V1UserFeatures. # noqa: E501
1196
+ :type: bool
1197
+ """
1198
+
1199
+ self._gcs_fuse = gcs_fuse
1200
+
1062
1201
  @property
1063
1202
  def instant_capacity_reservation(self) -> 'bool':
1064
1203
  """Gets the instant_capacity_reservation of this V1UserFeatures. # noqa: E501
@@ -1101,6 +1240,27 @@ class V1UserFeatures(object):
1101
1240
 
1102
1241
  self._job_artifacts_v2 = job_artifacts_v2
1103
1242
 
1243
+ @property
1244
+ def kubernetes_clusters(self) -> 'bool':
1245
+ """Gets the kubernetes_clusters of this V1UserFeatures. # noqa: E501
1246
+
1247
+
1248
+ :return: The kubernetes_clusters of this V1UserFeatures. # noqa: E501
1249
+ :rtype: bool
1250
+ """
1251
+ return self._kubernetes_clusters
1252
+
1253
+ @kubernetes_clusters.setter
1254
+ def kubernetes_clusters(self, kubernetes_clusters: 'bool'):
1255
+ """Sets the kubernetes_clusters of this V1UserFeatures.
1256
+
1257
+
1258
+ :param kubernetes_clusters: The kubernetes_clusters of this V1UserFeatures. # noqa: E501
1259
+ :type: bool
1260
+ """
1261
+
1262
+ self._kubernetes_clusters = kubernetes_clusters
1263
+
1104
1264
  @property
1105
1265
  def lambda_labs(self) -> 'bool':
1106
1266
  """Gets the lambda_labs of this V1UserFeatures. # noqa: E501
@@ -1185,6 +1345,27 @@ class V1UserFeatures(object):
1185
1345
 
1186
1346
  self._lit_logger = lit_logger
1187
1347
 
1348
+ @property
1349
+ def litcr_byoc_gcp(self) -> 'bool':
1350
+ """Gets the litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1351
+
1352
+
1353
+ :return: The litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1354
+ :rtype: bool
1355
+ """
1356
+ return self._litcr_byoc_gcp
1357
+
1358
+ @litcr_byoc_gcp.setter
1359
+ def litcr_byoc_gcp(self, litcr_byoc_gcp: 'bool'):
1360
+ """Sets the litcr_byoc_gcp of this V1UserFeatures.
1361
+
1362
+
1363
+ :param litcr_byoc_gcp: The litcr_byoc_gcp of this V1UserFeatures. # noqa: E501
1364
+ :type: bool
1365
+ """
1366
+
1367
+ self._litcr_byoc_gcp = litcr_byoc_gcp
1368
+
1188
1369
  @property
1189
1370
  def machine_selector_v2(self) -> 'bool':
1190
1371
  """Gets the machine_selector_v2 of this V1UserFeatures. # noqa: E501
@@ -1437,6 +1618,27 @@ class V1UserFeatures(object):
1437
1618
 
1438
1619
  self._nerf_fs_nonpaying = nerf_fs_nonpaying
1439
1620
 
1621
+ @property
1622
+ def onboarding_v2(self) -> 'bool':
1623
+ """Gets the onboarding_v2 of this V1UserFeatures. # noqa: E501
1624
+
1625
+
1626
+ :return: The onboarding_v2 of this V1UserFeatures. # noqa: E501
1627
+ :rtype: bool
1628
+ """
1629
+ return self._onboarding_v2
1630
+
1631
+ @onboarding_v2.setter
1632
+ def onboarding_v2(self, onboarding_v2: 'bool'):
1633
+ """Sets the onboarding_v2 of this V1UserFeatures.
1634
+
1635
+
1636
+ :param onboarding_v2: The onboarding_v2 of this V1UserFeatures. # noqa: E501
1637
+ :type: bool
1638
+ """
1639
+
1640
+ self._onboarding_v2 = onboarding_v2
1641
+
1440
1642
  @property
1441
1643
  def org_level_member_permissions(self) -> 'bool':
1442
1644
  """Gets the org_level_member_permissions of this V1UserFeatures. # noqa: E501
@@ -1501,25 +1703,25 @@ class V1UserFeatures(object):
1501
1703
  self._paygo_free_storage_limit_check = paygo_free_storage_limit_check
1502
1704
 
1503
1705
  @property
1504
- def pipelines(self) -> 'bool':
1505
- """Gets the pipelines of this V1UserFeatures. # noqa: E501
1706
+ def persistent_disk(self) -> 'bool':
1707
+ """Gets the persistent_disk of this V1UserFeatures. # noqa: E501
1506
1708
 
1507
1709
 
1508
- :return: The pipelines of this V1UserFeatures. # noqa: E501
1710
+ :return: The persistent_disk of this V1UserFeatures. # noqa: E501
1509
1711
  :rtype: bool
1510
1712
  """
1511
- return self._pipelines
1713
+ return self._persistent_disk
1512
1714
 
1513
- @pipelines.setter
1514
- def pipelines(self, pipelines: 'bool'):
1515
- """Sets the pipelines of this V1UserFeatures.
1715
+ @persistent_disk.setter
1716
+ def persistent_disk(self, persistent_disk: 'bool'):
1717
+ """Sets the persistent_disk of this V1UserFeatures.
1516
1718
 
1517
1719
 
1518
- :param pipelines: The pipelines of this V1UserFeatures. # noqa: E501
1720
+ :param persistent_disk: The persistent_disk of this V1UserFeatures. # noqa: E501
1519
1721
  :type: bool
1520
1722
  """
1521
1723
 
1522
- self._pipelines = pipelines
1724
+ self._persistent_disk = persistent_disk
1523
1725
 
1524
1726
  @property
1525
1727
  def plugin_distributed(self) -> 'bool':
@@ -1941,6 +2143,27 @@ class V1UserFeatures(object):
1941
2143
 
1942
2144
  self._slurm = slurm
1943
2145
 
2146
+ @property
2147
+ def specialised_studios(self) -> 'bool':
2148
+ """Gets the specialised_studios of this V1UserFeatures. # noqa: E501
2149
+
2150
+
2151
+ :return: The specialised_studios of this V1UserFeatures. # noqa: E501
2152
+ :rtype: bool
2153
+ """
2154
+ return self._specialised_studios
2155
+
2156
+ @specialised_studios.setter
2157
+ def specialised_studios(self, specialised_studios: 'bool'):
2158
+ """Sets the specialised_studios of this V1UserFeatures.
2159
+
2160
+
2161
+ :param specialised_studios: The specialised_studios of this V1UserFeatures. # noqa: E501
2162
+ :type: bool
2163
+ """
2164
+
2165
+ self._specialised_studios = specialised_studios
2166
+
1944
2167
  @property
1945
2168
  def storage_overuse_deletion(self) -> 'bool':
1946
2169
  """Gets the storage_overuse_deletion of this V1UserFeatures. # noqa: E501
@@ -1983,6 +2206,27 @@ class V1UserFeatures(object):
1983
2206
 
1984
2207
  self._studio_config = studio_config
1985
2208
 
2209
+ @property
2210
+ def studio_sharing_v2(self) -> 'bool':
2211
+ """Gets the studio_sharing_v2 of this V1UserFeatures. # noqa: E501
2212
+
2213
+
2214
+ :return: The studio_sharing_v2 of this V1UserFeatures. # noqa: E501
2215
+ :rtype: bool
2216
+ """
2217
+ return self._studio_sharing_v2
2218
+
2219
+ @studio_sharing_v2.setter
2220
+ def studio_sharing_v2(self, studio_sharing_v2: 'bool'):
2221
+ """Sets the studio_sharing_v2 of this V1UserFeatures.
2222
+
2223
+
2224
+ :param studio_sharing_v2: The studio_sharing_v2 of this V1UserFeatures. # noqa: E501
2225
+ :type: bool
2226
+ """
2227
+
2228
+ self._studio_sharing_v2 = studio_sharing_v2
2229
+
1986
2230
  @property
1987
2231
  def studio_version_visibility(self) -> 'bool':
1988
2232
  """Gets the studio_version_visibility of this V1UserFeatures. # noqa: E501
@@ -2025,6 +2269,27 @@ class V1UserFeatures(object):
2025
2269
 
2026
2270
  self._trainium2 = trainium2
2027
2271
 
2272
+ @property
2273
+ def use_internal_data_connection_mounts(self) -> 'bool':
2274
+ """Gets the use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2275
+
2276
+
2277
+ :return: The use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2278
+ :rtype: bool
2279
+ """
2280
+ return self._use_internal_data_connection_mounts
2281
+
2282
+ @use_internal_data_connection_mounts.setter
2283
+ def use_internal_data_connection_mounts(self, use_internal_data_connection_mounts: 'bool'):
2284
+ """Sets the use_internal_data_connection_mounts of this V1UserFeatures.
2285
+
2286
+
2287
+ :param use_internal_data_connection_mounts: The use_internal_data_connection_mounts of this V1UserFeatures. # noqa: E501
2288
+ :type: bool
2289
+ """
2290
+
2291
+ self._use_internal_data_connection_mounts = use_internal_data_connection_mounts
2292
+
2028
2293
  @property
2029
2294
  def use_rclone_mounts_only(self) -> 'bool':
2030
2295
  """Gets the use_rclone_mounts_only of this V1UserFeatures. # noqa: E501
@@ -2130,6 +2395,27 @@ class V1UserFeatures(object):
2130
2395
 
2131
2396
  self._weka = weka
2132
2397
 
2398
+ @property
2399
+ def writable_s3_connections(self) -> 'bool':
2400
+ """Gets the writable_s3_connections of this V1UserFeatures. # noqa: E501
2401
+
2402
+
2403
+ :return: The writable_s3_connections of this V1UserFeatures. # noqa: E501
2404
+ :rtype: bool
2405
+ """
2406
+ return self._writable_s3_connections
2407
+
2408
+ @writable_s3_connections.setter
2409
+ def writable_s3_connections(self, writable_s3_connections: 'bool'):
2410
+ """Sets the writable_s3_connections of this V1UserFeatures.
2411
+
2412
+
2413
+ :param writable_s3_connections: The writable_s3_connections of this V1UserFeatures. # noqa: E501
2414
+ :type: bool
2415
+ """
2416
+
2417
+ self._writable_s3_connections = writable_s3_connections
2418
+
2133
2419
  def to_dict(self) -> dict:
2134
2420
  """Returns the model properties as a dict"""
2135
2421
  result = {}