lightning-sdk 2025.7.10__py3-none-any.whl → 2025.7.22__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 (93) hide show
  1. lightning_sdk/__init__.py +3 -2
  2. lightning_sdk/api/cloud_account_api.py +154 -0
  3. lightning_sdk/api/deployment_api.py +11 -0
  4. lightning_sdk/api/job_api.py +9 -0
  5. lightning_sdk/api/llm_api.py +11 -6
  6. lightning_sdk/api/mmt_api.py +9 -0
  7. lightning_sdk/api/pipeline_api.py +4 -3
  8. lightning_sdk/api/studio_api.py +19 -5
  9. lightning_sdk/cli/clusters_menu.py +3 -3
  10. lightning_sdk/cli/create.py +22 -10
  11. lightning_sdk/cli/deploy/_auth.py +19 -3
  12. lightning_sdk/cli/deploy/serve.py +18 -4
  13. lightning_sdk/cli/entrypoint.py +1 -1
  14. lightning_sdk/cli/start.py +37 -7
  15. lightning_sdk/deployment/deployment.py +8 -0
  16. lightning_sdk/job/base.py +37 -5
  17. lightning_sdk/job/job.py +28 -4
  18. lightning_sdk/job/v1.py +10 -1
  19. lightning_sdk/job/v2.py +15 -1
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +15 -1
  21. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +335 -0
  22. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +214 -0
  23. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +5 -1
  24. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +11 -11
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +15 -1
  26. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py +53 -1
  28. lightning_sdk/lightning_cloud/openapi/models/{v1_list_new_features_for_user_response.py → conversations_id_body1.py} +23 -23
  29. lightning_sdk/lightning_cloud/openapi/models/messages_id_body.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py +1 -27
  32. lightning_sdk/lightning_cloud/openapi/models/protobuf_null_value.py +102 -0
  33. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +1 -27
  35. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py +3 -55
  36. lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py +175 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_ai_pod_v1.py +123 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_artifact.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_session_daily_aggregated.py +357 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py +53 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +1 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload.py +3 -55
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_upgrade_trigger_record_response.py +97 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py +53 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py +29 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_function_tool.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py +29 -3
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_assistant_session_daily_aggregated_response.py +201 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +79 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_artifact.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_like_status.py +104 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_list_notification_dialogs_response.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoint_models_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +53 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py +1 -53
  61. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_tool.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_like_response.py +149 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_message_like_response.py +149 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +105 -261
  68. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +27 -1
  69. lightning_sdk/llm/llm.py +32 -5
  70. lightning_sdk/llm/public_assistants.json +3 -1
  71. lightning_sdk/machine.py +24 -1
  72. lightning_sdk/mmt/base.py +20 -2
  73. lightning_sdk/mmt/mmt.py +25 -3
  74. lightning_sdk/mmt/v1.py +7 -1
  75. lightning_sdk/mmt/v2.py +21 -2
  76. lightning_sdk/organization.py +4 -0
  77. lightning_sdk/pipeline/pipeline.py +16 -5
  78. lightning_sdk/pipeline/printer.py +5 -3
  79. lightning_sdk/pipeline/schedule.py +844 -1
  80. lightning_sdk/pipeline/steps.py +19 -4
  81. lightning_sdk/sandbox.py +4 -1
  82. lightning_sdk/serve.py +2 -0
  83. lightning_sdk/studio.py +79 -39
  84. lightning_sdk/teamspace.py +14 -8
  85. lightning_sdk/utils/resolve.py +29 -2
  86. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/METADATA +1 -1
  87. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/RECORD +92 -78
  88. lightning_sdk/api/cluster_api.py +0 -119
  89. /lightning_sdk/cli/{inspect.py → inspection.py} +0 -0
  90. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/LICENSE +0 -0
  91. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/WHEEL +0 -0
  92. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/entry_points.txt +0 -0
  93. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/top_level.txt +0 -0
@@ -49,6 +49,7 @@ class V1UserFeatures(object):
49
49
  'auto_fast_load': 'bool',
50
50
  'auto_join_orgs': 'bool',
51
51
  'b2c_experience': 'bool',
52
+ 'byo_machine_type': 'bool',
52
53
  'cap_add': 'list[str]',
53
54
  'cap_drop': 'list[str]',
54
55
  'capacity_reservation_byoc': 'bool',
