lightning-sdk 0.2.14__py3-none-any.whl → 0.2.16__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 (122) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +79 -0
  3. lightning_sdk/api/cluster_api.py +83 -1
  4. lightning_sdk/api/license_api.py +48 -0
  5. lightning_sdk/api/llm_api.py +73 -12
  6. lightning_sdk/api/studio_api.py +50 -1
  7. lightning_sdk/api/teamspace_api.py +127 -1
  8. lightning_sdk/api/utils.py +4 -0
  9. lightning_sdk/base_studio.py +83 -0
  10. lightning_sdk/cli/create.py +21 -1
  11. lightning_sdk/cli/delete.py +6 -8
  12. lightning_sdk/cli/deploy/__init__.py +0 -0
  13. lightning_sdk/cli/deploy/_auth.py +189 -0
  14. lightning_sdk/cli/deploy/devbox.py +157 -0
  15. lightning_sdk/cli/{serve.py → deploy/serve.py} +22 -281
  16. lightning_sdk/cli/download.py +69 -16
  17. lightning_sdk/cli/entrypoint.py +1 -1
  18. lightning_sdk/cli/open.py +21 -2
  19. lightning_sdk/cli/start.py +12 -3
  20. lightning_sdk/cli/teamspace_menu.py +9 -1
  21. lightning_sdk/cli/upload.py +2 -5
  22. lightning_sdk/lightning_cloud/openapi/__init__.py +29 -0
  23. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  24. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +121 -0
  25. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +9 -1
  26. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +226 -0
  27. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  28. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +178 -0
  29. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +984 -101
  30. lightning_sdk/lightning_cloud/openapi/api/product_license_service_api.py +525 -0
  31. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +93 -0
  32. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  33. lightning_sdk/lightning_cloud/openapi/models/__init__.py +28 -0
  34. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  35. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body.py +331 -0
  37. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body1.py +305 -0
  38. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +53 -1
  39. lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/models_id_body.py +123 -0
  42. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +183 -1
  43. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +107 -3
  47. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/update.py +79 -1
  49. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py +55 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +53 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +3 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +123 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +79 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +104 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_source_type.py +103 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +66 -66
  58. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_upload.py +149 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload.py +55 -3
  62. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +79 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_alerting_policy_response.py +175 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_event.py +487 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy.py +383 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_frequency.py +105 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_operation.py +105 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_severity.py +106 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_type.py +111 -0
  72. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +27 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py +4 -4
  74. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_external_search_user.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_ge_list_deployment_routing_telemetry_response.py +123 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_open_ports_response.py +123 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_content_response.py +123 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadata_response.py +331 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +1 -27
  82. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +27 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_job_type.py +1 -0
  84. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_events_response.py +123 -0
  85. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_policies_response.py +175 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_list_product_licenses_response.py +123 -0
  87. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  88. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +43 -17
  89. lightning_sdk/lightning_cloud/openapi/models/v1_modify_filesystem_volume_response.py +97 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +183 -1
  91. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +6 -6
  92. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_state.py +111 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py +53 -1
  94. lightning_sdk/lightning_cloud/openapi/models/v1_product_license.py +409 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_product_license_check_response.py +123 -0
  96. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  97. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +43 -17
  98. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +107 -3
  99. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +53 -1
  100. lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py +53 -1
  101. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +253 -0
  102. lightning_sdk/lightning_cloud/openapi/models/v1_secret_type.py +1 -0
  103. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +2 -0
  104. lightning_sdk/lightning_cloud/openapi/models/v1_sleep_server_response.py +97 -0
  105. lightning_sdk/lightning_cloud/openapi/models/v1_trigger_filesystem_upgrade_response.py +123 -0
  106. lightning_sdk/lightning_cloud/openapi/models/v1_upload_project_artifact_response.py +27 -1
  107. lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py +79 -1
  108. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +347 -113
  109. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  110. lightning_sdk/lightning_cloud/rest_client.py +4 -0
  111. lightning_sdk/llm/llm.py +132 -40
  112. lightning_sdk/services/__init__.py +1 -1
  113. lightning_sdk/services/license.py +236 -0
  114. lightning_sdk/studio.py +62 -1
  115. lightning_sdk/teamspace.py +68 -0
  116. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/METADATA +1 -1
  117. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/RECORD +122 -86
  118. /lightning_sdk/services/{finetune/__init__.py → finetune_llm.py} +0 -0
  119. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/LICENSE +0 -0
  120. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/WHEEL +0 -0
  121. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/entry_points.txt +0 -0
  122. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/top_level.txt +0 -0
