lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc1__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 (83) 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 +23 -13
  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 +18 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +2 -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/cluster_service_api.py +9 -1
  14. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +13 -1
  15. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  17. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  18. lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py +258 -0
  19. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  20. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  23. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +53 -1
  25. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  27. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py +103 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +29 -3
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_volume_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  56. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +149 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py +104 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +1 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_update_volume_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +317 -31
  70. lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.py +123 -0
  71. lightning_sdk/llm/llm.py +118 -115
  72. lightning_sdk/llm/public_assistants.json +8 -0
  73. lightning_sdk/pipeline/pipeline.py +17 -2
  74. lightning_sdk/pipeline/printer.py +11 -10
  75. lightning_sdk/pipeline/steps.py +4 -1
  76. lightning_sdk/pipeline/utils.py +29 -4
  77. lightning_sdk/studio.py +3 -0
  78. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/METADATA +1 -1
  79. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/RECORD +83 -64
  80. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/LICENSE +0 -0
  81. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/WHEEL +0 -0
  82. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/entry_points.txt +0 -0
  83. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/top_level.txt +0 -0
@@ -55,6 +55,7 @@ class V1ClusterSpec(object):
55
55
  'freeze_accelerators': 'bool',
56
56
  'google_cloud_v1': 'V1GoogleCloudDirectV1',
57
57
  'insurer_disabled': 'bool',
58
+ 'kubernetes_v1': 'V1KubernetesDirectV1',
58
59
  'lambda_labs_v1': 'V1LambdaLabsDirectV1',
59
60
  'lock_overprovisioning': 'bool',
60
61
  'locked_zones': 'list[str]',
@@ -90,6 +91,7 @@ class V1ClusterSpec(object):
90
91
  'freeze_accelerators': 'freezeAccelerators',
91
92
  'google_cloud_v1': 'googleCloudV1',
92
93
  'insurer_disabled': 'insurerDisabled',
94
+ 'kubernetes_v1': 'kubernetesV1',
93
95
  'lambda_labs_v1': 'lambdaLabsV1',
94
96
  'lock_overprovisioning': 'lockOverprovisioning',
95
97
  'locked_zones': 'lockedZones',
@@ -110,7 +112,7 @@ class V1ClusterSpec(object):
110
112
  'vultr_v1': 'vultrV1'
111
113
  }
112
114
 
113
- 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, 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
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
114
116
  """V1ClusterSpec - a model defined in Swagger""" # noqa: E501
115
117
  self._auth_token = None
116
118
  self._available_accelerators = None
@@ -126,6 +128,7 @@ class V1ClusterSpec(object):
126
128
  self._freeze_accelerators = None
127
129
  self._google_cloud_v1 = None
128
130
  self._insurer_disabled = None
131
+ self._kubernetes_v1 = None
129
132
  self._lambda_labs_v1 = None
130
133
  self._lock_overprovisioning = None
131
134
  self._locked_zones = None
@@ -173,6 +176,8 @@ class V1ClusterSpec(object):
173
176
  self.google_cloud_v1 = google_cloud_v1
174
177
  if insurer_disabled is not None:
175
178
  self.insurer_disabled = insurer_disabled
179
+ if kubernetes_v1 is not None:
180
+ self.kubernetes_v1 = kubernetes_v1
176
181
  if lambda_labs_v1 is not None:
177
182
  self.lambda_labs_v1 = lambda_labs_v1
178
183
  if lock_overprovisioning is not None:
@@ -506,6 +511,27 @@ class V1ClusterSpec(object):
506
511
 
507
512
  self._insurer_disabled = insurer_disabled
508
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
+
509
535
  @property
510
536
  def lambda_labs_v1(self) -> 'V1LambdaLabsDirectV1':
511
537
  """Gets the lambda_labs_v1 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 = {}
@@ -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
@@ -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