@@ -67,8 +68,10 @@ class V1UserFeatures(object):
67
68
  'deployment_persistent_disk': 'bool',
68
69
  'drive_v2': 'bool',
69
70
  'enterprise_compute_admin': 'bool',
71
+ 'f236': 'bool',
70
72
  'fair_share': 'bool',
71
73
  'featured_studios_admin': 'bool',
74
+ 'gcp_overprovisioning': 'bool',
72
75
  'gcs_connections_optimized': 'bool',
73
76
  'gcs_folders': 'bool',
74
77
  'gcs_fuse': 'bool',
@@ -76,27 +79,19 @@ class V1UserFeatures(object):
76
79
  'job_artifacts_v2': 'bool',
77
80
  'kubernetes_cluster_ui': 'bool',
78
81
  'kubernetes_clusters': 'bool',
79
- 'lambda_labs': 'bool',
80
- 'lambda_labs_studios': 'bool',
81
82
  'landing_studios': 'bool',
82
83
  'lit_logger': 'bool',
83
- 'machine_selector_v2': 'bool',
84
84
  'marketplace': 'bool',
85
85
  'mmt_fault_tolerance': 'bool',
86
86
  'mmt_strategy_selector': 'bool',
87
87
  'model_api_dashboard': 'bool',
88
88
  'model_api_dashboard_clickhouse': 'bool',
89
89
  'multicloud_folders': 'bool',
90
- 'multicloud_saas': 'bool',
91
90
  'multiple_studio_versions': 'bool',
92
- 'nebius': 'bool',
93
- 'nebius_cpu_studios': 'bool',
94
- 'nebius_gpu_studios': 'bool',
95
91
  'nerf_fs_nonpaying': 'bool',
96
92
  'onboarding_v2': 'bool',
97
93
  'org_level_member_permissions': 'bool',
98
94
  'org_usage_limits': 'bool',
99
- 'paygo_free_storage_limit_check': 'bool',
100
95
  'persistent_disk': 'bool',
101
96
  'plugin_distributed': 'bool',
102
97
  'plugin_inference': 'bool',
@@ -111,6 +106,7 @@ class V1UserFeatures(object):
111
106
  'project_selector': 'bool',
112
107
  'publish_pipelines': 'bool',
113
108
  'r2_data_connections': 'bool',
109
+ 'r2_uploads': 'bool',
114
110
  'reserved_machines_tab': 'bool',
115
111
  'restartable_jobs': 'bool',
116
112
  'runnable_public_studio_page': 'bool',
@@ -126,8 +122,6 @@ class V1UserFeatures(object):
126
122
  'trainium2': 'bool',
127
123
  'use_internal_data_connection_mounts': 'bool',
128
124
  'use_rclone_mounts_only': 'bool',
129
- 'voltage_park': 'bool',
130
- 'voltage_park_studios': 'bool',
131
125
  'vultr': 'bool',
132
126
  'weka': 'bool',
133
127
  'writable_s3_connections': 'bool'
@@ -142,6 +136,7 @@ class V1UserFeatures(object):
142
136
  'auto_fast_load': 'autoFastLoad',
143
137
  'auto_join_orgs': 'autoJoinOrgs',
144
138
  'b2c_experience': 'b2cExperience',
139
+ 'byo_machine_type': 'byoMachineType',
145
140
  'cap_add': 'capAdd',
146
141
  'cap_drop': 'capDrop',
147
142
  'capacity_reservation_byoc': 'capacityReservationByoc',
@@ -160,8 +155,10 @@ class V1UserFeatures(object):
160
155
  'deployment_persistent_disk': 'deploymentPersistentDisk',
161
156
  'drive_v2': 'driveV2',
162
157
  'enterprise_compute_admin': 'enterpriseComputeAdmin',
158
+ 'f236': 'f236',
163
159
  'fair_share': 'fairShare',
164
160
  'featured_studios_admin': 'featuredStudiosAdmin',
161
+ 'gcp_overprovisioning': 'gcpOverprovisioning',
165
162
  'gcs_connections_optimized': 'gcsConnectionsOptimized',
166
163
  'gcs_folders': 'gcsFolders',
167
164
  'gcs_fuse': 'gcsFuse',
@@ -169,27 +166,19 @@ class V1UserFeatures(object):
169
166
  'job_artifacts_v2': 'jobArtifactsV2',
