lightning-sdk 0.1.31__py3-none-any.whl → 0.1.33__py3-none-any.whl

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

lightning_sdk/lightning_cloud/openapi/models/v1_transaction.py

@@ -43,25 +43,30 @@ class V1Transaction(object):
     swagger_types = {
         'amount': 'float',
         'created_at': 'datetime',
+        'display_name': 'str',
         'type': 'str'
     }
 
     attribute_map = {
         'amount': 'amount',
         'created_at': 'createdAt',
+        'display_name': 'displayName',
         'type': 'type'
     }
 
-    def __init__(self, amount: 'float' =None, created_at: 'datetime' =None, type: 'str' =None):  # noqa: E501
+    def __init__(self, amount: 'float' =None, created_at: 'datetime' =None, display_name: 'str' =None, type: 'str' =None):  # noqa: E501
         """V1Transaction - a model defined in Swagger"""  # noqa: E501
         self._amount = None
         self._created_at = None
+        self._display_name = None
         self._type = None
         self.discriminator = None
         if amount is not None:
             self.amount = amount
         if created_at is not None:
             self.created_at = created_at
+        if display_name is not None:
+            self.display_name = display_name
         if type is not None:
             self.type = type
 
@@ -107,6 +112,27 @@ class V1Transaction(object):
 
         self._created_at = created_at
 
+    @property
+    def display_name(self) -> 'str':
+        """Gets the display_name of this V1Transaction.  # noqa: E501
+
+
+        :return: The display_name of this V1Transaction.  # noqa: E501
+        :rtype: str
+        """
+        return self._display_name
+
+    @display_name.setter
+    def display_name(self, display_name: 'str'):
+        """Sets the display_name of this V1Transaction.
+
+
+        :param display_name: The display_name of this V1Transaction.  # noqa: E501
+        :type: str
+        """
+
+        self._display_name = display_name
+
     @property
     def type(self) -> 'str':
         """Gets the type of this V1Transaction.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py

@@ -0,0 +1,97 @@
+# 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 V1UpdateSharedMetricsStreamResponse(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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1UpdateSharedMetricsStreamResponse - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    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(V1UpdateSharedMetricsStreamResponse, 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: 'V1UpdateSharedMetricsStreamResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1UpdateSharedMetricsStreamResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1UpdateSharedMetricsStreamResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -60,6 +60,7 @@ class V1UserFeatures(object):
         'custom_app_domain': 'bool',
         'custom_instance_types': 'bool',
         'default_one_cluster': 'bool',
+        'deployment_data_path': 'bool',
         'deployment_gallery': 'bool',
         'deployment_persistent_disk': 'bool',
         'docs_agent': 'bool',
@@ -79,6 +80,7 @@ class V1UserFeatures(object):
         'mmt_fault_tolerance': 'bool',
         'mmt_strategy_selector': 'bool',
         'model_store': 'bool',
+        'multiple_deployment_versions': 'bool',
         'org_level_member_permissions': 'bool',
         'plugin_biz_chat': 'bool',
         'plugin_distributed': 'bool',
@@ -134,6 +136,7 @@ class V1UserFeatures(object):
         'custom_app_domain': 'customAppDomain',
         'custom_instance_types': 'customInstanceTypes',
         'default_one_cluster': 'defaultOneCluster',
+        'deployment_data_path': 'deploymentDataPath',
         'deployment_gallery': 'deploymentGallery',
         'deployment_persistent_disk': 'deploymentPersistentDisk',
         'docs_agent': 'docsAgent',
@@ -153,6 +156,7 @@ class V1UserFeatures(object):
         'mmt_fault_tolerance': 'mmtFaultTolerance',
         'mmt_strategy_selector': 'mmtStrategySelector',
         'model_store': 'modelStore',
+        'multiple_deployment_versions': 'multipleDeploymentVersions',
         'org_level_member_permissions': 'orgLevelMemberPermissions',
         'plugin_biz_chat': 'pluginBizChat',
         'plugin_distributed': 'pluginDistributed',
@@ -188,7 +192,7 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
@@ -209,6 +213,7 @@ class V1UserFeatures(object):
         self._custom_app_domain = None
         self._custom_instance_types = None
         self._default_one_cluster = None
+        self._deployment_data_path = None
         self._deployment_gallery = None
         self._deployment_persistent_disk = None
         self._docs_agent = None
@@ -228,6 +233,7 @@ class V1UserFeatures(object):
         self._mmt_fault_tolerance = None
         self._mmt_strategy_selector = None
         self._model_store = None
+        self._multiple_deployment_versions = None
         self._org_level_member_permissions = None
         self._plugin_biz_chat = None
         self._plugin_distributed = None
@@ -300,6 +306,8 @@ class V1UserFeatures(object):
             self.custom_instance_types = custom_instance_types
         if default_one_cluster is not None:
             self.default_one_cluster = default_one_cluster
+        if deployment_data_path is not None:
+            self.deployment_data_path = deployment_data_path
         if deployment_gallery is not None:
             self.deployment_gallery = deployment_gallery
         if deployment_persistent_disk is not None:
@@ -338,6 +346,8 @@ class V1UserFeatures(object):
             self.mmt_strategy_selector = mmt_strategy_selector
         if model_store is not None:
             self.model_store = model_store
+        if multiple_deployment_versions is not None:
+            self.multiple_deployment_versions = multiple_deployment_versions
         if org_level_member_permissions is not None:
             self.org_level_member_permissions = org_level_member_permissions
         if plugin_biz_chat is not None:
@@ -804,6 +814,27 @@ class V1UserFeatures(object):
 
         self._default_one_cluster = default_one_cluster
 
+    @property
+    def deployment_data_path(self) -> 'bool':
+        """Gets the deployment_data_path of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The deployment_data_path of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._deployment_data_path
+
+    @deployment_data_path.setter
+    def deployment_data_path(self, deployment_data_path: 'bool'):
+        """Sets the deployment_data_path of this V1UserFeatures.
+
+
+        :param deployment_data_path: The deployment_data_path of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._deployment_data_path = deployment_data_path
+
     @property
     def deployment_gallery(self) -> 'bool':
         """Gets the deployment_gallery of this V1UserFeatures.  # noqa: E501
