lightning-sdk 0.2.23__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 (70) 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/llm_api.py +5 -11
  5. lightning_sdk/api/pipeline_api.py +31 -11
  6. lightning_sdk/api/studio_api.py +4 -0
  7. lightning_sdk/base_studio.py +22 -6
  8. lightning_sdk/deployment/deployment.py +17 -7
  9. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  11. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  12. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  13. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  14. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  15. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/models/__init__.py +11 -0
  17. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  20. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  21. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  24. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  40. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  41. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  43. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  46. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +185 -29
  58. lightning_sdk/llm/llm.py +113 -115
  59. lightning_sdk/llm/public_assistants.json +8 -0
  60. lightning_sdk/pipeline/pipeline.py +17 -2
  61. lightning_sdk/pipeline/printer.py +11 -10
  62. lightning_sdk/pipeline/steps.py +4 -1
  63. lightning_sdk/pipeline/utils.py +29 -4
  64. lightning_sdk/studio.py +3 -0
  65. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  66. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +70 -57
  67. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  68. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  69. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  70. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -46,7 +46,8 @@ class V1ConversationResponseChunk(object):
46
46
  'executable': 'bool',
47
47
  'id': 'str',
48
48
  'object': 'str',
49
- 'throughput': 'float'
49
+ 'throughput': 'float',
50
+ 'usage': 'V1TokenUsage'
50
51
  }
51
52
 
52
53
  attribute_map = {
@@ -55,10 +56,11 @@ class V1ConversationResponseChunk(object):
55
56
  'executable': 'executable',
56
57
  'id': 'id',
57
58
  'object': 'object',
58
- 'throughput': 'throughput'
59
+ 'throughput': 'throughput',
60
+ 'usage': 'usage'
59
61
  }
60
62
 
61
- def __init__(self, choices: 'list[V1ResponseChoice]' =None, conversation_id: 'str' =None, executable: 'bool' =None, id: 'str' =None, object: 'str' =None, throughput: 'float' =None): # noqa: E501
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
62
64
  """V1ConversationResponseChunk - a model defined in Swagger""" # noqa: E501
63
65
  self._choices = None
64
66
  self._conversation_id = None
@@ -66,6 +68,7 @@ class V1ConversationResponseChunk(object):
66
68
  self._id = None
67
69
  self._object = None
68
70
  self._throughput = None
71
+ self._usage = None
69
72
  self.discriminator = None
70
73
  if choices is not None:
71
74
  self.choices = choices
@@ -79,6 +82,8 @@ class V1ConversationResponseChunk(object):
79
82
  self.object = object
80
83
  if throughput is not None:
81
84
  self.throughput = throughput
85
+ if usage is not None:
86
+ self.usage = usage
82
87
 
83
88
  @property
84
89
  def choices(self) -> 'list[V1ResponseChoice]':
@@ -206,6 +211,27 @@ class V1ConversationResponseChunk(object):
206
211
 
207
212
  self._throughput = throughput
208
213
 
214
+ @property
215
+ def usage(self) -> 'V1TokenUsage':
216
+ """Gets the usage of this V1ConversationResponseChunk. # noqa: E501
217
+
218
+
219
+ :return: The usage of this V1ConversationResponseChunk. # noqa: E501
220
+ :rtype: V1TokenUsage
221
+ """
222
+ return self._usage
223
+
224
+ @usage.setter
225
+ def usage(self, usage: 'V1TokenUsage'):
226
+ """Sets the usage of this V1ConversationResponseChunk.
227
+
228
+
229
+ :param usage: The usage of this V1ConversationResponseChunk. # noqa: E501
230
+ :type: V1TokenUsage
231
+ """
232
+
233
+ self._usage = usage
234
+
209
235
  def to_dict(self) -> dict:
210
236
  """Returns the model properties as a dict"""
211
237
  result = {}
@@ -51,7 +51,8 @@ class V1CreateCloudSpaceEnvironmentTemplateRequest(object):
51
51
  'name': 'str',
52
52
  'org_id': 'str',
53
53
  'plugins': 'list[str]',