@@ -52,8 +52,10 @@ class DeploymentsIdBody(object):
52
52
  'desired_state': 'V1DeploymentState',
53
53
  'endpoint': 'V1Endpoint',
54
54
  'is_published': 'bool',
55
+ 'managed': 'bool',
55
56
  'managed_endpoint_id': 'str',
56
57
  'name': 'str',
58
+ 'oncall_notification': 'bool',
57
59
  'parameter_spec': 'V1ParameterizationSpec',
58
60
  'pipeline_id': 'str',
59
61
  'release_id': 'str',
@@ -80,8 +82,10 @@ class DeploymentsIdBody(object):
80
82
  'desired_state': 'desiredState',
81
83
  'endpoint': 'endpoint',
82
84
  'is_published': 'isPublished',
85
+ 'managed': 'managed',
83
86
  'managed_endpoint_id': 'managedEndpointId',
84
87
  'name': 'name',
88
+ 'oncall_notification': 'oncallNotification',
85
89
  'parameter_spec': 'parameterSpec',
86
90
  'pipeline_id': 'pipelineId',
87
91
  'release_id': 'releaseId',
@@ -96,7 +100,7 @@ class DeploymentsIdBody(object):
96
100
  'visibility': 'visibility'
97
101
  }
98
102
 
99
- def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, assistant_id: 'str' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, current_state: 'V1DeploymentState' =None, debug: 'bool' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, managed_endpoint_id: 'str' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pipeline_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None, visibility: 'V1ResourceVisibility' =None): # noqa: E501
103
+ def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, assistant_id: 'str' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, current_state: 'V1DeploymentState' =None, debug: 'bool' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, managed: 'bool' =None, managed_endpoint_id: 'str' =None, name: 'str' =None, oncall_notification: 'bool' =None, parameter_spec: 'V1ParameterizationSpec' =None, pipeline_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None, visibility: 'V1ResourceVisibility' =None): # noqa: E501
100
104
  """DeploymentsIdBody - a model defined in Swagger""" # noqa: E501
101
105
  self._api_standard = None
102
106
  self._apis = None
@@ -109,8 +113,10 @@ class DeploymentsIdBody(object):
109
113
  self._desired_state = None
110
114
  self._endpoint = None
111
115
  self._is_published = None
116
+ self._managed = None
112
117
  self._managed_endpoint_id = None
113
118
  self._name = None
119
+ self._oncall_notification = None
114
120
  self._parameter_spec = None
115
121
  self._pipeline_id = None
116
122
  self._release_id = None
@@ -146,10 +152,14 @@ class DeploymentsIdBody(object):
146
152
  self.endpoint = endpoint
147
153
  if is_published is not None:
148
154
  self.is_published = is_published
155
+ if managed is not None:
156
+ self.managed = managed
149
157
  if managed_endpoint_id is not None:
150
158
  self.managed_endpoint_id = managed_endpoint_id
151
159
  if name is not None:
152
160
  self.name = name
161
+ if oncall_notification is not None:
162
+ self.oncall_notification = oncall_notification
153
163
  if parameter_spec is not None:
154
164
  self.parameter_spec = parameter_spec
155
165
  if pipeline_id is not None:
@@ -406,6 +416,27 @@ class DeploymentsIdBody(object):
406
416
 
407
417
  self._is_published = is_published
408
418
 