@@ -1203,6 +1234,27 @@ class V1UserFeatures(object):
 
         self._model_store = model_store
 
+    @property
+    def multiple_deployment_versions(self) -> 'bool':
+        """Gets the multiple_deployment_versions of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The multiple_deployment_versions of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._multiple_deployment_versions
+
+    @multiple_deployment_versions.setter
+    def multiple_deployment_versions(self, multiple_deployment_versions: 'bool'):
+        """Sets the multiple_deployment_versions of this V1UserFeatures.
+
+
+        :param multiple_deployment_versions: The multiple_deployment_versions of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._multiple_deployment_versions = multiple_deployment_versions
+
     @property
     def org_level_member_permissions(self) -> 'bool':
         """Gets the org_level_member_permissions of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py

@@ -43,27 +43,42 @@ class V1ValidateDataConnectionResponse(object):
     swagger_types = {
         'aws': 'V1AwsDataConnection',
         'efs': 'V1EfsDataConnection',
-        'gcp': 'V1GcpDataConnection'
+        'efs_folder': 'V1EFSFolderDataConnection',
+        'gcp': 'V1GcpDataConnection',
+        'gcs_folder': 'V1GCSFolderDataConnection',
+        's3_folder': 'V1S3FolderDataConnection'
     }
 
     attribute_map = {
         'aws': 'aws',
         'efs': 'efs',
-        'gcp': 'gcp'
+        'efs_folder': 'efsFolder',
+        'gcp': 'gcp',
+        'gcs_folder': 'gcsFolder',
+        's3_folder': 's3Folder'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsDataConnection' =None, gcp: 'V1GcpDataConnection' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
         """V1ValidateDataConnectionResponse - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._efs = None
+        self._efs_folder = None
         self._gcp = None
+        self._gcs_folder = None
+        self._s3_folder = None
         self.discriminator = None
         if aws is not None:
             self.aws = aws
         if efs is not None:
             self.efs = efs
+        if efs_folder is not None:
+            self.efs_folder = efs_folder
         if gcp is not None:
             self.gcp = gcp
+        if gcs_folder is not None:
+            self.gcs_folder = gcs_folder
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
 
     @property
     def aws(self) -> 'V1AwsDataConnection':
@@ -107,6 +122,27 @@ class V1ValidateDataConnectionResponse(object):
 
         self._efs = efs
 
+    @property
+    def efs_folder(self) -> 'V1EFSFolderDataConnection':
+        """Gets the efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1EFSFolderDataConnection
+        """
+        return self._efs_folder
+
+    @efs_folder.setter
+    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
+        """Sets the efs_folder of this V1ValidateDataConnectionResponse.
+
+
+        :param efs_folder: The efs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1EFSFolderDataConnection
+        """
+
+        self._efs_folder = efs_folder
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this V1ValidateDataConnectionResponse.  # noqa: E501
@@ -128,6 +164,48 @@ class V1ValidateDataConnectionResponse(object):
 
         self._gcp = gcp
 
+    @property
+    def gcs_folder(self) -> 'V1GCSFolderDataConnection':
+        """Gets the gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1GCSFolderDataConnection
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'V1GCSFolderDataConnection'):
+        """Sets the gcs_folder of this V1ValidateDataConnectionResponse.
+
+
+        :param gcs_folder: The gcs_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1GCSFolderDataConnection
+        """
+
+        self._gcs_folder = gcs_folder
+
+    @property
+    def s3_folder(self) -> 'V1S3FolderDataConnection':
+        """Gets the s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+
+
+        :return: The s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :rtype: V1S3FolderDataConnection
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'V1S3FolderDataConnection'):
+        """Sets the s3_folder of this V1ValidateDataConnectionResponse.
+
+
+        :param s3_folder: The s3_folder of this V1ValidateDataConnectionResponse.  # noqa: E501
+        :type: V1S3FolderDataConnection
+        """
+
+        self._s3_folder = s3_folder
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/validate.py

