lightning-sdk 0.2.5__py3-none-any.whl → 0.2.7__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 (65) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/ai_hub_api.py +1 -0
  3. lightning_sdk/api/lit_container_api.py +84 -24
  4. lightning_sdk/api/teamspace_api.py +8 -9
  5. lightning_sdk/api/utils.py +0 -1
  6. lightning_sdk/cli/docker.py +1 -1
  7. lightning_sdk/cli/download.py +10 -2
  8. lightning_sdk/cli/serve.py +26 -19
  9. lightning_sdk/cli/upload.py +41 -6
  10. lightning_sdk/lightning_cloud/openapi/__init__.py +16 -0
  11. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +9 -9
  12. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +202 -0
  13. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  14. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +15 -3
  15. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  16. lightning_sdk/lightning_cloud/openapi/models/alerts_config_billing.py +175 -0
  17. lightning_sdk/lightning_cloud/openapi/models/alerts_config_studios.py +149 -0
  18. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +27 -1
  20. lightning_sdk/lightning_cloud/openapi/models/externalv1_cluster.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +53 -1
  22. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py +201 -0
  25. lightning_sdk/lightning_cloud/openapi/models/v1_alert_method.py +102 -0
  26. lightning_sdk/lightning_cloud/openapi/models/v1_alerts_config.py +149 -0
  27. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics.py +617 -0
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +131 -1
  32. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  33. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_create_server_alert_response.py +97 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +81 -3
  38. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +1 -29
  39. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_cold_start_metrics_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_registry_repository_image_artifact_versions_response.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +29 -3
  42. lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_report_cloud_space_instance_stop_at_response.py +97 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_reservation_details.py +201 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert.py +201 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_phase.py +104 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py +103 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +105 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_subnet_spec.py +149 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +131 -79
  54. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +29 -29
  55. lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +1 -27
  56. lightning_sdk/lit_container.py +19 -4
  57. lightning_sdk/models.py +1 -1
  58. lightning_sdk/serve.py +86 -15
  59. lightning_sdk/teamspace.py +32 -18
  60. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/METADATA +1 -1
  61. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/RECORD +65 -49
  62. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/LICENSE +0 -0
  63. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/WHEEL +0 -0
  64. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/entry_points.txt +0 -0
  65. {lightning_sdk-0.2.5.dist-info → lightning_sdk-0.2.7.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,97 @@
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 V1CreateServerAlertResponse(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
+ }
45
+
46
+ attribute_map = {
47
+ }
48
+
49
+ def __init__(self): # noqa: E501
50
+ """V1CreateServerAlertResponse - a model defined in Swagger""" # noqa: E501
51
+ self.discriminator = None
52
+
53
+ def to_dict(self) -> dict:
54
+ """Returns the model properties as a dict"""
55
+ result = {}
56
+
57
+ for attr, _ in six.iteritems(self.swagger_types):
58
+ value = getattr(self, attr)
59
+ if isinstance(value, list):
60
+ result[attr] = list(map(
61
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
62
+ value
63
+ ))
64
+ elif hasattr(value, "to_dict"):
65
+ result[attr] = value.to_dict()
66
+ elif isinstance(value, dict):
67
+ result[attr] = dict(map(
68
+ lambda item: (item[0], item[1].to_dict())
69
+ if hasattr(item[1], "to_dict") else item,
70
+ value.items()
71
+ ))
72
+ else:
73
+ result[attr] = value
74
+ if issubclass(V1CreateServerAlertResponse, dict):
75
+ for key, value in self.items():
76
+ result[key] = value
77
+
78
+ return result
79
+
80
+ def to_str(self) -> str:
81
+ """Returns the string representation of the model"""
82
+ return pprint.pformat(self.to_dict())
83
+
84
+ def __repr__(self) -> str:
85
+ """For `print` and `pprint`"""
86
+ return self.to_str()
87
+
88
+ def __eq__(self, other: 'V1CreateServerAlertResponse') -> bool:
89
+ """Returns true if both objects are equal"""
90
+ if not isinstance(other, V1CreateServerAlertResponse):
91
+ return False
92
+
93
+ return self.__dict__ == other.__dict__
94
+
95
+ def __ne__(self, other: 'V1CreateServerAlertResponse') -> bool:
96
+ """Returns true if both objects are not equal"""
97
+ return not self == other
@@ -47,6 +47,7 @@ class V1DataConnection(object):
47
47
  'cluster_id': 'str',
48
48
  'created_at': 'datetime',
49
49
  'efs': 'V1EfsConfig',
50
+ 'error': 'str',
50
51
  'filestore': 'V1FilestoreDataConnection',
51
52
  'gcp': 'V1GcpDataConnection',
52
53
  'gcs_folder': 'V1GCSFolderDataConnection',
@@ -74,6 +75,7 @@ class V1DataConnection(object):
74
75
  'cluster_id': 'clusterId',
75
76
  'created_at': 'createdAt',
76
77
  'efs': 'efs',
78
+ 'error': 'error',
77
79
  'filestore': 'filestore',
78
80
  'gcp': 'gcp',
79
81
  'gcs_folder': 'gcsFolder',
@@ -94,7 +96,7 @@ class V1DataConnection(object):
94
96
  'writable': 'writable'
95
97
  }
