lightning-sdk 0.2.15__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 (74) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +7 -1
  3. lightning_sdk/api/cluster_api.py +83 -1
  4. lightning_sdk/api/llm_api.py +24 -5
  5. lightning_sdk/api/studio_api.py +3 -0
  6. lightning_sdk/api/teamspace_api.py +127 -1
  7. lightning_sdk/api/utils.py +4 -0
  8. lightning_sdk/base_studio.py +14 -1
  9. lightning_sdk/cli/create.py +21 -1
  10. lightning_sdk/cli/deploy/__init__.py +0 -0
  11. lightning_sdk/cli/deploy/_auth.py +189 -0
  12. lightning_sdk/cli/deploy/devbox.py +157 -0
  13. lightning_sdk/cli/{serve.py → deploy/serve.py} +11 -322
  14. lightning_sdk/cli/download.py +44 -16
  15. lightning_sdk/cli/entrypoint.py +1 -1
  16. lightning_sdk/cli/open.py +21 -2
  17. lightning_sdk/cli/start.py +12 -3
  18. lightning_sdk/cli/upload.py +2 -4
  19. lightning_sdk/lightning_cloud/openapi/__init__.py +18 -0
  20. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +121 -0
  21. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +105 -0
  22. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  23. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +747 -105
  24. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +93 -0
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +18 -0
  26. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +53 -1
  28. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body.py +331 -0
  29. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body1.py +305 -0
  30. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/models_id_body.py +123 -0
  32. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +105 -1
  33. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +29 -3
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_source_type.py +103 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_alerting_policy_response.py +175 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +53 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_event.py +487 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy.py +383 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_frequency.py +105 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_operation.py +105 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_severity.py +106 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_type.py +111 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_ge_list_deployment_routing_telemetry_response.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_open_ports_response.py +123 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_content_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadata_response.py +331 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +1 -27
  51. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_events_response.py +123 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_policies_response.py +175 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +27 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +105 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +27 -1
  58. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +29 -3
  59. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +53 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +253 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +1 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_sleep_server_response.py +97 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +1 -27
  64. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +79 -27
  65. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  66. lightning_sdk/llm/llm.py +52 -8
  67. lightning_sdk/studio.py +32 -1
  68. lightning_sdk/teamspace.py +68 -0
  69. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/METADATA +1 -1
  70. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/RECORD +74 -53
  71. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/LICENSE +0 -0
  72. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/WHEEL +0 -0
  73. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/entry_points.txt +0 -0
  74. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.16.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,175 @@
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 V1ListDeploymentAlertingPoliciesResponse(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
+ 'alerting_policies': 'list[V1DeploymentAlertingPolicy]',
45
+ 'deployment_id': 'str',
46
+ 'project_id': 'str'
47
+ }
48
+
49
+ attribute_map = {
50
+ 'alerting_policies': 'alertingPolicies',
51
+ 'deployment_id': 'deploymentId',
52
+ 'project_id': 'projectId'
53
+ }
54
+
55
+ def __init__(self, alerting_policies: 'list[V1DeploymentAlertingPolicy]' =None, deployment_id: 'str' =None, project_id: 'str' =None): # noqa: E501
56
+ """V1ListDeploymentAlertingPoliciesResponse - a model defined in Swagger""" # noqa: E501
57
+ self._alerting_policies = None
58
+ self._deployment_id = None
59
+ self._project_id = None
60
+ self.discriminator = None
61
+ if alerting_policies is not None:
62
+ self.alerting_policies = alerting_policies
63
+ if deployment_id is not None:
64
+ self.deployment_id = deployment_id
65
+ if project_id is not None:
66
+ self.project_id = project_id
67
+
68
+ @property
69
+ def alerting_policies(self) -> 'list[V1DeploymentAlertingPolicy]':
70
+ """Gets the alerting_policies of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
71
+
72
+
73
+ :return: The alerting_policies of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
74
+ :rtype: list[V1DeploymentAlertingPolicy]
75
+ """
76
+ return self._alerting_policies
77
+
78
+ @alerting_policies.setter
79
+ def alerting_policies(self, alerting_policies: 'list[V1DeploymentAlertingPolicy]'):
80
+ """Sets the alerting_policies of this V1ListDeploymentAlertingPoliciesResponse.
81
+
82
+
83
+ :param alerting_policies: The alerting_policies of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
84
+ :type: list[V1DeploymentAlertingPolicy]
85
+ """
86
+
87
+ self._alerting_policies = alerting_policies
88
+
89
+ @property
90
+ def deployment_id(self) -> 'str':
91
+ """Gets the deployment_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
92
+
93
+
94
+ :return: The deployment_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
95
+ :rtype: str
96
+ """
97
+ return self._deployment_id
98
+
99
+ @deployment_id.setter
100
+ def deployment_id(self, deployment_id: 'str'):
101
+ """Sets the deployment_id of this V1ListDeploymentAlertingPoliciesResponse.
102
+
103
+
104
+ :param deployment_id: The deployment_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
105
+ :type: str
106
+ """
107
+
108
+ self._deployment_id = deployment_id
109
+
110
+ @property
111
+ def project_id(self) -> 'str':
112
+ """Gets the project_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
113
+
114
+
115
+ :return: The project_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
116
+ :rtype: str
117
+ """
118
+ return self._project_id
119
+
120
+ @project_id.setter
121
+ def project_id(self, project_id: 'str'):
122
+ """Sets the project_id of this V1ListDeploymentAlertingPoliciesResponse.
123
+
124
+
125
+ :param project_id: The project_id of this V1ListDeploymentAlertingPoliciesResponse. # noqa: E501
126
+ :type: str
127
+ """
128
+
129
+ self._project_id = project_id
130
+
131
+ def to_dict(self) -> dict:
132
+ """Returns the model properties as a dict"""
133
+ result = {}
134
+
135
+ for attr, _ in six.iteritems(self.swagger_types):
136
+ value = getattr(self, attr)
137
+ if isinstance(value, list):
138
+ result[attr] = list(map(
139
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
140
+ value
141
+ ))
142
+ elif hasattr(value, "to_dict"):
143
+ result[attr] = value.to_dict()
144
+ elif isinstance(value, dict):
145
+ result[attr] = dict(map(
146
+ lambda item: (item[0], item[1].to_dict())
147
+ if hasattr(item[1], "to_dict") else item,
148
+ value.items()
149
+ ))
150
+ else:
151
+ result[attr] = value
152
+ if issubclass(V1ListDeploymentAlertingPoliciesResponse, dict):
153
+ for key, value in self.items():
154
+ result[key] = value
155
+
156
+ return result
157
+
158
+ def to_str(self) -> str:
159
+ """Returns the string representation of the model"""
160
+ return pprint.pformat(self.to_dict())
161
+
162
+ def __repr__(self) -> str:
163
+ """For `print` and `pprint`"""
164
+ return self.to_str()
165
+
166
+ def __eq__(self, other: 'V1ListDeploymentAlertingPoliciesResponse') -> bool:
167
+ """Returns true if both objects are equal"""
168
+ if not isinstance(other, V1ListDeploymentAlertingPoliciesResponse):
169
+ return False
170
+
171
+ return self.__dict__ == other.__dict__
172
+
173
+ def __ne__(self, other: 'V1ListDeploymentAlertingPoliciesResponse') -> bool:
174
+ """Returns true if both objects are not equal"""
175
+ return not self == other
@@ -44,6 +44,7 @@ class V1Membership(object):
44
44
  'balance': 'float',
