lightning-sdk 0.2.22__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 (101) 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/license_api.py +2 -2
  5. lightning_sdk/api/llm_api.py +7 -11
  6. lightning_sdk/api/pipeline_api.py +31 -10
  7. lightning_sdk/api/studio_api.py +6 -0
  8. lightning_sdk/base_studio.py +22 -6
  9. lightning_sdk/cli/entrypoint.py +15 -13
  10. lightning_sdk/cli/start.py +5 -2
  11. lightning_sdk/deployment/deployment.py +17 -7
  12. lightning_sdk/lightning_cloud/openapi/__init__.py +20 -0
  13. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  14. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  15. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
  16. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  17. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +97 -0
  18. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  19. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
  20. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  21. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  22. lightning_sdk/lightning_cloud/openapi/models/__init__.py +19 -0
  23. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +81 -3
  25. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
  26. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  28. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  29. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  30. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  32. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  34. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +2 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +105 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  54. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  55. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  58. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +853 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -27
  63. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
  68. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
  69. lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  71. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  72. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  74. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
  77. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  80. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +234 -104
  83. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
  84. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
  85. lightning_sdk/llm/llm.py +113 -115
  86. lightning_sdk/llm/public_assistants.json +8 -0
  87. lightning_sdk/pipeline/__init__.py +11 -2
  88. lightning_sdk/pipeline/pipeline.py +54 -14
  89. lightning_sdk/pipeline/printer.py +36 -16
  90. lightning_sdk/pipeline/schedule.py +2 -1
  91. lightning_sdk/pipeline/{types.py → steps.py} +77 -56
  92. lightning_sdk/pipeline/utils.py +65 -3
  93. lightning_sdk/sandbox.py +157 -0
  94. lightning_sdk/services/license.py +12 -6
  95. lightning_sdk/studio.py +10 -1
  96. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  97. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +101 -79
  98. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  99. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  100. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  101. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -47,15 +47,12 @@ class V1UserFeatures(object):
47
47
  'ai_hub_monetization': 'bool',
48
48
  'auto_fast_load': 'bool',
49
49
  'auto_join_orgs': 'bool',
50
- 'auto_top_up': 'bool',
51
- 'auto_top_up_teamspace': 'bool',
52
50
  'b2c_experience': 'bool',
53
51
  'cap_add': 'list[str]',
54
52
  'cap_drop': 'list[str]',
55
53
  'capacity_reservation_byoc': 'bool',
56
54
  'capacity_reservation_dry_run': 'bool',
57
55
  'chat_models': 'bool',
58
- 'cloud_space_environment_templates': 'bool',
59
56
  'cloudy_vibe_code': 'bool',
60
57
  'code_tab': 'bool',
61
58
  'collab_screen_sharing': 'bool',
@@ -63,6 +60,7 @@ class V1UserFeatures(object):
63
60
  'control_center_monitoring': 'bool',
64
61
  'cost_attribution_settings': 'bool',
65
62
  'custom_app_domain': 'bool',
63
+ 'data_connection_flushing_v2': 'bool',
66
64
  'datasets': 'bool',
67
65
  'default_one_cluster': 'bool',
68
66
  'deployment_persistent_disk': 'bool',
@@ -70,13 +68,17 @@ class V1UserFeatures(object):
70
68
  'enterprise_compute_admin': 'bool',
71
69
  'fair_share': 'bool',
72
70
  'featured_studios_admin': 'bool',
71
+ 'gcs_connections_optimized': 'bool',
73
72
  'gcs_folders': 'bool',
74
73
  'instant_capacity_reservation': 'bool',
75
74
  'job_artifacts_v2': 'bool',
75
+ 'kubernetes_clusters': 'bool',
76
76
  'lambda_labs': 'bool',
77
77
  'lambda_labs_studios': 'bool',
78
78
  'landing_studios': 'bool',
79
79
  'lit_logger': 'bool',
80
+ 'litcr_byoc_gcp': 'bool',
81
+ 'machine_selector_v2': 'bool',
80
82
  'marketplace': 'bool',
