lightning-sdk 0.2.19__py3-none-any.whl → 0.2.21__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 (62) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +7 -2
  3. lightning_sdk/api/license_api.py +28 -6
  4. lightning_sdk/api/llm_api.py +118 -26
  5. lightning_sdk/api/studio_api.py +5 -0
  6. lightning_sdk/cli/configure.py +34 -27
  7. lightning_sdk/cli/connect.py +2 -2
  8. lightning_sdk/cli/deploy/_auth.py +11 -19
  9. lightning_sdk/cli/entrypoint.py +20 -2
  10. lightning_sdk/deployment/deployment.py +17 -3
  11. lightning_sdk/lightning_cloud/login.py +2 -2
  12. lightning_sdk/lightning_cloud/openapi/__init__.py +2 -3
  13. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +1 -5
  14. lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py +11 -1
  15. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +121 -0
  16. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +0 -85
  17. lightning_sdk/lightning_cloud/openapi/models/__init__.py +2 -3
  18. lightning_sdk/lightning_cloud/openapi/models/alertingevents_id_body.py +409 -0
  19. lightning_sdk/lightning_cloud/openapi/models/id_codeconfig_body.py +29 -3
  20. lightning_sdk/lightning_cloud/openapi/models/update.py +105 -1
  21. lightning_sdk/lightning_cloud/openapi/models/v1_author.py +201 -0
  22. lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py +53 -1
  23. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +53 -1
  25. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +105 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +79 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +79 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_type.py +1 -0
  31. lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadata_response.py +79 -1
  32. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py +105 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +79 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +53 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_notification_type.py +1 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +105 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +131 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset_type.py +2 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_transaction.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +79 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_usage.py +27 -27
  46. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +79 -443
  47. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +499 -31
  48. lightning_sdk/lightning_cloud/rest_client.py +13 -11
  49. lightning_sdk/lightning_cloud/source_code/logs_socket_api.py +8 -3
  50. lightning_sdk/llm/llm.py +52 -1
  51. lightning_sdk/pipeline/pipeline.py +1 -1
  52. lightning_sdk/services/license.py +78 -22
  53. lightning_sdk/services/utilities.py +15 -1
  54. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/METADATA +1 -1
  55. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/RECORD +59 -60
  56. lightning_sdk/lightning_cloud/openapi/models/v1_ebs.py +0 -279
  57. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py +0 -201
  58. lightning_sdk/lightning_cloud/openapi/models/v1_reservation_billing_session.py +0 -279
  59. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/LICENSE +0 -0
  60. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/WHEEL +0 -0
  61. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/entry_points.txt +0 -0
  62. {lightning_sdk-0.2.19.dist-info → lightning_sdk-0.2.21.dist-info}/top_level.txt +0 -0
@@ -44,6 +44,7 @@ class V1Transaction(object):
44
44
  'amount': 'float',
45
45
  'created_at': 'datetime',
46
46
  'display_name': 'str',
47
+ 'project_name': 'str',
47
48
  'type': 'str'
48
49
  }
49
50
 
@@ -51,14 +52,16 @@ class V1Transaction(object):
51
52
  'amount': 'amount',
52
53
  'created_at': 'createdAt',
53
54
  'display_name': 'displayName',
55
+ 'project_name': 'projectName',
54
56
  'type': 'type'
55
57
  }
56
58
 
57
- def __init__(self, amount: 'float' =None, created_at: 'datetime' =None, display_name: 'str' =None, type: 'str' =None): # noqa: E501
59
+ def __init__(self, amount: 'float' =None, created_at: 'datetime' =None, display_name: 'str' =None, project_name: 'str' =None, type: 'str' =None): # noqa: E501
58
60
  """V1Transaction - a model defined in Swagger""" # noqa: E501
59
61
  self._amount = None
60
62
  self._created_at = None
61
63
  self._display_name = None
64
+ self._project_name = None
62
65
  self._type = None
63
66
  self.discriminator = None
64
67
  if amount is not None:
@@ -67,6 +70,8 @@ class V1Transaction(object):
67
70
  self.created_at = created_at
68
71
  if display_name is not None:
69
72
  self.display_name = display_name
73
+ if project_name is not None:
74
+ self.project_name = project_name
70
75
  if type is not None:
71
76
  self.type = type
72
77
 
@@ -133,6 +138,27 @@ class V1Transaction(object):
133
138
 
134
139
  self._display_name = display_name
135
140
 
141
+ @property
142
+ def project_name(self) -> 'str':
143
+ """Gets the project_name of this V1Transaction. # noqa: E501
144
+
145
+
146
+ :return: The project_name of this V1Transaction. # noqa: E501
147
+ :rtype: str
148
+ """
149
+ return self._project_name
150
+
151
+ @project_name.setter
152
+ def project_name(self, project_name: 'str'):
153
+ """Sets the project_name of this V1Transaction.
154
+
155
+
156
+ :param project_name: The project_name of this V1Transaction. # noqa: E501
157
+ :type: str
158
+ """
159
+
160
+ self._project_name = project_name
161
+
136
162
  @property
137
163
  def type(self) -> 'str':