45
45
  'created_at': 'datetime',
46
46
  'creator_id': 'str',
47
+ 'current_storage_bytes': 'str',
47
48
  'datastore_count': 'str',
48
49
  'description': 'str',
49
50
  'display_name': 'str',
@@ -66,6 +67,7 @@ class V1Membership(object):
66
67
  'balance': 'balance',
67
68
  'created_at': 'createdAt',
68
69
  'creator_id': 'creatorId',
70
+ 'current_storage_bytes': 'currentStorageBytes',
69
71
  'datastore_count': 'datastoreCount',
70
72
  'description': 'description',
71
73
  'display_name': 'displayName',
@@ -84,11 +86,12 @@ class V1Membership(object):
84
86
  'user_id': 'userId'
85
87
  }
86
88
 
87
- def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
89
+ def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
88
90
  """V1Membership - a model defined in Swagger""" # noqa: E501
89
91
  self._balance = None
90
92
  self._created_at = None
91
93
  self._creator_id = None
94
+ self._current_storage_bytes = None
92
95
  self._datastore_count = None
93
96
  self._description = None
94
97
  self._display_name = None
@@ -112,6 +115,8 @@ class V1Membership(object):
112
115
  self.created_at = created_at
113
116
  if creator_id is not None:
114
117
  self.creator_id = creator_id