96
98
 
97
- def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None): # noqa: E501
99
+ def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, error: 'str' =None, filestore: 'V1FilestoreDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None): # noqa: E501
98
100
  """V1DataConnection - a model defined in Swagger""" # noqa: E501
99
101
  self._access_cluster_ids = None
100
102
  self._accessible = None
@@ -102,6 +104,7 @@ class V1DataConnection(object):
102
104
  self._cluster_id = None
103
105
  self._created_at = None
104
106
  self._efs = None
107
+ self._error = None
105
108
  self._filestore = None
106
109
  self._gcp = None
107
110
  self._gcs_folder = None
@@ -133,6 +136,8 @@ class V1DataConnection(object):
133
136
  self.created_at = created_at
134
137
  if efs is not None:
135
138
  self.efs = efs
139
+ if error is not None:
140
+ self.error = error
136
141
  if filestore is not None:
137
142
  self.filestore = filestore
138
143
  if gcp is not None:
@@ -296,6 +301,27 @@ class V1DataConnection(object):
296
301
 
297
302
  self._efs = efs
298
303
 
304
+ @property
305
+ def error(self) -> 'str':
306
+ """Gets the error of this V1DataConnection. # noqa: E501
307
+
308
+
309
+ :return: The error of this V1DataConnection. # noqa: E501
310
+ :rtype: str
311
+ """
312
+ return self._error
313
+
314
+ @error.setter
315
+ def error(self, error: 'str'):
316
+ """Sets the error of this V1DataConnection.
317
+
318
+
319
+ :param error: The error of this V1DataConnection. # noqa: E501
320
+ :type: str
321
+ """
322
+
323
+ self._error = error
324
+
299
325
  @property
300
326
  def filestore(self) -> 'V1FilestoreDataConnection':