81
83
  'mmt_fault_tolerance': 'bool',
82
84
  'mmt_strategy_selector': 'bool',
@@ -91,8 +93,8 @@ 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
- 'plg_control_center': 'bool',
96
98
  'plugin_distributed': 'bool',
97
99
  'plugin_inference': 'bool',
98
100
  'plugin_label_studio': 'bool',
@@ -113,15 +115,18 @@ class V1UserFeatures(object):
113
115
  'show_dev_admin': 'bool',
114
116
  'single_wallet': 'bool',
115
117
  'slurm': 'bool',
118
+ 'specialised_studios': 'bool',
116
119
  'storage_overuse_deletion': 'bool',
117
120
  'studio_config': 'bool',
118
121
  'studio_version_visibility': 'bool',
119
122
  'trainium2': 'bool',
123
+ 'use_internal_data_connection_mounts': 'bool',
120
124
  'use_rclone_mounts_only': 'bool',
121
125
  'voltage_park': 'bool',
122
126
  'voltage_park_studios': 'bool',
123
127
  'vultr': 'bool',
124
- 'weka': 'bool'
128
+ 'weka': 'bool',
129
+ 'writable_s3_connections': 'bool'
125
130
  }
126
131
 
127
132
  attribute_map = {
@@ -131,15 +136,12 @@ class V1UserFeatures(object):
131
136
  'ai_hub_monetization': 'aiHubMonetization',
132
137
  'auto_fast_load': 'autoFastLoad',
133
138
  'auto_join_orgs': 'autoJoinOrgs',
134
- 'auto_top_up': 'autoTopUp',
135
- 'auto_top_up_teamspace': 'autoTopUpTeamspace',
136
139
  'b2c_experience': 'b2cExperience',
137
140
  'cap_add': 'capAdd',
138
141
  'cap_drop': 'capDrop',
139
142
  'capacity_reservation_byoc': 'capacityReservationByoc',
140
143
  'capacity_reservation_dry_run': 'capacityReservationDryRun',
141
144
  'chat_models': 'chatModels',
142
- 'cloud_space_environment_templates': 'cloudSpaceEnvironmentTemplates',
143
145
  'cloudy_vibe_code': 'cloudyVibeCode',
144
146
  'code_tab': 'codeTab',
145
147
  'collab_screen_sharing': 'collabScreenSharing',
@@ -147,6 +149,7 @@ class V1UserFeatures(object):
147
149
  'control_center_monitoring': 'controlCenterMonitoring',
148
150
  'cost_attribution_settings': 'costAttributionSettings',
149
151
  'custom_app_domain': 'customAppDomain',
152
+ 'data_connection_flushing_v2': 'dataConnectionFlushingV2',
150
153
  'datasets': 'datasets',
151
154
  'default_one_cluster': 'defaultOneCluster',
152
155
  'deployment_persistent_disk': 'deploymentPersistentDisk',
@@ -154,13 +157,17 @@ class V1UserFeatures(object):
154
157
  'enterprise_compute_admin': 'enterpriseComputeAdmin',
155
158
  'fair_share': 'fairShare',
156
159
  'featured_studios_admin': 'featuredStudiosAdmin',
160
+ 'gcs_connections_optimized': 'gcsConnectionsOptimized',
157
161
  'gcs_folders': 'gcsFolders',
158
162
  'instant_capacity_reservation': 'instantCapacityReservation',
159
163
  'job_artifacts_v2': 'jobArtifactsV2',
164
+ 'kubernetes_clusters': 'kubernetesClusters',
160
165
  'lambda_labs': 'lambdaLabs',
161
166
  'lambda_labs_studios': 'lambdaLabsStudios',
162
167
  'landing_studios': 'landingStudios',
163
168
  'lit_logger': 'litLogger',
169
+ 'litcr_byoc_gcp': 'litcrByocGcp',
170
+ 'machine_selector_v2': 'machineSelectorV2',
164
171
  'marketplace': 'marketplace',
165
172
  'mmt_fault_tolerance': 'mmtFaultTolerance',
166
173
  'mmt_strategy_selector': 'mmtStrategySelector',
@@ -175,8 +182,8 @@ class V1UserFeatures(object):
175
182
  'org_level_member_permissions': 'orgLevelMemberPermissions',
176
183
  'org_usage_limits': 'orgUsageLimits',
177
184
  'paygo_free_storage_limit_check': 'paygoFreeStorageLimitCheck',
185
+ 'persistent_disk': 'persistentDisk',
178
186
  'pipelines': 'pipelines',
179
- 'plg_control_center': 'plgControlCenter',
180
187
  'plugin_distributed': 'pluginDistributed',
181
188
  'plugin_inference': 'pluginInference',
182
189
  'plugin_label_studio': 'pluginLabelStudio',
@@ -197,18 +204,21 @@ class V1UserFeatures(object):
197
204
  'show_dev_admin': 'showDevAdmin',
198
205
  'single_wallet': 'singleWallet',
199
206
  'slurm': 'slurm',
207
+ 'specialised_studios': 'specialisedStudios',
200
208
  'storage_overuse_deletion': 'storageOveruseDeletion',
201
209
  'studio_config': 'studioConfig',
202
210
  'studio_version_visibility': 'studioVersionVisibility',
203
211
  'trainium2': 'trainium2',
212
+ 'use_internal_data_connection_mounts': 'useInternalDataConnectionMounts',
204
213
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
205
214
  'voltage_park': 'voltagePark',
206
215
  'voltage_park_studios': 'voltageParkStudios',
207
216
  'vultr': 'vultr',
208
- 'weka': 'weka'
217
+ 'weka': 'weka',
218
+ 'writable_s3_connections': 'writableS3Connections'
209
219
  }