419
+ @property
420
+ def managed(self) -> 'bool':
421
+ """Gets the managed of this DeploymentsIdBody. # noqa: E501
422
+
423
+
424
+ :return: The managed of this DeploymentsIdBody. # noqa: E501
425
+ :rtype: bool
426
+ """
427
+ return self._managed
428
+
429
+ @managed.setter
430
+ def managed(self, managed: 'bool'):
431
+ """Sets the managed of this DeploymentsIdBody.
432
+
433
+
434
+ :param managed: The managed of this DeploymentsIdBody. # noqa: E501
435
+ :type: bool
436
+ """
437
+
438
+ self._managed = managed
439
+
409
440
  @property
410
441
  def managed_endpoint_id(self) -> 'str':
411
442
  """Gets the managed_endpoint_id of this DeploymentsIdBody. # noqa: E501
@@ -448,6 +479,27 @@ class DeploymentsIdBody(object):
448
479
 
449
480
  self._name = name
450
481
 
482
+ @property
483
+ def oncall_notification(self) -> 'bool':
484
+ """Gets the oncall_notification of this DeploymentsIdBody. # noqa: E501
485
+
486
+
487
+ :return: The oncall_notification of this DeploymentsIdBody. # noqa: E501
488
+ :rtype: bool
489
+ """
490
+ return self._oncall_notification
491
+
492
+ @oncall_notification.setter
493
+ def oncall_notification(self, oncall_notification: 'bool'):
494
+ """Sets the oncall_notification of this DeploymentsIdBody.
495
+
496
+
497
+ :param oncall_notification: The oncall_notification of this DeploymentsIdBody. # noqa: E501
498
+ :type: bool
499
+ """
500
+
501
+ self._oncall_notification = oncall_notification
502
+
451
503
  @property
452
504
  def parameter_spec(self) -> 'V1ParameterizationSpec':
453
505
  """Gets the parameter_spec of this DeploymentsIdBody. # noqa: E501
@@ -51,6 +51,7 @@ class EndpointsIdBody(object):
51
51
  'openai': 'V1UpstreamOpenAI',
52
52
  'ports': 'list[str]',
53
53
  'prewarm': 'V1EndpointPrewarm',
54
+ 'proxy': 'bool',
54
55
  'updated_at': 'datetime',
55
56
  'urls': 'list[str]',
56
57
  'user_id': 'str'
@@ -67,12 +68,13 @@ class EndpointsIdBody(object):
67
68
  'openai': 'openai',
68
69
  'ports': 'ports',
69
70
  'prewarm': 'prewarm',
71
+ 'proxy': 'proxy',
70
72
  'updated_at': 'updatedAt',
71
73
  'urls': 'urls',
72
74
  'user_id': 'userId'
73
75
  }
74
76
 
75
- def __init__(self, auth: 'V1EndpointAuth' =None, cloudspace: 'V1UpstreamCloudSpace' =None, created_at: 'datetime' =None, custom_domain: 'str' =None, job: 'V1UpstreamJob' =None, managed: 'V1UpstreamManaged' =None, name: 'str' =None, openai: 'V1UpstreamOpenAI' =None, ports: 'list[str]' =None, prewarm: 'V1EndpointPrewarm' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None): # noqa: E501
77
+ def __init__(self, auth: 'V1EndpointAuth' =None, cloudspace: 'V1UpstreamCloudSpace' =None, created_at: 'datetime' =None, custom_domain: 'str' =None, job: 'V1UpstreamJob' =None, managed: 'V1UpstreamManaged' =None, name: 'str' =None, openai: 'V1UpstreamOpenAI' =None, ports: 'list[str]' =None, prewarm: 'V1EndpointPrewarm' =None, proxy: 'bool' =None, updated_at: 'datetime' =None, urls: 'list[str]' =None, user_id: 'str' =None): # noqa: E501
76
78
  """EndpointsIdBody - a model defined in Swagger""" # noqa: E501
77
79
  self._auth = None
78
80
  self._cloudspace = None
@@ -84,6 +86,7 @@ class EndpointsIdBody(object):
84
86
  self._openai = None
85
87
  self._ports = None
