lightning-sdk 0.2.11__py3-none-any.whl → 0.2.13__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 (61) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +35 -3
  3. lightning_sdk/api/lit_container_api.py +13 -7
  4. lightning_sdk/api/llm_api.py +46 -0
  5. lightning_sdk/api/studio_api.py +17 -0
  6. lightning_sdk/cli/entrypoint.py +1 -1
  7. lightning_sdk/cli/serve.py +221 -62
  8. lightning_sdk/deployment/deployment.py +53 -7
  9. lightning_sdk/lightning_cloud/openapi/__init__.py +11 -1
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  11. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +4 -4
  12. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +13 -1
  13. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +4 -4
  14. lightning_sdk/lightning_cloud/openapi/api/git_credentials_service_api.py +497 -0
  15. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +124 -0
  16. lightning_sdk/lightning_cloud/openapi/models/__init__.py +10 -1
  17. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  20. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +27 -1
  22. lightning_sdk/lightning_cloud/openapi/models/update.py +65 -195
  23. lightning_sdk/lightning_cloud/openapi/models/update1.py +357 -0
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  25. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +79 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_create_git_credentials_request.py +175 -0
  31. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cloud_space_environment_template_response.py +1 -53
  32. lightning_sdk/lightning_cloud/openapi/models/v1_delete_git_credentials_response.py +97 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py +2 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +1 -27
  36. lightning_sdk/lightning_cloud/openapi/models/v1_git_credentials.py +227 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_job_resource.py +279 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_job_type.py +108 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_list_git_credentials_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_list_job_resources_response.py +123 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_nebius_direct_v1.py +149 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +55 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +29 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_reservation_billing_session.py +279 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_resources.py +55 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +1 -27
  47. lightning_sdk/lightning_cloud/openapi/models/v1_usage.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -1
  49. lightning_sdk/llm/__init__.py +3 -0
  50. lightning_sdk/llm/llm.py +118 -0
  51. lightning_sdk/plugin.py +19 -0
  52. lightning_sdk/serve.py +4 -6
  53. lightning_sdk/studio.py +33 -0
  54. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/METADATA +1 -1
  55. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/RECORD +60 -47
  56. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +0 -253
  57. /lightning_sdk/cli/{docker.py → docker_cli.py} +0 -0
  58. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/LICENSE +0 -0
  59. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/WHEEL +0 -0
  60. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/entry_points.txt +0 -0
  61. {lightning_sdk-0.2.11.dist-info → lightning_sdk-0.2.13.dist-info}/top_level.txt +0 -0
@@ -46,8 +46,10 @@ class V1Resources(object):
46
46
  'extra': 'dict(str, str)',
47
47
  'gpu': 'int',
48
48
  'gpu_type': 'str',
49
+ 'iops': 'str',
49
50
  'memory_mb': 'str',
50
- 'storage_gb': 'str'
51
+ 'storage_gb': 'str',
52
+ 'throughput': 'str'
51
53
  }
52
54
 
53
55
  attribute_map = {
@@ -56,19 +58,23 @@ class V1Resources(object):
56
58
  'extra': 'extra',
57
59
  'gpu': 'gpu',
58
60
  'gpu_type': 'gpuType',
61
+ 'iops': 'iops',
59
62
  'memory_mb': 'memoryMb',
60
- 'storage_gb': 'storageGb'
63
+ 'storage_gb': 'storageGb',
64
+ 'throughput': 'throughput'
61
65
  }
62
66
 
63
- def __init__(self, cpu: 'int' =None, cpus: 'str' =None, extra: 'dict(str, str)' =None, gpu: 'int' =None, gpu_type: 'str' =None, memory_mb: 'str' =None, storage_gb: 'str' =None): # noqa: E501
67
+ def __init__(self, cpu: 'int' =None, cpus: 'str' =None, extra: 'dict(str, str)' =None, gpu: 'int' =None, gpu_type: 'str' =None, iops: 'str' =None, memory_mb: 'str' =None, storage_gb: 'str' =None, throughput: 'str' =None): # noqa: E501
64
68
  """V1Resources - a model defined in Swagger""" # noqa: E501
65
69
  self._cpu = None
66
70
  self._cpus = None
67
71
  self._extra = None