54
- 'setup_script_text': 'str'
54
+ 'setup_script_text': 'str',
55
+ 'specialized_view': 'V1CloudSpaceSpecializedView'
55
56
  }
56
57
 
57
58
  attribute_map = {
@@ -65,10 +66,11 @@ class V1CreateCloudSpaceEnvironmentTemplateRequest(object):
65
66
  'name': 'name',
66
67
  'org_id': 'orgId',
67
68
  'plugins': 'plugins',
68
- 'setup_script_text': 'setupScriptText'
69
+ 'setup_script_text': 'setupScriptText',
70
+ 'specialized_view': 'specializedView'
69
71
  }
70
72
 
71
- def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, description: 'str' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, icon: 'str' =None, initial_setup_script_text: 'str' =None, machine_image_version: 'str' =None, name: 'str' =None, org_id: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None): # noqa: E501
73
+ def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, description: 'str' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, icon: 'str' =None, initial_setup_script_text: 'str' =None, machine_image_version: 'str' =None, name: 'str' =None, org_id: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None, specialized_view: 'V1CloudSpaceSpecializedView' =None): # noqa: E501
72
74
  """V1CreateCloudSpaceEnvironmentTemplateRequest - a model defined in Swagger""" # noqa: E501
73
75
  self._allowed_machines = None
74
76
  self._default_machine = None
@@ -81,6 +83,7 @@ class V1CreateCloudSpaceEnvironmentTemplateRequest(object):
81
83
  self._org_id = None
82
84
  self._plugins = None
83
85
  self._setup_script_text = None
86
+ self._specialized_view = None
84
87
  self.discriminator = None
85
88
  if allowed_machines is not None:
86
89
  self.allowed_machines = allowed_machines
@@ -104,6 +107,8 @@ class V1CreateCloudSpaceEnvironmentTemplateRequest(object):
104
107
  self.plugins = plugins
105
108
  if setup_script_text is not None:
106
109
  self.setup_script_text = setup_script_text
110
+ if specialized_view is not None:
111
+ self.specialized_view = specialized_view
107
112
 
108
113
  @property
109
114
  def allowed_machines(self) -> 'list[str]':
@@ -336,6 +341,27 @@ class V1CreateCloudSpaceEnvironmentTemplateRequest(object):
336
341
 
337
342
  self._setup_script_text = setup_script_text
338
343
 
344
+ @property
345
+ def specialized_view(self) -> 'V1CloudSpaceSpecializedView':
346
+ """Gets the specialized_view of this V1CreateCloudSpaceEnvironmentTemplateRequest. # noqa: E501
347
+
348
+
349
+ :return: The specialized_view of this V1CreateCloudSpaceEnvironmentTemplateRequest. # noqa: E501
350
+ :rtype: V1CloudSpaceSpecializedView
351
+ """
352
+ return self._specialized_view
353
+
354
+ @specialized_view.setter
355
+ def specialized_view(self, specialized_view: 'V1CloudSpaceSpecializedView'):
356
+ """Sets the specialized_view of this V1CreateCloudSpaceEnvironmentTemplateRequest.
357
+
358
+
359
+ :param specialized_view: The specialized_view of this V1CreateCloudSpaceEnvironmentTemplateRequest. # noqa: E501
360
+ :type: V1CloudSpaceSpecializedView
361
+ """
362
+
363
+ self._specialized_view = specialized_view
364
+
339
365
  def to_dict(self) -> dict:
340
366
  """Returns the model properties as a dict"""
341
367
  result = {}
@@ -41,9 +41,12 @@ class V1CreateOrganizationRequest(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'allow_credits_auto_replenish': 'bool',
44
45
  'allow_marketplace': 'bool',
45
46
  'auto_invite_by_domain': 'bool',
46
47
  'auto_join_domains': 'list[str]',
48
+ 'auto_replenish_amount': 'float',
49
+ 'auto_replenish_threshold': 'float',
47
50
  'description': 'str',
48
51
  'display_name': 'str',
49
52
  'domain': 'str',
@@ -57,9 +60,12 @@ class V1CreateOrganizationRequest(object):
57
60
  }