86
88
  self._prewarm = None
89
+ self._proxy = None
87
90
  self._updated_at = None
88
91
  self._urls = None
89
92
  self._user_id = None
@@ -108,6 +111,8 @@ class EndpointsIdBody(object):
108
111
  self.ports = ports
109
112
  if prewarm is not None:
110
113
  self.prewarm = prewarm
114
+ if proxy is not None:
115
+ self.proxy = proxy
111
116
  if updated_at is not None:
112
117
  self.updated_at = updated_at
113
118
  if urls is not None:
@@ -325,6 +330,27 @@ class EndpointsIdBody(object):
325
330
 
326
331
  self._prewarm = prewarm
327
332
 
333
+ @property
334
+ def proxy(self) -> 'bool':
335
+ """Gets the proxy of this EndpointsIdBody. # noqa: E501
336
+
337
+
338
+ :return: The proxy of this EndpointsIdBody. # noqa: E501
339
+ :rtype: bool
340
+ """
341
+ return self._proxy
342
+
343
+ @proxy.setter
344
+ def proxy(self, proxy: 'bool'):
345
+ """Sets the proxy of this EndpointsIdBody.
346
+
347
+
348
+ :param proxy: The proxy of this EndpointsIdBody. # noqa: E501
349
+ :type: bool
350
+ """
351
+
352
+ self._proxy = proxy
353
+
328
354
  @property
329
355
  def updated_at(self) -> 'datetime':
330
356
  """Gets the updated_at of this EndpointsIdBody. # noqa: E501
@@ -42,21 +42,26 @@ class ModelIdVersionsBody(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'cluster_id': 'str',
45
+ 'metadata': 'dict(str, str)',
45
46
  'version': 'str'
46
47
  }
47
48
 
48
49
  attribute_map = {
49
50
  'cluster_id': 'clusterId',
51
+ 'metadata': 'metadata',
50
52
  'version': 'version'
51
53
  }
52
54
 
53
- def __init__(self, cluster_id: 'str' =None, version: 'str' =None): # noqa: E501
55
+ def __init__(self, cluster_id: 'str' =None, metadata: 'dict(str, str)' =None, version: 'str' =None): # noqa: E501
54
56
  """ModelIdVersionsBody - a model defined in Swagger""" # noqa: E501
55
57
  self._cluster_id = None
58
+ self._metadata = None
56
59
  self._version = None
57
60
  self.discriminator = None
58
61
  if cluster_id is not None:
59
62
  self.cluster_id = cluster_id
63
+ if metadata is not None:
64
+ self.metadata = metadata
60
65
  if version is not None:
61
66
  self.version = version
62
67
 
@@ -81,6 +86,27 @@ class ModelIdVersionsBody(object):
81
86
 
82
87
  self._cluster_id = cluster_id
83
88
 
89
+ @property
90
+ def metadata(self) -> 'dict(str, str)':
91
+ """Gets the metadata of this ModelIdVersionsBody. # noqa: E501
92
+
93
+
94
+ :return: The metadata of this ModelIdVersionsBody. # noqa: E501
95
+ :rtype: dict(str, str)
96
+ """
97
+ return self._metadata
98
+
99
+ @metadata.setter
100
+ def metadata(self, metadata: 'dict(str, str)'):
101
+ """Sets the metadata of this ModelIdVersionsBody.
102
+
103
+
104
+ :param metadata: The metadata of this ModelIdVersionsBody. # noqa: E501
105
+ :type: dict(str, str)
106
+ """
107
+
108
+ self._metadata = metadata
109
+
84
110
  @property
85
111
  def version(self) -> 'str':
86
112
  """Gets the version of this ModelIdVersionsBody. # noqa: E501