68
72
  self._gpu = None
69
73
  self._gpu_type = None
74
+ self._iops = None
70
75
  self._memory_mb = None
71
76
  self._storage_gb = None
77
+ self._throughput = None
72
78
  self.discriminator = None
73
79
  if cpu is not None:
74
80
  self.cpu = cpu
@@ -80,10 +86,14 @@ class V1Resources(object):
80
86
  self.gpu = gpu
81
87
  if gpu_type is not None:
82
88
  self.gpu_type = gpu_type
89
+ if iops is not None:
90
+ self.iops = iops
83
91
  if memory_mb is not None:
84
92
  self.memory_mb = memory_mb
85
93
  if storage_gb is not None:
86
94
  self.storage_gb = storage_gb
95
+ if throughput is not None:
96
+ self.throughput = throughput
87
97
 
88
98
  @property
89
99
  def cpu(self) -> 'int':
@@ -190,6 +200,27 @@ class V1Resources(object):
190
200
 
191
201
  self._gpu_type = gpu_type
192
202
 
203
+ @property
204
+ def iops(self) -> 'str':
205
+ """Gets the iops of this V1Resources. # noqa: E501
206
+
207
+
208
+ :return: The iops of this V1Resources. # noqa: E501
209
+ :rtype: str
210
+ """
211
+ return self._iops
212
+
213
+ @iops.setter
214
+ def iops(self, iops: 'str'):
215
+ """Sets the iops of this V1Resources.
216
+
217
+
218
+ :param iops: The iops of this V1Resources. # noqa: E501
219
+ :type: str
220
+ """
221
+
222
+ self._iops = iops
223
+
193
224
  @property
194
225
  def memory_mb(self) -> 'str':
195
226
  """Gets the memory_mb of this V1Resources. # noqa: E501
@@ -232,6 +263,27 @@ class V1Resources(object):
232
263
 
233
264
  self._storage_gb = storage_gb
234
265
 
266
+ @property
267
+ def throughput(self) -> 'str':
268
+ """Gets the throughput of this V1Resources. # noqa: E501
269
+
270
+
271
+ :return: The throughput of this V1Resources. # noqa: E501
272
+ :rtype: str
273
+ """
274
+ return self._throughput
275
+
276
+ @throughput.setter
277
+ def throughput(self, throughput: 'str'):
278
+ """Sets the throughput of this V1Resources.
279
+
280
+
281
+ :param throughput: The throughput of this V1Resources. # noqa: E501
282
+ :type: str
283
+ """
284
+
285
+ self._throughput = throughput
286
+
235
287
  def to_dict(self) -> dict:
236
288
  """Returns the model properties as a dict"""
237
289
  result = {}
@@ -50,7 +50,6 @@ class V1UpdateUserRequest(object):
50
50
  'experimentation_id': 'str',
51
51
  'first_name': 'str',
52
52
  'general_audience_mode': 'bool',
53
- 'git_credentials_id': 'str',
54
53
  'last_name': 'str',
55
54
  'non_developer_mode': 'bool',
56
55
  'opted_in_marketing_emails': 'bool',
@@ -78,7 +77,6 @@ class V1UpdateUserRequest(object):
78
77
  'experimentation_id': 'experimentationId',
79
78
  'first_name': 'firstName',
80
79
  'general_audience_mode': 'generalAudienceMode',
81
- 'git_credentials_id': 'gitCredentialsId',
82
80
  'last_name': 'lastName',
83
81
  'non_developer_mode': 'nonDeveloperMode',
84
82
  'opted_in_marketing_emails': 'optedInMarketingEmails',
@@ -96,7 +94,7 @@ class V1UpdateUserRequest(object):
96
94
  'website': 'website'
97
95
  }
98
96
 
99
- def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, git_credentials_id: 'str' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None): # noqa: E501
97
+ def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, discounted_pro_plan: 'bool' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None): # noqa: E501
100
98
  """V1UpdateUserRequest - a model defined in Swagger""" # noqa: E501
101
99
  self._agree_to_terms_and_conditions = None
102
100
  self._complete_sign_up = None
@@ -107,7 +105,6 @@ class V1UpdateUserRequest(object):
107
105
  self._experimentation_id = None
108
106
  self._first_name = None
109
107
  self._general_audience_mode = None
110
- self._git_credentials_id = None
111
108
  self._last_name = None
112
109
  self._non_developer_mode = None
113
110
  self._opted_in_marketing_emails = None
@@ -142,8 +139,6 @@ class V1UpdateUserRequest(object):
142
139
  self.first_name = first_name
143
140
  if general_audience_mode is not None:
144
141
  self.general_audience_mode = general_audience_mode
145
- if git_credentials_id is not None:
146
- self.git_credentials_id = git_credentials_id
147
142
  if last_name is not None:
148
143
  self.last_name = last_name
149
144
  if non_developer_mode is not None:
@@ -364,27 +359,6 @@ class V1UpdateUserRequest(object):
364
359
 
365
360
  self._general_audience_mode = general_audience_mode
366
361
 
367
- @property
368
- def git_credentials_id(self) -> 'str':
369
- """Gets the git_credentials_id of this V1UpdateUserRequest. # noqa: E501
370
-
371
-
372
- :return: The git_credentials_id of this V1UpdateUserRequest. # noqa: E501
373
- :rtype: str
374
- """
375
- return self._git_credentials_id
376
-
377
- @git_credentials_id.setter
378
- def git_credentials_id(self, git_credentials_id: 'str'):
379
- """Sets the git_credentials_id of this V1UpdateUserRequest.
380
-
381
-
382
- :param git_credentials_id: The git_credentials_id of this V1UpdateUserRequest. # noqa: E501
383
- :type: str
384
- """
385
-
386
- self._git_credentials_id = git_credentials_id
387
-
388
362
  @property
389
363
  def last_name(self) -> 'str':
390
364
  """Gets the last_name of this V1UpdateUserRequest. # noqa: E501