210
220
 
211
- 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, auto_top_up: 'bool' =None, auto_top_up_teamspace: '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, 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, 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, plg_control_center: '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
212
222
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
213
223
  self._accurate_billing = None
214
224
  self._affiliate_links = None
@@ -216,15 +226,12 @@ class V1UserFeatures(object):
216
226
  self._ai_hub_monetization = None
217
227
  self._auto_fast_load = None
218
228
  self._auto_join_orgs = None
219
- self._auto_top_up = None
220
- self._auto_top_up_teamspace = None
221
229
  self._b2c_experience = None
222
230
  self._cap_add = None
223
231
  self._cap_drop = None
224
232
  self._capacity_reservation_byoc = None
225
233
  self._capacity_reservation_dry_run = None
226
234
  self._chat_models = None
227
- self._cloud_space_environment_templates = None
228
235
  self._cloudy_vibe_code = None
229
236
  self._code_tab = None
230
237
  self._collab_screen_sharing = None
@@ -232,6 +239,7 @@ class V1UserFeatures(object):
232
239
  self._control_center_monitoring = None
233
240
  self._cost_attribution_settings = None
234
241
  self._custom_app_domain = None
242
+ self._data_connection_flushing_v2 = None
235
243
  self._datasets = None
236
244
  self._default_one_cluster = None
237
245
  self._deployment_persistent_disk = None
@@ -239,13 +247,17 @@ class V1UserFeatures(object):
239
247
  self._enterprise_compute_admin = None
240
248
  self._fair_share = None
241
249
  self._featured_studios_admin = None
250
+ self._gcs_connections_optimized = None
242
251
  self._gcs_folders = None
243
252
  self._instant_capacity_reservation = None
244
253
  self._job_artifacts_v2 = None
254
+ self._kubernetes_clusters = None
245
255
  self._lambda_labs = None
246
256
  self._lambda_labs_studios = None
247
257
  self._landing_studios = None
248
258
  self._lit_logger = None
259
+ self._litcr_byoc_gcp = None
260
+ self._machine_selector_v2 = None
249
261
  self._marketplace = None
250
262
  self._mmt_fault_tolerance = None
251
263
  self._mmt_strategy_selector = None