@@ -0,0 +1,123 @@
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 ModelsIdBody(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
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'model': 'V1ManagedModel'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'model': 'model'
49
+ }
50
+
51
+ def __init__(self, model: 'V1ManagedModel' =None): # noqa: E501
52
+ """ModelsIdBody - a model defined in Swagger""" # noqa: E501
53
+ self._model = None
54
+ self.discriminator = None
55
+ if model is not None:
56
+ self.model = model
57
+
58
+ @property
59
+ def model(self) -> 'V1ManagedModel':
60
+ """Gets the model of this ModelsIdBody. # noqa: E501
61
+
62
+
63
+ :return: The model of this ModelsIdBody. # noqa: E501
64
+ :rtype: V1ManagedModel
65
+ """
66
+ return self._model
67
+
68
+ @model.setter
69
+ def model(self, model: 'V1ManagedModel'):
70
+ """Sets the model of this ModelsIdBody.
71
+
72
+
73
+ :param model: The model of this ModelsIdBody. # noqa: E501
74
+ :type: V1ManagedModel
75
+ """
76
+
77
+ self._model = model
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(ModelsIdBody, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'ModelsIdBody') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, ModelsIdBody):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'ModelsIdBody') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -44,16 +44,22 @@ class OrgsIdBody(object):
44
44
  'alerts_config': 'V1AlertsConfig',
45
45
  'allow_aws_saas': 'bool',
46
46
  'allow_budgeting': 'bool',
47
+ 'allow_credits_auto_replenish': 'bool',
48
+ 'allow_dgx_saas': 'bool',
47
49
  'allow_external_project_duplication': 'bool',
48
50
  'allow_gcp_saas': 'bool',
49
51
  'allow_guest': 'bool',
50
52
  'allow_lambda_saas': 'bool',
53
+ 'allow_lightning_saas': 'bool',
51
54
  'allow_marketplace': 'bool',
52
55
  'allow_member_invitations': 'bool',
53
56
  'allow_member_teamspace_creation': 'bool',
54
57
  'allow_vultr_saas': 'bool',
55
58
  'auto_invite_by_domain': 'bool',
56
59
  'auto_join_domains': 'list[str]',
60
+ 'auto_replenish_amount': 'float',
61
+ 'auto_replenish_threshold': 'float',
62
+ 'auto_switch_machine': 'bool',
57
63
  'default_machine_type': 'str',
58
64
  'description': 'str',
59
65
  'display_name': 'str',
@@ -65,6 +71,7 @@ class OrgsIdBody(object):
65
71
  'preferred_deployment_provider': 'str',
66
72
  'preferred_studio_provider': 'str',
67
73
  'start_studios_on_spot_instance': 'bool',
74
+ 'switch_to_default_machine_on_idle': 'bool',
68
75
  'teamspace_default_credits': 'float',
69
76
  'twitter_username': 'str',
70
77
  'workload_max_run_duration': 'str'
@@ -74,16 +81,22 @@ class OrgsIdBody(object):
74
81
  'alerts_config': 'alertsConfig',
75
82
  'allow_aws_saas': 'allowAwsSaas',
76
83
  'allow_budgeting': 'allowBudgeting',
84
+ 'allow_credits_auto_replenish': 'allowCreditsAutoReplenish',
85
+ 'allow_dgx_saas': 'allowDgxSaas',
77
86
  'allow_external_project_duplication': 'allowExternalProjectDuplication',
78
87
  'allow_gcp_saas': 'allowGcpSaas',
79
88
  'allow_guest': 'allowGuest',
80
89
  'allow_lambda_saas': 'allowLambdaSaas',
90
+ 'allow_lightning_saas': 'allowLightningSaas',
81
91
  'allow_marketplace': 'allowMarketplace',
82
92
  'allow_member_invitations': 'allowMemberInvitations',
83
93
  'allow_member_teamspace_creation': 'allowMemberTeamspaceCreation',
84
94
  'allow_vultr_saas': 'allowVultrSaas',
85
95
  'auto_invite_by_domain': 'autoInviteByDomain',
86
96
  'auto_join_domains': 'autoJoinDomains',
97
+ 'auto_replenish_amount': 'autoReplenishAmount',
98
+ 'auto_replenish_threshold': 'autoReplenishThreshold',
99
+ 'auto_switch_machine': 'autoSwitchMachine',
87
100
  'default_machine_type': 'defaultMachineType',