118
+ if current_storage_bytes is not None:
119
+ self.current_storage_bytes = current_storage_bytes
115
120
  if datastore_count is not None:
116
121
  self.datastore_count = datastore_count
117
122
  if description is not None:
@@ -208,6 +213,27 @@ class V1Membership(object):
208
213
 
209
214
  self._creator_id = creator_id
210
215
 
216
+ @property
217
+ def current_storage_bytes(self) -> 'str':
218
+ """Gets the current_storage_bytes of this V1Membership. # noqa: E501
219
+
220
+
221
+ :return: The current_storage_bytes of this V1Membership. # noqa: E501
222
+ :rtype: str
223
+ """
224
+ return self._current_storage_bytes
225
+
226
+ @current_storage_bytes.setter
227
+ def current_storage_bytes(self, current_storage_bytes: 'str'):
228
+ """Sets the current_storage_bytes of this V1Membership.
229
+
230
+
231
+ :param current_storage_bytes: The current_storage_bytes of this V1Membership. # noqa: E501
232
+ :type: str
233
+ """
234
+
235
+ self._current_storage_bytes = current_storage_bytes
236
+
211
237
  @property
212
238
  def datastore_count(self) -> 'str':
213
239
  """Gets the datastore_count of this V1Membership. # noqa: E501
@@ -44,6 +44,7 @@ class V1Organization(object):
44
44
  'alerts_config': 'V1AlertsConfig',
45
45
  'allow_aws_saas': 'bool',
46
46
  'allow_budgeting': 'bool',
47
+ 'allow_credits_auto_replenish': 'bool',
47
48
  'allow_dgx_saas': 'bool',
48
49
  'allow_external_project_duplication': 'bool',
49
50
  'allow_gcp_saas': 'bool',
@@ -57,6 +58,8 @@ class V1Organization(object):
57
58
  'auto_invite_by_domain': 'bool',
58
59
  'auto_join_domain_validations': 'dict(str, V1AutoJoinDomainValidation)',
59
60
  'auto_join_domains': 'list[str]',
61
+ 'auto_replenish_amount': 'float',
62
+ 'auto_replenish_threshold': 'float',
60
63
  'auto_switch_machine': 'bool',
61
64
  'created_at': 'datetime',
62
65
  'default_machine_image_version': 'str',
@@ -76,6 +79,7 @@ class V1Organization(object):
76
79
  'preferred_deployment_provider': 'str',
77
80
  'preferred_studio_provider': 'str',
78
81
  'start_studios_on_spot_instance': 'bool',
82
+ 'switch_to_default_machine_on_idle': 'bool',
79
83
  'teamspace_default_credits': 'float',
80
84
  'twitter_username': 'str',
81
85
  'updated_at': 'datetime',
@@ -86,6 +90,7 @@ class V1Organization(object):
86
90
  'alerts_config': 'alertsConfig',
87
91
  'allow_aws_saas': 'allowAwsSaas',
88
92
  'allow_budgeting': 'allowBudgeting',
93
+ 'allow_credits_auto_replenish': 'allowCreditsAutoReplenish',
89
94
  'allow_dgx_saas': 'allowDgxSaas',
90
95
  'allow_external_project_duplication': 'allowExternalProjectDuplication',
91
96
  'allow_gcp_saas': 'allowGcpSaas',
@@ -99,6 +104,8 @@ class V1Organization(object):
99
104
  'auto_invite_by_domain': 'autoInviteByDomain',
100
105
  'auto_join_domain_validations': 'autoJoinDomainValidations',
101
106
  'auto_join_domains': 'autoJoinDomains',
107
+ 'auto_replenish_amount': 'autoReplenishAmount',
108
+ 'auto_replenish_threshold': 'autoReplenishThreshold',
102
109
  'auto_switch_machine': 'autoSwitchMachine',
103
110
  'created_at': 'createdAt',
104
111
  'default_machine_image_version': 'defaultMachineImageVersion',
@@ -118,17 +125,19 @@ class V1Organization(object):
118
125
  'preferred_deployment_provider': 'preferredDeploymentProvider',
119
126
  'preferred_studio_provider': 'preferredStudioProvider',
120
127
  'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
128
+ 'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle',
121
129
  'teamspace_default_credits': 'teamspaceDefaultCredits',
122
130
  'twitter_username': 'twitterUsername',
123
131
  'updated_at': 'updatedAt',
124
132
  'workload_max_run_duration': 'workloadMaxRunDuration'
125
133
  }
126
134
 
127
- def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: '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_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, location: 'str' =None, name: '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, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None): # noqa: E501
135
+ 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_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, location: 'str' =None, name: '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, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None): # noqa: E501
128
136
  """V1Organization - a model defined in Swagger""" # noqa: E501
129
137
  self._alerts_config = None
130
138
  self._allow_aws_saas = None
131
139
  self._allow_budgeting = None
140
+ self._allow_credits_auto_replenish = None
132
141
  self._allow_dgx_saas = None
133
142
  self._allow_external_project_duplication = None
134
143
  self._allow_gcp_saas = None
@@ -142,6 +151,8 @@ class V1Organization(object):
142
151
  self._auto_invite_by_domain = None
143
152
  self._auto_join_domain_validations = None
144
153
  self._auto_join_domains = None
154
+ self._auto_replenish_amount = None
155
+ self._auto_replenish_threshold = None
145
156
  self._auto_switch_machine = None
146
157
  self._created_at = None
147
158
  self._default_machine_image_version = None
@@ -161,6 +172,7 @@ class V1Organization(object):
161
172
  self._preferred_deployment_provider = None
162
173
  self._preferred_studio_provider = None
163
174
  self._start_studios_on_spot_instance = None
175
+ self._switch_to_default_machine_on_idle = None
164
176
  self._teamspace_default_credits = None
165
177
  self._twitter_username = None
166
178
  self._updated_at = None
@@ -172,6 +184,8 @@ class V1Organization(object):
172
184
  self.allow_aws_saas = allow_aws_saas
173
185
  if allow_budgeting is not None:
174
186
  self.allow_budgeting = allow_budgeting
187
+ if allow_credits_auto_replenish is not None:
188
+ self.allow_credits_auto_replenish = allow_credits_auto_replenish
175
189
  if allow_dgx_saas is not None:
176
190
  self.allow_dgx_saas = allow_dgx_saas
177
191
  if allow_external_project_duplication is not None:
@@ -198,6 +212,10 @@ class V1Organization(object):
198
212
  self.auto_join_domain_validations = auto_join_domain_validations
199
213
  if auto_join_domains is not None:
200
214
  self.auto_join_domains = auto_join_domains
215
+ if auto_replenish_amount is not None:
216
+ self.auto_replenish_amount = auto_replenish_amount
217
+ if auto_replenish_threshold is not None:
218
+ self.auto_replenish_threshold = auto_replenish_threshold
201
219
  if auto_switch_machine is not None:
202
220
  self.auto_switch_machine = auto_switch_machine
203
221
  if created_at is not None:
@@ -236,6 +254,8 @@ class V1Organization(object):
236
254
  self.preferred_studio_provider = preferred_studio_provider
237
255
  if start_studios_on_spot_instance is not None:
238
256
  self.start_studios_on_spot_instance = start_studios_on_spot_instance
257
+ if switch_to_default_machine_on_idle is not None:
258
+ self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
239
259
  if teamspace_default_credits is not None:
240
260
  self.teamspace_default_credits = teamspace_default_credits
241
261
  if twitter_username is not None:
@@ -308,6 +328,27 @@ class V1Organization(object):
308
328
 
309
329
  self._allow_budgeting = allow_budgeting
310
330
 
331
+ @property
332
+ def allow_credits_auto_replenish(self) -> 'bool':
333
+ """Gets the allow_credits_auto_replenish of this V1Organization. # noqa: E501
334
+
335
+
336
+ :return: The allow_credits_auto_replenish of this V1Organization. # noqa: E501
337
+ :rtype: bool
338
+ """
339
+ return self._allow_credits_auto_replenish
340
+
341
+ @allow_credits_auto_replenish.setter
342
+ def allow_credits_auto_replenish(self, allow_credits_auto_replenish: 'bool'):
343
+ """Sets the allow_credits_auto_replenish of this V1Organization.
344
+
345
+
346
+ :param allow_credits_auto_replenish: The allow_credits_auto_replenish of this V1Organization. # noqa: E501
347
+ :type: bool
348
+ """
349
+
350
+ self._allow_credits_auto_replenish = allow_credits_auto_replenish
351
+
311
352
  @property
312
353
  def allow_dgx_saas(self) -> 'bool':
313
354
  """Gets the allow_dgx_saas of this V1Organization. # noqa: E501
