lightning-sdk 0.2.24rc0__py3-none-any.whl → 2025.7.9__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 (42) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/cluster_api.py +15 -0
  3. lightning_sdk/api/llm_api.py +18 -2
  4. lightning_sdk/api/pipeline_api.py +15 -7
  5. lightning_sdk/api/studio_api.py +0 -17
  6. lightning_sdk/lightning_cloud/openapi/__init__.py +6 -0
  7. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  8. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +9 -1
  9. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +13 -1
  10. lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py +258 -0
  11. lightning_sdk/lightning_cloud/openapi/models/__init__.py +5 -0
  12. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +53 -1
  13. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +1 -27
  14. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  15. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  16. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  17. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py +103 -0
  20. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +29 -3
  21. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  22. lightning_sdk/lightning_cloud/openapi/models/v1_get_volume_response.py +123 -0
  23. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +29 -3
  24. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +1 -27
  25. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +3 -29
  26. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  27. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py +104 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +1 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +1 -27
  31. lightning_sdk/lightning_cloud/openapi/models/v1_update_volume_response.py +123 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +183 -53
  33. lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.py +123 -0
  34. lightning_sdk/llm/llm.py +6 -1
  35. lightning_sdk/machine.py +1 -4
  36. lightning_sdk/pipeline/pipeline.py +1 -15
  37. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/METADATA +1 -1
  38. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/RECORD +42 -36
  39. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/LICENSE +0 -0
  40. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/WHEEL +0 -0
  41. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/entry_points.txt +0 -0
  42. {lightning_sdk-0.2.24rc0.dist-info → lightning_sdk-2025.7.9.dist-info}/top_level.txt +0 -0
@@ -50,8 +50,10 @@ class AssistantIdConversationsBody(object):
50
50
  'message': 'V1Message',
51
51
  'metadata': 'dict(str, str)',
52
52
  'name': 'str',
53
+ 'parent_conversation_id': 'str',
53
54
  'parent_message_id': 'str',
54
55
  'reasoning_effort': 'str',
56
+ 'sent_at': 'datetime',
55
57
  'store': 'bool',
56
58
  'stream': 'bool',
57
59
  'system_prompt': 'str'
@@ -67,14 +69,16 @@ class AssistantIdConversationsBody(object):
67
69
  'message': 'message',
68
70
  'metadata': 'metadata',
69
71
  'name': 'name',
72
+ 'parent_conversation_id': 'parentConversationId',
70
73
  'parent_message_id': 'parentMessageId',
71
74
  'reasoning_effort': 'reasoningEffort',
75
+ 'sent_at': 'sentAt',
72
76
  'store': 'store',
73
77
  'stream': 'stream',
74
78
  'system_prompt': 'systemPrompt'
75
79
  }
76
80
 
77
- def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None): # noqa: E501
81
+ def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None): # noqa: E501
78
82
  """AssistantIdConversationsBody - a model defined in Swagger""" # noqa: E501
79
83
  self._auto_name = None
80
84
  self._billing_project_id = None
@@ -85,8 +89,10 @@ class AssistantIdConversationsBody(object):
85
89
  self._message = None
86
90
  self._metadata = None
87
91
  self._name = None
92
+ self._parent_conversation_id = None
88
93
  self._parent_message_id = None
89
94
  self._reasoning_effort = None
95
+ self._sent_at = None
90
96
  self._store = None
91
97
  self._stream = None
92
98
  self._system_prompt = None
@@ -109,10 +115,14 @@ class AssistantIdConversationsBody(object):
109
115
  self.metadata = metadata
110
116
  if name is not None:
111
117
  self.name = name
118
+ if parent_conversation_id is not None:
119
+ self.parent_conversation_id = parent_conversation_id
112
120
  if parent_message_id is not None:
113
121
  self.parent_message_id = parent_message_id
114
122
  if reasoning_effort is not None:
115
123
  self.reasoning_effort = reasoning_effort
124
+ if sent_at is not None:
125
+ self.sent_at = sent_at
116
126
  if store is not None:
117
127
  self.store = store
118
128
  if stream is not None:
@@ -309,6 +319,27 @@ class AssistantIdConversationsBody(object):
309
319
 
310
320
  self._name = name
311
321
 