@@ -50,6 +50,7 @@ class V1Usage(object):
50
50
  'id': 'str',
51
51
  'name': 'str',
52
52
  'project_id': 'str',
53
+ 'reservations': 'list[V1ReservationBillingSession]',
53
54
  'resource_type': 'str',
54
55
  'saved_cost': 'float',
55
56
  'session_ended_at': 'datetime',
@@ -68,6 +69,7 @@ class V1Usage(object):
68
69
  'id': 'id',
69
70
  'name': 'name',
70
71
  'project_id': 'projectId',
72
+ 'reservations': 'reservations',
71
73
  'resource_type': 'resourceType',
72
74
  'saved_cost': 'savedCost',
73
75
  'session_ended_at': 'sessionEndedAt',
@@ -76,7 +78,7 @@ class V1Usage(object):
76
78
  'user_id': 'userId'
77
79
  }
78
80
 
79
- def __init__(self, billed_time_seconds: 'str' =None, cluster_id: 'str' =None, cost: 'float' =None, created_at: 'datetime' =None, deleted_at: 'datetime' =None, free: 'bool' =None, id: 'str' =None, name: 'str' =None, project_id: 'str' =None, resource_type: 'str' =None, saved_cost: 'float' =None, session_ended_at: 'datetime' =None, session_started_at: 'datetime' =None, spot: 'bool' =None, user_id: 'str' =None): # noqa: E501
81
+ def __init__(self, billed_time_seconds: 'str' =None, cluster_id: 'str' =None, cost: 'float' =None, created_at: 'datetime' =None, deleted_at: 'datetime' =None, free: 'bool' =None, id: 'str' =None, name: 'str' =None, project_id: 'str' =None, reservations: 'list[V1ReservationBillingSession]' =None, resource_type: 'str' =None, saved_cost: 'float' =None, session_ended_at: 'datetime' =None, session_started_at: 'datetime' =None, spot: 'bool' =None, user_id: 'str' =None): # noqa: E501
80
82
  """V1Usage - a model defined in Swagger""" # noqa: E501
81
83
  self._billed_time_seconds = None
82
84
  self._cluster_id = None
@@ -87,6 +89,7 @@ class V1Usage(object):
87
89
  self._id = None
88
90
  self._name = None
89
91
  self._project_id = None
92
+ self._reservations = None
90
93
  self._resource_type = None
91
94
  self._saved_cost = None
92
95
  self._session_ended_at = None
