lightning-sdk 0.2.22__py3-none-any.whl → 0.2.24rc0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (101) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +9 -2
  3. lightning_sdk/api/deployment_api.py +9 -9
  4. lightning_sdk/api/license_api.py +2 -2
  5. lightning_sdk/api/llm_api.py +7 -11
  6. lightning_sdk/api/pipeline_api.py +31 -10
  7. lightning_sdk/api/studio_api.py +6 -0
  8. lightning_sdk/base_studio.py +22 -6
  9. lightning_sdk/cli/entrypoint.py +15 -13
  10. lightning_sdk/cli/start.py +5 -2
  11. lightning_sdk/deployment/deployment.py +17 -7
  12. lightning_sdk/lightning_cloud/openapi/__init__.py +20 -0
  13. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  14. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  15. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
  16. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  17. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +97 -0
  18. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  19. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
  20. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  21. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  22. lightning_sdk/lightning_cloud/openapi/models/__init__.py +19 -0
  23. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +81 -3
  25. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
  26. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  28. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  29. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  30. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  32. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  34. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +2 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +105 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  54. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  55. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  58. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +853 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -27
  63. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
  68. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
  69. lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  71. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  72. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  74. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
  77. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  80. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +234 -104
  83. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
  84. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
  85. lightning_sdk/llm/llm.py +113 -115
  86. lightning_sdk/llm/public_assistants.json +8 -0
  87. lightning_sdk/pipeline/__init__.py +11 -2
  88. lightning_sdk/pipeline/pipeline.py +54 -14
  89. lightning_sdk/pipeline/printer.py +36 -16
  90. lightning_sdk/pipeline/schedule.py +2 -1
  91. lightning_sdk/pipeline/{types.py → steps.py} +77 -56
  92. lightning_sdk/pipeline/utils.py +65 -3
  93. lightning_sdk/sandbox.py +157 -0
  94. lightning_sdk/services/license.py +12 -6
  95. lightning_sdk/studio.py +10 -1
  96. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  97. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +101 -79
  98. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  99. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  100. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  101. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -44,8 +44,10 @@ class V1ClusterSpec(object):
44
44
  'auth_token': 'str',
45
45
  'available_accelerators': 'list[str]',
46
46
  'aws_v1': 'V1AWSDirectV1',
47
+ 'cloud_pricing_enabled': 'bool',
47
48
  'cloudflare_v1': 'V1CloudflareV1',
48
49
  'cluster_type': 'V1ClusterType',
50
+ 'compute_cluster_ids': 'list[str]',
49
51
  'deletion_options': 'V1ClusterDeletionOptions',
50
52
  'desired_state': 'V1ClusterState',
51
53
  'domain': 'str',
@@ -53,7 +55,9 @@ class V1ClusterSpec(object):
53
55
  'freeze_accelerators': 'bool',
54
56
  'google_cloud_v1': 'V1GoogleCloudDirectV1',
55
57
  'insurer_disabled': 'bool',
58
+ 'kubernetes_v1': 'V1KubernetesDirectV1',
56
59
  'lambda_labs_v1': 'V1LambdaLabsDirectV1',
60
+ 'lock_overprovisioning': 'bool',
57
61
  'locked_zones': 'list[str]',
58
62
  'monitor_deletion_disabled': 'bool',
59
63
  'nebius_v1': 'V1NebiusDirectV1',
@@ -76,8 +80,10 @@ class V1ClusterSpec(object):
76
80
  'auth_token': 'authToken',
77
81
  'available_accelerators': 'availableAccelerators',
78
82
  'aws_v1': 'awsV1',
83
+ 'cloud_pricing_enabled': 'cloudPricingEnabled',
79
84
  'cloudflare_v1': 'cloudflareV1',
80
85
  'cluster_type': 'clusterType',
86
+ 'compute_cluster_ids': 'computeClusterIds',
81
87
  'deletion_options': 'deletionOptions',
82
88
  'desired_state': 'desiredState',
83
89
  'domain': 'domain',
@@ -85,7 +91,9 @@ class V1ClusterSpec(object):
85
91
  'freeze_accelerators': 'freezeAccelerators',
86
92
  'google_cloud_v1': 'googleCloudV1',
87
93
  'insurer_disabled': 'insurerDisabled',