@@ -260,8 +272,8 @@ class V1UserFeatures(object):
260
272
  self._org_level_member_permissions = None
261
273
  self._org_usage_limits = None
262
274
  self._paygo_free_storage_limit_check = None
275
+ self._persistent_disk = None
263
276
  self._pipelines = None
264
- self._plg_control_center = None
265
277
  self._plugin_distributed = None
266
278
  self._plugin_inference = None
267
279
  self._plugin_label_studio = None
@@ -282,15 +294,18 @@ class V1UserFeatures(object):
282
294
  self._show_dev_admin = None
283
295
  self._single_wallet = None
284
296
  self._slurm = None
297
+ self._specialised_studios = None
285
298
  self._storage_overuse_deletion = None
286
299
  self._studio_config = None
287
300
  self._studio_version_visibility = None
288
301
  self._trainium2 = None
302
+ self._use_internal_data_connection_mounts = None
289
303
  self._use_rclone_mounts_only = None
290
304
  self._voltage_park = None
291
305
  self._voltage_park_studios = None
292
306
  self._vultr = None
293
307
  self._weka = None
308
+ self._writable_s3_connections = None
294
309
  self.discriminator = None
295
310
  if accurate_billing is not None:
296
311
  self.accurate_billing = accurate_billing
@@ -304,10 +319,6 @@ class V1UserFeatures(object):
304
319
  self.auto_fast_load = auto_fast_load
305
320
  if auto_join_orgs is not None:
306
321
  self.auto_join_orgs = auto_join_orgs
307
- if auto_top_up is not None:
308
- self.auto_top_up = auto_top_up
309
- if auto_top_up_teamspace is not None:
310
- self.auto_top_up_teamspace = auto_top_up_teamspace
311
322
  if b2c_experience is not None:
312
323
  self.b2c_experience = b2c_experience
313
324
  if cap_add is not None:
@@ -320,8 +331,6 @@ class V1UserFeatures(object):
320
331
  self.capacity_reservation_dry_run = capacity_reservation_dry_run
321
332
  if chat_models is not None:
322
333
  self.chat_models = chat_models
323
- if cloud_space_environment_templates is not None:
324
- self.cloud_space_environment_templates = cloud_space_environment_templates
325
334
  if cloudy_vibe_code is not None:
326
335
  self.cloudy_vibe_code = cloudy_vibe_code
327
336
  if code_tab is not None:
@@ -336,6 +345,8 @@ class V1UserFeatures(object):
336
345
  self.cost_attribution_settings = cost_attribution_settings
337
346
  if custom_app_domain is not None:
338
347
  self.custom_app_domain = custom_app_domain
348
+ if data_connection_flushing_v2 is not None:
349
+ self.data_connection_flushing_v2 = data_connection_flushing_v2
339
350
  if datasets is not None:
340
351
  self.datasets = datasets
341
352
  if default_one_cluster is not None:
@@ -350,12 +361,16 @@ class V1UserFeatures(object):
350
361
  self.fair_share = fair_share
351
362
  if featured_studios_admin is not None:
352
363
  self.featured_studios_admin = featured_studios_admin
364
+ if gcs_connections_optimized is not None:
365
+ self.gcs_connections_optimized = gcs_connections_optimized
353
366
  if gcs_folders is not None:
354
367
  self.gcs_folders = gcs_folders
355
368
  if instant_capacity_reservation is not None:
356
369
  self.instant_capacity_reservation = instant_capacity_reservation
357
370
  if job_artifacts_v2 is not None:
358
371
  self.job_artifacts_v2 = job_artifacts_v2
372
+ if kubernetes_clusters is not None:
373
+ self.kubernetes_clusters = kubernetes_clusters
359
374
  if lambda_labs is not None:
360
375
  self.lambda_labs = lambda_labs
361
376
  if lambda_labs_studios is not None:
@@ -364,6 +379,10 @@ class V1UserFeatures(object):
364
379
  self.landing_studios = landing_studios
365
380
  if lit_logger is not None:
366
381
  self.lit_logger = lit_logger