@@ -583,6 +624,48 @@ class V1Organization(object):
583
624
 
584
625
  self._auto_join_domains = auto_join_domains
585
626
 
627
+ @property
628
+ def auto_replenish_amount(self) -> 'float':
629
+ """Gets the auto_replenish_amount of this V1Organization. # noqa: E501
630
+
631
+
632
+ :return: The auto_replenish_amount of this V1Organization. # noqa: E501
633
+ :rtype: float
634
+ """
635
+ return self._auto_replenish_amount
636
+
637
+ @auto_replenish_amount.setter
638
+ def auto_replenish_amount(self, auto_replenish_amount: 'float'):
639
+ """Sets the auto_replenish_amount of this V1Organization.
640
+
641
+
642
+ :param auto_replenish_amount: The auto_replenish_amount of this V1Organization. # noqa: E501
643
+ :type: float
644
+ """
645
+
646
+ self._auto_replenish_amount = auto_replenish_amount
647
+
648
+ @property
649
+ def auto_replenish_threshold(self) -> 'float':
650
+ """Gets the auto_replenish_threshold of this V1Organization. # noqa: E501
651
+
652
+
653
+ :return: The auto_replenish_threshold of this V1Organization. # noqa: E501
654
+ :rtype: float
655
+ """
656
+ return self._auto_replenish_threshold
657
+
658
+ @auto_replenish_threshold.setter
659
+ def auto_replenish_threshold(self, auto_replenish_threshold: 'float'):
660
+ """Sets the auto_replenish_threshold of this V1Organization.
661
+
662
+
663
+ :param auto_replenish_threshold: The auto_replenish_threshold of this V1Organization. # noqa: E501
664
+ :type: float
665
+ """
666
+
667
+ self._auto_replenish_threshold = auto_replenish_threshold
668
+
586
669
  @property