138
164
  """Gets the type of this V1Transaction. # noqa: E501
@@ -42,6 +42,9 @@ class V1UpdateUserRequest(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'agree_to_terms_and_conditions': 'bool',
45
+ 'allow_credits_auto_replenish': 'bool',
46
+ 'auto_replenish_amount': 'float',
47
+ 'auto_replenish_threshold': 'float',
45
48
  'complete_sign_up': 'bool',
46
49
  'completed_project_onboarding': 'bool',
47
50
  'country': 'str',
@@ -69,6 +72,9 @@ class V1UpdateUserRequest(object):
69
72
 
70
73
  attribute_map = {
71
74
  'agree_to_terms_and_conditions': 'agreeToTermsAndConditions',
75
+ 'allow_credits_auto_replenish': 'allowCreditsAutoReplenish',
76
+ 'auto_replenish_amount': 'autoReplenishAmount',
77
+ 'auto_replenish_threshold': 'autoReplenishThreshold',
72
78
  'complete_sign_up': 'completeSignUp',
73
79
  'completed_project_onboarding': 'completedProjectOnboarding',
74
80
  'country': 'country',
@@ -94,9 +100,12 @@ class V1UpdateUserRequest(object):
94
100
  'website': 'website'
95
101
  }
96
102
 
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
103
+ def __init__(self, agree_to_terms_and_conditions: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =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
98
104
  """V1UpdateUserRequest - a model defined in Swagger""" # noqa: E501
99
105
  self._agree_to_terms_and_conditions = None
106
+ self._allow_credits_auto_replenish = None
107
+ self._auto_replenish_amount = None
108
+ self._auto_replenish_threshold = None
100
109
  self._complete_sign_up = None
101
110
  self._completed_project_onboarding = None
102
111
  self._country = None
@@ -123,6 +132,12 @@ class V1UpdateUserRequest(object):
123
132
  self.discriminator = None
124
133
  if agree_to_terms_and_conditions is not None:
125
134
  self.agree_to_terms_and_conditions = agree_to_terms_and_conditions
135
+ if allow_credits_auto_replenish is not None:
136
+ self.allow_credits_auto_replenish = allow_credits_auto_replenish
137
+ if auto_replenish_amount is not None:
138
+ self.auto_replenish_amount = auto_replenish_amount
139
+ if auto_replenish_threshold is not None:
140
+ self.auto_replenish_threshold = auto_replenish_threshold
126
141
  if complete_sign_up is not None:
127
142
  self.complete_sign_up = complete_sign_up
128
143
  if completed_project_onboarding is not None:
@@ -191,6 +206,69 @@ class V1UpdateUserRequest(object):
191
206
 
192
207
  self._agree_to_terms_and_conditions = agree_to_terms_and_conditions
193
208
 
209
+ @property
210
+ def allow_credits_auto_replenish(self) -> 'bool':
211
+ """Gets the allow_credits_auto_replenish of this V1UpdateUserRequest. # noqa: E501
212
+
213
+
214
+ :return: The allow_credits_auto_replenish of this V1UpdateUserRequest. # noqa: E501
215
+ :rtype: bool
216
+ """
217
+ return self._allow_credits_auto_replenish
218
+
219
+ @allow_credits_auto_replenish.setter
220
+ def allow_credits_auto_replenish(self, allow_credits_auto_replenish: 'bool'):
221
+ """Sets the allow_credits_auto_replenish of this V1UpdateUserRequest.
222
+
223
+
224
+ :param allow_credits_auto_replenish: The allow_credits_auto_replenish of this V1UpdateUserRequest. # noqa: E501
225
+ :type: bool
226
+ """
227
+
228
+ self._allow_credits_auto_replenish = allow_credits_auto_replenish
229
+
230
+ @property
231
+ def auto_replenish_amount(self) -> 'float':
232
+ """Gets the auto_replenish_amount of this V1UpdateUserRequest. # noqa: E501
233
+
234
+
235
+ :return: The auto_replenish_amount of this V1UpdateUserRequest. # noqa: E501
236
+ :rtype: float
237
+ """
238
+ return self._auto_replenish_amount
239
+
240
+ @auto_replenish_amount.setter
241
+ def auto_replenish_amount(self, auto_replenish_amount: 'float'):
242
+ """Sets the auto_replenish_amount of this V1UpdateUserRequest.
243
+
244
+
245
+ :param auto_replenish_amount: The auto_replenish_amount of this V1UpdateUserRequest. # noqa: E501
246
+ :type: float
247
+ """
248
+
249
+ self._auto_replenish_amount = auto_replenish_amount
250
+
251
+ @property
252
+ def auto_replenish_threshold(self) -> 'float':
253
+ """Gets the auto_replenish_threshold of this V1UpdateUserRequest. # noqa: E501
254
+
255
+
256
+ :return: The auto_replenish_threshold of this V1UpdateUserRequest. # noqa: E501
257
+ :rtype: float
258
+ """
259
+ return self._auto_replenish_threshold
260
+
261
+ @auto_replenish_threshold.setter
262
+ def auto_replenish_threshold(self, auto_replenish_threshold: 'float'):
263
+ """Sets the auto_replenish_threshold of this V1UpdateUserRequest.
264
+
265
+
266
+ :param auto_replenish_threshold: The auto_replenish_threshold of this V1UpdateUserRequest. # noqa: E501
267
+ :type: float
268
+ """
269
+
270
+ self._auto_replenish_threshold = auto_replenish_threshold
271
+
194
272
  @property
195
273
  def complete_sign_up(self) -> 'bool':
196
274
  """Gets the complete_sign_up of this V1UpdateUserRequest. # noqa: E501