@@ -112,6 +115,8 @@ class V1Usage(object):
112
115
  self.name = name
113
116
  if project_id is not None:
114
117
  self.project_id = project_id
118
+ if reservations is not None:
119
+ self.reservations = reservations
115
120
  if resource_type is not None:
116
121
  self.resource_type = resource_type
117
122
  if saved_cost is not None:
@@ -314,6 +319,27 @@ class V1Usage(object):
314
319
 
315
320
  self._project_id = project_id
316
321
 
322
+ @property
323
+ def reservations(self) -> 'list[V1ReservationBillingSession]':
324
+ """Gets the reservations of this V1Usage. # noqa: E501
325
+
326
+
327
+ :return: The reservations of this V1Usage. # noqa: E501
328
+ :rtype: list[V1ReservationBillingSession]
329
+ """
330
+ return self._reservations
331
+
332
+ @reservations.setter
333
+ def reservations(self, reservations: 'list[V1ReservationBillingSession]'):
334
+ """Sets the reservations of this V1Usage.
335
+
336
+
337
+ :param reservations: The reservations of this V1Usage. # noqa: E501
338
+ :type: list[V1ReservationBillingSession]
339
+ """
340
+
341
+ self._reservations = reservations
342
+
317
343
  @property
318
344
  def resource_type(self) -> 'str':