587
670
  def auto_switch_machine(self) -> 'bool':
588
671
  """Gets the auto_switch_machine of this V1Organization. # noqa: E501
@@ -982,6 +1065,27 @@ class V1Organization(object):
982
1065
 
983
1066
  self._start_studios_on_spot_instance = start_studios_on_spot_instance
984
1067
 
1068
+ @property
1069
+ def switch_to_default_machine_on_idle(self) -> 'bool':
1070
+ """Gets the switch_to_default_machine_on_idle of this V1Organization. # noqa: E501
1071
+
1072
+
1073
+ :return: The switch_to_default_machine_on_idle of this V1Organization. # noqa: E501
1074
+ :rtype: bool
1075
+ """
1076
+ return self._switch_to_default_machine_on_idle
1077
+
1078
+ @switch_to_default_machine_on_idle.setter
1079
+ def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
1080
+ """Sets the switch_to_default_machine_on_idle of this V1Organization.
1081
+
1082
+
1083
+ :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this V1Organization. # noqa: E501
1084
+ :type: bool
1085
+ """
1086
+
1087
+ self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
1088
+
985
1089
  @property
986
1090
  def teamspace_default_credits(self) -> 'float':
987
1091
  """Gets the teamspace_default_credits of this V1Organization. # noqa: E501