@@ -48,9 +48,9 @@ class V1Usage(object):
48
48
  'deleted_at': 'datetime',
49
49
  'free': 'bool',
50
50
  'id': 'str',
51
+ 'instance_type': 'str',
51
52
  'name': 'str',
52
53
  'project_id': 'str',
53
- 'reservations': 'list[V1ReservationBillingSession]',
54
54
  'resource_type': 'str',
55
55
  'saved_cost': 'float',
56
56
  'session_ended_at': 'datetime',
@@ -70,9 +70,9 @@ class V1Usage(object):
70
70
  'deleted_at': 'deletedAt',
71
71
  'free': 'free',
72
72
  'id': 'id',
73
+ 'instance_type': 'instanceType',
73
74
  'name': 'name',
74
75
  'project_id': 'projectId',
75
- 'reservations': 'reservations',
76
76
  'resource_type': 'resourceType',
77
77
  'saved_cost': 'savedCost',
78
78
  'session_ended_at': 'sessionEndedAt',
@@ -84,7 +84,7 @@ class V1Usage(object):
84
84
  'user_id': 'userId'
85
85
  }
86
86
 
87
- 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, total_completion_tokens: 'str' =None, total_num_messages: 'str' =None, total_prompt_tokens: 'str' =None, user_id: 'str' =None): # noqa: E501
87
+ 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, instance_type: '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, total_completion_tokens: 'str' =None, total_num_messages: 'str' =None, total_prompt_tokens: 'str' =None, user_id: 'str' =None): # noqa: E501
88
88
  """V1Usage - a model defined in Swagger""" # noqa: E501
89
89
  self._billed_time_seconds = None
90
90
  self._cluster_id = None
@@ -93,9 +93,9 @@ class V1Usage(object):
93
93
  self._deleted_at = None
94
94
  self._free = None
95
95
  self._id = None
96
+ self._instance_type = None
96
97
  self._name = None
97
98
  self._project_id = None
98
- self._reservations = None
99
99
  self._resource_type = None
100
100
  self._saved_cost = None
101
101
  self._session_ended_at = None
@@ -120,12 +120,12 @@ class V1Usage(object):
120
120
  self.free = free
121
121
  if id is not None:
122
122
  self.id = id
123
+ if instance_type is not None:
124
+ self.instance_type = instance_type
123
125
  if name is not None:
124
126
  self.name = name
125
127
  if project_id is not None:
126
128
  self.project_id = project_id
127
- if reservations is not None:
128
- self.reservations = reservations
129
129
  if resource_type is not None:
130
130
  self.resource_type = resource_type
131
131
  if saved_cost is not None:
@@ -292,6 +292,27 @@ class V1Usage(object):
292
292
 
293
293
  self._id = id
294
294
 
295
+ @property
296
+ def instance_type(self) -> 'str':
297
+ """Gets the instance_type of this V1Usage. # noqa: E501
298
+
299
+
300
+ :return: The instance_type of this V1Usage. # noqa: E501
301
+ :rtype: str
302
+ """
303
+ return self._instance_type
304
+
305
+ @instance_type.setter
306
+ def instance_type(self, instance_type: 'str'):
307
+ """Sets the instance_type of this V1Usage.
308
+
309
+
310
+ :param instance_type: The instance_type of this V1Usage. # noqa: E501
311
+ :type: str
312
+ """
313
+
314
+ self._instance_type = instance_type
315
+
295
316
  @property
296
317
  def name(self) -> 'str':
297
318
  """Gets the name of this V1Usage. # noqa: E501
@@ -334,27 +355,6 @@ class V1Usage(object):
334
355
 
335
356
  self._project_id = project_id
336
357
 
337
- @property
338
- def reservations(self) -> 'list[V1ReservationBillingSession]':
339
- """Gets the reservations of this V1Usage. # noqa: E501
340
-
341
-
342
- :return: The reservations of this V1Usage. # noqa: E501
343
- :rtype: list[V1ReservationBillingSession]
344
- """
345
- return self._reservations
346
-
347
- @reservations.setter
348
- def reservations(self, reservations: 'list[V1ReservationBillingSession]'):
349
- """Sets the reservations of this V1Usage.
350
-
351
-
352
- :param reservations: The reservations of this V1Usage. # noqa: E501
353
- :type: list[V1ReservationBillingSession]
354
- """
355
-
356
- self._reservations = reservations
357
-
358
358
  @property
359
359
  def resource_type(self) -> 'str':
360
360
  """Gets the resource_type of this V1Usage. # noqa: E501