58
61
 
59
62
  attribute_map = {
63
+ 'allow_credits_auto_replenish': 'allowCreditsAutoReplenish',
60
64
  'allow_marketplace': 'allowMarketplace',
61
65
  'auto_invite_by_domain': 'autoInviteByDomain',
62
66
  'auto_join_domains': 'autoJoinDomains',
67
+ 'auto_replenish_amount': 'autoReplenishAmount',
68
+ 'auto_replenish_threshold': 'autoReplenishThreshold',
63
69
  'description': 'description',
64
70
  'display_name': 'displayName',
65
71
  'domain': 'domain',
@@ -72,11 +78,14 @@ class V1CreateOrganizationRequest(object):
72
78
  'type': 'type'
73
79
  }
74
80
 
75
- def __init__(self, allow_marketplace: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, start_studios_on_spot_instance: 'bool' =None, twitter_username: 'str' =None, type: 'str' =None): # noqa: E501
81
+ def __init__(self, allow_credits_auto_replenish: 'bool' =None, allow_marketplace: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, start_studios_on_spot_instance: 'bool' =None, twitter_username: 'str' =None, type: 'str' =None): # noqa: E501
76
82
  """V1CreateOrganizationRequest - a model defined in Swagger""" # noqa: E501
83
+ self._allow_credits_auto_replenish = None
77
84
  self._allow_marketplace = None
78
85
  self._auto_invite_by_domain = None
79
86
  self._auto_join_domains = None
87
+ self._auto_replenish_amount = None
88
+ self._auto_replenish_threshold = None
80
89
  self._description = None
81
90
  self._display_name = None
82
91
  self._domain = None
@@ -88,12 +97,18 @@ class V1CreateOrganizationRequest(object):
88
97
  self._twitter_username = None
89
98
  self._type = None
90
99
  self.discriminator = None
100
+ if allow_credits_auto_replenish is not None:
101
+ self.allow_credits_auto_replenish = allow_credits_auto_replenish
91
102
  if allow_marketplace is not None:
92
103
  self.allow_marketplace = allow_marketplace
93
104
  if auto_invite_by_domain is not None:
94
105
  self.auto_invite_by_domain = auto_invite_by_domain
95
106
  if auto_join_domains is not None:
96
107
  self.auto_join_domains = auto_join_domains
108
+ if auto_replenish_amount is not None:
109
+ self.auto_replenish_amount = auto_replenish_amount
110
+ if auto_replenish_threshold is not None:
111
+ self.auto_replenish_threshold = auto_replenish_threshold
97
112
  if description is not None:
98
113
  self.description = description
99
114
  if display_name is not None:
@@ -115,6 +130,27 @@ class V1CreateOrganizationRequest(object):
115
130
  if type is not None:
116
131
  self.type = type
117
132
 
133
+ @property
134
+ def allow_credits_auto_replenish(self) -> 'bool':
135
+ """Gets the allow_credits_auto_replenish of this V1CreateOrganizationRequest. # noqa: E501
136
+
137
+
138
+ :return: The allow_credits_auto_replenish of this V1CreateOrganizationRequest. # noqa: E501
139
+ :rtype: bool
140
+ """
141
+ return self._allow_credits_auto_replenish
142
+
143
+ @allow_credits_auto_replenish.setter
144
+ def allow_credits_auto_replenish(self, allow_credits_auto_replenish: 'bool'):
145
+ """Sets the allow_credits_auto_replenish of this V1CreateOrganizationRequest.
146
+
147
+
148
+ :param allow_credits_auto_replenish: The allow_credits_auto_replenish of this V1CreateOrganizationRequest. # noqa: E501
149
+ :type: bool
150
+ """
151
+
152
+ self._allow_credits_auto_replenish = allow_credits_auto_replenish
153
+
118
154
  @property
119
155
  def allow_marketplace(self) -> 'bool':