301
327
  """Gets the filestore of this V1DataConnection. # noqa: E501
@@ -45,12 +45,15 @@ class V1GoogleCloudDirectV1(object):
45
45
  'compute_project_role': 'str',
46
46
  'credentials_secret_id': 'str',
47
47
  'credentials_service_account_email': 'str',
48
+ 'custom_subnet_mode': 'bool',
48
49
  'primary_region': 'str',
49
50
  'project_id': 'str',
50
51
  'project_sa_enabled': 'bool',
51
52
  'regions': 'list[str]',
52
53
  'service_account_email': 'str',
53
- 'source_cidr_ips': 'list[str]'
54
+ 'source_cidr_ips': 'list[str]',
55
+ 'subnets': 'list[V1SubnetSpec]',
56
+ 'vpc_networks': 'list[str]'
54
57
  }
55
58
 
56
59
  attribute_map = {
@@ -58,26 +61,32 @@ class V1GoogleCloudDirectV1(object):
58
61
  'compute_project_role': 'computeProjectRole',
59
62
  'credentials_secret_id': 'credentialsSecretId',
60
63
  'credentials_service_account_email': 'credentialsServiceAccountEmail',
64
+ 'custom_subnet_mode': 'customSubnetMode',
61
65
  'primary_region': 'primaryRegion',
62
66
  'project_id': 'projectId',
63
67
  'project_sa_enabled': 'projectSaEnabled',
64
68
  'regions': 'regions',
65
69
  'service_account_email': 'serviceAccountEmail',
66
- 'source_cidr_ips': 'sourceCidrIps'
70
+ 'source_cidr_ips': 'sourceCidrIps',
71
+ 'subnets': 'subnets',
72
+ 'vpc_networks': 'vpcNetworks'
67
73
  }
68
74
 
69
- def __init__(self, bucket_name: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, project_sa_enabled: 'bool' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None): # noqa: E501
75
+ def __init__(self, bucket_name: 'str' =None, compute_project_role: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, custom_subnet_mode: 'bool' =None, primary_region: 'str' =None, project_id: 'str' =None, project_sa_enabled: 'bool' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None, subnets: 'list[V1SubnetSpec]' =None, vpc_networks: 'list[str]' =None): # noqa: E501
70
76
  """V1GoogleCloudDirectV1 - a model defined in Swagger""" # noqa: E501
71
77
  self._bucket_name = None
72
78
  self._compute_project_role = None
73
79
  self._credentials_secret_id = None
74
80
  self._credentials_service_account_email = None
81
+ self._custom_subnet_mode = None
75
82
  self._primary_region = None
76
83
  self._project_id = None
77
84
  self._project_sa_enabled = None
78
85
  self._regions = None
79
86
  self._service_account_email = None
80
87
  self._source_cidr_ips = None
88
+ self._subnets = None
89
+ self._vpc_networks = None
81
90
  self.discriminator = None
82
91
  if bucket_name is not None:
83
92
  self.bucket_name = bucket_name
@@ -87,6 +96,8 @@ class V1GoogleCloudDirectV1(object):
87
96
  self.credentials_secret_id = credentials_secret_id
88
97
  if credentials_service_account_email is not None:
89
98
  self.credentials_service_account_email = credentials_service_account_email
99
+ if custom_subnet_mode is not None:
100
+ self.custom_subnet_mode = custom_subnet_mode
90
101
  if primary_region is not None:
91
102
  self.primary_region = primary_region
92
103
  if project_id is not None:
@@ -99,6 +110,10 @@ class V1GoogleCloudDirectV1(object):
99
110
  self.service_account_email = service_account_email
100
111
  if source_cidr_ips is not None:
101
112
  self.source_cidr_ips = source_cidr_ips
113
+ if subnets is not None:
114
+ self.subnets = subnets
115
+ if vpc_networks is not None:
116
+ self.vpc_networks = vpc_networks
102
117
 
103
118
  @property
104
119
  def bucket_name(self) -> 'str':
@@ -186,6 +201,27 @@ class V1GoogleCloudDirectV1(object):
186
201
 
187
202
  self._credentials_service_account_email = credentials_service_account_email
188
203
 
204
+ @property
205
+ def custom_subnet_mode(self) -> 'bool':
206
+ """Gets the custom_subnet_mode of this V1GoogleCloudDirectV1. # noqa: E501
207
+
208
+
209
+ :return: The custom_subnet_mode of this V1GoogleCloudDirectV1. # noqa: E501
210
+ :rtype: bool
211
+ """
212
+ return self._custom_subnet_mode
213
+
214
+ @custom_subnet_mode.setter
215
+ def custom_subnet_mode(self, custom_subnet_mode: 'bool'):
216
+ """Sets the custom_subnet_mode of this V1GoogleCloudDirectV1.
217
+
218
+
219
+ :param custom_subnet_mode: The custom_subnet_mode of this V1GoogleCloudDirectV1. # noqa: E501
220
+ :type: bool
221
+ """
222
+
223
+ self._custom_subnet_mode = custom_subnet_mode
224
+
189
225
  @property
190
226
  def primary_region(self) -> 'str':
191
227
  """Gets the primary_region of this V1GoogleCloudDirectV1. # noqa: E501
@@ -316,6 +352,48 @@ class V1GoogleCloudDirectV1(object):
316
352
 
317
353
  self._source_cidr_ips = source_cidr_ips
318
354
 