88
101
  'description': 'description',
89
102
  'display_name': 'displayName',
@@ -95,26 +108,33 @@ class OrgsIdBody(object):
95
108
  'preferred_deployment_provider': 'preferredDeploymentProvider',
96
109
  'preferred_studio_provider': 'preferredStudioProvider',
97
110
  'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
111
+ 'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle',
98
112
  'teamspace_default_credits': 'teamspaceDefaultCredits',
99
113
  'twitter_username': 'twitterUsername',
100
114
  'workload_max_run_duration': 'workloadMaxRunDuration'
101
115
  }
102
116
 
103
- def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, default_machine_type: '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, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, workload_max_run_duration: 'str' =None): # noqa: E501
117
+ def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, default_machine_type: '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, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, workload_max_run_duration: 'str' =None): # noqa: E501
104
118
  """OrgsIdBody - a model defined in Swagger""" # noqa: E501
105
119
  self._alerts_config = None
106
120
  self._allow_aws_saas = None
107
121
  self._allow_budgeting = None
122
+ self._allow_credits_auto_replenish = None
123
+ self._allow_dgx_saas = None
108
124
  self._allow_external_project_duplication = None
109
125
  self._allow_gcp_saas = None
110
126
  self._allow_guest = None
111
127
  self._allow_lambda_saas = None
128
+ self._allow_lightning_saas = None
112
129
  self._allow_marketplace = None
113
130
  self._allow_member_invitations = None
114
131
  self._allow_member_teamspace_creation = None
115
132
  self._allow_vultr_saas = None
116
133
  self._auto_invite_by_domain = None
117
134
  self._auto_join_domains = None
135
+ self._auto_replenish_amount = None
136
+ self._auto_replenish_threshold = None
137
+ self._auto_switch_machine = None
118
138
  self._default_machine_type = None
119
139
  self._description = None
120
140
  self._display_name = None
@@ -126,6 +146,7 @@ class OrgsIdBody(object):
126
146
  self._preferred_deployment_provider = None
127
147
  self._preferred_studio_provider = None
128
148
  self._start_studios_on_spot_instance = None
149
+ self._switch_to_default_machine_on_idle = None
129
150
  self._teamspace_default_credits = None
130
151
  self._twitter_username = None
131
152
  self._workload_max_run_duration = None
@@ -136,6 +157,10 @@ class OrgsIdBody(object):
136
157
  self.allow_aws_saas = allow_aws_saas
137
158
  if allow_budgeting is not None:
138
159
  self.allow_budgeting = allow_budgeting
160
+ if allow_credits_auto_replenish is not None:
161
+ self.allow_credits_auto_replenish = allow_credits_auto_replenish
162
+ if allow_dgx_saas is not None:
163
+ self.allow_dgx_saas = allow_dgx_saas
139
164
  if allow_external_project_duplication is not None:
140
165
  self.allow_external_project_duplication = allow_external_project_duplication
141
166
  if allow_gcp_saas is not None:
@@ -144,6 +169,8 @@ class OrgsIdBody(object):
144
169
  self.allow_guest = allow_guest
145
170
  if allow_lambda_saas is not None:
146
171
  self.allow_lambda_saas = allow_lambda_saas
172
+ if allow_lightning_saas is not None:
173
+ self.allow_lightning_saas = allow_lightning_saas
147
174
  if allow_marketplace is not None:
148
175
  self.allow_marketplace = allow_marketplace
149
176
  if allow_member_invitations is not None:
@@ -156,6 +183,12 @@ class OrgsIdBody(object):
156
183
  self.auto_invite_by_domain = auto_invite_by_domain
157
184
  if auto_join_domains is not None:
158
185
  self.auto_join_domains = auto_join_domains
186
+ if auto_replenish_amount is not None:
187
+ self.auto_replenish_amount = auto_replenish_amount
188
+ if auto_replenish_threshold is not None:
189
+ self.auto_replenish_threshold = auto_replenish_threshold
190
+ if auto_switch_machine is not None:
191
+ self.auto_switch_machine = auto_switch_machine
159
192
  if default_machine_type is not None:
160
193
  self.default_machine_type = default_machine_type
161
194
  if description is not None:
@@ -178,6 +211,8 @@ class OrgsIdBody(object):
178
211
  self.preferred_studio_provider = preferred_studio_provider
179
212
  if start_studios_on_spot_instance is not None:
180
213
  self.start_studios_on_spot_instance = start_studios_on_spot_instance
214
+ if switch_to_default_machine_on_idle is not None:
215
+ self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
181
216
  if teamspace_default_credits is not None:
182
217
  self.teamspace_default_credits = teamspace_default_credits
183
218
  if twitter_username is not None:
@@ -248,6 +283,48 @@ class OrgsIdBody(object):
248
283
 
249
284
  self._allow_budgeting = allow_budgeting
250
285
 
286
+ @property
287
+ def allow_credits_auto_replenish(self) -> 'bool':
288
+ """Gets the allow_credits_auto_replenish of this OrgsIdBody. # noqa: E501
289
+
290
+
291
+ :return: The allow_credits_auto_replenish of this OrgsIdBody. # noqa: E501
292
+ :rtype: bool
293
+ """
294
+ return self._allow_credits_auto_replenish
295
+
296
+ @allow_credits_auto_replenish.setter
297
+ def allow_credits_auto_replenish(self, allow_credits_auto_replenish: 'bool'):
298
+ """Sets the allow_credits_auto_replenish of this OrgsIdBody.
299
+
300
+
301
+ :param allow_credits_auto_replenish: The allow_credits_auto_replenish of this OrgsIdBody. # noqa: E501
302
+ :type: bool
303
+ """
304
+
305
+ self._allow_credits_auto_replenish = allow_credits_auto_replenish
306
+
307
+ @property
308
+ def allow_dgx_saas(self) -> 'bool':
309
+ """Gets the allow_dgx_saas of this OrgsIdBody. # noqa: E501
310
+
311
+
312
+ :return: The allow_dgx_saas of this OrgsIdBody. # noqa: E501
313
+ :rtype: bool
314
+ """
315
+ return self._allow_dgx_saas
316
+
317
+ @allow_dgx_saas.setter
318
+ def allow_dgx_saas(self, allow_dgx_saas: 'bool'):
319
+ """Sets the allow_dgx_saas of this OrgsIdBody.
320
+
321
+
322
+ :param allow_dgx_saas: The allow_dgx_saas of this OrgsIdBody. # noqa: E501
323
+ :type: bool
324
+ """
325
+
326
+ self._allow_dgx_saas = allow_dgx_saas
327
+
251
328
  @property
252
329
  def allow_external_project_duplication(self) -> 'bool':
253
330
  """Gets the allow_external_project_duplication of this OrgsIdBody. # noqa: E501
@@ -332,6 +409,27 @@ class OrgsIdBody(object):
332
409
 
333
410
  self._allow_lambda_saas = allow_lambda_saas
334
411
 
412
+ @property
413
+ def allow_lightning_saas(self) -> 'bool':
414
+ """Gets the allow_lightning_saas of this OrgsIdBody. # noqa: E501
415
+
416
+
417
+ :return: The allow_lightning_saas of this OrgsIdBody. # noqa: E501
418
+ :rtype: bool
419
+ """
420
+ return self._allow_lightning_saas
421
+
422
+ @allow_lightning_saas.setter
423
+ def allow_lightning_saas(self, allow_lightning_saas: 'bool'):
424
+ """Sets the allow_lightning_saas of this OrgsIdBody.
425
+
426
+
427
+ :param allow_lightning_saas: The allow_lightning_saas of this OrgsIdBody. # noqa: E501
428
+ :type: bool
429
+ """
430
+
431
+ self._allow_lightning_saas = allow_lightning_saas
432
+
335
433
  @property
336
434
  def allow_marketplace(self) -> 'bool':
337
435
  """Gets the allow_marketplace of this OrgsIdBody. # noqa: E501