382
+ if litcr_byoc_gcp is not None:
383
+ self.litcr_byoc_gcp = litcr_byoc_gcp
384
+ if machine_selector_v2 is not None:
385
+ self.machine_selector_v2 = machine_selector_v2
367
386
  if marketplace is not None:
368
387
  self.marketplace = marketplace
369
388
  if mmt_fault_tolerance is not None:
@@ -392,10 +411,10 @@ class V1UserFeatures(object):
392
411
  self.org_usage_limits = org_usage_limits
393
412
  if paygo_free_storage_limit_check is not None:
394
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
395
416
  if pipelines is not None:
396
417
  self.pipelines = pipelines
397
- if plg_control_center is not None:
398
- self.plg_control_center = plg_control_center
399
418
  if plugin_distributed is not None:
400
419
  self.plugin_distributed = plugin_distributed
401
420
  if plugin_inference is not None:
@@ -436,6 +455,8 @@ class V1UserFeatures(object):
436
455
  self.single_wallet = single_wallet
437
456
  if slurm is not None:
438
457
  self.slurm = slurm
458
+ if specialised_studios is not None:
459
+ self.specialised_studios = specialised_studios
439
460
  if storage_overuse_deletion is not None:
440
461
  self.storage_overuse_deletion = storage_overuse_deletion
441
462
  if studio_config is not None:
@@ -444,6 +465,8 @@ class V1UserFeatures(object):
444
465
  self.studio_version_visibility = studio_version_visibility
445
466
  if trainium2 is not None:
446
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
447
470
  if use_rclone_mounts_only is not None:
448
471
  self.use_rclone_mounts_only = use_rclone_mounts_only
449
472
  if voltage_park is not None:
@@ -454,6 +477,8 @@ class V1UserFeatures(object):
454
477
  self.vultr = vultr
455
478
  if weka is not None:
456
479
  self.weka = weka
480
+ if writable_s3_connections is not None:
481
+ self.writable_s3_connections = writable_s3_connections
457
482
 
458
483
  @property
459
484
  def accurate_billing(self) -> 'bool':
@@ -581,48 +606,6 @@ class V1UserFeatures(object):
581
606
 
582
607
  self._auto_join_orgs = auto_join_orgs
583
608
 
584
- @property
585
- def auto_top_up(self) -> 'bool':
586
- """Gets the auto_top_up of this V1UserFeatures. # noqa: E501
587
-
588
-
589
- :return: The auto_top_up of this V1UserFeatures. # noqa: E501
590
- :rtype: bool
591
- """
592
- return self._auto_top_up
593
-
594
- @auto_top_up.setter
595
- def auto_top_up(self, auto_top_up: 'bool'):
596
- """Sets the auto_top_up of this V1UserFeatures.
597
-
598
-
599
- :param auto_top_up: The auto_top_up of this V1UserFeatures. # noqa: E501
600
- :type: bool
601
- """
602
-
603
- self._auto_top_up = auto_top_up
604
-
605
- @property
606
- def auto_top_up_teamspace(self) -> 'bool':
607
- """Gets the auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
608
-
609
-
610
- :return: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
611
- :rtype: bool
612
- """
613
- return self._auto_top_up_teamspace
614
-
615
- @auto_top_up_teamspace.setter
616
- def auto_top_up_teamspace(self, auto_top_up_teamspace: 'bool'):
617
- """Sets the auto_top_up_teamspace of this V1UserFeatures.
618
-
619
-
620
- :param auto_top_up_teamspace: The auto_top_up_teamspace of this V1UserFeatures. # noqa: E501
621
- :type: bool
622
- """
623
-
624
- self._auto_top_up_teamspace = auto_top_up_teamspace
625
-
626
609
  @property
627
610
  def b2c_experience(self) -> 'bool':