@@ -46,6 +46,7 @@ class V1Project(object):
46
46
  'creator_id': 'str',
47
47
  'description': 'str',
48
48
  'display_name': 'str',
49
+ 'gcp_bucket_name': 'str',
49
50
  'id': 'str',
50
51
  'is_default': 'bool',
51
52
  'lock_out_uploads': 'bool',
@@ -66,6 +67,7 @@ class V1Project(object):
66
67
  'creator_id': 'creatorId',
67
68
  'description': 'description',
68
69
  'display_name': 'displayName',
70
+ 'gcp_bucket_name': 'gcpBucketName',
69
71
  'id': 'id',
70
72
  'is_default': 'isDefault',
71
73
  'lock_out_uploads': 'lockOutUploads',
@@ -80,13 +82,14 @@ class V1Project(object):
80
82
  'updated_at': 'updatedAt'
81
83
  }
82
84
 
83
- def __init__(self, abac_enabled: 'bool' =None, created_at: 'datetime' =None, creator_id: 'str' =None, description: 'str' =None, display_name: 'str' =None, id: 'str' =None, is_default: 'bool' =None, lock_out_uploads: 'bool' =None, name: 'str' =None, number_of_files_uploads: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, private: 'bool' =None, project_settings: 'V1ProjectSettings' =None, quotas: 'V1Quotas' =None, total_size_uploads_bytes: 'str' =None, updated_at: 'datetime' =None): # noqa: E501
85
+ def __init__(self, abac_enabled: 'bool' =None, created_at: 'datetime' =None, creator_id: 'str' =None, description: 'str' =None, display_name: 'str' =None, gcp_bucket_name: 'str' =None, id: 'str' =None, is_default: 'bool' =None, lock_out_uploads: 'bool' =None, name: 'str' =None, number_of_files_uploads: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, private: 'bool' =None, project_settings: 'V1ProjectSettings' =None, quotas: 'V1Quotas' =None, total_size_uploads_bytes: 'str' =None, updated_at: 'datetime' =None): # noqa: E501
84
86
  """V1Project - a model defined in Swagger""" # noqa: E501
85
87
  self._abac_enabled = None
86
88
  self._created_at = None
87
89
  self._creator_id = None
88
90
  self._description = None
89
91
  self._display_name = None
92
+ self._gcp_bucket_name = None
90
93
  self._id = None
91
94
  self._is_default = None
92
95
  self._lock_out_uploads = None