322
+ @property
323
+ def parent_conversation_id(self) -> 'str':
324
+ """Gets the parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
325
+
326
+
327
+ :return: The parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
328
+ :rtype: str
329
+ """
330
+ return self._parent_conversation_id
331
+
332
+ @parent_conversation_id.setter
333
+ def parent_conversation_id(self, parent_conversation_id: 'str'):
334
+ """Sets the parent_conversation_id of this AssistantIdConversationsBody.
335
+
336
+
337
+ :param parent_conversation_id: The parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
338
+ :type: str
339
+ """
340
+
341
+ self._parent_conversation_id = parent_conversation_id
342
+
312
343
  @property
313
344
  def parent_message_id(self) -> 'str':
314
345
  """Gets the parent_message_id of this AssistantIdConversationsBody. # noqa: E501
@@ -351,6 +382,27 @@ class AssistantIdConversationsBody(object):
351
382
 
352
383
  self._reasoning_effort = reasoning_effort
353
384
 
385
+ @property
386
+ def sent_at(self) -> 'datetime':
387
+ """Gets the sent_at of this AssistantIdConversationsBody. # noqa: E501
388
+
389
+
390
+ :return: The sent_at of this AssistantIdConversationsBody. # noqa: E501
391
+ :rtype: datetime
392
+ """
393
+ return self._sent_at
394
+
395
+ @sent_at.setter
396
+ def sent_at(self, sent_at: 'datetime'):
397
+ """Sets the sent_at of this AssistantIdConversationsBody.
398
+
399
+
400
+ :param sent_at: The sent_at of this AssistantIdConversationsBody. # noqa: E501
401
+ :type: datetime
402
+ """
403
+
404
+ self._sent_at = sent_at
405
+
354
406
  @property
355
407
  def store(self) -> 'bool':
356
408
  """Gets the store of this AssistantIdConversationsBody. # noqa: E501
@@ -48,7 +48,6 @@ class Externalv1UserStatus(object):
48
48
  'completed_signup': 'bool',
49
49
  'has_received_free_months': 'bool',
50
50
  'installed_grid': 'bool',
51
- 'should_auto_verify': 'bool',
52
51
  'verified': 'bool',
53
52
  'verified_at': 'datetime'
54
53
  }
@@ -61,12 +60,11 @@ class Externalv1UserStatus(object):
61
60
  'completed_signup': 'completedSignup',
62
61
  'has_received_free_months': 'hasReceivedFreeMonths',
63
62
  'installed_grid': 'installedGrid',
64
- 'should_auto_verify': 'shouldAutoVerify',
65
63
  'verified': 'verified',
66
64
  'verified_at': 'verifiedAt'
67
65
  }
68
66
 
69
- def __init__(self, acked_storage_violation: 'bool' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, should_auto_verify: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None): # noqa: E501
67
+ def __init__(self, acked_storage_violation: 'bool' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None): # noqa: E501
70
68
  """Externalv1UserStatus - a model defined in Swagger""" # noqa: E501
71
69
  self._acked_storage_violation = None
72
70
  self._auto_verify_reason = None
@@ -75,7 +73,6 @@ class Externalv1UserStatus(object):
75
73
  self._completed_signup = None
76
74
  self._has_received_free_months = None
77
75
  self._installed_grid = None
78
- self._should_auto_verify = None
79
76
  self._verified = None
80
77
  self._verified_at = None
81
78
  self.discriminator = None
@@ -93,8 +90,6 @@ class Externalv1UserStatus(object):
93
90
  self.has_received_free_months = has_received_free_months
94
91
  if installed_grid is not None:
95
92
  self.installed_grid = installed_grid
96
- if should_auto_verify is not None:
97
- self.should_auto_verify = should_auto_verify
98
93
  if verified is not None:
99
94
  self.verified = verified
100
95
  if verified_at is not None:
@@ -247,27 +242,6 @@ class Externalv1UserStatus(object):
247
242
 
248
243
  self._installed_grid = installed_grid
249
244
 
250
- @property
251
- def should_auto_verify(self) -> 'bool':
252
- """Gets the should_auto_verify of this Externalv1UserStatus. # noqa: E501
253
-
254
-
255
- :return: The should_auto_verify of this Externalv1UserStatus. # noqa: E501
256
- :rtype: bool
257
- """
258
- return self._should_auto_verify
259
-
260
- @should_auto_verify.setter
261
- def should_auto_verify(self, should_auto_verify: 'bool'):
262
- """Sets the should_auto_verify of this Externalv1UserStatus.
263
-
264
-
265
- :param should_auto_verify: The should_auto_verify of this Externalv1UserStatus. # noqa: E501
266
- :type: bool
267
- """
268
-
269
- self._should_auto_verify = should_auto_verify
270
-
271
245
  @property