355
+ @property
356
+ def subnets(self) -> 'list[V1SubnetSpec]':
357
+ """Gets the subnets of this V1GoogleCloudDirectV1. # noqa: E501
358
+
359
+
360
+ :return: The subnets of this V1GoogleCloudDirectV1. # noqa: E501
361
+ :rtype: list[V1SubnetSpec]
362
+ """
363
+ return self._subnets
364
+
365
+ @subnets.setter
366
+ def subnets(self, subnets: 'list[V1SubnetSpec]'):
367
+ """Sets the subnets of this V1GoogleCloudDirectV1.
368
+
369
+
370
+ :param subnets: The subnets of this V1GoogleCloudDirectV1. # noqa: E501
371
+ :type: list[V1SubnetSpec]
372
+ """
373
+
374
+ self._subnets = subnets
375
+
376
+ @property
377
+ def vpc_networks(self) -> 'list[str]':
378
+ """Gets the vpc_networks of this V1GoogleCloudDirectV1. # noqa: E501
379
+
380
+
381
+ :return: The vpc_networks of this V1GoogleCloudDirectV1. # noqa: E501
382
+ :rtype: list[str]
383
+ """
384
+ return self._vpc_networks
385
+
386
+ @vpc_networks.setter
387
+ def vpc_networks(self, vpc_networks: 'list[str]'):
388
+ """Sets the vpc_networks of this V1GoogleCloudDirectV1.
389
+
390
+
391
+ :param vpc_networks: The vpc_networks of this V1GoogleCloudDirectV1. # noqa: E501
392
+ :type: list[str]
393
+ """
394
+
395
+ self._vpc_networks = vpc_networks
396
+
319
397
  def to_dict(self) -> dict:
320
398
  """Returns the model properties as a dict"""
321
399
  result = {}
@@ -42,29 +42,24 @@ class V1LambdaLabsDirectV1(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'credentials_secret_id': 'str',
45
- 'parent_cluster_id': 'str',
46
45
  'primary_region': 'str',
47
46
  'regions': 'list[str]'
48
47
  }
49
48
 
50
49
  attribute_map = {
51
50
  'credentials_secret_id': 'credentialsSecretId',
52
- 'parent_cluster_id': 'parentClusterId',
53
51
  'primary_region': 'primaryRegion',
54
52
  'regions': 'regions'
55
53
  }
56
54
 
57
- def __init__(self, credentials_secret_id: 'str' =None, parent_cluster_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None): # noqa: E501
55
+ def __init__(self, credentials_secret_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None): # noqa: E501
58
56
  """V1LambdaLabsDirectV1 - a model defined in Swagger""" # noqa: E501
59
57
  self._credentials_secret_id = None
60
- self._parent_cluster_id = None
61
58
  self._primary_region = None
62
59
  self._regions = None
63
60
  self.discriminator = None
64
61
  if credentials_secret_id is not None:
65
62
  self.credentials_secret_id = credentials_secret_id
66
- if parent_cluster_id is not None:
67
- self.parent_cluster_id = parent_cluster_id
68
63
  if primary_region is not None:
69
64
  self.primary_region = primary_region
70
65
  if regions is not None:
@@ -93,29 +88,6 @@ class V1LambdaLabsDirectV1(object):
93
88
 
94
89
  self._credentials_secret_id = credentials_secret_id
95
90
 
96
- @property
97
- def parent_cluster_id(self) -> 'str':
98
- """Gets the parent_cluster_id of this V1LambdaLabsDirectV1. # noqa: E501
99
-
100
- Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that. # noqa: E501
101
-
102
- :return: The parent_cluster_id of this V1LambdaLabsDirectV1. # noqa: E501
103
- :rtype: str
104
- """
105
- return self._parent_cluster_id
106
-
107
- @parent_cluster_id.setter
108
- def parent_cluster_id(self, parent_cluster_id: 'str'):
109
- """Sets the parent_cluster_id of this V1LambdaLabsDirectV1.
110
-
111
- Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that. # noqa: E501
112
-
113
- :param parent_cluster_id: The parent_cluster_id of this V1LambdaLabsDirectV1. # noqa: E501
114
- :type: str
115
- """
116
-
117
- self._parent_cluster_id = parent_cluster_id
118
-
119
91
  @property
