lightning-sdk 0.1.41__py3-none-any.whl → 0.1.42__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/ai_hub.py +8 -3
  3. lightning_sdk/api/ai_hub_api.py +3 -3
  4. lightning_sdk/api/deployment_api.py +6 -6
  5. lightning_sdk/api/job_api.py +32 -6
  6. lightning_sdk/api/mmt_api.py +59 -19
  7. lightning_sdk/api/studio_api.py +37 -19
  8. lightning_sdk/api/teamspace_api.py +34 -29
  9. lightning_sdk/api/utils.py +46 -34
  10. lightning_sdk/cli/ai_hub.py +3 -3
  11. lightning_sdk/cli/entrypoint.py +3 -1
  12. lightning_sdk/cli/mmt.py +11 -10
  13. lightning_sdk/cli/run.py +9 -8
  14. lightning_sdk/cli/serve.py +130 -0
  15. lightning_sdk/deployment/deployment.py +18 -12
  16. lightning_sdk/job/base.py +118 -24
  17. lightning_sdk/job/job.py +87 -9
  18. lightning_sdk/job/v1.py +75 -18
  19. lightning_sdk/job/v2.py +51 -15
  20. lightning_sdk/job/work.py +36 -7
  21. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  22. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +215 -5
  23. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +218 -0
  24. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +226 -0
  25. lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py +21 -1
  26. lightning_sdk/lightning_cloud/openapi/models/__init__.py +12 -0
  27. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  29. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +29 -3
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +15 -67
  32. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +17 -69
  33. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +29 -81
  34. lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py +17 -69
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +17 -17
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event_type.py +104 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +29 -81
  50. lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py +29 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_trainium_system_metrics.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -53
  54. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  56. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  57. lightning_sdk/mmt/__init__.py +3 -0
  58. lightning_sdk/{_mmt → mmt}/base.py +20 -14
  59. lightning_sdk/{_mmt → mmt}/mmt.py +46 -17
  60. lightning_sdk/mmt/v1.py +129 -0
  61. lightning_sdk/{_mmt → mmt}/v2.py +16 -21
  62. lightning_sdk/plugin.py +43 -16
  63. lightning_sdk/services/file_endpoint.py +11 -5
  64. lightning_sdk/studio.py +16 -9
  65. lightning_sdk/teamspace.py +21 -8
  66. lightning_sdk/utils/resolve.py +18 -0
  67. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/METADATA +3 -1
  68. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/RECORD +72 -59
  69. lightning_sdk/_mmt/__init__.py +0 -3
  70. lightning_sdk/_mmt/v1.py +0 -69
  71. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/LICENSE +0 -0
  72. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/WHEEL +0 -0
  73. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/entry_points.txt +0 -0
  74. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/top_level.txt +0 -0
@@ -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 IdVisibilityBody(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
+ 'visibility': 'V1ResourceVisibility'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'visibility': 'visibility'
49
+ }
50
+
51
+ def __init__(self, visibility: 'V1ResourceVisibility' =None): # noqa: E501
52
+ """IdVisibilityBody - a model defined in Swagger""" # noqa: E501
53
+ self._visibility = None
54
+ self.discriminator = None
55
+ if visibility is not None:
56
+ self.visibility = visibility
57
+
58
+ @property
59
+ def visibility(self) -> 'V1ResourceVisibility':
60
+ """Gets the visibility of this IdVisibilityBody. # noqa: E501
61
+
62
+
63
+ :return: The visibility of this IdVisibilityBody. # noqa: E501
64
+ :rtype: V1ResourceVisibility
65
+ """
66
+ return self._visibility
67
+
68
+ @visibility.setter
69
+ def visibility(self, visibility: 'V1ResourceVisibility'):
70
+ """Sets the visibility of this IdVisibilityBody.
71
+
72
+
73
+ :param visibility: The visibility of this IdVisibilityBody. # noqa: E501
74
+ :type: V1ResourceVisibility
75
+ """
76
+
77
+ self._visibility = visibility
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(IdVisibilityBody, 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: 'IdVisibilityBody') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, IdVisibilityBody):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'IdVisibilityBody') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -41,19 +41,24 @@ class ModelIdVersionsBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'cluster_id': 'str'
44
+ 'cluster_id': 'str',
45
+ 'version': 'str'
45
46
  }
46
47
 
47
48
  attribute_map = {
48
- 'cluster_id': 'clusterId'
49
+ 'cluster_id': 'clusterId',
50
+ 'version': 'version'
49
51
  }
50
52
 