272
246
  def verified(self) -> 'bool':
273
247
  """Gets the verified of this Externalv1UserStatus. # noqa: E501
@@ -41,6 +41,7 @@ class ProjectIdSchedulesBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'action_type': 'V1ScheduleActionType',
44
45
  'cron_expression': 'str',
45
46
  'display_name': 'str',
46
47
  'parent_resource_id': 'str',
@@ -49,6 +50,7 @@ class ProjectIdSchedulesBody(object):
49
50
  }
50
51
 
51
52
  attribute_map = {
53
+ 'action_type': 'actionType',
52
54
  'cron_expression': 'cronExpression',
53
55
  'display_name': 'displayName',
54
56
  'parent_resource_id': 'parentResourceId',
@@ -56,14 +58,17 @@ class ProjectIdSchedulesBody(object):
56
58
  'resource_type': 'resourceType'
57
59
  }
58
60
 
59
- def __init__(self, cron_expression: 'str' =None, display_name: 'str' =None, parent_resource_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None): # noqa: E501
61
+ def __init__(self, action_type: 'V1ScheduleActionType' =None, cron_expression: 'str' =None, display_name: 'str' =None, parent_resource_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None): # noqa: E501
60
62
  """ProjectIdSchedulesBody - a model defined in Swagger""" # noqa: E501
63
+ self._action_type = None
61
64
  self._cron_expression = None
62
65
  self._display_name = None
63
66
  self._parent_resource_id = None
64
67
  self._resource_id = None
65
68
  self._resource_type = None
66
69
  self.discriminator = None
70
+ if action_type is not None:
71
+ self.action_type = action_type
67
72
  if cron_expression is not None:
68
73
  self.cron_expression = cron_expression
69
74
  if display_name is not None:
@@ -75,6 +80,27 @@ class ProjectIdSchedulesBody(object):
75
80
  if resource_type is not None:
76
81
  self.resource_type = resource_type
77
82
 
83
+ @property
84
+ def action_type(self) -> 'V1ScheduleActionType':
85
+ """Gets the action_type of this ProjectIdSchedulesBody. # noqa: E501
86
+
87
+
88
+ :return: The action_type of this ProjectIdSchedulesBody. # noqa: E501
89
+ :rtype: V1ScheduleActionType
90
+ """
91
+ return self._action_type
92
+
93
+ @action_type.setter
94
+ def action_type(self, action_type: 'V1ScheduleActionType'):
95
+ """Sets the action_type of this ProjectIdSchedulesBody.
96
+
97
+
98
+ :param action_type: The action_type of this ProjectIdSchedulesBody. # noqa: E501
99
+ :type: V1ScheduleActionType
100
+ """
101
+
102
+ self._action_type = action_type
103
+
78
104
  @property
79
105
  def cron_expression(self) -> 'str':
80
106
  """Gets the cron_expression of this ProjectIdSchedulesBody. # noqa: E501
@@ -41,6 +41,7 @@ class SchedulesIdBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'action_type': 'V1ScheduleActionType',
44
45
  'created_at': 'datetime',
45
46
  'cron_expression': 'str',
46
47
  'display_name': 'str',
@@ -56,6 +57,7 @@ class SchedulesIdBody(object):
56
57
  }
57
58
 
58
59
  attribute_map = {
60
+ 'action_type': 'actionType',
59
61
  'created_at': 'createdAt',
60
62
  'cron_expression': 'cronExpression',
61
63
  'display_name': 'displayName',
@@ -70,8 +72,9 @@ class SchedulesIdBody(object):
70
72
  'user_id': 'userId'
71
73
  }
72
74
 