94
+ 'kubernetes_v1': 'kubernetesV1',
88
95
  'lambda_labs_v1': 'lambdaLabsV1',
96
+ 'lock_overprovisioning': 'lockOverprovisioning',
89
97
  'locked_zones': 'lockedZones',
90
98
  'monitor_deletion_disabled': 'monitorDeletionDisabled',
91
99
  'nebius_v1': 'nebiusV1',
@@ -104,13 +112,15 @@ class V1ClusterSpec(object):
104
112
  'vultr_v1': 'vultrV1'
105
113
  }
106
114
 
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, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =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
115
+ def __init__(self, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloud_pricing_enabled: 'bool' =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, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, lock_overprovisioning: 'bool' =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
108
116
  """V1ClusterSpec - a model defined in Swagger""" # noqa: E501
109
117
  self._auth_token = None
110
118
  self._available_accelerators = None
111
119
  self._aws_v1 = None
120
+ self._cloud_pricing_enabled = None
112
121
  self._cloudflare_v1 = None
113
122
  self._cluster_type = None
123
+ self._compute_cluster_ids = None
114
124
  self._deletion_options = None
115
125
  self._desired_state = None
116
126
  self._domain = None
@@ -118,7 +128,9 @@ class V1ClusterSpec(object):
118
128
  self._freeze_accelerators = None
119
129
  self._google_cloud_v1 = None
120
130
  self._insurer_disabled = None
131
+ self._kubernetes_v1 = None
121
132
  self._lambda_labs_v1 = None
133
+ self._lock_overprovisioning = None
122
134
  self._locked_zones = None
123
135
  self._monitor_deletion_disabled = None
124
136
  self._nebius_v1 = None
@@ -142,10 +154,14 @@ class V1ClusterSpec(object):
142
154
  self.available_accelerators = available_accelerators
143
155
  if aws_v1 is not None:
144
156
  self.aws_v1 = aws_v1
157
+ if cloud_pricing_enabled is not None:
158
+ self.cloud_pricing_enabled = cloud_pricing_enabled
145
159
  if cloudflare_v1 is not None:
146
160
  self.cloudflare_v1 = cloudflare_v1
147
161
  if cluster_type is not None:
148
162
  self.cluster_type = cluster_type
163
+ if compute_cluster_ids is not None:
164
+ self.compute_cluster_ids = compute_cluster_ids
149
165
  if deletion_options is not None:
150
166
  self.deletion_options = deletion_options
151
167
  if desired_state is not None:
@@ -160,8 +176,12 @@ class V1ClusterSpec(object):
160
176
  self.google_cloud_v1 = google_cloud_v1
161
177
  if insurer_disabled is not None:
162
178
  self.insurer_disabled = insurer_disabled
179
+ if kubernetes_v1 is not None:
180
+ self.kubernetes_v1 = kubernetes_v1
163
181
  if lambda_labs_v1 is not None:
164
182
  self.lambda_labs_v1 = lambda_labs_v1
183
+ if lock_overprovisioning is not None:
184
+ self.lock_overprovisioning = lock_overprovisioning
165
185
  if locked_zones is not None:
166
186
  self.locked_zones = locked_zones
167
187
  if monitor_deletion_disabled is not None:
@@ -258,6 +278,27 @@ class V1ClusterSpec(object):
258
278
 
259
279
  self._aws_v1 = aws_v1
260
280
 
281
+ @property
282
+ def cloud_pricing_enabled(self) -> 'bool':
283
+ """Gets the cloud_pricing_enabled of this V1ClusterSpec. # noqa: E501
284
+
285
+
286
+ :return: The cloud_pricing_enabled of this V1ClusterSpec. # noqa: E501
287
+ :rtype: bool
288
+ """
289
+ return self._cloud_pricing_enabled
290
+
291
+ @cloud_pricing_enabled.setter
292
+ def cloud_pricing_enabled(self, cloud_pricing_enabled: 'bool'):
293
+ """Sets the cloud_pricing_enabled of this V1ClusterSpec.
294
+
295
+
296
+ :param cloud_pricing_enabled: The cloud_pricing_enabled of this V1ClusterSpec. # noqa: E501
297
+ :type: bool
298
+ """
299
+
300
+ self._cloud_pricing_enabled = cloud_pricing_enabled
301
+
261
302
  @property
262
303
  def cloudflare_v1(self) -> 'V1CloudflareV1':
263
304
  """Gets the cloudflare_v1 of this V1ClusterSpec. # noqa: E501
@@ -300,6 +341,27 @@ class V1ClusterSpec(object):
300
341
 
301
342
  self._cluster_type = cluster_type
302
343
 
344
+ @property
345
+ def compute_cluster_ids(self) -> 'list[str]':
346
+ """Gets the compute_cluster_ids of this V1ClusterSpec. # noqa: E501
347
+
348
+
349
+ :return: The compute_cluster_ids of this V1ClusterSpec. # noqa: E501
350
+ :rtype: list[str]
351
+ """
352
+ return self._compute_cluster_ids
353
+
354
+ @compute_cluster_ids.setter
355
+ def compute_cluster_ids(self, compute_cluster_ids: 'list[str]'):
356
+ """Sets the compute_cluster_ids of this V1ClusterSpec.
357
+
358
+
359
+ :param compute_cluster_ids: The compute_cluster_ids of this V1ClusterSpec. # noqa: E501
360
+ :type: list[str]
361
+ """
362
+
363
+ self._compute_cluster_ids = compute_cluster_ids
364
+
303
365
  @property
304
366
  def deletion_options(self) -> 'V1ClusterDeletionOptions':
305
367
  """Gets the deletion_options of this V1ClusterSpec. # noqa: E501
@@ -449,6 +511,27 @@ class V1ClusterSpec(object):
449
511
 
450
512
  self._insurer_disabled = insurer_disabled
451
513
 
514
+ @property
515
+ def kubernetes_v1(self) -> 'V1KubernetesDirectV1':
516
+ """Gets the kubernetes_v1 of this V1ClusterSpec. # noqa: E501
517
+
518
+
519
+ :return: The kubernetes_v1 of this V1ClusterSpec. # noqa: E501
520
+ :rtype: V1KubernetesDirectV1
521
+ """
522
+ return self._kubernetes_v1
523
+
524
+ @kubernetes_v1.setter
525
+ def kubernetes_v1(self, kubernetes_v1: 'V1KubernetesDirectV1'):
526
+ """Sets the kubernetes_v1 of this V1ClusterSpec.
527
+
528
+
529
+ :param kubernetes_v1: The kubernetes_v1 of this V1ClusterSpec. # noqa: E501
530
+ :type: V1KubernetesDirectV1
531
+ """
532
+
533
+ self._kubernetes_v1 = kubernetes_v1
534
+
452
535
  @property
453
536
  def lambda_labs_v1(self) -> 'V1LambdaLabsDirectV1':
454
537
  """Gets the lambda_labs_v1 of this V1ClusterSpec. # noqa: E501
@@ -470,6 +553,27 @@ class V1ClusterSpec(object):
470
553
 
471
554
  self._lambda_labs_v1 = lambda_labs_v1
472
555
 
556
+ @property
557
+ def lock_overprovisioning(self) -> 'bool':
558
+ """Gets the lock_overprovisioning of this V1ClusterSpec. # noqa: E501
559
+
560
+
561
+ :return: The lock_overprovisioning of this V1ClusterSpec. # noqa: E501
562
+ :rtype: bool
563
+ """
564
+ return self._lock_overprovisioning
565
+
566
+ @lock_overprovisioning.setter
567
+ def lock_overprovisioning(self, lock_overprovisioning: 'bool'):
568
+ """Sets the lock_overprovisioning of this V1ClusterSpec.
569
+
570
+
571
+ :param lock_overprovisioning: The lock_overprovisioning of this V1ClusterSpec. # noqa: E501
572
+ :type: bool
573
+ """
574
+
575
+ self._lock_overprovisioning = lock_overprovisioning
576
+
473
577
  @property
474
578
  def locked_zones(self) -> 'list[str]':
475
579
  """Gets the locked_zones of this V1ClusterSpec. # noqa: E501
@@ -44,6 +44,7 @@ class V1ClusterStatus(object):
44
44
  'aws_v1_region_status': 'list[V1AWSDirectV1Status]',
45
45
  'dependency_service_health': 'list[V1ServiceHealth]',
46
46
  'gcp_v1_status': 'V1GoogleCloudDirectV1Status',
47
+ 'kubernetes_v1_status': 'V1KubernetesDirectV1Status',
47
48
  'last_healthcheck_timestamp': 'datetime',
48
49
  'phase': 'V1ClusterState',
49
50
  'prometheus_endpoint': 'str',
@@ -58,6 +59,7 @@ class V1ClusterStatus(object):
58
59
  'aws_v1_region_status': 'awsV1RegionStatus',
59
60
  'dependency_service_health': 'dependencyServiceHealth',
60
61
  'gcp_v1_status': 'gcpV1Status',
62
+ 'kubernetes_v1_status': 'kubernetesV1Status',
61
63
  'last_healthcheck_timestamp': 'lastHealthcheckTimestamp',
62
64
  'phase': 'phase',
63
65
  'prometheus_endpoint': 'prometheusEndpoint',
@@ -68,11 +70,12 @@ class V1ClusterStatus(object):
68
70
  'storage_endpoint': 'storageEndpoint'
69
71
  }