120
156
  """Gets the allow_marketplace of this V1CreateOrganizationRequest. # noqa: E501
@@ -178,6 +214,48 @@ class V1CreateOrganizationRequest(object):
178
214
 
179
215
  self._auto_join_domains = auto_join_domains
180
216
 
217
+ @property
218
+ def auto_replenish_amount(self) -> 'float':
219
+ """Gets the auto_replenish_amount of this V1CreateOrganizationRequest. # noqa: E501
220
+
221
+
222
+ :return: The auto_replenish_amount of this V1CreateOrganizationRequest. # noqa: E501
223
+ :rtype: float
224
+ """
225
+ return self._auto_replenish_amount
226
+
227
+ @auto_replenish_amount.setter
228
+ def auto_replenish_amount(self, auto_replenish_amount: 'float'):
229
+ """Sets the auto_replenish_amount of this V1CreateOrganizationRequest.
230
+
231
+
232
+ :param auto_replenish_amount: The auto_replenish_amount of this V1CreateOrganizationRequest. # noqa: E501
233
+ :type: float
234
+ """
235
+
236
+ self._auto_replenish_amount = auto_replenish_amount
237
+
238
+ @property
239
+ def auto_replenish_threshold(self) -> 'float':
240
+ """Gets the auto_replenish_threshold of this V1CreateOrganizationRequest. # noqa: E501
241
+
242
+
243
+ :return: The auto_replenish_threshold of this V1CreateOrganizationRequest. # noqa: E501
244
+ :rtype: float
245
+ """
246
+ return self._auto_replenish_threshold
247
+
248
+ @auto_replenish_threshold.setter
249
+ def auto_replenish_threshold(self, auto_replenish_threshold: 'float'):
250
+ """Sets the auto_replenish_threshold of this V1CreateOrganizationRequest.
251
+
252
+
253
+ :param auto_replenish_threshold: The auto_replenish_threshold of this V1CreateOrganizationRequest. # noqa: E501
254
+ :type: float
255
+ """
256
+
257
+ self._auto_replenish_threshold = auto_replenish_threshold
258
+
181
259
  @property
182
260
  def description(self) -> 'str':
183
261
  """Gets the description of this V1CreateOrganizationRequest. # noqa: E501
@@ -41,12 +41,13 @@ class V1DeploymentStatus(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'deleting_replicas': 'str',
45
- 'failing_replicas': 'str',
44
+ 'deleting_replicas': 'int',
45
+ 'failing_replicas': 'int',
46
46
  'first_job_state_current_release': 'str',
47
47
  'message': 'str',
48
- 'pending_replicas': 'str',
49
- 'ready_replicas': 'str',
48
+ 'pending_replicas': 'int',
49
+ 'queued_replicas': 'int',
50
+ 'ready_replicas': 'int',
50
51
  'urls': 'list[str]'
51
52
  }
52
53
 
@@ -56,17 +57,19 @@ class V1DeploymentStatus(object):
56
57
  'first_job_state_current_release': 'firstJobStateCurrentRelease',
57
58
  'message': 'message',
58
59
  'pending_replicas': 'pendingReplicas',
60
+ 'queued_replicas': 'queuedReplicas',
59
61
  'ready_replicas': 'readyReplicas',
60
62
  'urls': 'urls'
61
63
  }
62
64
 
63
- def __init__(self, deleting_replicas: 'str' =None, failing_replicas: 'str' =None, first_job_state_current_release: 'str' =None, message: 'str' =None, pending_replicas: 'str' =None, ready_replicas: 'str' =None, urls: 'list[str]' =None): # noqa: E501
65
+ def __init__(self, deleting_replicas: 'int' =None, failing_replicas: 'int' =None, first_job_state_current_release: 'str' =None, message: 'str' =None, pending_replicas: 'int' =None, queued_replicas: 'int' =None, ready_replicas: 'int' =None, urls: 'list[str]' =None): # noqa: E501
64
66
  """V1DeploymentStatus - a model defined in Swagger""" # noqa: E501
65
67
  self._deleting_replicas = None
66
68
  self._failing_replicas = None
67
69
  self._first_job_state_current_release = None
68
70
  self._message = None
69
71
  self._pending_replicas = None