170
167
  'kubernetes_cluster_ui': 'kubernetesClusterUi',
171
168
  'kubernetes_clusters': 'kubernetesClusters',
172
- 'lambda_labs': 'lambdaLabs',
173
- 'lambda_labs_studios': 'lambdaLabsStudios',
174
169
  'landing_studios': 'landingStudios',
175
170
  'lit_logger': 'litLogger',
176
- 'machine_selector_v2': 'machineSelectorV2',
177
171
  'marketplace': 'marketplace',
178
172
  'mmt_fault_tolerance': 'mmtFaultTolerance',
179
173
  'mmt_strategy_selector': 'mmtStrategySelector',
180
174
  'model_api_dashboard': 'modelApiDashboard',
181
175
  'model_api_dashboard_clickhouse': 'modelApiDashboardClickhouse',
182
176
  'multicloud_folders': 'multicloudFolders',
183
- 'multicloud_saas': 'multicloudSaas',
184
177
  'multiple_studio_versions': 'multipleStudioVersions',
185
- 'nebius': 'nebius',
186
- 'nebius_cpu_studios': 'nebiusCpuStudios',
187
- 'nebius_gpu_studios': 'nebiusGpuStudios',
188
178
  'nerf_fs_nonpaying': 'nerfFsNonpaying',
189
179
  'onboarding_v2': 'onboardingV2',
190
180
  'org_level_member_permissions': 'orgLevelMemberPermissions',
191
181
  'org_usage_limits': 'orgUsageLimits',
192
- 'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
193
182
  'persistent_disk': 'persistentDisk',
194
183
  'plugin_distributed': 'pluginDistributed',
195
184
  'plugin_inference': 'pluginInference',
@@ -204,6 +193,7 @@ class V1UserFeatures(object):
204
193
  'project_selector': 'projectSelector',
205
194
  'publish_pipelines': 'publishPipelines',
206
195
  'r2_data_connections': 'r2DataConnections',
196
+ 'r2_uploads': 'r2Uploads',
207
197
  'reserved_machines_tab': 'reservedMachinesTab',
208
198
  'restartable_jobs': 'restartableJobs',
209
199
  'runnable_public_studio_page': 'runnablePublicStudioPage',
@@ -219,14 +209,12 @@ class V1UserFeatures(object):
219
209
  'trainium2': 'trainium2',
220
210
  'use_internal_data_connection_mounts': 'useInternalDataConnectionMounts',
221
211
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
222
- 'voltage_park': 'voltagePark',
223
- 'voltage_park_studios': 'voltageParkStudios',
224
212
  'vultr': 'vultr',
225
213
  'weka': 'weka',
226
214
  'writable_s3_connections': 'writableS3Connections'
227
215
  }
228
216
 