51
- def __init__(self, cluster_id: 'str' =None): # noqa: E501
53
+ def __init__(self, cluster_id: 'str' =None, version: 'str' =None): # noqa: E501
52
54
  """ModelIdVersionsBody - a model defined in Swagger""" # noqa: E501
53
55
  self._cluster_id = None
56
+ self._version = None
54
57
  self.discriminator = None
55
58
  if cluster_id is not None:
56
59
  self.cluster_id = cluster_id
60
+ if version is not None:
61
+ self.version = version
57
62
 
58
63
  @property
59
64
  def cluster_id(self) -> 'str':
@@ -76,6 +81,27 @@ class ModelIdVersionsBody(object):
76
81
 
77
82
  self._cluster_id = cluster_id
78
83
 
84
+ @property
85
+ def version(self) -> 'str':
86
+ """Gets the version of this ModelIdVersionsBody. # noqa: E501
87
+
88
+
89
+ :return: The version of this ModelIdVersionsBody. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._version
93
+
94
+ @version.setter
95
+ def version(self, version: 'str'):
96
+ """Sets the version of this ModelIdVersionsBody.
97
+
98
+
99
+ :param version: The version of this ModelIdVersionsBody. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._version = version
104
+
79
105
  def to_dict(self) -> dict:
80
106
  """Returns the model properties as a dict"""
81
107
  result = {}