72
+ self._queued_replicas = None
70
73
  self._ready_replicas = None
71
74
  self._urls = None
72
75
  self.discriminator = None
@@ -80,49 +83,51 @@ class V1DeploymentStatus(object):
80
83
  self.message = message
81
84
  if pending_replicas is not None:
82
85
  self.pending_replicas = pending_replicas
86
+ if queued_replicas is not None:
87
+ self.queued_replicas = queued_replicas
83
88
  if ready_replicas is not None:
84
89
  self.ready_replicas = ready_replicas
85
90
  if urls is not None:
86
91
  self.urls = urls
87
92
 
88
93
  @property
89
- def deleting_replicas(self) -> 'str':
94
+ def deleting_replicas(self) -> 'int':
90
95
  """Gets the deleting_replicas of this V1DeploymentStatus. # noqa: E501
91
96
 
92
97
 
93
98
  :return: The deleting_replicas of this V1DeploymentStatus. # noqa: E501
94
- :rtype: str
99
+ :rtype: int
95
100
  """
96
101
  return self._deleting_replicas
97
102
 
98
103
  @deleting_replicas.setter
99
- def deleting_replicas(self, deleting_replicas: 'str'):
104
+ def deleting_replicas(self, deleting_replicas: 'int'):
100
105
  """Sets the deleting_replicas of this V1DeploymentStatus.
101
106
 
102
107
 
103
108
  :param deleting_replicas: The deleting_replicas of this V1DeploymentStatus. # noqa: E501
104
- :type: str
109
+ :type: int
105
110
  """
106
111
 
107
112
  self._deleting_replicas = deleting_replicas
108
113
 
109
114
  @property
110
- def failing_replicas(self) -> 'str':
115
+ def failing_replicas(self) -> 'int':
111
116
  """Gets the failing_replicas of this V1DeploymentStatus. # noqa: E501
112
117
 
113
118
 
114
119
  :return: The failing_replicas of this V1DeploymentStatus. # noqa: E501
115
- :rtype: str
120
+ :rtype: int
116
121
  """
117
122
  return self._failing_replicas
118
123
 
119
124
  @failing_replicas.setter
120
- def failing_replicas(self, failing_replicas: 'str'):
125
+ def failing_replicas(self, failing_replicas: 'int'):
121
126
  """Sets the failing_replicas of this V1DeploymentStatus.
122
127
 
123
128
 
124
129
  :param failing_replicas: The failing_replicas of this V1DeploymentStatus. # noqa: E501
125
- :type: str
130
+ :type: int
126
131
  """
127
132
 
128
133
  self._failing_replicas = failing_replicas
@@ -170,43 +175,64 @@ class V1DeploymentStatus(object):
170
175
  self._message = message
171
176
 
172
177
  @property
173
- def pending_replicas(self) -> 'str':
178
+ def pending_replicas(self) -> 'int':
174
179
  """Gets the pending_replicas of this V1DeploymentStatus. # noqa: E501
175
180
 
176
181
 
177
182
  :return: The pending_replicas of this V1DeploymentStatus. # noqa: E501
178
- :rtype: str
183
+ :rtype: int
179
184
  """
180
185
  return self._pending_replicas
181
186
 
182
187
  @pending_replicas.setter
183
- def pending_replicas(self, pending_replicas: 'str'):
188
+ def pending_replicas(self, pending_replicas: 'int'):
184
189
  """Sets the pending_replicas of this V1DeploymentStatus.
185
190
 
186
191
 
187
192
  :param pending_replicas: The pending_replicas of this V1DeploymentStatus. # noqa: E501
188
- :type: str
193
+ :type: int
189
194
  """
190
195
 
191
196
  self._pending_replicas = pending_replicas
192
197
 
193
198
  @property