70
72
 
71
- def __init__(self, aws_v1_region_status: 'list[V1AWSDirectV1Status]' =None, dependency_service_health: 'list[V1ServiceHealth]' =None, gcp_v1_status: 'V1GoogleCloudDirectV1Status' =None, last_healthcheck_timestamp: 'datetime' =None, phase: 'V1ClusterState' =None, prometheus_endpoint: 'str' =None, queue_endpoint: 'str' =None, reason: 'str' =None, slurm_v1_status: 'V1SlurmV1Status' =None, ssh_gateway_endpoint: 'str' =None, storage_endpoint: 'str' =None): # noqa: E501
73
+ def __init__(self, aws_v1_region_status: 'list[V1AWSDirectV1Status]' =None, dependency_service_health: 'list[V1ServiceHealth]' =None, gcp_v1_status: 'V1GoogleCloudDirectV1Status' =None, kubernetes_v1_status: 'V1KubernetesDirectV1Status' =None, last_healthcheck_timestamp: 'datetime' =None, phase: 'V1ClusterState' =None, prometheus_endpoint: 'str' =None, queue_endpoint: 'str' =None, reason: 'str' =None, slurm_v1_status: 'V1SlurmV1Status' =None, ssh_gateway_endpoint: 'str' =None, storage_endpoint: 'str' =None): # noqa: E501
72
74
  """V1ClusterStatus - a model defined in Swagger""" # noqa: E501