@@ -42,6 +42,7 @@ class ProjectIdMultimachinejobsBody(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'cluster_id': 'str',
45
+ 'fault_tolerance': 'V1MultiMachineJobFaultTolerance',
45
46
  'machines': 'int',
46
47
  'name': 'str',
47
48
  'spec': 'V1JobSpec'
@@ -49,20 +50,24 @@ class ProjectIdMultimachinejobsBody(object):
49
50
 
50
51
  attribute_map = {
51
52
  'cluster_id': 'clusterId',
53
+ 'fault_tolerance': 'faultTolerance',
52
54
  'machines': 'machines',
53
55
  'name': 'name',
54
56
  'spec': 'spec'
55
57
  }
56
58
 
57
- def __init__(self, cluster_id: 'str' =None, machines: 'int' =None, name: 'str' =None, spec: 'V1JobSpec' =None): # noqa: E501
59
+ def __init__(self, cluster_id: 'str' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, machines: 'int' =None, name: 'str' =None, spec: 'V1JobSpec' =None): # noqa: E501
58
60
  """ProjectIdMultimachinejobsBody - a model defined in Swagger""" # noqa: E501
59
61
  self._cluster_id = None
62
+ self._fault_tolerance = None
60
63
  self._machines = None
61
64
  self._name = None
62
65
  self._spec = None
63
66
  self.discriminator = None
64
67
  if cluster_id is not None:
65
68
  self.cluster_id = cluster_id
69
+ if fault_tolerance is not None:
70
+ self.fault_tolerance = fault_tolerance
66
71
  if machines is not None:
67
72
  self.machines = machines
68
73
  if name is not None:
@@ -91,6 +96,27 @@ class ProjectIdMultimachinejobsBody(object):
91
96
 
92
97
  self._cluster_id = cluster_id
93
98
 
99
+ @property
100
+ def fault_tolerance(self) -> 'V1MultiMachineJobFaultTolerance':
101
+ """Gets the fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
102
+
103
+
104
+ :return: The fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
105
+ :rtype: V1MultiMachineJobFaultTolerance
106
+ """
107
+ return self._fault_tolerance
108
+
109
+ @fault_tolerance.setter
110
+ def fault_tolerance(self, fault_tolerance: 'V1MultiMachineJobFaultTolerance'):
111
+ """Sets the fault_tolerance of this ProjectIdMultimachinejobsBody.
112
+
113
+
114
+ :param fault_tolerance: The fault_tolerance of this ProjectIdMultimachinejobsBody. # noqa: E501
115
+ :type: V1MultiMachineJobFaultTolerance
116
+ """
117
+
118
+ self._fault_tolerance = fault_tolerance
119
+
94
120
  @property
95
121
  def machines(self) -> 'int':
96
122
  """Gets the machines of this ProjectIdMultimachinejobsBody. # noqa: E501
@@ -41,92 +41,40 @@ class ProjectIdSnowflakeBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'account': 'str',
45
- 'password': 'str',
46
- 'username': 'str'
44
+ 'credential_id': 'str'
47
45
  }
48
46
 
49
47
  attribute_map = {
50
- 'account': 'account',
51
- 'password': 'password',
52
- 'username': 'username'
48
+ 'credential_id': 'credentialId'
53
49
  }
54
50
 
55
- def __init__(self, account: 'str' =None, password: 'str' =None, username: 'str' =None): # noqa: E501
51
+ def __init__(self, credential_id: 'str' =None): # noqa: E501
56
52
  """ProjectIdSnowflakeBody - a model defined in Swagger""" # noqa: E501
57
- self._account = None
58
- self._password = None
59
- self._username = None
53
+ self._credential_id = None
60
54
  self.discriminator = None
61
- if account is not None:
62
- self.account = account
63
- if password is not None:
64
- self.password = password
65
- if username is not None:
66
- self.username = username
55
+ if credential_id is not None:
56
+ self.credential_id = credential_id
67
57
 
68
58
  @property
69
- def account(self) -> 'str':
70
- """Gets the account of this ProjectIdSnowflakeBody. # noqa: E501
59
+ def credential_id(self) -> 'str':
60
+ """Gets the credential_id of this ProjectIdSnowflakeBody. # noqa: E501
71
61
 
72
62
 
73
- :return: The account of this ProjectIdSnowflakeBody. # noqa: E501
63
+ :return: The credential_id of this ProjectIdSnowflakeBody. # noqa: E501
74
64
  :rtype: str
75
65
  """
76
- return self._account
66
+ return self._credential_id
77
67
 
78
- @account.setter
79
- def account(self, account: 'str'):
80
- """Sets the account of this ProjectIdSnowflakeBody.
68
+ @credential_id.setter
69
+ def credential_id(self, credential_id: 'str'):
70
+ """Sets the credential_id of this ProjectIdSnowflakeBody.
81
71
 
82
72
 
83
- :param account: The account of this ProjectIdSnowflakeBody. # noqa: E501
73
+ :param credential_id: The credential_id of this ProjectIdSnowflakeBody. # noqa: E501
84
74
  :type: str
85
75
  """
86
76
 
87
- self._account = account
88
-
89
- @property
90
- def password(self) -> 'str':
91
- """Gets the password of this ProjectIdSnowflakeBody. # noqa: E501
92
-
93
-
94
- :return: The password of this ProjectIdSnowflakeBody. # noqa: E501
95
- :rtype: str
96
- """
97
- return self._password
98
-
99
- @password.setter
100
- def password(self, password: 'str'):
101
- """Sets the password of this ProjectIdSnowflakeBody.
102
-
103
-
104
- :param password: The password of this ProjectIdSnowflakeBody. # noqa: E501
105
- :type: str
106
- """
107
-
108
- self._password = password
109
-
110
- @property
111
- def username(self) -> 'str':
112
- """Gets the username of this ProjectIdSnowflakeBody. # noqa: E501
113
-
114
-
115
- :return: The username of this ProjectIdSnowflakeBody. # noqa: E501
116
- :rtype: str
117
- """
118
- return self._username
119
-
120
- @username.setter
121
- def username(self, username: 'str'):
122
- """Sets the username of this ProjectIdSnowflakeBody.
123
-
124
-
125
- :param username: The username of this ProjectIdSnowflakeBody. # noqa: E501
126
- :type: str
127
- """
128
-
129
- self._username = username
77
+ self._credential_id = credential_id
130
78
 
131
79
  def to_dict(self) -> dict:
132
80
  """Returns the model properties as a dict"""
@@ -41,60 +41,29 @@ class QueryQueryIdBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'account': 'str',
45
44
  'connection_id': 'str',
46
- 'password': 'str',
47
- 'query': 'str',
48
- 'username': 'str'
45
+ 'credential_id': 'str',
46
+ 'query': 'str'
49
47
  }
50
48
 
51
49
  attribute_map = {
52
- 'account': 'account',
53
50
  'connection_id': 'connectionId',
54
- 'password': 'password',
55
- 'query': 'query',
56
- 'username': 'username'
51
+ 'credential_id': 'credentialId',
52
+ 'query': 'query'
57
53
  }
58
54
 
59
- def __init__(self, account: 'str' =None, connection_id: 'str' =None, password: 'str' =None, query: 'str' =None, username: 'str' =None): # noqa: E501
55
+ def __init__(self, connection_id: 'str' =None, credential_id: 'str' =None, query: 'str' =None): # noqa: E501
60
56
  """QueryQueryIdBody - a model defined in Swagger""" # noqa: E501
61
- self._account = None
62
57
  self._connection_id = None
63
- self._password = None
58
+ self._credential_id = None
64
59
  self._query = None
65
- self._username = None
66
60
  self.discriminator = None
67
- if account is not None:
68
- self.account = account
69
61
  if connection_id is not None:
70
62
  self.connection_id = connection_id
71
- if password is not None:
72
- self.password = password
63
+ if credential_id is not None:
64
+ self.credential_id = credential_id
73
65
  if query is not None:
74
66
  self.query = query
75
- if username is not None:
76
- self.username = username
77
-
78
- @property
79
- def account(self) -> 'str':
80
- """Gets the account of this QueryQueryIdBody. # noqa: E501
81
-
82
-
83
- :return: The account of this QueryQueryIdBody. # noqa: E501
84
- :rtype: str
85
- """
86
- return self._account
87
-
88
- @account.setter
89
- def account(self, account: 'str'):
90
- """Sets the account of this QueryQueryIdBody.
91
-
92
-
93
- :param account: The account of this QueryQueryIdBody. # noqa: E501
94
- :type: str
95
- """
96
-
97
- self._account = account
98
67
 
99
68
  @property
100
69
  def connection_id(self) -> 'str':
@@ -118,25 +87,25 @@ class QueryQueryIdBody(object):
118
87
  self._connection_id = connection_id
119
88
 
120
89
  @property
121
- def password(self) -> 'str':
122
- """Gets the password of this QueryQueryIdBody. # noqa: E501
90
+ def credential_id(self) -> 'str':
91
+ """Gets the credential_id of this QueryQueryIdBody. # noqa: E501
123
92
 
124
93
 
125
- :return: The password of this QueryQueryIdBody. # noqa: E501
94
+ :return: The credential_id of this QueryQueryIdBody. # noqa: E501
126
95
  :rtype: str
127
96
  """
128
- return self._password
97
+ return self._credential_id
129
98
 
130
- @password.setter
131
- def password(self, password: 'str'):
132
- """Sets the password of this QueryQueryIdBody.
99
+ @credential_id.setter
100
+ def credential_id(self, credential_id: 'str'):
101
+ """Sets the credential_id of this QueryQueryIdBody.
133
102
 
134
103
 
135
- :param password: The password of this QueryQueryIdBody. # noqa: E501
104
+ :param credential_id: The credential_id of this QueryQueryIdBody. # noqa: E501
136
105
  :type: str
137
106
  """
138
107
 
139
- self._password = password
108
+ self._credential_id = credential_id
140
109
 
141
110
  @property
142
111
  def query(self) -> 'str':
@@ -159,27 +128,6 @@ class QueryQueryIdBody(object):
159
128
 
160
129
  self._query = query
161
130
 
162
- @property
163
- def username(self) -> 'str':
164
- """Gets the username of this QueryQueryIdBody. # noqa: E501
165
-
166
-
167
- :return: The username of this QueryQueryIdBody. # noqa: E501
168
- :rtype: str
169
- """
170
- return self._username
171
-
172
- @username.setter
173
- def username(self, username: 'str'):
174
- """Sets the username of this QueryQueryIdBody.
175
-
176
-
177
- :param username: The username of this QueryQueryIdBody. # noqa: E501
178
- :type: str
179
- """
180
-
181
- self._username = username
182
-
183
131
  def to_dict(self) -> dict:
184
132
  """Returns the model properties as a dict"""
185
133
  result = {}
@@ -41,85 +41,54 @@ class SnowflakeExportBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'account': 'str',
45
44
  'cluster_id': 'str',
46
45
  'compress': 'bool',
47
46
  'connection_id': 'str',
47
+ 'credential_id': 'str',
48
48
  'format': 'str',
49
49
  'include_header': 'bool',
50
- 'password': 'str',
51
50
  'query': 'str',
52
- 'query_name': 'str',
53
- 'username': 'str'
51
+ 'query_name': 'str'
54
52
  }
55
53
 
56
54
  attribute_map = {
57
- 'account': 'account',
58
55
  'cluster_id': 'clusterId',
59
56
  'compress': 'compress',
60
57
  'connection_id': 'connectionId',
58
+ 'credential_id': 'credentialId',
61
59
  'format': 'format',
62
60
  'include_header': 'includeHeader',
63
- 'password': 'password',
64
61
  'query': 'query',
65
- 'query_name': 'queryName',
66
- 'username': 'username'
62
+ 'query_name': 'queryName'
67
63
  }
68
64
 
69
- def __init__(self, account: 'str' =None, cluster_id: 'str' =None, compress: 'bool' =None, connection_id: 'str' =None, format: 'str' =None, include_header: 'bool' =None, password: 'str' =None, query: 'str' =None, query_name: 'str' =None, username: 'str' =None): # noqa: E501
65
+ def __init__(self, cluster_id: 'str' =None, compress: 'bool' =None, connection_id: 'str' =None, credential_id: 'str' =None, format: 'str' =None, include_header: 'bool' =None, query: 'str' =None, query_name: 'str' =None): # noqa: E501
70
66
  """SnowflakeExportBody - a model defined in Swagger""" # noqa: E501
71
- self._account = None
72
67
  self._cluster_id = None
73
68
  self._compress = None
74
69
  self._connection_id = None
70
+ self._credential_id = None
75
71
  self._format = None
76
72
  self._include_header = None
77
- self._password = None
78
73
  self._query = None
79
74
  self._query_name = None
80
- self._username = None
81
75
  self.discriminator = None
82
- if account is not None:
83
- self.account = account
84
76
  if cluster_id is not None:
85
77
  self.cluster_id = cluster_id
86
78
  if compress is not None:
87
79
  self.compress = compress
88
80
  if connection_id is not None:
89
81
  self.connection_id = connection_id
82
+ if credential_id is not None:
83
+ self.credential_id = credential_id
90
84
  if format is not None:
91
85
  self.format = format
92
86
  if include_header is not None:
93
87
  self.include_header = include_header
94
- if password is not None:
95
- self.password = password
96
88
  if query is not None:
97
89
  self.query = query
98
90
  if query_name is not None:
99
91
  self.query_name = query_name
100
- if username is not None:
101
- self.username = username
102
-
103
- @property
104
- def account(self) -> 'str':
105
- """Gets the account of this SnowflakeExportBody. # noqa: E501
106
-
107
-
108
- :return: The account of this SnowflakeExportBody. # noqa: E501
109
- :rtype: str
110
- """
111
- return self._account
112
-
113
- @account.setter
114
- def account(self, account: 'str'):
115
- """Sets the account of this SnowflakeExportBody.
116
-
117
-
118
- :param account: The account of this SnowflakeExportBody. # noqa: E501
119
- :type: str
120
- """
121
-
122
- self._account = account
123
92
 
124
93
  @property
125
94
  def cluster_id(self) -> 'str':
@@ -184,6 +153,27 @@ class SnowflakeExportBody(object):
184
153
 
185
154
  self._connection_id = connection_id
186
155
 
156
+ @property
157
+ def credential_id(self) -> 'str':
158
+ """Gets the credential_id of this SnowflakeExportBody. # noqa: E501
159
+
160
+
161
+ :return: The credential_id of this SnowflakeExportBody. # noqa: E501
162
+ :rtype: str
163
+ """
164
+ return self._credential_id
165
+
166
+ @credential_id.setter
167
+ def credential_id(self, credential_id: 'str'):
168
+ """Sets the credential_id of this SnowflakeExportBody.
169
+
170
+
171
+ :param credential_id: The credential_id of this SnowflakeExportBody. # noqa: E501
172
+ :type: str
173
+ """
174
+
175
+ self._credential_id = credential_id
176
+
187
177
  @property
188
178
  def format(self) -> 'str':
189
179
  """Gets the format of this SnowflakeExportBody. # noqa: E501
@@ -226,27 +216,6 @@ class SnowflakeExportBody(object):
226
216
 
227
217
  self._include_header = include_header
228
218
 
229
- @property
230
- def password(self) -> 'str':
231
- """Gets the password of this SnowflakeExportBody. # noqa: E501
232
-
233
-
234
- :return: The password of this SnowflakeExportBody. # noqa: E501
235
- :rtype: str
236
- """
237
- return self._password
238
-
239
- @password.setter
240
- def password(self, password: 'str'):
241
- """Sets the password of this SnowflakeExportBody.
242
-
243
-
244
- :param password: The password of this SnowflakeExportBody. # noqa: E501
245
- :type: str
246
- """
247
-
248
- self._password = password
249
-
250
219
  @property
251
220
  def query(self) -> 'str':
252
221
  """Gets the query of this SnowflakeExportBody. # noqa: E501
@@ -289,27 +258,6 @@ class SnowflakeExportBody(object):
289
258
 
290
259
  self._query_name = query_name
291
260
 
292
- @property
293
- def username(self) -> 'str':
294
- """Gets the username of this SnowflakeExportBody. # noqa: E501
295
-
296
-
297
- :return: The username of this SnowflakeExportBody. # noqa: E501
298
- :rtype: str
299
- """
300
- return self._username
301
-
302
- @username.setter
303
- def username(self, username: 'str'):
304
- """Sets the username of this SnowflakeExportBody.
305
-
306
-
307
- :param username: The username of this SnowflakeExportBody. # noqa: E501
308
- :type: str
309
- """
310
-
311
- self._username = username
312
-
313
261
  def to_dict(self) -> dict:
314
262
  """Returns the model properties as a dict"""
315
263
  result = {}