319
345
  """Gets the resource_type of this V1Usage. # noqa: E501
@@ -56,6 +56,7 @@ class V1UserFeatures(object):
56
56
  'cloud_space_environment_templates': 'bool',
57
57
  'code_tab': 'bool',
58
58
  'collab_screen_sharing': 'bool',
59
+ 'concurrent_gpu_limit': 'bool',
59
60
  'cost_attribution_settings': 'bool',
60
61
  'custom_app_domain': 'bool',
61
62
  'custom_instance_types': 'bool',
@@ -72,6 +73,7 @@ class V1UserFeatures(object):
72
73
  'fair_share': 'bool',
73
74
  'featured_studios_admin': 'bool',
74
75
  'filestore': 'bool',
76
+ 'gcp_local_disk_binding': 'bool',
75
77
  'inactive_notify_delete': 'bool',
76
78
  'instant_capacity_reservation': 'bool',
77
79
  'job_artifacts_v2': 'bool',
@@ -83,6 +85,7 @@ class V1UserFeatures(object):
83
85
  'mmt_strategy_selector': 'bool',
84
86
  'multicloud_saas': 'bool',
85
87
  'multiple_studio_versions': 'bool',
88
+ 'nerf_fs_nonpaying': 'bool',
86
89
  'org_admin_alerts': 'bool',
87
90
  'org_level_member_permissions': 'bool',
88
91
  'org_usage_limits': 'bool',
@@ -98,6 +101,7 @@ class V1UserFeatures(object):
98
101
  'product_generator': 'bool',
99
102
  'project_selector': 'bool',
100
103
  'publish_pipelines': 'bool',
104
+ 'r2_data_connections': 'bool',
101
105
  'restartable_jobs': 'bool',
102
106
  'runnable_public_studio_page': 'bool',
103
107
  'security_docs': 'bool',
@@ -107,6 +111,7 @@ class V1UserFeatures(object):
107
111
  'slurm_machine_selector': 'bool',
108
112
  'stop_ide_container_on_shutdown': 'bool',
109
113
  'studio_config': 'bool',
114
+ 'studio_deployment': 'bool',
110
115
  'studio_on_stop': 'bool',
111
116
  'studio_version_visibility': 'bool',
112
117
  'studios_dashboard': 'bool',
@@ -114,6 +119,7 @@ class V1UserFeatures(object):
114
119
  'teamspace_storage_tab': 'bool',
115
120
  'trainium2': 'bool',
116
121
  'use_rclone_mounts_only': 'bool',
122
+ 'voltage_park': 'bool',
117
123
  'vultr': 'bool',
118
124
  'weka': 'bool'
119
125
  }
@@ -134,6 +140,7 @@ class V1UserFeatures(object):
134
140
  'cloud_space_environment_templates': 'cloudSpaceEnvironmentTemplates',
135
141
  'code_tab': 'codeTab',
136
142
  'collab_screen_sharing': 'collabScreenSharing',
143
+ 'concurrent_gpu_limit': 'concurrentGpuLimit',
137
144
  'cost_attribution_settings': 'costAttributionSettings',
138
145
  'custom_app_domain': 'customAppDomain',
139
146
  'custom_instance_types': 'customInstanceTypes',
@@ -150,6 +157,7 @@ class V1UserFeatures(object):
150
157
  'fair_share': 'fairShare',
151
158
  'featured_studios_admin': 'featuredStudiosAdmin',
152
159
  'filestore': 'filestore',
160
+ 'gcp_local_disk_binding': 'gcpLocalDiskBinding',
153
161
  'inactive_notify_delete': 'inactiveNotifyDelete',
154
162
  'instant_capacity_reservation': 'instantCapacityReservation',
155
163
  'job_artifacts_v2': 'jobArtifactsV2',
@@ -161,6 +169,7 @@ class V1UserFeatures(object):
161
169
  'mmt_strategy_selector': 'mmtStrategySelector',
162
170
  'multicloud_saas': 'multicloudSaas',
163
171
  'multiple_studio_versions': 'multipleStudioVersions',
172
+ 'nerf_fs_nonpaying': 'nerfFsNonpaying',
164
173
  'org_admin_alerts': 'orgAdminAlerts',
165
174
  'org_level_member_permissions': 'orgLevelMemberPermissions',
166
175
  'org_usage_limits': 'orgUsageLimits',
@@ -176,6 +185,7 @@ class V1UserFeatures(object):
176
185
  'product_generator': 'productGenerator',
177
186
  'project_selector': 'projectSelector',
178
187
  'publish_pipelines': 'publishPipelines',
188
+ 'r2_data_connections': 'r2DataConnections',
179
189
  'restartable_jobs': 'restartableJobs',
180
190
  'runnable_public_studio_page': 'runnablePublicStudioPage',
181
191
  'security_docs': 'securityDocs',
@@ -185,6 +195,7 @@ class V1UserFeatures(object):
185
195
  'slurm_machine_selector': 'slurmMachineSelector',
186
196
  'stop_ide_container_on_shutdown': 'stopIdeContainerOnShutdown',
187
197
  'studio_config': 'studioConfig',
198
+ 'studio_deployment': 'studioDeployment',
188
199
  'studio_on_stop': 'studioOnStop',
189
200
  'studio_version_visibility': 'studioVersionVisibility',
190
201
  'studios_dashboard': 'studiosDashboard',
@@ -192,11 +203,12 @@ class V1UserFeatures(object):
192
203
  'teamspace_storage_tab': 'teamspaceStorageTab',
193
204
  'trainium2': 'trainium2',
194
205
  'use_rclone_mounts_only': 'useRcloneMountsOnly',
206
+ 'voltage_park': 'voltagePark',
195
207
  'vultr': 'vultr',
196
208
  'weka': 'weka'
197
209
  }
198
210
 
199
- def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, dgx_cloud: 'bool' =None, doc_helper_chat: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, inactive_notify_delete: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, org_admin_alerts: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, studios_dashboard: 'bool' =None, studios_dashboard_system_metrics: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
211
+ def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byoc_litcr: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloud_space_environment_templates: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, concurrent_gpu_limit: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_alerts: 'bool' =None, deployment_persistent_disk: 'bool' =None, dgx_cloud: 'bool' =None, doc_helper_chat: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_storage_limits: 'bool' =None, enterprise_compute_admin: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, filestore: 'bool' =None, gcp_local_disk_binding: 'bool' =None, inactive_notify_delete: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, lambda_labs: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multicloud_saas: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_admin_alerts: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, pipelines: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, r2_data_connections: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, stop_ide_container_on_shutdown: 'bool' =None, studio_config: 'bool' =None, studio_deployment: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, studios_dashboard: 'bool' =None, studios_dashboard_system_metrics: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, voltage_park: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None): # noqa: E501
200
212
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
201
213
  self._affiliate_links = None
202
214
  self._agents_v2 = None
@@ -213,6 +225,7 @@ class V1UserFeatures(object):
213
225
  self._cloud_space_environment_templates = None
214
226
  self._code_tab = None
215
227
  self._collab_screen_sharing = None
228
+ self._concurrent_gpu_limit = None
216
229
  self._cost_attribution_settings = None
217
230
  self._custom_app_domain = None
218
231
  self._custom_instance_types = None
@@ -229,6 +242,7 @@ class V1UserFeatures(object):
229
242
  self._fair_share = None
230
243
  self._featured_studios_admin = None
231
244
  self._filestore = None
245
+ self._gcp_local_disk_binding = None
232
246
  self._inactive_notify_delete = None
233
247
  self._instant_capacity_reservation = None
234
248
  self._job_artifacts_v2 = None
@@ -240,6 +254,7 @@ class V1UserFeatures(object):
240
254
  self._mmt_strategy_selector = None
241
255
  self._multicloud_saas = None
242
256
  self._multiple_studio_versions = None
257
+ self._nerf_fs_nonpaying = None
243
258
  self._org_admin_alerts = None
244
259
  self._org_level_member_permissions = None
245
260
  self._org_usage_limits = None
@@ -255,6 +270,7 @@ class V1UserFeatures(object):
255
270
  self._product_generator = None
256
271
  self._project_selector = None
257
272
  self._publish_pipelines = None
273
+ self._r2_data_connections = None
258
274
  self._restartable_jobs = None
259
275
  self._runnable_public_studio_page = None
260
276
  self._security_docs = None
@@ -264,6 +280,7 @@ class V1UserFeatures(object):
264
280
  self._slurm_machine_selector = None
265
281
  self._stop_ide_container_on_shutdown = None
266
282
  self._studio_config = None
283
+ self._studio_deployment = None
267
284
  self._studio_on_stop = None
268
285
  self._studio_version_visibility = None
269
286
  self._studios_dashboard = None
@@ -271,6 +288,7 @@ class V1UserFeatures(object):
271
288
  self._teamspace_storage_tab = None
272
289
  self._trainium2 = None
273
290
  self._use_rclone_mounts_only = None
291
+ self._voltage_park = None
274
292
  self._vultr = None
275
293
  self._weka = None
276
294
  self.discriminator = None
@@ -304,6 +322,8 @@ class V1UserFeatures(object):
304
322
  self.code_tab = code_tab
305
323
  if collab_screen_sharing is not None:
306
324
  self.collab_screen_sharing = collab_screen_sharing
325
+ if concurrent_gpu_limit is not None:
326
+ self.concurrent_gpu_limit = concurrent_gpu_limit
307
327
  if cost_attribution_settings is not None:
308
328
  self.cost_attribution_settings = cost_attribution_settings
309
329
  if custom_app_domain is not None:
@@ -336,6 +356,8 @@ class V1UserFeatures(object):
336
356
  self.featured_studios_admin = featured_studios_admin
337
357
  if filestore is not None:
338
358
  self.filestore = filestore
359
+ if gcp_local_disk_binding is not None:
360
+ self.gcp_local_disk_binding = gcp_local_disk_binding
339
361
  if inactive_notify_delete is not None:
340
362
  self.inactive_notify_delete = inactive_notify_delete
341
363
  if instant_capacity_reservation is not None:
@@ -358,6 +380,8 @@ class V1UserFeatures(object):
358
380
  self.multicloud_saas = multicloud_saas
359
381
  if multiple_studio_versions is not None:
360
382
  self.multiple_studio_versions = multiple_studio_versions
383
+ if nerf_fs_nonpaying is not None:
384
+ self.nerf_fs_nonpaying = nerf_fs_nonpaying
361
385
  if org_admin_alerts is not None:
362
386
  self.org_admin_alerts = org_admin_alerts
363
387
  if org_level_member_permissions is not None:
@@ -388,6 +412,8 @@ class V1UserFeatures(object):
388
412
  self.project_selector = project_selector
389
413
  if publish_pipelines is not None:
390
414
  self.publish_pipelines = publish_pipelines
415
+ if r2_data_connections is not None:
416
+ self.r2_data_connections = r2_data_connections
391
417
  if restartable_jobs is not None:
392
418
  self.restartable_jobs = restartable_jobs
393
419
  if runnable_public_studio_page is not None:
@@ -406,6 +432,8 @@ class V1UserFeatures(object):
406
432
  self.stop_ide_container_on_shutdown = stop_ide_container_on_shutdown
407
433
  if studio_config is not None:
408
434
  self.studio_config = studio_config
435
+ if studio_deployment is not None:
436
+ self.studio_deployment = studio_deployment
409
437
  if studio_on_stop is not None:
410
438
  self.studio_on_stop = studio_on_stop
411
439
  if studio_version_visibility is not None:
@@ -420,6 +448,8 @@ class V1UserFeatures(object):
420
448
  self.trainium2 = trainium2
421
449
  if use_rclone_mounts_only is not None:
422
450
  self.use_rclone_mounts_only = use_rclone_mounts_only
451
+ if voltage_park is not None:
452
+ self.voltage_park = voltage_park
423
453
  if vultr is not None:
424
454
  self.vultr = vultr
425
455
  if weka is not None:
@@ -740,6 +770,27 @@ class V1UserFeatures(object):
740
770
 
741
771
  self._collab_screen_sharing = collab_screen_sharing
742
772
 
773
+ @property
774
+ def concurrent_gpu_limit(self) -> 'bool':
775
+ """Gets the concurrent_gpu_limit of this V1UserFeatures. # noqa: E501
776
+
777
+
778
+ :return: The concurrent_gpu_limit of this V1UserFeatures. # noqa: E501
779
+ :rtype: bool
780
+ """
781
+ return self._concurrent_gpu_limit
782
+
783
+ @concurrent_gpu_limit.setter
784
+ def concurrent_gpu_limit(self, concurrent_gpu_limit: 'bool'):
785
+ """Sets the concurrent_gpu_limit of this V1UserFeatures.
786
+
787
+
788
+ :param concurrent_gpu_limit: The concurrent_gpu_limit of this V1UserFeatures. # noqa: E501
789
+ :type: bool
790
+ """
791
+
792
+ self._concurrent_gpu_limit = concurrent_gpu_limit
793
+
743
794
  @property
744
795
  def cost_attribution_settings(self) -> 'bool':
745
796
  """Gets the cost_attribution_settings of this V1UserFeatures. # noqa: E501