@@ -110,6 +113,8 @@ class V1Project(object):
110
113
  self.description = description
111
114
  if display_name is not None:
112
115
  self.display_name = display_name
116
+ if gcp_bucket_name is not None:
117
+ self.gcp_bucket_name = gcp_bucket_name
113
118
  if id is not None:
114
119
  self.id = id
115
120
  if is_default is not None:
@@ -240,6 +245,27 @@ class V1Project(object):
240
245
 
241
246
  self._display_name = display_name
242
247
 
248
+ @property
249
+ def gcp_bucket_name(self) -> 'str':
250
+ """Gets the gcp_bucket_name of this V1Project. # noqa: E501
251
+
252
+
253
+ :return: The gcp_bucket_name of this V1Project. # noqa: E501
254
+ :rtype: str
255
+ """
256
+ return self._gcp_bucket_name
257
+
258
+ @gcp_bucket_name.setter
259
+ def gcp_bucket_name(self, gcp_bucket_name: 'str'):
260
+ """Sets the gcp_bucket_name of this V1Project.
261
+
262
+
263
+ :param gcp_bucket_name: The gcp_bucket_name of this V1Project. # noqa: E501
264
+ :type: str
265
+ """
266
+
267
+ self._gcp_bucket_name = gcp_bucket_name
268
+
243
269
  @property
244
270
  def id(self) -> 'str':
245
271
  """Gets the id of this V1Project. # noqa: E501
@@ -45,6 +45,7 @@ class V1ProjectMembership(object):
45
45
  'balance': 'float',
46
46
  'created_at': 'datetime',
47
47
  'creator_id': 'str',
48
+ 'current_storage_bytes': 'str',
48
49
  'datastore_count': 'str',
49
50
  'description': 'str',
50
51
  'display_name': 'str',
@@ -75,6 +76,7 @@ class V1ProjectMembership(object):
75
76
  'balance': 'balance',
76
77
  'created_at': 'createdAt',
77
78
  'creator_id': 'creatorId',
79
+ 'current_storage_bytes': 'currentStorageBytes',
78
80
  'datastore_count': 'datastoreCount',
79
81
  'description': 'description',
80
82
  'display_name': 'displayName',
@@ -100,12 +102,13 @@ class V1ProjectMembership(object):
100
102
  'username': 'username'
101
103
  }
102
104
 
103
- def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None): # noqa: E501
105
+ def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None): # noqa: E501
104
106
  """V1ProjectMembership - a model defined in Swagger""" # noqa: E501
105
107
  self._avatar_url = None
106
108
  self._balance = None
107
109
  self._created_at = None
108
110
  self._creator_id = None
111
+ self._current_storage_bytes = None
109
112
  self._datastore_count = None
110
113
  self._description = None
111
114
  self._display_name = None
@@ -138,6 +141,8 @@ class V1ProjectMembership(object):
138
141
  self.created_at = created_at
139
142
  if creator_id is not None:
140
143
  self.creator_id = creator_id
144
+ if current_storage_bytes is not None:
145
+ self.current_storage_bytes = current_storage_bytes
141
146
  if datastore_count is not None:
142
147
  self.datastore_count = datastore_count
143
148
  if description is not None:
@@ -269,6 +274,27 @@ class V1ProjectMembership(object):
269
274
 
270
275
  self._creator_id = creator_id
271
276
 
277
+ @property
278
+ def current_storage_bytes(self) -> 'str':
279
+ """Gets the current_storage_bytes of this V1ProjectMembership. # noqa: E501
280
+
281
+
282
+ :return: The current_storage_bytes of this V1ProjectMembership. # noqa: E501
283
+ :rtype: str
284
+ """
285
+ return self._current_storage_bytes
286
+
287
+ @current_storage_bytes.setter
288
+ def current_storage_bytes(self, current_storage_bytes: 'str'):
289
+ """Sets the current_storage_bytes of this V1ProjectMembership.
290
+
291
+
292
+ :param current_storage_bytes: The current_storage_bytes of this V1ProjectMembership. # noqa: E501
293
+ :type: str
294
+ """
295
+
296
+ self._current_storage_bytes = current_storage_bytes
297
+
272
298
  @property
273
299
  def datastore_count(self) -> 'str':
274
300
  """Gets the datastore_count of this V1ProjectMembership. # noqa: E501