628
611
  """Gets the b2c_experience of this V1UserFeatures. # noqa: E501
@@ -749,27 +732,6 @@ class V1UserFeatures(object):
749
732
 
750
733
  self._chat_models = chat_models
751
734
 
752
- @property
753
- def cloud_space_environment_templates(self) -> 'bool':
754
- """Gets the cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
755
-
756
-
757
- :return: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
758
- :rtype: bool
759
- """
760
- return self._cloud_space_environment_templates
761
-
762
- @cloud_space_environment_templates.setter
763
- def cloud_space_environment_templates(self, cloud_space_environment_templates: 'bool'):
764
- """Sets the cloud_space_environment_templates of this V1UserFeatures.
765
-
766
-
767
- :param cloud_space_environment_templates: The cloud_space_environment_templates of this V1UserFeatures. # noqa: E501
768
- :type: bool
769
- """
770
-
771
- self._cloud_space_environment_templates = cloud_space_environment_templates
772
-
773
735
  @property
774
736
  def cloudy_vibe_code(self) -> 'bool':
775
737
  """Gets the cloudy_vibe_code of this V1UserFeatures. # noqa: E501
@@ -917,6 +879,27 @@ class V1UserFeatures(object):
917
879
 
918
880
  self._custom_app_domain = custom_app_domain
919
881
 
882
+ @property
883
+ def data_connection_flushing_v2(self) -> 'bool':
884
+ """Gets the data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
885
+
886
+
887
+ :return: The data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
888
+ :rtype: bool
889
+ """
890
+ return self._data_connection_flushing_v2
891
+
892
+ @data_connection_flushing_v2.setter
893
+ def data_connection_flushing_v2(self, data_connection_flushing_v2: 'bool'):
894
+ """Sets the data_connection_flushing_v2 of this V1UserFeatures.
895
+
896
+
897
+ :param data_connection_flushing_v2: The data_connection_flushing_v2 of this V1UserFeatures. # noqa: E501
898
+ :type: bool
899
+ """
900
+
901
+ self._data_connection_flushing_v2 = data_connection_flushing_v2
902
+
920
903
  @property
921
904
  def datasets(self) -> 'bool':
922
905
  """Gets the datasets of this V1UserFeatures. # noqa: E501
@@ -1064,6 +1047,27 @@ class V1UserFeatures(object):
1064
1047
 
1065
1048
  self._featured_studios_admin = featured_studios_admin
1066
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
+
1067
1071
  @property
1068
1072
  def gcs_folders(self) -> 'bool':
1069
1073
  """Gets the gcs_folders of this V1UserFeatures. # noqa: E501
@@ -1127,6 +1131,27 @@ class V1UserFeatures(object):
1127
1131
 
1128
1132
  self._job_artifacts_v2 = job_artifacts_v2
1129
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
+
1130
1155
  @property
1131
1156
  def lambda_labs(self) -> 'bool':
1132
1157
  """Gets the lambda_labs of this V1UserFeatures. # noqa: E501
@@ -1211,6 +1236,48 @@ class V1UserFeatures(object):
1211
1236
 
1212
1237
  self._lit_logger = lit_logger
1213
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
+
1260
+ @property
1261
+ def machine_selector_v2(self) -> 'bool':
1262
+ """Gets the machine_selector_v2 of this V1UserFeatures. # noqa: E501
1263
+
1264
+
1265
+ :return: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1266
+ :rtype: bool
1267
+ """
1268
+ return self._machine_selector_v2
1269
+
1270
+ @machine_selector_v2.setter
1271
+ def machine_selector_v2(self, machine_selector_v2: 'bool'):
1272
+ """Sets the machine_selector_v2 of this V1UserFeatures.
1273
+
1274
+
1275
+ :param machine_selector_v2: The machine_selector_v2 of this V1UserFeatures. # noqa: E501
1276
+ :type: bool
1277
+ """
1278
+
1279
+ self._machine_selector_v2 = machine_selector_v2
1280
+
1214
1281
  @property
1215
1282
  def marketplace(self) -> 'bool':
1216
1283
  """Gets the marketplace of this V1UserFeatures. # noqa: E501
@@ -1506,46 +1573,46 @@ class V1UserFeatures(object):
1506
1573
  self._paygo_free_storage_limit_check = paygo_free_storage_limit_check