@@ -45,7 +45,10 @@ class Validate(object):
         'check_is_public': 'bool',
         'cluster_ids': 'list[str]',
         'efs': 'V1EfsDataConnection',
-        'gcp': 'V1GcpDataConnection'
+        'efs_folder': 'V1EFSFolderDataConnection',
+        'gcp': 'V1GcpDataConnection',
+        'gcs_folder': 'V1GCSFolderDataConnection',
+        's3_folder': 'V1S3FolderDataConnection'
     }
 
     attribute_map = {
@@ -53,16 +56,22 @@ class Validate(object):
         'check_is_public': 'checkIsPublic',
         'cluster_ids': 'clusterIds',
         'efs': 'efs',
-        'gcp': 'gcp'
+        'efs_folder': 'efsFolder',
+        'gcp': 'gcp',
+        'gcs_folder': 'gcsFolder',
+        's3_folder': 's3Folder'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, check_is_public: 'bool' =None, cluster_ids: 'list[str]' =None, efs: 'V1EfsDataConnection' =None, gcp: 'V1GcpDataConnection' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, check_is_public: 'bool' =None, cluster_ids: 'list[str]' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, s3_folder: 'V1S3FolderDataConnection' =None):  # noqa: E501
         """Validate - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._check_is_public = None
         self._cluster_ids = None
         self._efs = None
+        self._efs_folder = None
         self._gcp = None
+        self._gcs_folder = None
+        self._s3_folder = None
         self.discriminator = None
         if aws is not None:
             self.aws = aws
@@ -72,8 +81,14 @@ class Validate(object):
             self.cluster_ids = cluster_ids
         if efs is not None:
             self.efs = efs
+        if efs_folder is not None:
+            self.efs_folder = efs_folder
         if gcp is not None:
             self.gcp = gcp
+        if gcs_folder is not None:
+            self.gcs_folder = gcs_folder
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
 
     @property
     def aws(self) -> 'V1AwsDataConnection':
@@ -159,6 +174,27 @@ class Validate(object):
 
         self._efs = efs
 
+    @property
+    def efs_folder(self) -> 'V1EFSFolderDataConnection':
+        """Gets the efs_folder of this Validate.  # noqa: E501
+
+
+        :return: The efs_folder of this Validate.  # noqa: E501
+        :rtype: V1EFSFolderDataConnection
+        """
+        return self._efs_folder
+
+    @efs_folder.setter
+    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
+        """Sets the efs_folder of this Validate.
+
+
+        :param efs_folder: The efs_folder of this Validate.  # noqa: E501
+        :type: V1EFSFolderDataConnection
+        """
+
+        self._efs_folder = efs_folder
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this Validate.  # noqa: E501
@@ -180,6 +216,48 @@ class Validate(object):
 
         self._gcp = gcp
 
+    @property
+    def gcs_folder(self) -> 'V1GCSFolderDataConnection':
+        """Gets the gcs_folder of this Validate.  # noqa: E501
+
+
+        :return: The gcs_folder of this Validate.  # noqa: E501
+        :rtype: V1GCSFolderDataConnection
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'V1GCSFolderDataConnection'):
+        """Sets the gcs_folder of this Validate.
+
+
+        :param gcs_folder: The gcs_folder of this Validate.  # noqa: E501
+        :type: V1GCSFolderDataConnection
+        """
+
+        self._gcs_folder = gcs_folder
+
+    @property
+    def s3_folder(self) -> 'V1S3FolderDataConnection':
+        """Gets the s3_folder of this Validate.  # noqa: E501
+
+
+        :return: The s3_folder of this Validate.  # noqa: E501
+        :rtype: V1S3FolderDataConnection
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'V1S3FolderDataConnection'):
+        """Sets the s3_folder of this Validate.
+
+
+        :param s3_folder: The s3_folder of this Validate.  # noqa: E501
+        :type: V1S3FolderDataConnection
+        """
+
+        self._s3_folder = s3_folder
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/rest_client.py

@@ -32,6 +32,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     JobsServiceApi,
     AssistantsServiceApi,
     StorageServiceApi,
+    DeploymentTemplatesServiceApi,
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.source_code.logs_socket_api import LightningLogsSocketAPI
@@ -89,6 +90,7 @@ class GridRestClient(
         JobsServiceApi,
         AssistantsServiceApi,
         StorageServiceApi,
+        DeploymentTemplatesServiceApi,
 ):
 
     def __init__(self, api_client: Optional[ApiClient] = None):