lightning-sdk 0.1.41__py3-none-any.whl → 0.1.42__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py

@@ -41,60 +41,29 @@ class SnowflakeQueryBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'account': 'str',
         'connection_id': 'str',
-        'password': 'str',
-        'query': 'str',
-        'username': 'str'
+        'credential_id': 'str',
+        'query': 'str'
     }
 
     attribute_map = {
-        'account': 'account',
         'connection_id': 'connectionId',
-        'password': 'password',
-        'query': 'query',
-        'username': 'username'
+        'credential_id': 'credentialId',
+        'query': 'query'
     }
 
-    def __init__(self, account: 'str' =None, connection_id: 'str' =None, password: 'str' =None, query: 'str' =None, username: 'str' =None):  # noqa: E501
+    def __init__(self, connection_id: 'str' =None, credential_id: 'str' =None, query: 'str' =None):  # noqa: E501
         """SnowflakeQueryBody - a model defined in Swagger"""  # noqa: E501
-        self._account = None
         self._connection_id = None
-        self._password = None
+        self._credential_id = None
         self._query = None
-        self._username = None
         self.discriminator = None
-        if account is not None:
-            self.account = account
         if connection_id is not None:
             self.connection_id = connection_id
-        if password is not None:
-            self.password = password
+        if credential_id is not None:
+            self.credential_id = credential_id
         if query is not None:
             self.query = query
-        if username is not None:
-            self.username = username
-
-    @property
-    def account(self) -> 'str':
-        """Gets the account of this SnowflakeQueryBody.  # noqa: E501
-
-
-        :return: The account of this SnowflakeQueryBody.  # noqa: E501
-        :rtype: str
-        """
-        return self._account
-
-    @account.setter
-    def account(self, account: 'str'):
-        """Sets the account of this SnowflakeQueryBody.
-
-
-        :param account: The account of this SnowflakeQueryBody.  # noqa: E501
-        :type: str
-        """
-
-        self._account = account
 
     @property
     def connection_id(self) -> 'str':
@@ -118,25 +87,25 @@ class SnowflakeQueryBody(object):
         self._connection_id = connection_id
 
     @property
-    def password(self) -> 'str':
-        """Gets the password of this SnowflakeQueryBody.  # noqa: E501
+    def credential_id(self) -> 'str':
+        """Gets the credential_id of this SnowflakeQueryBody.  # noqa: E501
 
 
-        :return: The password of this SnowflakeQueryBody.  # noqa: E501
+        :return: The credential_id of this SnowflakeQueryBody.  # noqa: E501
         :rtype: str
         """
-        return self._password
+        return self._credential_id
 
-    @password.setter
-    def password(self, password: 'str'):
-        """Sets the password of this SnowflakeQueryBody.
+    @credential_id.setter
+    def credential_id(self, credential_id: 'str'):
+        """Sets the credential_id of this SnowflakeQueryBody.
 
 
-        :param password: The password of this SnowflakeQueryBody.  # noqa: E501
+        :param credential_id: The credential_id of this SnowflakeQueryBody.  # noqa: E501
         :type: str
         """
 
-        self._password = password
+        self._credential_id = credential_id
 
     @property
     def query(self) -> 'str':
@@ -159,27 +128,6 @@ class SnowflakeQueryBody(object):
 
         self._query = query
 
-    @property
-    def username(self) -> 'str':
-        """Gets the username of this SnowflakeQueryBody.  # noqa: E501
-
-
-        :return: The username of this SnowflakeQueryBody.  # noqa: E501
-        :rtype: str
-        """
-        return self._username
-
-    @username.setter
-    def username(self, username: 'str'):
-        """Sets the username of this SnowflakeQueryBody.
-
-
-        :param username: The username of this SnowflakeQueryBody.  # noqa: E501
-        :type: str
-        """
-
-        self._username = username
-
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py