@@ -1076,6 +1127,27 @@ class V1UserFeatures(object):
1076
1127
 
1077
1128
  self._filestore = filestore
1078
1129
 
1130
+ @property
1131
+ def gcp_local_disk_binding(self) -> 'bool':
1132
+ """Gets the gcp_local_disk_binding of this V1UserFeatures. # noqa: E501
1133
+
1134
+
1135
+ :return: The gcp_local_disk_binding of this V1UserFeatures. # noqa: E501
1136
+ :rtype: bool
1137
+ """
1138
+ return self._gcp_local_disk_binding
1139
+
1140
+ @gcp_local_disk_binding.setter
1141
+ def gcp_local_disk_binding(self, gcp_local_disk_binding: 'bool'):
1142
+ """Sets the gcp_local_disk_binding of this V1UserFeatures.
1143
+
1144
+
1145
+ :param gcp_local_disk_binding: The gcp_local_disk_binding of this V1UserFeatures. # noqa: E501
1146
+ :type: bool
1147
+ """
1148
+
1149
+ self._gcp_local_disk_binding = gcp_local_disk_binding
1150
+
1079
1151
  @property
1080
1152
  def inactive_notify_delete(self) -> 'bool':
1081
1153
  """Gets the inactive_notify_delete of this V1UserFeatures. # noqa: E501
@@ -1307,6 +1379,27 @@ class V1UserFeatures(object):
1307
1379
 
1308
1380
  self._multiple_studio_versions = multiple_studio_versions
1309
1381
 
1382
+ @property
1383
+ def nerf_fs_nonpaying(self) -> 'bool':
1384
+ """Gets the nerf_fs_nonpaying of this V1UserFeatures. # noqa: E501
1385
+
1386
+
1387
+ :return: The nerf_fs_nonpaying of this V1UserFeatures. # noqa: E501
1388
+ :rtype: bool
1389
+ """
1390
+ return self._nerf_fs_nonpaying
1391
+
1392
+ @nerf_fs_nonpaying.setter
1393
+ def nerf_fs_nonpaying(self, nerf_fs_nonpaying: 'bool'):
1394
+ """Sets the nerf_fs_nonpaying of this V1UserFeatures.
1395
+
1396
+
1397
+ :param nerf_fs_nonpaying: The nerf_fs_nonpaying of this V1UserFeatures. # noqa: E501
1398
+ :type: bool
1399
+ """
1400
+
1401
+ self._nerf_fs_nonpaying = nerf_fs_nonpaying
1402
+
1310
1403
  @property
1311
1404
  def org_admin_alerts(self) -> 'bool':
1312
1405
  """Gets the org_admin_alerts of this V1UserFeatures. # noqa: E501