194
- def ready_replicas(self) -> 'str':
199
+ def queued_replicas(self) -> 'int':
200
+ """Gets the queued_replicas of this V1DeploymentStatus. # noqa: E501
201
+
202
+
203
+ :return: The queued_replicas of this V1DeploymentStatus. # noqa: E501
204
+ :rtype: int
205
+ """
206
+ return self._queued_replicas
207
+
208
+ @queued_replicas.setter
209
+ def queued_replicas(self, queued_replicas: 'int'):
210
+ """Sets the queued_replicas of this V1DeploymentStatus.
211
+
212
+
213
+ :param queued_replicas: The queued_replicas of this V1DeploymentStatus. # noqa: E501
214
+ :type: int
215
+ """
216
+
217
+ self._queued_replicas = queued_replicas
218
+
219
+ @property
220
+ def ready_replicas(self) -> 'int':
195
221
  """Gets the ready_replicas of this V1DeploymentStatus. # noqa: E501
196
222
 
197
223
 
198
224
  :return: The ready_replicas of this V1DeploymentStatus. # noqa: E501
199
- :rtype: str
225
+ :rtype: int
200
226
  """
201
227
  return self._ready_replicas
202
228
 
203
229
  @ready_replicas.setter
204
- def ready_replicas(self, ready_replicas: 'str'):
230
+ def ready_replicas(self, ready_replicas: 'int'):
205
231
  """Sets the ready_replicas of this V1DeploymentStatus.
206
232
 
207
233
 
208
234
  :param ready_replicas: The ready_replicas of this V1DeploymentStatus. # noqa: E501
209
- :type: str
235
+ :type: int
210
236
  """
211
237
 
212
238
  self._ready_replicas = ready_replicas
@@ -52,6 +52,7 @@ class V1ExternalClusterSpec(object):
52
52
  'domain': 'str',
53
53
  'driver': 'V1CloudProvider',
54
54
  'google_cloud_v1': 'V1GoogleCloudDirectV1',
55
+ 'kubernetes_v1': 'V1KubernetesDirectV1',
55
56
  'lambda_labs_v1': 'V1LambdaLabsDirectV1',
56
57
  'locked_zones': 'list[str]',
57
58
  'monitor_deletion_disabled': 'bool',
@@ -83,6 +84,7 @@ class V1ExternalClusterSpec(object):
83
84
  'domain': 'domain',
84
85
  'driver': 'driver',
85
86
  'google_cloud_v1': 'googleCloudV1',
87
+ 'kubernetes_v1': 'kubernetesV1',
86
88
  'lambda_labs_v1': 'lambdaLabsV1',
87
89
  'locked_zones': 'lockedZones',
88
90
  'monitor_deletion_disabled': 'monitorDeletionDisabled',
@@ -102,7 +104,7 @@ class V1ExternalClusterSpec(object):
102
104
  'vultr_v1': 'vultrV1'
103
105
  }
104
106
 
105
- def __init__(self, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, locked_zones: 'list[str]' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None): # noqa: E501
107
+ def __init__(self, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, locked_zones: 'list[str]' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None): # noqa: E501
106
108
  """V1ExternalClusterSpec - a model defined in Swagger""" # noqa: E501
107
109
  self._auth_token = None
108
110
  self._available_accelerators = None
@@ -115,6 +117,7 @@ class V1ExternalClusterSpec(object):
115
117
  self._domain = None
116
118
  self._driver = None
117
119
  self._google_cloud_v1 = None
120
+ self._kubernetes_v1 = None
118
121
  self._lambda_labs_v1 = None
119
122
  self._locked_zones = None
120
123
  self._monitor_deletion_disabled = None
@@ -155,6 +158,8 @@ class V1ExternalClusterSpec(object):
155
158
  self.driver = driver
156
159
  if google_cloud_v1 is not None:
157
160
  self.google_cloud_v1 = google_cloud_v1
161
+ if kubernetes_v1 is not None:
162
+ self.kubernetes_v1 = kubernetes_v1
158
163
  if lambda_labs_v1 is not None:
159
164
  self.lambda_labs_v1 = lambda_labs_v1
160
165
  if locked_zones is not None:
@@ -423,6 +428,27 @@ class V1ExternalClusterSpec(object):
423
428
 
424
429
  self._google_cloud_v1 = google_cloud_v1
425
430
 