120
92
  def primary_region(self) -> 'str':
121
93
  """Gets the primary_region of this V1LambdaLabsDirectV1. # 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 V1ListCloudSpaceColdStartMetricsResponse(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
+ 'metrics': 'list[V1CloudSpaceColdStartMetrics]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'metrics': 'metrics'
49
+ }
50
+
51
+ def __init__(self, metrics: 'list[V1CloudSpaceColdStartMetrics]' =None): # noqa: E501
52
+ """V1ListCloudSpaceColdStartMetricsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._metrics = None
54
+ self.discriminator = None
55
+ if metrics is not None:
56
+ self.metrics = metrics
57
+
58
+ @property
59
+ def metrics(self) -> 'list[V1CloudSpaceColdStartMetrics]':
60
+ """Gets the metrics of this V1ListCloudSpaceColdStartMetricsResponse. # noqa: E501
61
+
62
+
63
+ :return: The metrics of this V1ListCloudSpaceColdStartMetricsResponse. # noqa: E501
64
+ :rtype: list[V1CloudSpaceColdStartMetrics]
65
+ """
66
+ return self._metrics
67
+
68
+ @metrics.setter
69
+ def metrics(self, metrics: 'list[V1CloudSpaceColdStartMetrics]'):
70
+ """Sets the metrics of this V1ListCloudSpaceColdStartMetricsResponse.
71
+
72
+
73
+ :param metrics: The metrics of this V1ListCloudSpaceColdStartMetricsResponse. # noqa: E501
74
+ :type: list[V1CloudSpaceColdStartMetrics]
75
+ """
76
+
77
+ self._metrics = metrics
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(V1ListCloudSpaceColdStartMetricsResponse, 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: 'V1ListCloudSpaceColdStartMetricsResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1ListCloudSpaceColdStartMetricsResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1ListCloudSpaceColdStartMetricsResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -42,6 +42,7 @@ class V1ListLitRegistryRepositoryImageArtifactVersionsResponse(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'artifacts': 'list[V1LitRegistryArtifact]',
45
+ 'cluster_id': 'str',
45
46
  'latest_artifact': 'V1LitRegistryArtifact',
46
47
  'latest_tag_name': 'str',
47
48
  'lit_repo_name': 'str',
@@ -50,15 +51,17 @@ class V1ListLitRegistryRepositoryImageArtifactVersionsResponse(object):
50
51
 
51
52
  attribute_map = {
52
53
  'artifacts': 'artifacts',
54
+ 'cluster_id': 'clusterId',
53
55
  'latest_artifact': 'latestArtifact',
54
56
  'latest_tag_name': 'latestTagName',
55
57
  'lit_repo_name': 'litRepoName',
56
58
  'project_id': 'projectId'
57
59
  }
58
60
 
59
- def __init__(self, artifacts: 'list[V1LitRegistryArtifact]' =None, latest_artifact: 'V1LitRegistryArtifact' =None, latest_tag_name: 'str' =None, lit_repo_name: 'str' =None, project_id: 'str' =None): # noqa: E501
61
+ def __init__(self, artifacts: 'list[V1LitRegistryArtifact]' =None, cluster_id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, latest_tag_name: 'str' =None, lit_repo_name: 'str' =None, project_id: 'str' =None): # noqa: E501
60
62
  """V1ListLitRegistryRepositoryImageArtifactVersionsResponse - a model defined in Swagger""" # noqa: E501
61
63
  self._artifacts = None
64
+ self._cluster_id = None
62
65
  self._latest_artifact = None
63
66
  self._latest_tag_name = None
64
67
  self._lit_repo_name = None
@@ -66,6 +69,8 @@ class V1ListLitRegistryRepositoryImageArtifactVersionsResponse(object):
66
69
  self.discriminator = None
67
70
  if artifacts is not None:
68
71
  self.artifacts = artifacts
72
+ if cluster_id is not None:
73
+ self.cluster_id = cluster_id
69
74
  if latest_artifact is not None:
70
75
  self.latest_artifact = latest_artifact
71
76
  if latest_tag_name is not None:
@@ -98,6 +103,27 @@ class V1ListLitRegistryRepositoryImageArtifactVersionsResponse(object):
98
103
 
99
104
  self._artifacts = artifacts
100
105
 
106
+ @property
107
+ def cluster_id(self) -> 'str':
108
+ """Gets the cluster_id of this V1ListLitRegistryRepositoryImageArtifactVersionsResponse. # noqa: E501
109
+
110
+
111
+ :return: The cluster_id of this V1ListLitRegistryRepositoryImageArtifactVersionsResponse. # noqa: E501
112
+ :rtype: str
113
+ """
114
+ return self._cluster_id
115
+
116
+ @cluster_id.setter
117
+ def cluster_id(self, cluster_id: 'str'):
118
+ """Sets the cluster_id of this V1ListLitRegistryRepositoryImageArtifactVersionsResponse.
119
+
120
+
121
+ :param cluster_id: The cluster_id of this V1ListLitRegistryRepositoryImageArtifactVersionsResponse. # noqa: E501
122
+ :type: str
123
+ """
124
+
125
+ self._cluster_id = cluster_id
126
+
101
127
  @property
102
128
  def latest_artifact(self) -> 'V1LitRegistryArtifact':
103
129
  """Gets the latest_artifact of this V1ListLitRegistryRepositoryImageArtifactVersionsResponse. # noqa: E501
@@ -50,7 +50,8 @@ class V1Message(object):
50
50
  'executable': 'bool',
51
51
  'id': 'str',
52
52
  'model': 'str',
53
- 'prompt_tokens': 'str'
53
+ 'prompt_tokens': 'str',
54
+ 'throughput': 'float'
54
55
  }
55
56
 
56
57
  attribute_map = {
@@ -63,10 +64,11 @@ class V1Message(object):
63
64
  'executable': 'executable',
64
65
  'id': 'id',
65
66
  'model': 'model',
66
- 'prompt_tokens': 'promptTokens'
67
+ 'prompt_tokens': 'promptTokens',
68
+ 'throughput': 'throughput'
67
69
  }
68
70
 
69
- def __init__(self, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: 'list[V1MessageContent]' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, executable: 'bool' =None, id: 'str' =None, model: 'str' =None, prompt_tokens: 'str' =None): # noqa: E501
71
+ def __init__(self, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: 'list[V1MessageContent]' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, executable: 'bool' =None, id: 'str' =None, model: 'str' =None, prompt_tokens: 'str' =None, throughput: 'float' =None): # noqa: E501
70
72
  """V1Message - a model defined in Swagger""" # noqa: E501
71
73
  self._assistant_id = None
72
74
  self._author = None
@@ -78,6 +80,7 @@ class V1Message(object):
78
80
  self._id = None
79
81
  self._model = None
80
82
  self._prompt_tokens = None
83
+ self._throughput = None
81
84
  self.discriminator = None
82
85
  if assistant_id is not None:
83
86
  self.assistant_id = assistant_id
@@ -99,6 +102,8 @@ class V1Message(object):
99
102
  self.model = model
100
103
  if prompt_tokens is not None:
101
104
  self.prompt_tokens = prompt_tokens
105
+ if throughput is not None:
106
+ self.throughput = throughput
102
107
 
103
108
  @property
104
109
  def assistant_id(self) -> 'str':
@@ -310,6 +315,27 @@ class V1Message(object):
310
315
 
311
316
  self._prompt_tokens = prompt_tokens
312
317
 
318
+ @property
319
+ def throughput(self) -> 'float':
320
+ """Gets the throughput of this V1Message. # noqa: E501
321
+
322
+
323
+ :return: The throughput of this V1Message. # noqa: E501
324
+ :rtype: float
325
+ """
326
+ return self._throughput
327
+
328
+ @throughput.setter
329
+ def throughput(self, throughput: 'float'):
330
+ """Sets the throughput of this V1Message.
331
+
332
+
333
+ :param throughput: The throughput of this V1Message. # noqa: E501
334
+ :type: float
335
+ """
336
+
337
+ self._throughput = throughput
338
+
313
339
  def to_dict(self) -> dict:
314
340
  """Returns the model properties as a dict"""
315
341
  result = {}