@@ -41,25 +41,51 @@ class V1GetModelFileUrlResponse(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'filepath': 'str',
         'size': 'str',
         'url': 'str'
     }
 
     attribute_map = {
+        'filepath': 'filepath',
         'size': 'size',
         'url': 'url'
     }
 
-    def __init__(self, size: 'str' =None, url: 'str' =None):  # noqa: E501
+    def __init__(self, filepath: 'str' =None, size: 'str' =None, url: 'str' =None):  # noqa: E501
         """V1GetModelFileUrlResponse - a model defined in Swagger"""  # noqa: E501
+        self._filepath = None
         self._size = None
         self._url = None
         self.discriminator = None
+        if filepath is not None:
+            self.filepath = filepath
         if size is not None:
             self.size = size
         if url is not None:
             self.url = url
 
+    @property
+    def filepath(self) -> 'str':
+        """Gets the filepath of this V1GetModelFileUrlResponse.  # noqa: E501
+
+
+        :return: The filepath of this V1GetModelFileUrlResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._filepath
+
+    @filepath.setter
+    def filepath(self, filepath: 'str'):
+        """Sets the filepath of this V1GetModelFileUrlResponse.
+
+
+        :param filepath: The filepath of this V1GetModelFileUrlResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._filepath = filepath
+
     @property
     def size(self) -> 'str':
         """Gets the size of this V1GetModelFileUrlResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py

@@ -42,7 +42,7 @@ class V1GetModelFilesResponse(object):
     """
     swagger_types = {
         'filepaths': 'list[str]',
-        'filesizes': 'list[str]',
+        'files': 'list[V1ModelFile]',
         'model_id': 'str',
         'project_id': 'str',
         'size_bytes': 'str',
@@ -52,7 +52,7 @@ class V1GetModelFilesResponse(object):
 
     attribute_map = {
         'filepaths': 'filepaths',
-        'filesizes': 'filesizes',
+        'files': 'files',
         'model_id': 'modelId',
         'project_id': 'projectId',
         'size_bytes': 'sizeBytes',
@@ -60,10 +60,10 @@ class V1GetModelFilesResponse(object):
         'version': 'version'
     }
 
-    def __init__(self, filepaths: 'list[str]' =None, filesizes: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None):  # noqa: E501
+    def __init__(self, filepaths: 'list[str]' =None, files: 'list[V1ModelFile]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None):  # noqa: E501
         """V1GetModelFilesResponse - a model defined in Swagger"""  # noqa: E501
         self._filepaths = None
-        self._filesizes = None
+        self._files = None
         self._model_id = None
         self._project_id = None
         self._size_bytes = None
@@ -72,8 +72,8 @@ class V1GetModelFilesResponse(object):
         self.discriminator = None
         if filepaths is not None:
             self.filepaths = filepaths
-        if filesizes is not None:
-            self.filesizes = filesizes
+        if files is not None:
+            self.files = files
         if model_id is not None:
             self.model_id = model_id
         if project_id is not None:
@@ -107,25 +107,25 @@ class V1GetModelFilesResponse(object):
         self._filepaths = filepaths
 
     @property
-    def filesizes(self) -> 'list[str]':
-        """Gets the filesizes of this V1GetModelFilesResponse.  # noqa: E501
+    def files(self) -> 'list[V1ModelFile]':
+        """Gets the files of this V1GetModelFilesResponse.  # noqa: E501
 
 
-        :return: The filesizes of this V1GetModelFilesResponse.  # noqa: E501
-        :rtype: list[str]
+        :return: The files of this V1GetModelFilesResponse.  # noqa: E501
+        :rtype: list[V1ModelFile]
         """
-        return self._filesizes
+        return self._files
 
-    @filesizes.setter
-    def filesizes(self, filesizes: 'list[str]'):
-        """Sets the filesizes of this V1GetModelFilesResponse.
+    @files.setter
+    def files(self, files: 'list[V1ModelFile]'):
+        """Sets the files of this V1GetModelFilesResponse.
 
 
-        :param filesizes: The filesizes of this V1GetModelFilesResponse.  # noqa: E501
-        :type: list[str]
+        :param files: The files of this V1GetModelFilesResponse.  # noqa: E501
+        :type: list[V1ModelFile]
         """
 
-        self._filesizes = filesizes
+        self._files = files
 
     @property
     def model_id(self) -> 'str':

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py

@@ -0,0 +1,149 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1GetModelFilesUrlResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'files': 'list[V1ModelFile]',
+        'offset': 'str'
+    }
+
+    attribute_map = {
+        'files': 'files',
+        'offset': 'offset'
+    }
+
+    def __init__(self, files: 'list[V1ModelFile]' =None, offset: 'str' =None):  # noqa: E501
+        """V1GetModelFilesUrlResponse - a model defined in Swagger"""  # noqa: E501
+        self._files = None
+        self._offset = None
+        self.discriminator = None
+        if files is not None:
+            self.files = files
+        if offset is not None:
+            self.offset = offset
+
+    @property
+    def files(self) -> 'list[V1ModelFile]':
+        """Gets the files of this V1GetModelFilesUrlResponse.  # noqa: E501
+
+
+        :return: The files of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :rtype: list[V1ModelFile]
+        """
+        return self._files
+
+    @files.setter
+    def files(self, files: 'list[V1ModelFile]'):
+        """Sets the files of this V1GetModelFilesUrlResponse.
+
+
+        :param files: The files of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :type: list[V1ModelFile]
+        """
+
+        self._files = files
+
+    @property
+    def offset(self) -> 'str':
+        """Gets the offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+
+
+        :return: The offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._offset
+
+    @offset.setter
+    def offset(self, offset: 'str'):
+        """Sets the offset of this V1GetModelFilesUrlResponse.
+
+
+        :param offset: The offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._offset = offset
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1GetModelFilesUrlResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1GetModelFilesUrlResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1GetModelFilesUrlResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1GetModelFilesUrlResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py

@@ -43,6 +43,7 @@ class V1GetProjectBalanceResponse(object):
     swagger_types = {
         'account_id': 'str',
         'balance': 'float',
+        'next_free_credits_grant_at': 'datetime',
         'project_id': 'str',
         'transactions': 'list[V1Transaction]'
     }
@@ -50,14 +51,16 @@ class V1GetProjectBalanceResponse(object):
     attribute_map = {
         'account_id': 'accountId',
         'balance': 'balance',
+        'next_free_credits_grant_at': 'nextFreeCreditsGrantAt',
         'project_id': 'projectId',
         'transactions': 'transactions'
     }
 
-    def __init__(self, account_id: 'str' =None, balance: 'float' =None, project_id: 'str' =None, transactions: 'list[V1Transaction]' =None):  # noqa: E501
+    def __init__(self, account_id: 'str' =None, balance: 'float' =None, next_free_credits_grant_at: 'datetime' =None, project_id: 'str' =None, transactions: 'list[V1Transaction]' =None):  # noqa: E501
         """V1GetProjectBalanceResponse - a model defined in Swagger"""  # noqa: E501
         self._account_id = None
         self._balance = None
+        self._next_free_credits_grant_at = None
         self._project_id = None
         self._transactions = None
         self.discriminator = None
@@ -65,6 +68,8 @@ class V1GetProjectBalanceResponse(object):
             self.account_id = account_id
         if balance is not None:
             self.balance = balance
+        if next_free_credits_grant_at is not None:
+            self.next_free_credits_grant_at = next_free_credits_grant_at
         if project_id is not None:
             self.project_id = project_id
         if transactions is not None:
@@ -112,6 +117,27 @@ class V1GetProjectBalanceResponse(object):
 
         self._balance = balance
 
+    @property
+    def next_free_credits_grant_at(self) -> 'datetime':
+        """Gets the next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+
+
+        :return: The next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._next_free_credits_grant_at
+
+    @next_free_credits_grant_at.setter
+    def next_free_credits_grant_at(self, next_free_credits_grant_at: 'datetime'):
+        """Sets the next_free_credits_grant_at of this V1GetProjectBalanceResponse.
+
+
+        :param next_free_credits_grant_at: The next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+        :type: datetime
+        """
+
+        self._next_free_credits_grant_at = next_free_credits_grant_at
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1GetProjectBalanceResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ListMultiMachineJobEventsResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'events': 'list[V1MultiMachineJobEvent]'
+    }
+
+    attribute_map = {
+        'events': 'events'
+    }
+
+    def __init__(self, events: 'list[V1MultiMachineJobEvent]' =None):  # noqa: E501
+        """V1ListMultiMachineJobEventsResponse - a model defined in Swagger"""  # noqa: E501
+        self._events = None
+        self.discriminator = None
+        if events is not None:
+            self.events = events
+
+    @property
+    def events(self) -> 'list[V1MultiMachineJobEvent]':
+        """Gets the events of this V1ListMultiMachineJobEventsResponse.  # noqa: E501
+
+
+        :return: The events of this V1ListMultiMachineJobEventsResponse.  # noqa: E501
+        :rtype: list[V1MultiMachineJobEvent]
+        """
+        return self._events
+
+    @events.setter
+    def events(self, events: 'list[V1MultiMachineJobEvent]'):
+        """Sets the events of this V1ListMultiMachineJobEventsResponse.
+
+
+        :param events: The events of this V1ListMultiMachineJobEventsResponse.  # noqa: E501
+        :type: list[V1MultiMachineJobEvent]
+        """
+
+        self._events = events
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ListMultiMachineJobEventsResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ListMultiMachineJobEventsResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ListMultiMachineJobEventsResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ListMultiMachineJobEventsResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py

@@ -66,6 +66,7 @@ class V1MetricsStream(object):
         'user_id': 'str',
         'version': 'datetime',
         'version_number': 'int',
+        'visibility': 'V1ResourceVisibility',
         'work_id': 'str'
     }
 
@@ -95,10 +96,11 @@ class V1MetricsStream(object):
         'user_id': 'userId',
         'version': 'version',
         'version_number': 'versionNumber',
+        'visibility': 'visibility',
         'work_id': 'workId'
     }
 
-    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, work_id: 'str' =None):  # noqa: E501
+    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, visibility: 'V1ResourceVisibility' =None, work_id: 'str' =None):  # noqa: E501
         """V1MetricsStream - a model defined in Swagger"""  # noqa: E501
         self._app_id = None
         self._cloudspace_id = None
@@ -125,6 +127,7 @@ class V1MetricsStream(object):
         self._user_id = None
         self._version = None
         self._version_number = None
+        self._visibility = None
         self._work_id = None
         self.discriminator = None
         if app_id is not None:
@@ -177,6 +180,8 @@ class V1MetricsStream(object):
             self.version = version
         if version_number is not None:
             self.version_number = version_number
+        if visibility is not None:
+            self.visibility = visibility
         if work_id is not None:
             self.work_id = work_id
 
@@ -705,6 +710,27 @@ class V1MetricsStream(object):
 
         self._version_number = version_number
 
+    @property
+    def visibility(self) -> 'V1ResourceVisibility':
+        """Gets the visibility of this V1MetricsStream.  # noqa: E501
+
+
+        :return: The visibility of this V1MetricsStream.  # noqa: E501
+        :rtype: V1ResourceVisibility
+        """
+        return self._visibility
+
+    @visibility.setter
+    def visibility(self, visibility: 'V1ResourceVisibility'):
+        """Sets the visibility of this V1MetricsStream.
+
+
+        :param visibility: The visibility of this V1MetricsStream.  # noqa: E501
+        :type: V1ResourceVisibility
+        """
+
+        self._visibility = visibility
+
     @property
     def work_id(self) -> 'str':
         """Gets the work_id of this V1MetricsStream.  # noqa: E501