@@ -1622,6 +1715,27 @@ class V1UserFeatures(object):
1622
1715
 
1623
1716
  self._publish_pipelines = publish_pipelines
1624
1717
 
1718
+ @property
1719
+ def r2_data_connections(self) -> 'bool':
1720
+ """Gets the r2_data_connections of this V1UserFeatures. # noqa: E501
1721
+
1722
+
1723
+ :return: The r2_data_connections of this V1UserFeatures. # noqa: E501
1724
+ :rtype: bool
1725
+ """
1726
+ return self._r2_data_connections
1727
+
1728
+ @r2_data_connections.setter
1729
+ def r2_data_connections(self, r2_data_connections: 'bool'):
1730
+ """Sets the r2_data_connections of this V1UserFeatures.
1731
+
1732
+
1733
+ :param r2_data_connections: The r2_data_connections of this V1UserFeatures. # noqa: E501
1734
+ :type: bool
1735
+ """
1736
+
1737
+ self._r2_data_connections = r2_data_connections
1738
+
1625
1739
  @property
1626
1740
  def restartable_jobs(self) -> 'bool':
1627
1741
  """Gets the restartable_jobs of this V1UserFeatures. # noqa: E501
@@ -1811,6 +1925,27 @@ class V1UserFeatures(object):
1811
1925
 
1812
1926
  self._studio_config = studio_config
1813
1927
 
1928
+ @property
1929
+ def studio_deployment(self) -> 'bool':
1930
+ """Gets the studio_deployment of this V1UserFeatures. # noqa: E501
1931
+
1932
+
1933
+ :return: The studio_deployment of this V1UserFeatures. # noqa: E501
1934
+ :rtype: bool
1935
+ """
1936
+ return self._studio_deployment
1937
+
1938
+ @studio_deployment.setter
1939
+ def studio_deployment(self, studio_deployment: 'bool'):
1940
+ """Sets the studio_deployment of this V1UserFeatures.
1941
+
1942
+
1943
+ :param studio_deployment: The studio_deployment of this V1UserFeatures. # noqa: E501
1944
+ :type: bool
1945
+ """
1946
+
1947
+ self._studio_deployment = studio_deployment
1948
+
1814
1949
  @property
1815
1950
  def studio_on_stop(self) -> 'bool':
1816
1951
  """Gets the studio_on_stop of this V1UserFeatures. # noqa: E501