431
+ @property
432
+ def kubernetes_v1(self) -> 'V1KubernetesDirectV1':
433
+ """Gets the kubernetes_v1 of this V1ExternalClusterSpec. # noqa: E501
434
+
435
+
436
+ :return: The kubernetes_v1 of this V1ExternalClusterSpec. # noqa: E501
437
+ :rtype: V1KubernetesDirectV1
438
+ """
439
+ return self._kubernetes_v1
440
+
441
+ @kubernetes_v1.setter
442
+ def kubernetes_v1(self, kubernetes_v1: 'V1KubernetesDirectV1'):
443
+ """Sets the kubernetes_v1 of this V1ExternalClusterSpec.
444
+
445
+
446
+ :param kubernetes_v1: The kubernetes_v1 of this V1ExternalClusterSpec. # noqa: E501
447
+ :type: V1KubernetesDirectV1
448
+ """
449
+
450
+ self._kubernetes_v1 = kubernetes_v1
451
+
426
452
  @property
427
453
  def lambda_labs_v1(self) -> 'V1LambdaLabsDirectV1':
428
454
  """Gets the lambda_labs_v1 of this V1ExternalClusterSpec. # noqa: E501
@@ -41,6 +41,8 @@ class V1GetJobStatsResponse(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'queued_deployments': 'int',
45
+ 'queued_jobs': 'int',
44
46
  'running_deployments': 'int',
45
47
  'running_jobs': 'int',
46
48
  'running_pipelines': 'int',
@@ -48,19 +50,27 @@ class V1GetJobStatsResponse(object):
48
50
  }
49
51
 
50
52
  attribute_map = {
53
+ 'queued_deployments': 'queuedDeployments',
54
+ 'queued_jobs': 'queuedJobs',
51
55
  'running_deployments': 'runningDeployments',
52
56
  'running_jobs': 'runningJobs',
53
57
  'running_pipelines': 'runningPipelines',
54
58
  'total_jobs': 'totalJobs'
55
59
  }
56
60
 
57
- def __init__(self, running_deployments: 'int' =None, running_jobs: 'int' =None, running_pipelines: 'int' =None, total_jobs: 'int' =None): # noqa: E501
61
+ def __init__(self, queued_deployments: 'int' =None, queued_jobs: 'int' =None, running_deployments: 'int' =None, running_jobs: 'int' =None, running_pipelines: 'int' =None, total_jobs: 'int' =None): # noqa: E501
58
62
  """V1GetJobStatsResponse - a model defined in Swagger""" # noqa: E501
63
+ self._queued_deployments = None
64
+ self._queued_jobs = None
59
65
  self._running_deployments = None
60
66
  self._running_jobs = None
61
67
  self._running_pipelines = None
62
68
  self._total_jobs = None
63
69
  self.discriminator = None
70
+ if queued_deployments is not None:
71
+ self.queued_deployments = queued_deployments
72
+ if queued_jobs is not None:
73
+ self.queued_jobs = queued_jobs
64
74
  if running_deployments is not None:
65
75
  self.running_deployments = running_deployments
66
76
  if running_jobs is not None:
@@ -70,6 +80,48 @@ class V1GetJobStatsResponse(object):
70
80
  if total_jobs is not None:
71
81
  self.total_jobs = total_jobs
72
82
 
83
+ @property
84
+ def queued_deployments(self) -> 'int':
85
+ """Gets the queued_deployments of this V1GetJobStatsResponse. # noqa: E501
86
+
87
+
88
+ :return: The queued_deployments of this V1GetJobStatsResponse. # noqa: E501
89
+ :rtype: int
90
+ """
91
+ return self._queued_deployments
92
+
93
+ @queued_deployments.setter
94
+ def queued_deployments(self, queued_deployments: 'int'):
95
+ """Sets the queued_deployments of this V1GetJobStatsResponse.
96
+
97
+
98
+ :param queued_deployments: The queued_deployments of this V1GetJobStatsResponse. # noqa: E501
99
+ :type: int
100
+ """
101
+
102
+ self._queued_deployments = queued_deployments
103
+
104
+ @property
105
+ def queued_jobs(self) -> 'int':
106
+ """Gets the queued_jobs of this V1GetJobStatsResponse. # noqa: E501
107
+
108
+
109
+ :return: The queued_jobs of this V1GetJobStatsResponse. # noqa: E501
110
+ :rtype: int
111
+ """
112
+ return self._queued_jobs
113
+
114
+ @queued_jobs.setter
115
+ def queued_jobs(self, queued_jobs: 'int'):
116
+ """Sets the queued_jobs of this V1GetJobStatsResponse.
117
+
118
+
119
+ :param queued_jobs: The queued_jobs of this V1GetJobStatsResponse. # noqa: E501
120
+ :type: int
121
+ """
122
+
123
+ self._queued_jobs = queued_jobs
124
+
73
125
  @property
74
126
  def running_deployments(self) -> 'int':
75
127
  """Gets the running_deployments of this V1GetJobStatsResponse. # noqa: E501
