lightning-sdk 0.1.3__py3-none-any.whl → 0.1.46__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/{v1_download_model_response.py → snowflake_query_body.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1DownloadModelResponse(object):
+class SnowflakeQueryBody(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -41,92 +41,92 @@ class V1DownloadModelResponse(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'download_url': 'str',
-        'metadata': 'dict(str, str)',
-        'version': 'str'
+        'connection_id': 'str',
+        'credential_id': 'str',
+        'query': 'str'
     }
 
     attribute_map = {
-        'download_url': 'downloadUrl',
-        'metadata': 'metadata',
-        'version': 'version'
+        'connection_id': 'connectionId',
+        'credential_id': 'credentialId',
+        'query': 'query'
     }
 
-    def __init__(self, download_url: 'str' =None, metadata: 'dict(str, str)' =None, version: 'str' =None):  # noqa: E501
-        """V1DownloadModelResponse - a model defined in Swagger"""  # noqa: E501
-        self._download_url = None
-        self._metadata = None
-        self._version = None
+    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._connection_id = None
+        self._credential_id = None
+        self._query = None
         self.discriminator = None
-        if download_url is not None:
-            self.download_url = download_url
-        if metadata is not None:
-            self.metadata = metadata
-        if version is not None:
-            self.version = version
+        if connection_id is not None:
+            self.connection_id = connection_id
+        if credential_id is not None:
+            self.credential_id = credential_id
+        if query is not None:
+            self.query = query
 
     @property
-    def download_url(self) -> 'str':
-        """Gets the download_url of this V1DownloadModelResponse.  # noqa: E501
+    def connection_id(self) -> 'str':
+        """Gets the connection_id of this SnowflakeQueryBody.  # noqa: E501
 
 
-        :return: The download_url of this V1DownloadModelResponse.  # noqa: E501
+        :return: The connection_id of this SnowflakeQueryBody.  # noqa: E501
         :rtype: str
         """
-        return self._download_url
+        return self._connection_id
 
-    @download_url.setter
-    def download_url(self, download_url: 'str'):
-        """Sets the download_url of this V1DownloadModelResponse.
+    @connection_id.setter
+    def connection_id(self, connection_id: 'str'):
+        """Sets the connection_id of this SnowflakeQueryBody.
 
 
-        :param download_url: The download_url of this V1DownloadModelResponse.  # noqa: E501
+        :param connection_id: The connection_id of this SnowflakeQueryBody.  # noqa: E501
         :type: str
         """
 
-        self._download_url = download_url
+        self._connection_id = connection_id
 
     @property
-    def metadata(self) -> 'dict(str, str)':
-        """Gets the metadata of this V1DownloadModelResponse.  # noqa: E501
+    def credential_id(self) -> 'str':
+        """Gets the credential_id of this SnowflakeQueryBody.  # noqa: E501
 
 
-        :return: The metadata of this V1DownloadModelResponse.  # noqa: E501
-        :rtype: dict(str, str)
+        :return: The credential_id of this SnowflakeQueryBody.  # noqa: E501
+        :rtype: str
         """
-        return self._metadata
+        return self._credential_id
 
-    @metadata.setter
-    def metadata(self, metadata: 'dict(str, str)'):
-        """Sets the metadata of this V1DownloadModelResponse.
+    @credential_id.setter
+    def credential_id(self, credential_id: 'str'):
+        """Sets the credential_id of this SnowflakeQueryBody.
 
 
-        :param metadata: The metadata of this V1DownloadModelResponse.  # noqa: E501
-        :type: dict(str, str)
+        :param credential_id: The credential_id of this SnowflakeQueryBody.  # noqa: E501
+        :type: str
         """
 
-        self._metadata = metadata
+        self._credential_id = credential_id
 
     @property
-    def version(self) -> 'str':
-        """Gets the version of this V1DownloadModelResponse.  # noqa: E501
+    def query(self) -> 'str':
+        """Gets the query of this SnowflakeQueryBody.  # noqa: E501
 
 
-        :return: The version of this V1DownloadModelResponse.  # noqa: E501
+        :return: The query of this SnowflakeQueryBody.  # noqa: E501
         :rtype: str
         """
-        return self._version
+        return self._query
 
-    @version.setter
-    def version(self, version: 'str'):
-        """Sets the version of this V1DownloadModelResponse.
+    @query.setter
+    def query(self, query: 'str'):
+        """Sets the query of this SnowflakeQueryBody.
 
 
-        :param version: The version of this V1DownloadModelResponse.  # noqa: E501
+        :param query: The query of this SnowflakeQueryBody.  # noqa: E501
         :type: str
         """
 
-        self._version = version
+        self._query = query
 
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
@@ -149,7 +149,7 @@ class V1DownloadModelResponse(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1DownloadModelResponse, dict):
+        if issubclass(SnowflakeQueryBody, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -163,13 +163,13 @@ class V1DownloadModelResponse(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1DownloadModelResponse') -> bool:
+    def __eq__(self, other: 'SnowflakeQueryBody') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1DownloadModelResponse):
+        if not isinstance(other, SnowflakeQueryBody):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1DownloadModelResponse') -> bool:
+    def __ne__(self, other: 'SnowflakeQueryBody') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other

lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py

@@ -44,6 +44,7 @@ class StorageCompleteBody(object):
         'cluster_id': 'str',
         'filename': 'str',
         'parts': 'list[V1CompleteUpload]',
+        'suppress_reindex': 'bool',
         'upload_id': 'str'
     }
 
@@ -51,14 +52,16 @@ class StorageCompleteBody(object):
         'cluster_id': 'clusterId',
         'filename': 'filename',
         'parts': 'parts',
+        'suppress_reindex': 'suppressReindex',
         'upload_id': 'uploadId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, upload_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, filename: 'str' =None, parts: 'list[V1CompleteUpload]' =None, suppress_reindex: 'bool' =None, upload_id: 'str' =None):  # noqa: E501
         """StorageCompleteBody - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._filename = None
         self._parts = None
+        self._suppress_reindex = None
         self._upload_id = None
         self.discriminator = None
         if cluster_id is not None:
@@ -67,6 +70,8 @@ class StorageCompleteBody(object):
             self.filename = filename
         if parts is not None:
             self.parts = parts
+        if suppress_reindex is not None:
+            self.suppress_reindex = suppress_reindex
         if upload_id is not None:
             self.upload_id = upload_id
 
@@ -133,6 +138,27 @@ class StorageCompleteBody(object):
 
         self._parts = parts
 
+    @property
+    def suppress_reindex(self) -> 'bool':
+        """Gets the suppress_reindex of this StorageCompleteBody.  # noqa: E501
+
+
+        :return: The suppress_reindex of this StorageCompleteBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._suppress_reindex
+
+    @suppress_reindex.setter
+    def suppress_reindex(self, suppress_reindex: 'bool'):
+        """Sets the suppress_reindex of this StorageCompleteBody.
+
+
+        :param suppress_reindex: The suppress_reindex of this StorageCompleteBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._suppress_reindex = suppress_reindex
+
     @property
     def upload_id(self) -> 'str':
         """Gets the upload_id of this StorageCompleteBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/update.py

@@ -42,36 +42,56 @@ class Update(object):
     """
     swagger_types = {
         'aws': 'V1AwsDataConnection',
+        'efs': 'V1EfsConfig',
         'gcp': 'V1GcpDataConnection',
+        'gcs_folder': 'V1GCSFolderDataConnection',
         'name': 'str',
         'run_cmds': 'list[str]',
+        's3_folder': 'V1S3FolderDataConnection',
+        'snowflake': 'V1SnowflakeDataConnection',
         'writable': 'bool'
     }
 
     attribute_map = {
         'aws': 'aws',
+        'efs': 'efs',
         'gcp': 'gcp',
+        'gcs_folder': 'gcsFolder',
         'name': 'name',
         'run_cmds': 'runCmds',
+        's3_folder': 's3Folder',
+        'snowflake': 'snowflake',
         'writable': 'writable'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, gcp: 'V1GcpDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """Update - a model defined in Swagger"""  # noqa: E501
         self._aws = None
+        self._efs = None
         self._gcp = None
+        self._gcs_folder = None
         self._name = None
         self._run_cmds = None
+        self._s3_folder = None
+        self._snowflake = None
         self._writable = None
         self.discriminator = None
         if aws is not None:
             self.aws = aws
+        if efs is not None:
+            self.efs = efs
         if gcp is not None:
             self.gcp = gcp
+        if gcs_folder is not None:
+            self.gcs_folder = gcs_folder
         if name is not None:
             self.name = name
         if run_cmds is not None:
             self.run_cmds = run_cmds
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
+        if snowflake is not None:
+            self.snowflake = snowflake
         if writable is not None:
             self.writable = writable
 
@@ -96,6 +116,27 @@ class Update(object):
 
         self._aws = aws
 
+    @property
+    def efs(self) -> 'V1EfsConfig':
+        """Gets the efs of this Update.  # noqa: E501
+
+
+        :return: The efs of this Update.  # noqa: E501
+        :rtype: V1EfsConfig
+        """
+        return self._efs
+
+    @efs.setter
+    def efs(self, efs: 'V1EfsConfig'):
+        """Sets the efs of this Update.
+
+
+        :param efs: The efs of this Update.  # noqa: E501
+        :type: V1EfsConfig
+        """
+
+        self._efs = efs
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this Update.  # noqa: E501
@@ -117,6 +158,27 @@ class Update(object):
 
         self._gcp = gcp
 
+    @property
+    def gcs_folder(self) -> 'V1GCSFolderDataConnection':
+        """Gets the gcs_folder of this Update.  # noqa: E501
+
+
+        :return: The gcs_folder of this Update.  # noqa: E501
+        :rtype: V1GCSFolderDataConnection
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'V1GCSFolderDataConnection'):
+        """Sets the gcs_folder of this Update.
+
+
+        :param gcs_folder: The gcs_folder of this Update.  # noqa: E501
+        :type: V1GCSFolderDataConnection
+        """
+
+        self._gcs_folder = gcs_folder
+
     @property
     def name(self) -> 'str':
         """Gets the name of this Update.  # noqa: E501
@@ -159,6 +221,48 @@ class Update(object):
 
         self._run_cmds = run_cmds
 
+    @property
+    def s3_folder(self) -> 'V1S3FolderDataConnection':
+        """Gets the s3_folder of this Update.  # noqa: E501
+
+
+        :return: The s3_folder of this Update.  # noqa: E501
+        :rtype: V1S3FolderDataConnection
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'V1S3FolderDataConnection'):
+        """Sets the s3_folder of this Update.
+
+
+        :param s3_folder: The s3_folder of this Update.  # noqa: E501
+        :type: V1S3FolderDataConnection
+        """
+
+        self._s3_folder = s3_folder
+
+    @property
+    def snowflake(self) -> 'V1SnowflakeDataConnection':
+        """Gets the snowflake of this Update.  # noqa: E501
+
+
+        :return: The snowflake of this Update.  # noqa: E501
+        :rtype: V1SnowflakeDataConnection
+        """
+        return self._snowflake
+
+    @snowflake.setter
+    def snowflake(self, snowflake: 'V1SnowflakeDataConnection'):
+        """Sets the snowflake of this Update.
+
+
+        :param snowflake: The snowflake of this Update.  # noqa: E501
+        :type: V1SnowflakeDataConnection
+        """
+
+        self._snowflake = snowflake
+
     @property
     def writable(self) -> 'bool':
         """Gets the writable of this Update.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body.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 UploadIdCompleteBody(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 = {
+        'filepath': 'str',
+        'parts': 'list[V1CompletedPart]'
+    }
+
+    attribute_map = {
+        'filepath': 'filepath',
+        'parts': 'parts'
+    }
+
+    def __init__(self, filepath: 'str' =None, parts: 'list[V1CompletedPart]' =None):  # noqa: E501
+        """UploadIdCompleteBody - a model defined in Swagger"""  # noqa: E501
+        self._filepath = None
+        self._parts = None
+        self.discriminator = None
+        if filepath is not None:
+            self.filepath = filepath
+        if parts is not None:
+            self.parts = parts
+
+    @property
+    def filepath(self) -> 'str':
+        """Gets the filepath of this UploadIdCompleteBody.  # noqa: E501
+
+
+        :return: The filepath of this UploadIdCompleteBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._filepath
+
+    @filepath.setter
+    def filepath(self, filepath: 'str'):
+        """Sets the filepath of this UploadIdCompleteBody.
+
+
+        :param filepath: The filepath of this UploadIdCompleteBody.  # noqa: E501
+        :type: str
+        """
+
+        self._filepath = filepath
+
+    @property
+    def parts(self) -> 'list[V1CompletedPart]':
+        """Gets the parts of this UploadIdCompleteBody.  # noqa: E501
+
+
+        :return: The parts of this UploadIdCompleteBody.  # noqa: E501
+        :rtype: list[V1CompletedPart]
+        """
+        return self._parts
+
+    @parts.setter
+    def parts(self, parts: 'list[V1CompletedPart]'):
+        """Sets the parts of this UploadIdCompleteBody.
+
+
+        :param parts: The parts of this UploadIdCompleteBody.  # noqa: E501
+        :type: list[V1CompletedPart]
+        """
+
+        self._parts = parts
+
+    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(UploadIdCompleteBody, 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: 'UploadIdCompleteBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, UploadIdCompleteBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'UploadIdCompleteBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other