@@ -1958,6 +2093,27 @@ class V1UserFeatures(object):
1958
2093
 
1959
2094
  self._use_rclone_mounts_only = use_rclone_mounts_only
1960
2095
 
2096
+ @property
2097
+ def voltage_park(self) -> 'bool':
2098
+ """Gets the voltage_park of this V1UserFeatures. # noqa: E501
2099
+
2100
+
2101
+ :return: The voltage_park of this V1UserFeatures. # noqa: E501
2102
+ :rtype: bool
2103
+ """
2104
+ return self._voltage_park
2105
+
2106
+ @voltage_park.setter
2107
+ def voltage_park(self, voltage_park: 'bool'):
2108
+ """Sets the voltage_park of this V1UserFeatures.
2109
+
2110
+
2111
+ :param voltage_park: The voltage_park of this V1UserFeatures. # noqa: E501
2112
+ :type: bool
2113
+ """
2114
+
2115
+ self._voltage_park = voltage_park
2116
+
1961
2117
  @property
1962
2118
  def vultr(self) -> 'bool':
1963
2119
  """Gets the vultr of this V1UserFeatures. # noqa: E501
@@ -0,0 +1,3 @@
1
+ from lightning_sdk.llm.llm import LLM
2
+
3
+ __all__ = ["LLM"]