@@ -47,7 +47,6 @@ class V1InstanceOverprovisioningSpec(object):
47
47
  'maximum_count': 'int',
48
48
  'minimum_count': 'int',
49
49
  'region': 'str',
50
- 'resource_type': 'str',
51
50
  'volume_size': 'int'
52
51
  }
53
52
 
@@ -58,11 +57,10 @@ class V1InstanceOverprovisioningSpec(object):
58
57
  'maximum_count': 'maximumCount',
59
58
  'minimum_count': 'minimumCount',
60
59
  'region': 'region',
61
- 'resource_type': 'resourceType',
62
60
  'volume_size': 'volumeSize'
63
61
  }
64
62
 
65
- def __init__(self, instance_type: 'str' =None, is_spot: 'bool' =None, machine_image_version: 'str' =None, maximum_count: 'int' =None, minimum_count: 'int' =None, region: 'str' =None, resource_type: 'str' =None, volume_size: 'int' =None): # noqa: E501
63
+ def __init__(self, instance_type: 'str' =None, is_spot: 'bool' =None, machine_image_version: 'str' =None, maximum_count: 'int' =None, minimum_count: 'int' =None, region: 'str' =None, volume_size: 'int' =None): # noqa: E501
66
64
  """V1InstanceOverprovisioningSpec - a model defined in Swagger""" # noqa: E501
67
65
  self._instance_type = None
68
66
  self._is_spot = None
@@ -70,7 +68,6 @@ class V1InstanceOverprovisioningSpec(object):
70
68
  self._maximum_count = None
71
69
  self._minimum_count = None
72
70
  self._region = None
73
- self._resource_type = None
74
71
  self._volume_size = None
75
72
  self.discriminator = None
76
73
  if instance_type is not None:
@@ -85,8 +82,6 @@ class V1InstanceOverprovisioningSpec(object):
85
82
  self.minimum_count = minimum_count
86
83
  if region is not None:
87
84
  self.region = region
88
- if resource_type is not None:
89
- self.resource_type = resource_type
90
85
  if volume_size is not None:
91
86
  self.volume_size = volume_size
92
87
 
@@ -218,27 +213,6 @@ class V1InstanceOverprovisioningSpec(object):
218
213
 
219
214
  self._region = region
220
215
 
221
- @property
222
- def resource_type(self) -> 'str':
223
- """Gets the resource_type of this V1InstanceOverprovisioningSpec. # noqa: E501
224
-
225
-
226
- :return: The resource_type of this V1InstanceOverprovisioningSpec. # noqa: E501
227
- :rtype: str
228
- """
229
- return self._resource_type
230
-
231
- @resource_type.setter
232
- def resource_type(self, resource_type: 'str'):
233
- """Sets the resource_type of this V1InstanceOverprovisioningSpec.
234
-
235
-
236
- :param resource_type: The resource_type of this V1InstanceOverprovisioningSpec. # noqa: E501
237
- :type: str
238
- """
239
-
240
- self._resource_type = resource_type
241
-
242
216
  @property
243
217
  def volume_size(self) -> 'int':
244
218
  """Gets the volume_size of this V1InstanceOverprovisioningSpec. # noqa: E501