73
- def __init__(self, created_at: 'datetime' =None, cron_expression: 'str' =None, display_name: 'str' =None, name: 'str' =None, next: 'datetime' =None, parent_resource_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None, state: 'str' =None, total: 'int' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
75
+ def __init__(self, action_type: 'V1ScheduleActionType' =None, created_at: 'datetime' =None, cron_expression: 'str' =None, display_name: 'str' =None, name: 'str' =None, next: 'datetime' =None, parent_resource_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None, state: 'str' =None, total: 'int' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
74
76
  """SchedulesIdBody - a model defined in Swagger""" # noqa: E501
77
+ self._action_type = None
75
78
  self._created_at = None
76
79
  self._cron_expression = None
77
80
  self._display_name = None
@@ -85,6 +88,8 @@ class SchedulesIdBody(object):
85
88
  self._updated_at = None
86
89
  self._user_id = None
87
90
  self.discriminator = None
91
+ if action_type is not None:
92
+ self.action_type = action_type
88
93
  if created_at is not None:
89
94
  self.created_at = created_at
90
95
  if cron_expression is not None:
@@ -110,6 +115,27 @@ class SchedulesIdBody(object):
110
115
  if user_id is not None:
111
116
  self.user_id = user_id
112
117
 
118
+ @property
119
+ def action_type(self) -> 'V1ScheduleActionType':
120
+ """Gets the action_type of this SchedulesIdBody. # noqa: E501
121
+
122
+
123
+ :return: The action_type of this SchedulesIdBody. # noqa: E501
124
+ :rtype: V1ScheduleActionType
125
+ """
126
+ return self._action_type
127
+
128
+ @action_type.setter
129
+ def action_type(self, action_type: 'V1ScheduleActionType'):
130
+ """Sets the action_type of this SchedulesIdBody.
131
+
132
+
133
+ :param action_type: The action_type of this SchedulesIdBody. # noqa: E501
134
+ :type: V1ScheduleActionType
135
+ """
136
+
137
+ self._action_type = action_type
138
+
113
139
  @property
114
140
  def created_at(self) -> 'datetime':
115
141
  """Gets the created_at of this SchedulesIdBody. # noqa: E501
@@ -42,6 +42,7 @@ class V1BillingTier(object):
42
42
  TEAMS = "Teams"
43
43
  ENTERPRISE = "Enterprise"
44
44
  DISCOUNTED_PRO = "Discounted_Pro"
45
+ ACADEMIC = "Academic"
45
46
  """
46
47
  Attributes:
47
48
  swagger_types (dict): The key is attribute name
@@ -49,6 +49,7 @@ class V1ClusterCapacityReservation(object):
49
49
  'full_cloud_provider_reservation_string': 'str',
50
50
  'id': 'str',
51
51
  'in_use': 'str',
52
+ 'in_use_aggregate': 'str',
52
53
  'instance_type': 'str',
53
54
  'match_pattern': 'str',
54
55
  'num_instances': 'str',
@@ -67,6 +68,7 @@ class V1ClusterCapacityReservation(object):
67
68
  'full_cloud_provider_reservation_string': 'fullCloudProviderReservationString',
68
69
  'id': 'id',
69
70
  'in_use': 'inUse',
71
+ 'in_use_aggregate': 'inUseAggregate',
70
72
  'instance_type': 'instanceType',
71
73
  'match_pattern': 'matchPattern',
72
74
  'num_instances': 'numInstances',
@@ -76,7 +78,7 @@ class V1ClusterCapacityReservation(object):
76
78
  'zone': 'zone'
77
79
  }
78
80
 
79
- def __init__(self, aggregate_availability: 'str' =None, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, cluster_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, id: 'str' =None, in_use: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, num_instances: 'str' =None, project_id: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, zone: 'str' =None): # noqa: E501
81
+ def __init__(self, aggregate_availability: 'str' =None, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, cluster_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, id: 'str' =None, in_use: 'str' =None, in_use_aggregate: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, num_instances: 'str' =None, project_id: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, zone: 'str' =None): # noqa: E501
80
82
  """V1ClusterCapacityReservation - a model defined in Swagger""" # noqa: E501
81
83
  self._aggregate_availability = None
82
84
  self._capacity_reservation_type = None
@@ -86,6 +88,7 @@ class V1ClusterCapacityReservation(object):
86
88
  self._full_cloud_provider_reservation_string = None
87
89
  self._id = None
88
90
  self._in_use = None
91
+ self._in_use_aggregate = None
89
92
  self._instance_type = None
90
93
  self._match_pattern = None
91
94
  self._num_instances = None
@@ -110,6 +113,8 @@ class V1ClusterCapacityReservation(object):
110
113
  self.id = id
111
114
  if in_use is not None:
112
115
  self.in_use = in_use
116
+ if in_use_aggregate is not None:
117
+ self.in_use_aggregate = in_use_aggregate
113
118
  if instance_type is not None:
114
119
  self.instance_type = instance_type
115
120
  if match_pattern is not None:
@@ -293,6 +298,27 @@ class V1ClusterCapacityReservation(object):
293
298
 
294
299
  self._in_use = in_use
295
300
 
301
+ @property
302
+ def in_use_aggregate(self) -> 'str':
303
+ """Gets the in_use_aggregate of this V1ClusterCapacityReservation. # noqa: E501
304
+
305
+
306
+ :return: The in_use_aggregate of this V1ClusterCapacityReservation. # noqa: E501
307
+ :rtype: str
308
+ """
309
+ return self._in_use_aggregate
310
+
311
+ @in_use_aggregate.setter
312
+ def in_use_aggregate(self, in_use_aggregate: 'str'):
313
+ """Sets the in_use_aggregate of this V1ClusterCapacityReservation.
314
+
315
+
316
+ :param in_use_aggregate: The in_use_aggregate of this V1ClusterCapacityReservation. # noqa: E501
317
+ :type: str
318
+ """
319
+
320
+ self._in_use_aggregate = in_use_aggregate
321
+
296
322
  @property
297
323
  def instance_type(self) -> 'str':
298
324
  """Gets the instance_type of this V1ClusterCapacityReservation. # noqa: E501
@@ -46,6 +46,7 @@ class V1ConversationResponseChunk(object):
46
46
  'executable': 'bool',
47
47
  'id': 'str',
48
48
  'object': 'str',
49
+ 'stats': 'dict(str, str)',
49
50
  'throughput': 'float',
50
51
  'usage': 'V1TokenUsage'
51
52
  }
@@ -56,17 +57,19 @@ class V1ConversationResponseChunk(object):
56
57
  'executable': 'executable',
57
58
  'id': 'id',
58
59
  'object': 'object',
60
+ 'stats': 'stats',
59
61
  'throughput': 'throughput',
60
62
  'usage': 'usage'
61
63
  }
62
64
 
63
- def __init__(self, choices: 'list[V1ResponseChoice]' =None, conversation_id: 'str' =None, executable: 'bool' =None, id: 'str' =None, object: 'str' =None, throughput: 'float' =None, usage: 'V1TokenUsage' =None): # noqa: E501
65
+ def __init__(self, choices: 'list[V1ResponseChoice]' =None, conversation_id: 'str' =None, executable: 'bool' =None, id: 'str' =None, object: 'str' =None, stats: 'dict(str, str)' =None, throughput: 'float' =None, usage: 'V1TokenUsage' =None): # noqa: E501
64
66
  """V1ConversationResponseChunk - a model defined in Swagger""" # noqa: E501
65
67
  self._choices = None
66
68
  self._conversation_id = None
67
69
  self._executable = None
68
70
  self._id = None
69
71
  self._object = None
72
+ self._stats = None
70
73
  self._throughput = None
71
74
  self._usage = None
72
75
  self.discriminator = None
@@ -80,6 +83,8 @@ class V1ConversationResponseChunk(object):
80
83
  self.id = id
81
84
  if object is not None:
82
85
  self.object = object
86
+ if stats is not None:
87
+ self.stats = stats
83
88
  if throughput is not None:
84
89
  self.throughput = throughput
85
90
  if usage is not None:
@@ -190,6 +195,27 @@ class V1ConversationResponseChunk(object):
190
195
 
191
196
  self._object = object
192
197
 
198
+ @property
199
+ def stats(self) -> 'dict(str, str)':
200
+ """Gets the stats of this V1ConversationResponseChunk. # noqa: E501
201
+
202
+
203
+ :return: The stats of this V1ConversationResponseChunk. # noqa: E501
204
+ :rtype: dict(str, str)
205
+ """
206
+ return self._stats
207
+
208
+ @stats.setter
209
+ def stats(self, stats: 'dict(str, str)'):
210
+ """Sets the stats of this V1ConversationResponseChunk.
211
+
212
+
213
+ :param stats: The stats of this V1ConversationResponseChunk. # noqa: E501
214
+ :type: dict(str, str)
215
+ """
216
+
217
+ self._stats = stats
218
+
193
219
  @property
194
220
  def throughput(self) -> 'float':
195
221
  """Gets the throughput of this V1ConversationResponseChunk. # noqa: E501
@@ -0,0 +1,103 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1DataConnectionTier(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+
37
+ """
38
+ allowed enum values
39
+ """
40
+ SSD = "DATA_CONNECTION_TIER_SSD"
41
+ HDD = "DATA_CONNECTION_TIER_HDD"
42
+ """
43
+ Attributes:
44
+ swagger_types (dict): The key is attribute name
45
+ and the value is attribute type.
46
+ attribute_map (dict): The key is attribute name
47
+ and the value is json key in definition.
48
+ """
49
+ swagger_types = {
50
+ }
51
+
52
+ attribute_map = {
53
+ }
54
+
55
+ def __init__(self): # noqa: E501
56
+ """V1DataConnectionTier - a model defined in Swagger""" # noqa: E501
57
+ self.discriminator = None
58
+
59
+ def to_dict(self) -> dict:
60
+ """Returns the model properties as a dict"""
61
+ result = {}
62
+
63
+ for attr, _ in six.iteritems(self.swagger_types):
64
+ value = getattr(self, attr)
65
+ if isinstance(value, list):
66
+ result[attr] = list(map(
67
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
68
+ value
69
+ ))
70
+ elif hasattr(value, "to_dict"):
71
+ result[attr] = value.to_dict()
72
+ elif isinstance(value, dict):
73
+ result[attr] = dict(map(
74
+ lambda item: (item[0], item[1].to_dict())
75
+ if hasattr(item[1], "to_dict") else item,
76
+ value.items()
77
+ ))
78
+ else:
79
+ result[attr] = value
80
+ if issubclass(V1DataConnectionTier, dict):
81
+ for key, value in self.items():
82
+ result[key] = value
83
+
84
+ return result
85
+
86
+ def to_str(self) -> str:
87
+ """Returns the string representation of the model"""
88
+ return pprint.pformat(self.to_dict())
89
+
90
+ def __repr__(self) -> str:
91
+ """For `print` and `pprint`"""
92
+ return self.to_str()
93
+
94
+ def __eq__(self, other: 'V1DataConnectionTier') -> bool:
95
+ """Returns true if both objects are equal"""
96
+ if not isinstance(other, V1DataConnectionTier):
97
+ return False
98
+
99
+ return self.__dict__ == other.__dict__
100
+
101
+ def __ne__(self, other: 'V1DataConnectionTier') -> bool:
102
+ """Returns true if both objects are not equal"""
103
+ return not self == other
@@ -44,22 +44,25 @@ class V1FilestoreDataConnection(object):
44
44
  'capacity_gb': 'str',
45
45
  'mount_ip': 'str',
46
46
  'region': 'str',
47
- 'source': 'str'
47
+ 'source': 'str',
48
+ 'tier': 'V1DataConnectionTier'
48
49
  }
49
50
 
50
51
  attribute_map = {
51
52
  'capacity_gb': 'capacityGb',
52
53
  'mount_ip': 'mountIp',
53
54
  'region': 'region',
54
- 'source': 'source'
55
+ 'source': 'source',
56
+ 'tier': 'tier'
55
57
  }
56
58
 
57
- def __init__(self, capacity_gb: 'str' =None, mount_ip: 'str' =None, region: 'str' =None, source: 'str' =None): # noqa: E501
59
+ def __init__(self, capacity_gb: 'str' =None, mount_ip: 'str' =None, region: 'str' =None, source: 'str' =None, tier: 'V1DataConnectionTier' =None): # noqa: E501
58
60
  """V1FilestoreDataConnection - a model defined in Swagger""" # noqa: E501
59
61
  self._capacity_gb = None
60
62
  self._mount_ip = None
61
63
  self._region = None
62
64
  self._source = None
65
+ self._tier = None
63
66
  self.discriminator = None
64
67
  if capacity_gb is not None:
65
68
  self.capacity_gb = capacity_gb
@@ -69,6 +72,8 @@ class V1FilestoreDataConnection(object):
69
72
  self.region = region
70
73
  if source is not None:
71
74
  self.source = source
75
+ if tier is not None:
76
+ self.tier = tier
72
77
 
73
78
  @property
74
79
  def capacity_gb(self) -> 'str':
@@ -154,6 +159,27 @@ class V1FilestoreDataConnection(object):
154
159
 
155
160
  self._source = source
156
161
 
162
+ @property
163
+ def tier(self) -> 'V1DataConnectionTier':
164
+ """Gets the tier of this V1FilestoreDataConnection. # noqa: E501
165
+
166
+
167
+ :return: The tier of this V1FilestoreDataConnection. # noqa: E501
168
+ :rtype: V1DataConnectionTier
169
+ """
170
+ return self._tier
171
+
172
+ @tier.setter
173
+ def tier(self, tier: 'V1DataConnectionTier'):
174
+ """Sets the tier of this V1FilestoreDataConnection.
175
+
176
+
177
+ :param tier: The tier of this V1FilestoreDataConnection. # noqa: E501
178
+ :type: V1DataConnectionTier
179
+ """
180
+
181
+ self._tier = tier
182
+
157
183
  def to_dict(self) -> dict:
158
184
  """Returns the model properties as a dict"""
159
185
  result = {}