1507
1574
 
1508
1575
  @property
1509
- def pipelines(self) -> 'bool':
1510
- """Gets the pipelines of this V1UserFeatures. # noqa: E501
1576
+ def persistent_disk(self) -> 'bool':
1577
+ """Gets the persistent_disk of this V1UserFeatures. # noqa: E501
1511
1578
 
1512
1579
 
1513
- :return: The pipelines of this V1UserFeatures. # noqa: E501
1580
+ :return: The persistent_disk of this V1UserFeatures. # noqa: E501
1514
1581
  :rtype: bool
1515
1582
  """
1516
- return self._pipelines
1583
+ return self._persistent_disk
1517
1584
 
1518
- @pipelines.setter
1519
- def pipelines(self, pipelines: 'bool'):
1520
- """Sets the pipelines of this V1UserFeatures.
1585
+ @persistent_disk.setter
1586
+ def persistent_disk(self, persistent_disk: 'bool'):
1587
+ """Sets the persistent_disk of this V1UserFeatures.
1521
1588
 
1522
1589
 
1523
- :param pipelines: The pipelines of this V1UserFeatures. # noqa: E501
1590
+ :param persistent_disk: The persistent_disk of this V1UserFeatures. # noqa: E501
1524
1591
  :type: bool
1525
1592
  """
1526
1593
 
1527
- self._pipelines = pipelines
1594
+ self._persistent_disk = persistent_disk
1528
1595
 
1529
1596
  @property
1530
- def plg_control_center(self) -> 'bool':
1531
- """Gets the plg_control_center of this V1UserFeatures. # noqa: E501
1597
+ def pipelines(self) -> 'bool':
1598
+ """Gets the pipelines of this V1UserFeatures. # noqa: E501
1532
1599
 
1533
1600
 
1534
- :return: The plg_control_center of this V1UserFeatures. # noqa: E501
1601
+ :return: The pipelines of this V1UserFeatures. # noqa: E501
1535
1602
  :rtype: bool
1536
1603
  """
1537
- return self._plg_control_center
1604
+ return self._pipelines
1538
1605
 
1539
- @plg_control_center.setter
1540
- def plg_control_center(self, plg_control_center: 'bool'):
1541
- """Sets the plg_control_center of this V1UserFeatures.
1606
+ @pipelines.setter
1607
+ def pipelines(self, pipelines: 'bool'):
1608
+ """Sets the pipelines of this V1UserFeatures.
1542
1609
 
1543
1610
 
1544
- :param plg_control_center: The plg_control_center of this V1UserFeatures. # noqa: E501
1611
+ :param pipelines: The pipelines of this V1UserFeatures. # noqa: E501
1545
1612
  :type: bool
1546
1613
  """
1547
1614
 
1548
- self._plg_control_center = plg_control_center
1615
+ self._pipelines = pipelines
1549
1616
 
1550
1617
  @property
1551
1618
  def plugin_distributed(self) -> 'bool':
@@ -1967,6 +2034,27 @@ class V1UserFeatures(object):
1967
2034
 
1968
2035
  self._slurm = slurm
1969
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
+
1970
2058
  @property
1971
2059
  def storage_overuse_deletion(self) -> 'bool':
1972
2060
  """Gets the storage_overuse_deletion of this V1UserFeatures. # noqa: E501
@@ -2051,6 +2139,27 @@ class V1UserFeatures(object):
2051
2139
 
2052
2140
  self._trainium2 = trainium2
2053
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
+
2054
2163
  @property
2055
2164
  def use_rclone_mounts_only(self) -> 'bool':
2056
2165
  """Gets the use_rclone_mounts_only of this V1UserFeatures. # noqa: E501
@@ -2156,6 +2265,27 @@ class V1UserFeatures(object):
2156
2265
 
2157
2266
  self._weka = weka
2158
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
+
2159
2289
  def to_dict(self) -> dict:
2160
2290
  """Returns the model properties as a dict"""
2161
2291
  result = {}