73
75
  self._aws_v1_region_status = None
74
76
  self._dependency_service_health = None
75
77
  self._gcp_v1_status = None
78
+ self._kubernetes_v1_status = None
76
79
  self._last_healthcheck_timestamp = None
77
80
  self._phase = None
78
81
  self._prometheus_endpoint = None
@@ -88,6 +91,8 @@ class V1ClusterStatus(object):
88
91
  self.dependency_service_health = dependency_service_health
89
92
  if gcp_v1_status is not None:
90
93
  self.gcp_v1_status = gcp_v1_status
94
+ if kubernetes_v1_status is not None:
95
+ self.kubernetes_v1_status = kubernetes_v1_status
91
96
  if last_healthcheck_timestamp is not None:
92
97
  self.last_healthcheck_timestamp = last_healthcheck_timestamp
93
98
  if phase is not None:
@@ -168,6 +173,27 @@ class V1ClusterStatus(object):
168
173
 
169
174
  self._gcp_v1_status = gcp_v1_status
170
175
 
176
+ @property
177
+ def kubernetes_v1_status(self) -> 'V1KubernetesDirectV1Status':
178
+ """Gets the kubernetes_v1_status of this V1ClusterStatus. # noqa: E501
179
+
180
+
181
+ :return: The kubernetes_v1_status of this V1ClusterStatus. # noqa: E501
182
+ :rtype: V1KubernetesDirectV1Status
183
+ """
184
+ return self._kubernetes_v1_status
185
+
186
+ @kubernetes_v1_status.setter
187
+ def kubernetes_v1_status(self, kubernetes_v1_status: 'V1KubernetesDirectV1Status'):
188
+ """Sets the kubernetes_v1_status of this V1ClusterStatus.
189
+
190
+
191
+ :param kubernetes_v1_status: The kubernetes_v1_status of this V1ClusterStatus. # noqa: E501
192
+ :type: V1KubernetesDirectV1Status
193
+ """
194
+
195
+ self._kubernetes_v1_status = kubernetes_v1_status
196
+
171
197
  @property