229
- def __init__(self, academic_tier: '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, 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_cluster_ui: 'bool' =None, kubernetes_clusters: '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, model_api_dashboard_clickhouse: '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
217
+ def __init__(self, academic_tier: '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, byo_machine_type: '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, 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, f236: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: '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_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, model_api_dashboard_clickhouse: 'bool' =None, multicloud_folders: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: '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, r2_uploads: '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, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None): # noqa: E501
230
218
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
231
219
  self._academic_tier = None
232
220
  self._add_data_v2 = None
@@ -236,6 +224,7 @@ class V1UserFeatures(object):
236
224
  self._auto_fast_load = None
237
225
  self._auto_join_orgs = None
238
226
  self._b2c_experience = None
227
+ self._byo_machine_type = None
239
228
  self._cap_add = None
240
229
  self._cap_drop = None
241
230
  self._capacity_reservation_byoc = None
@@ -254,8 +243,10 @@ class V1UserFeatures(object):
254
243
  self._deployment_persistent_disk = None
255
244
  self._drive_v2 = None
256
245
  self._enterprise_compute_admin = None
246
+ self._f236 = None
257
247
  self._fair_share = None
258
248
  self._featured_studios_admin = None
249
+ self._gcp_overprovisioning = None
259
250
  self._gcs_connections_optimized = None
260
251
  self._gcs_folders = None
261
252
  self._gcs_fuse = None
@@ -263,27 +254,19 @@ class V1UserFeatures(object):
263
254
  self._job_artifacts_v2 = None
264
255
  self._kubernetes_cluster_ui = None
265
256
  self._kubernetes_clusters = None
266
- self._lambda_labs = None
267
- self._lambda_labs_studios = None
268
257
  self._landing_studios = None
269
258
  self._lit_logger = None
270
- self._machine_selector_v2 = None
271
259
  self._marketplace = None
272
260
  self._mmt_fault_tolerance = None
273
261
  self._mmt_strategy_selector = None
274
262
  self._model_api_dashboard = None
275
263
  self._model_api_dashboard_clickhouse = None
276
264
  self._multicloud_folders = None
277
- self._multicloud_saas = None
278
265
  self._multiple_studio_versions = None
279
- self._nebius = None
280
- self._nebius_cpu_studios = None
281
- self._nebius_gpu_studios = None
282
266
  self._nerf_fs_nonpaying = None
283
267
  self._onboarding_v2 = None
284
268
  self._org_level_member_permissions = None
285
269
  self._org_usage_limits = None
286
- self._paygo_free_storage_limit_check = None
287
270
  self._persistent_disk = None
288
271
  self._plugin_distributed = None
289
272
  self._plugin_inference = None
@@ -298,6 +281,7 @@ class V1UserFeatures(object):
298
281
  self._project_selector = None
299
282
  self._publish_pipelines = None
300
283
  self._r2_data_connections = None
284
+ self._r2_uploads = None
301
285
  self._reserved_machines_tab = None
302
286
  self._restartable_jobs = None
303
287
  self._runnable_public_studio_page = None
@@ -313,8 +297,6 @@ class V1UserFeatures(object):
313
297
  self._trainium2 = None
314
298
  self._use_internal_data_connection_mounts = None
315
299
  self._use_rclone_mounts_only = None
316
- self._voltage_park = None
317
- self._voltage_park_studios = None
318
300
  self._vultr = None
319
301
  self._weka = None
320
302
  self._writable_s3_connections = None
@@ -335,6 +317,8 @@ class V1UserFeatures(object):
335
317
  self.auto_join_orgs = auto_join_orgs
336
318
  if b2c_experience is not None:
337
319
  self.b2c_experience = b2c_experience
320
+ if byo_machine_type is not None:
321
+ self.byo_machine_type = byo_machine_type
338
322
  if cap_add is not None:
339
323
  self.cap_add = cap_add
340
324
  if cap_drop is not None:
@@ -371,10 +355,14 @@ class V1UserFeatures(object):
371
355
  self.drive_v2 = drive_v2
372
356
  if enterprise_compute_admin is not None:
373
357
  self.enterprise_compute_admin = enterprise_compute_admin
358
+ if f236 is not None:
359
+ self.f236 = f236
374
360
  if fair_share is not None:
375
361
  self.fair_share = fair_share
376
362
  if featured_studios_admin is not None:
377
363
  self.featured_studios_admin = featured_studios_admin
364
+ if gcp_overprovisioning is not None:
365
+ self.gcp_overprovisioning = gcp_overprovisioning
378
366
  if gcs_connections_optimized is not None:
379
367
  self.gcs_connections_optimized = gcs_connections_optimized
380
368
  if gcs_folders is not None:
@@ -389,16 +377,10 @@ class V1UserFeatures(object):
389
377
  self.kubernetes_cluster_ui = kubernetes_cluster_ui
390
378
  if kubernetes_clusters is not None:
391
379
  self.kubernetes_clusters = kubernetes_clusters
392
- if lambda_labs is not None:
393
- self.lambda_labs = lambda_labs
394
- if lambda_labs_studios is not None:
395
- self.lambda_labs_studios = lambda_labs_studios
396
380
  if landing_studios is not None:
397
381
  self.landing_studios = landing_studios
398
382
  if lit_logger is not None:
399
383
  self.lit_logger = lit_logger
400
- if machine_selector_v2 is not None:
401
- self.machine_selector_v2 = machine_selector_v2
402
384
  if marketplace is not None:
403
385
  self.marketplace = marketplace
404
386
  if mmt_fault_tolerance is not None:
@@ -411,16 +393,8 @@ class V1UserFeatures(object):
411
393
  self.model_api_dashboard_clickhouse = model_api_dashboard_clickhouse
412
394
  if multicloud_folders is not None:
413
395
  self.multicloud_folders = multicloud_folders
414
- if multicloud_saas is not None:
415
- self.multicloud_saas = multicloud_saas
416
396
  if multiple_studio_versions is not None:
417
397
  self.multiple_studio_versions = multiple_studio_versions
418
- if nebius is not None:
419
- self.nebius = nebius
420
- if nebius_cpu_studios is not None:
421
- self.nebius_cpu_studios = nebius_cpu_studios
422
- if nebius_gpu_studios is not None:
423
- self.nebius_gpu_studios = nebius_gpu_studios
424
398
  if nerf_fs_nonpaying is not None:
425
399
  self.nerf_fs_nonpaying = nerf_fs_nonpaying
426
400
  if onboarding_v2 is not None:
@@ -429,8 +403,6 @@ class V1UserFeatures(object):
429
403
  self.org_level_member_permissions = org_level_member_permissions
430
404
  if org_usage_limits is not None:
431
405
  self.org_usage_limits = org_usage_limits
432
- if paygo_free_storage_limit_check is not None:
433
- self.paygo_free_storage_limit_check = paygo_free_storage_limit_check
434
406
  if persistent_disk is not None:
435
407
  self.persistent_disk = persistent_disk
436
408
  if plugin_distributed is not None:
@@ -459,6 +431,8 @@ class V1UserFeatures(object):
459
431
  self.publish_pipelines = publish_pipelines
460
432
  if r2_data_connections is not None:
461
433
  self.r2_data_connections = r2_data_connections
434
+ if r2_uploads is not None:
435
+ self.r2_uploads = r2_uploads
462
436
  if reserved_machines_tab is not None:
463
437
  self.reserved_machines_tab = reserved_machines_tab
464
438
  if restartable_jobs is not None:
@@ -489,10 +463,6 @@ class V1UserFeatures(object):
489
463
  self.use_internal_data_connection_mounts = use_internal_data_connection_mounts
490
464
  if use_rclone_mounts_only is not None:
491
465
  self.use_rclone_mounts_only = use_rclone_mounts_only
492
- if voltage_park is not None:
493
- self.voltage_park = voltage_park
494
- if voltage_park_studios is not None:
495
- self.voltage_park_studios = voltage_park_studios
496
466
  if vultr is not None:
497
467
  self.vultr = vultr
498
468
  if weka is not None:
@@ -668,6 +638,27 @@ class V1UserFeatures(object):
668
638
 
669
639
  self._b2c_experience = b2c_experience
670
640
 
641
+ @property
642
+ def byo_machine_type(self) -> 'bool':
643
+ """Gets the byo_machine_type of this V1UserFeatures. # noqa: E501
644
+
645
+
646
+ :return: The byo_machine_type of this V1UserFeatures. # noqa: E501
647
+ :rtype: bool
648
+ """
649
+ return self._byo_machine_type
650
+
651
+ @byo_machine_type.setter
652
+ def byo_machine_type(self, byo_machine_type: 'bool'):
653
+ """Sets the byo_machine_type of this V1UserFeatures.
654
+
655
+
656
+ :param byo_machine_type: The byo_machine_type of this V1UserFeatures. # noqa: E501
657
+ :type: bool
658
+ """
659
+
660
+ self._byo_machine_type = byo_machine_type
661
+
671
662
  @property
672
663
  def cap_add(self) -> 'list[str]':
673
664
  """Gets the cap_add of this V1UserFeatures. # noqa: E501
@@ -1046,6 +1037,27 @@ class V1UserFeatures(object):
1046
1037
 
1047
1038
  self._enterprise_compute_admin = enterprise_compute_admin
1048
1039
 
1040
+ @property
1041
+ def f236(self) -> 'bool':
1042
+ """Gets the f236 of this V1UserFeatures. # noqa: E501
1043
+
1044
+
1045
+ :return: The f236 of this V1UserFeatures. # noqa: E501
1046
+ :rtype: bool
1047
+ """
1048
+ return self._f236
1049
+
1050
+ @f236.setter
1051
+ def f236(self, f236: 'bool'):
1052
+ """Sets the f236 of this V1UserFeatures.
1053
+
1054
+
1055
+ :param f236: The f236 of this V1UserFeatures. # noqa: E501
1056
+ :type: bool
1057
+ """
1058
+
1059
+ self._f236 = f236
1060
+
1049
1061
  @property
1050
1062
  def fair_share(self) -> 'bool':
1051
1063
  """Gets the fair_share of this V1UserFeatures. # noqa: E501
@@ -1088,6 +1100,27 @@ class V1UserFeatures(object):
1088
1100
 
1089
1101
  self._featured_studios_admin = featured_studios_admin
1090
1102
 
1103
+ @property
1104
+ def gcp_overprovisioning(self) -> 'bool':
1105
+ """Gets the gcp_overprovisioning of this V1UserFeatures. # noqa: E501
1106
+
1107
+
1108
+ :return: The gcp_overprovisioning of this V1UserFeatures. # noqa: E501
1109
+ :rtype: bool
1110
+ """
1111
+ return self._gcp_overprovisioning
1112
+
1113
+ @gcp_overprovisioning.setter
1114
+ def gcp_overprovisioning(self, gcp_overprovisioning: 'bool'):
1115
+ """Sets the gcp_overprovisioning of this V1UserFeatures.
1116
+
1117
+
1118
+ :param gcp_overprovisioning: The gcp_overprovisioning of this V1UserFeatures. # noqa: E501
1119
+ :type: bool
1120
+ """
1121
+
1122
+ self._gcp_overprovisioning = gcp_overprovisioning
1123
+
1091
1124
  @property
1092
1125
  def gcs_connections_optimized(self) -> 'bool':
1093
1126
  """Gets the gcs_connections_optimized of this V1UserFeatures. # noqa: E501
@@ -1235,48 +1268,6 @@ class V1UserFeatures(object):
1235
1268
 
1236
1269
  self._kubernetes_clusters = kubernetes_clusters
1237
1270
 
1238
- @property
1239
- def lambda_labs(self) -> 'bool':
1240
- """Gets the lambda_labs of this V1UserFeatures. # noqa: E501
1241
-
1242
-
1243
- :return: The lambda_labs of this V1UserFeatures. # noqa: E501
1244
- :rtype: bool
1245
- """
1246
- return self._lambda_labs
1247
-
1248
- @lambda_labs.setter
1249
- def lambda_labs(self, lambda_labs: 'bool'):
1250
- """Sets the lambda_labs of this V1UserFeatures.
1251
-
1252
-
1253
- :param lambda_labs: The lambda_labs of this V1UserFeatures. # noqa: E501
1254
- :type: bool
1255
- """
1256
-
1257
- self._lambda_labs = lambda_labs
1258
-
1259
- @property
1260
- def lambda_labs_studios(self) -> 'bool':
1261
- """Gets the lambda_labs_studios of this V1UserFeatures. # noqa: E501
1262
-
1263
-
1264
- :return: The lambda_labs_studios of this V1UserFeatures. # noqa: E501
1265
- :rtype: bool
1266
- """
1267
- return self._lambda_labs_studios
1268
-
1269
- @lambda_labs_studios.setter
1270
- def lambda_labs_studios(self, lambda_labs_studios: 'bool'):
1271
- """Sets the lambda_labs_studios of this V1UserFeatures.
1272
-
1273
-
1274
- :param lambda_labs_studios: The lambda_labs_studios of this V1UserFeatures. # noqa: E501
1275
- :type: bool
1276
- """
1277
-
1278
- self._lambda_labs_studios = lambda_labs_studios
1279
-
1280
1271
  @property
1281
1272
  def landing_studios(self) -> 'bool':
1282
1273
  """Gets the landing_studios of this V1UserFeatures. # noqa: E501
@@ -1319,27 +1310,6 @@ class V1UserFeatures(object):
1319
1310
 
1320
1311
  self._lit_logger = lit_logger
1321
1312
 
1322
- @property
1323
- def machine_selector_v2(self) -> 'bool':
1324
- """Gets the machine_selector_v2 of this V1UserFeatures. # noqa: E501
1325
-
1326
-
1327
- :return: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1328
- :rtype: bool
1329
- """
1330
- return self._machine_selector_v2
1331
-
1332
- @machine_selector_v2.setter
1333
- def machine_selector_v2(self, machine_selector_v2: 'bool'):
1334
- """Sets the machine_selector_v2 of this V1UserFeatures.
1335
-
1336
-
1337
- :param machine_selector_v2: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1338
- :type: bool
1339
- """
1340
-
1341
- self._machine_selector_v2 = machine_selector_v2
1342
-
1343
1313
  @property
1344
1314
  def marketplace(self) -> 'bool':
1345
1315
  """Gets the marketplace of this V1UserFeatures. # noqa: E501
@@ -1466,27 +1436,6 @@ class V1UserFeatures(object):
1466
1436
 
1467
1437
  self._multicloud_folders = multicloud_folders
1468
1438
 
1469
- @property
1470
- def multicloud_saas(self) -> 'bool':
1471
- """Gets the multicloud_saas of this V1UserFeatures. # noqa: E501
1472
-
1473
-
1474
- :return: The multicloud_saas of this V1UserFeatures. # noqa: E501
1475
- :rtype: bool
1476
- """
1477
- return self._multicloud_saas
1478
-
1479
- @multicloud_saas.setter
1480
- def multicloud_saas(self, multicloud_saas: 'bool'):
1481
- """Sets the multicloud_saas of this V1UserFeatures.
1482
-
1483
-
1484
- :param multicloud_saas: The multicloud_saas of this V1UserFeatures. # noqa: E501
1485
- :type: bool
1486
- """
1487
-
1488
- self._multicloud_saas = multicloud_saas
1489
-
1490
1439
  @property
1491
1440
  def multiple_studio_versions(self) -> 'bool':
1492
1441
  """Gets the multiple_studio_versions of this V1UserFeatures. # noqa: E501
@@ -1508,69 +1457,6 @@ class V1UserFeatures(object):
1508
1457
 
1509
1458
  self._multiple_studio_versions = multiple_studio_versions
1510
1459
 
1511
- @property
1512
- def nebius(self) -> 'bool':
1513
- """Gets the nebius of this V1UserFeatures. # noqa: E501
1514
-
1515
-
1516
- :return: The nebius of this V1UserFeatures. # noqa: E501
1517
- :rtype: bool
1518
- """
1519
- return self._nebius
1520
-
1521
- @nebius.setter
1522
- def nebius(self, nebius: 'bool'):
1523
- """Sets the nebius of this V1UserFeatures.
1524
-
1525
-
1526
- :param nebius: The nebius of this V1UserFeatures. # noqa: E501
1527
- :type: bool
1528
- """
1529
-
1530
- self._nebius = nebius
1531
-
1532
- @property
1533
- def nebius_cpu_studios(self) -> 'bool':
1534
- """Gets the nebius_cpu_studios of this V1UserFeatures. # noqa: E501
1535
-
1536
-
1537
- :return: The nebius_cpu_studios of this V1UserFeatures. # noqa: E501
1538
- :rtype: bool
1539
- """
1540
- return self._nebius_cpu_studios
1541
-
1542
- @nebius_cpu_studios.setter
1543
- def nebius_cpu_studios(self, nebius_cpu_studios: 'bool'):
1544
- """Sets the nebius_cpu_studios of this V1UserFeatures.
1545
-
1546
-
1547
- :param nebius_cpu_studios: The nebius_cpu_studios of this V1UserFeatures. # noqa: E501
1548
- :type: bool
1549
- """
1550
-
1551
- self._nebius_cpu_studios = nebius_cpu_studios
1552
-
1553
- @property
1554
- def nebius_gpu_studios(self) -> 'bool':
1555
- """Gets the nebius_gpu_studios of this V1UserFeatures. # noqa: E501
1556
-
1557
-
1558
- :return: The nebius_gpu_studios of this V1UserFeatures. # noqa: E501
1559
- :rtype: bool
1560
- """
1561
- return self._nebius_gpu_studios
1562
-
1563
- @nebius_gpu_studios.setter
1564
- def nebius_gpu_studios(self, nebius_gpu_studios: 'bool'):
1565
- """Sets the nebius_gpu_studios of this V1UserFeatures.
1566
-
1567
-
1568
- :param nebius_gpu_studios: The nebius_gpu_studios of this V1UserFeatures. # noqa: E501
1569
- :type: bool
1570
- """
1571
-
1572
- self._nebius_gpu_studios = nebius_gpu_studios
1573
-
1574
1460
  @property
1575
1461
  def nerf_fs_nonpaying(self) -> 'bool':
1576
1462
  """Gets the nerf_fs_nonpaying of this V1UserFeatures. # noqa: E501
@@ -1655,27 +1541,6 @@ class V1UserFeatures(object):
1655
1541
 
1656
1542
  self._org_usage_limits = org_usage_limits
1657
1543
 
1658
- @property
1659
- def paygo_free_storage_limit_check(self) -> 'bool':
1660
- """Gets the paygo_free_storage_limit_check of this V1UserFeatures. # noqa: E501
1661
-
1662
-
1663
- :return: The paygo_free_storage_limit_check of this V1UserFeatures. # noqa: E501
1664
- :rtype: bool
1665
- """
1666
- return self._paygo_free_storage_limit_check
1667
-
1668
- @paygo_free_storage_limit_check.setter
1669
- def paygo_free_storage_limit_check(self, paygo_free_storage_limit_check: 'bool'):
1670
- """Sets the paygo_free_storage_limit_check of this V1UserFeatures.
1671
-
1672
-
1673
- :param paygo_free_storage_limit_check: The paygo_free_storage_limit_check of this V1UserFeatures. # noqa: E501
1674
- :type: bool
1675
- """
1676
-
1677
- self._paygo_free_storage_limit_check = paygo_free_storage_limit_check
1678
-
1679
1544
  @property
1680
1545
  def persistent_disk(self) -> 'bool':
1681
1546
  """Gets the persistent_disk of this V1UserFeatures. # noqa: E501
@@ -1970,6 +1835,27 @@ class V1UserFeatures(object):
1970
1835
 
1971
1836
  self._r2_data_connections = r2_data_connections
1972
1837
 
1838
+ @property
1839
+ def r2_uploads(self) -> 'bool':
1840
+ """Gets the r2_uploads of this V1UserFeatures. # noqa: E501
1841
+
1842
+
1843
+ :return: The r2_uploads of this V1UserFeatures. # noqa: E501
1844
+ :rtype: bool
1845
+ """
1846
+ return self._r2_uploads
1847
+
1848
+ @r2_uploads.setter
1849
+ def r2_uploads(self, r2_uploads: 'bool'):
1850
+ """Sets the r2_uploads of this V1UserFeatures.
1851
+
1852
+
1853
+ :param r2_uploads: The r2_uploads of this V1UserFeatures. # noqa: E501
1854
+ :type: bool
1855
+ """
1856
+
1857
+ self._r2_uploads = r2_uploads
1858
+
1973
1859
  @property
1974
1860
  def reserved_machines_tab(self) -> 'bool':
1975
1861
  """Gets the reserved_machines_tab of this V1UserFeatures. # noqa: E501
@@ -2285,48 +2171,6 @@ class V1UserFeatures(object):
2285
2171
 
2286
2172
  self._use_rclone_mounts_only = use_rclone_mounts_only
2287
2173
 
2288
- @property
2289
- def voltage_park(self) -> 'bool':
2290
- """Gets the voltage_park of this V1UserFeatures. # noqa: E501
2291
-
2292
-
2293
- :return: The voltage_park of this V1UserFeatures. # noqa: E501
2294
- :rtype: bool
2295
- """
2296
- return self._voltage_park
2297
-
2298
- @voltage_park.setter
2299
- def voltage_park(self, voltage_park: 'bool'):
2300
- """Sets the voltage_park of this V1UserFeatures.
2301
-
2302
-
2303
- :param voltage_park: The voltage_park of this V1UserFeatures. # noqa: E501
2304
- :type: bool
2305
- """
2306
-
2307
- self._voltage_park = voltage_park
2308
-
2309
- @property
2310
- def voltage_park_studios(self) -> 'bool':
2311
- """Gets the voltage_park_studios of this V1UserFeatures. # noqa: E501
2312
-
2313
-
2314
- :return: The voltage_park_studios of this V1UserFeatures. # noqa: E501
2315
- :rtype: bool
2316
- """
2317
- return self._voltage_park_studios
2318
-
2319
- @voltage_park_studios.setter
2320
- def voltage_park_studios(self, voltage_park_studios: 'bool'):
2321
- """Sets the voltage_park_studios of this V1UserFeatures.
2322
-
2323
-
2324
- :param voltage_park_studios: The voltage_park_studios of this V1UserFeatures. # noqa: E501
2325
- :type: bool
2326
- """
2327
-
2328
- self._voltage_park_studios = voltage_park_studios
2329
-
2330
2174
  @property
2331
2175
  def vultr(self) -> 'bool':
2332
2176
  """Gets the vultr of this V1UserFeatures. # noqa: E501