@@ -55,7 +55,8 @@ class V1ProjectSettings(object):
55
55
  'preferred_deployment_provider': 'str',
56
56
  'preferred_studio_provider': 'str',
57
57
  'same_compute_on_resume': 'bool',
58
- 'start_studio_on_spot_instance': 'bool'
58
+ 'start_studio_on_spot_instance': 'bool',
59
+ 'switch_to_default_machine_on_idle': 'bool'
59
60
  }
60
61
 
61
62
  attribute_map = {
@@ -73,10 +74,11 @@ class V1ProjectSettings(object):
73
74
  'preferred_deployment_provider': 'preferredDeploymentProvider',
74
75
  'preferred_studio_provider': 'preferredStudioProvider',
75
76
  'same_compute_on_resume': 'sameComputeOnResume',
76
- 'start_studio_on_spot_instance': 'startStudioOnSpotInstance'
77
+ 'start_studio_on_spot_instance': 'startStudioOnSpotInstance',
78
+ 'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle'
77
79
  }
78
80
 
79
- def __init__(self, allow_aws_saas: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_switch_machine: 'bool' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None): # noqa: E501
81
+ def __init__(self, allow_aws_saas: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_switch_machine: 'bool' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None): # noqa: E501
80
82
  """V1ProjectSettings - a model defined in Swagger""" # noqa: E501
81
83
  self._allow_aws_saas = None
82
84
  self._allow_dgx_saas = None
@@ -93,6 +95,7 @@ class V1ProjectSettings(object):
93
95
  self._preferred_studio_provider = None
94
96
  self._same_compute_on_resume = None
95
97
  self._start_studio_on_spot_instance = None
98
+ self._switch_to_default_machine_on_idle = None
96
99
  self.discriminator = None
97
100
  if allow_aws_saas is not None:
98
101
  self.allow_aws_saas = allow_aws_saas
@@ -124,6 +127,8 @@ class V1ProjectSettings(object):
124
127
  self.same_compute_on_resume = same_compute_on_resume
125
128
  if start_studio_on_spot_instance is not None:
126
129
  self.start_studio_on_spot_instance = start_studio_on_spot_instance
130
+ if switch_to_default_machine_on_idle is not None:
131
+ self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
127
132
 
128
133
  @property
129
134
  def allow_aws_saas(self) -> 'bool':
@@ -442,6 +447,27 @@ class V1ProjectSettings(object):
442
447
 
443
448
  self._start_studio_on_spot_instance = start_studio_on_spot_instance
444
449
 
450
+ @property
451
+ def switch_to_default_machine_on_idle(self) -> 'bool':
452
+ """Gets the switch_to_default_machine_on_idle of this V1ProjectSettings. # noqa: E501
453
+
454
+
455
+ :return: The switch_to_default_machine_on_idle of this V1ProjectSettings. # noqa: E501
456
+ :rtype: bool
457
+ """
458
+ return self._switch_to_default_machine_on_idle
459
+
460
+ @switch_to_default_machine_on_idle.setter
461
+ def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
462
+ """Sets the switch_to_default_machine_on_idle of this V1ProjectSettings.
463
+
464
+
465
+ :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this V1ProjectSettings. # noqa: E501
466
+ :type: bool
467
+ """
468
+
469
+ self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
470
+
445
471
  def to_dict(self) -> dict:
446
472
  """Returns the model properties as a dict"""
447
473
  result = {}