@@ -458,6 +556,69 @@ class OrgsIdBody(object):
458
556
 
459
557
  self._auto_join_domains = auto_join_domains
460
558
 
559
+ @property
560
+ def auto_replenish_amount(self) -> 'float':
561
+ """Gets the auto_replenish_amount of this OrgsIdBody. # noqa: E501
562
+
563
+
564
+ :return: The auto_replenish_amount of this OrgsIdBody. # noqa: E501
565
+ :rtype: float
566
+ """
567
+ return self._auto_replenish_amount
568
+
569
+ @auto_replenish_amount.setter
570
+ def auto_replenish_amount(self, auto_replenish_amount: 'float'):
571
+ """Sets the auto_replenish_amount of this OrgsIdBody.
572
+
573
+
574
+ :param auto_replenish_amount: The auto_replenish_amount of this OrgsIdBody. # noqa: E501
575
+ :type: float
576
+ """
577
+
578
+ self._auto_replenish_amount = auto_replenish_amount
579
+
580
+ @property
581
+ def auto_replenish_threshold(self) -> 'float':
582
+ """Gets the auto_replenish_threshold of this OrgsIdBody. # noqa: E501
583
+
584
+
585
+ :return: The auto_replenish_threshold of this OrgsIdBody. # noqa: E501
586
+ :rtype: float
587
+ """
588
+ return self._auto_replenish_threshold
589
+
590
+ @auto_replenish_threshold.setter
591
+ def auto_replenish_threshold(self, auto_replenish_threshold: 'float'):
592
+ """Sets the auto_replenish_threshold of this OrgsIdBody.
593
+
594
+
595
+ :param auto_replenish_threshold: The auto_replenish_threshold of this OrgsIdBody. # noqa: E501
596
+ :type: float
597
+ """
598
+
599
+ self._auto_replenish_threshold = auto_replenish_threshold
600
+
601
+ @property
602
+ def auto_switch_machine(self) -> 'bool':
603
+ """Gets the auto_switch_machine of this OrgsIdBody. # noqa: E501
604
+
605
+
606
+ :return: The auto_switch_machine of this OrgsIdBody. # noqa: E501
607
+ :rtype: bool
608
+ """
609
+ return self._auto_switch_machine
610
+
611
+ @auto_switch_machine.setter
612
+ def auto_switch_machine(self, auto_switch_machine: 'bool'):
613
+ """Sets the auto_switch_machine of this OrgsIdBody.
614
+
615
+
616
+ :param auto_switch_machine: The auto_switch_machine of this OrgsIdBody. # noqa: E501
617
+ :type: bool
618
+ """
619
+
620
+ self._auto_switch_machine = auto_switch_machine
621
+
461
622
  @property
462
623
  def default_machine_type(self) -> 'str':
463
624
  """Gets the default_machine_type of this OrgsIdBody. # noqa: E501
@@ -689,6 +850,27 @@ class OrgsIdBody(object):
689
850
 
690
851
  self._start_studios_on_spot_instance = start_studios_on_spot_instance
691
852
 
853
+ @property
854
+ def switch_to_default_machine_on_idle(self) -> 'bool':
855
+ """Gets the switch_to_default_machine_on_idle of this OrgsIdBody. # noqa: E501
856
+
857
+
858
+ :return: The switch_to_default_machine_on_idle of this OrgsIdBody. # noqa: E501
859
+ :rtype: bool
860
+ """
861
+ return self._switch_to_default_machine_on_idle
862
+
863
+ @switch_to_default_machine_on_idle.setter
864
+ def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
865
+ """Sets the switch_to_default_machine_on_idle of this OrgsIdBody.
866
+
867
+
868
+ :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this OrgsIdBody. # noqa: E501
869
+ :type: bool
870
+ """
871
+
872
+ self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
873
+
692
874
  @property
693
875
  def teamspace_default_credits(self) -> 'float':
694
876
  """Gets the teamspace_default_credits of this OrgsIdBody. # noqa: E501