172
198
  def last_healthcheck_timestamp(self) -> 'datetime':
173
199
  """Gets the last_healthcheck_timestamp of this V1ClusterStatus. # noqa: E501
@@ -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 = {}
@@ -52,6 +52,7 @@ class V1CreateDeploymentRequest(object):
52
52
  'name': 'str',
53
53
  'parameter_spec': 'V1ParameterizationSpec',
54
54
  'parent_template_id': 'str',
55
+ 'pipeline_reuse_deployment_between_runs': 'bool',
55
56
  'project_id': 'str',
56
57
  'recipients': 'V1DeploymentAlertingRecipients',
57
58
  'replicas': 'int',
@@ -71,6 +72,7 @@ class V1CreateDeploymentRequest(object):
71
72
  'name': 'name',
72
73
  'parameter_spec': 'parameterSpec',
73
74
  'parent_template_id': 'parentTemplateId',
75
+ 'pipeline_reuse_deployment_between_runs': 'pipelineReuseDeploymentBetweenRuns',
74
76
  'project_id': 'projectId',
75
77
  'recipients': 'recipients',
76
78
  'replicas': 'replicas',
@@ -78,7 +80,7 @@ class V1CreateDeploymentRequest(object):
78
80
  'strategy': 'strategy'
79
81
  }
80
82
 
81
- def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, from_litserve: 'bool' =None, from_onboarding: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, parent_template_id: 'str' =None, project_id: 'str' =None, recipients: 'V1DeploymentAlertingRecipients' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None): # noqa: E501
83
+ def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, from_litserve: 'bool' =None, from_onboarding: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, parent_template_id: 'str' =None, pipeline_reuse_deployment_between_runs: 'bool' =None, project_id: 'str' =None, recipients: 'V1DeploymentAlertingRecipients' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None): # noqa: E501
82
84
  """V1CreateDeploymentRequest - a model defined in Swagger""" # noqa: E501
83
85
  self._api_standard = None
84
86
  self._apis = None
@@ -91,6 +93,7 @@ class V1CreateDeploymentRequest(object):
91
93
  self._name = None
92
94
  self._parameter_spec = None
93
95
  self._parent_template_id = None
96
+ self._pipeline_reuse_deployment_between_runs = None
94
97
  self._project_id = None
95
98
  self._recipients = None
96
99
  self._replicas = None
@@ -119,6 +122,8 @@ class V1CreateDeploymentRequest(object):
119
122
  self.parameter_spec = parameter_spec
120
123
  if parent_template_id is not None:
121
124
  self.parent_template_id = parent_template_id
125
+ if pipeline_reuse_deployment_between_runs is not None:
126
+ self.pipeline_reuse_deployment_between_runs = pipeline_reuse_deployment_between_runs
122
127
  if project_id is not None:
123
128
  self.project_id = project_id
124
129
  if recipients is not None:
@@ -361,6 +366,27 @@ class V1CreateDeploymentRequest(object):
361
366
 
362
367
  self._parent_template_id = parent_template_id
363
368
 
369
+ @property
370
+ def pipeline_reuse_deployment_between_runs(self) -> 'bool':
371
+ """Gets the pipeline_reuse_deployment_between_runs of this V1CreateDeploymentRequest. # noqa: E501
372
+
373
+
374
+ :return: The pipeline_reuse_deployment_between_runs of this V1CreateDeploymentRequest. # noqa: E501
375
+ :rtype: bool
376
+ """
377
+ return self._pipeline_reuse_deployment_between_runs
378
+
379
+ @pipeline_reuse_deployment_between_runs.setter
380
+ def pipeline_reuse_deployment_between_runs(self, pipeline_reuse_deployment_between_runs: 'bool'):
381
+ """Sets the pipeline_reuse_deployment_between_runs of this V1CreateDeploymentRequest.
382
+
383
+
384
+ :param pipeline_reuse_deployment_between_runs: The pipeline_reuse_deployment_between_runs of this V1CreateDeploymentRequest. # noqa: E501
385
+ :type: bool
386
+ """
387
+
388
+ self._pipeline_reuse_deployment_between_runs = pipeline_reuse_deployment_between_runs
389
+
364
390
  @property
365
391
  def project_id(self) -> 'str':
366
392
  """Gets the project_id of this V1CreateDeploymentRequest. # noqa: E501
@@ -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