lightning-sdk 0.1.34__py3-none-any.whl → 0.1.35__py3-none-any.whl

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.34"
+__version__ = "0.1.35"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/ai_hub.py

@@ -23,12 +23,66 @@ class AIHub:
         self._api = AIHubApi()
         self._auth = None
 
+    def api_info(self, api_id: str) -> dict:
+        """Get full API template info such as input details.
+
+        Example:
+            ai_hub = AIHub()
+            api_info = ai_hub.api_info("api_12345")
+
+        Args:
+            api_id: The ID of the API for which information is requested.
+
+        Returns:
+            A dictionary containing detailed information about the API,
+            including its name, description, creation and update timestamps,
+            parameters, tags, job specifications, and autoscaling settings.
+        """
+        template = self._api.api_info(api_id)
+
+        api_arguments = [
+            {
+                "name": param.name,
+                "short_description": param.short_description,
+                "required": param.required,
+                "default": param.input.default_value,
+            }
+            for param in template.parameter_spec.parameters
+        ]
+
+        return {
+            "name": template.name,
+            "description": template.description,
+            "created_at": template.created_at,
+            "updated_at": template.updated_at,
+            "api_arguments": api_arguments,
+            "tags": [tag.name for tag in template.tags],
+            "job": {
+                "image": template.spec_v2.job.image,
+                "interruptible": template.spec_v2.job.spot,
+                "instance_type": template.spec_v2.job.instance_type,
+                "resources": template.spec_v2.job.resources,
+            },
+            "autoscaling": {
+                "enabled": template.spec_v2.autoscaling.enabled,
+                "min_replicas": template.spec_v2.autoscaling.min_replicas,
+                "max_replicas": template.spec_v2.autoscaling.max_replicas,
+            },
+        }
+
     def list_apis(self, search: Optional[str] = None) -> List[Dict[str, str]]:
         """Get a list of AI Hub API templates.
 
         Example:
-            api_hub = AIHub()
-            api_list = api_hub.list_apis(search="Llama")
+            ai_hub = AIHub()
+            api_list = ai_hub.list_apis(search="Llama")
+
+        Args:
+            search: A search query to filter the list of APIs. Defaults to None.
+
+        Returns:
+            A list of dictionaries, each containing information about an API,
+            such as its ID, name, description, creator's username, and creation timestamp.
         """
         search_query = search or ""
         api_templates = self._api.list_apis(search_query=search_query)
@@ -84,7 +138,7 @@ class AIHub:
 
         Args:
             api_id: The ID of the API you want to deploy.
-            cluster: The cluster where you want to deploy the API, such as "lightning-public-prod".
+            cluster: The cluster where you want to deploy the API, such as "lightning-public-prod". Defaults to None.
             name: Name for the deployed API. Defaults to None.
             teamspace: The team or group for deployment. Defaults to None.
             org: The organization for deployment. Defaults to None.

lightning_sdk/api/ai_hub_api.py

@@ -1,4 +1,5 @@
 import re
+import traceback
 from typing import List, Optional
 
 import backoff
@@ -6,6 +7,7 @@ import backoff
 from lightning_sdk.lightning_cloud.openapi.models import (
     CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
     V1Deployment,
+    V1DeploymentTemplate,
     V1ParameterizationSpec,
 )
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import (
@@ -18,6 +20,15 @@ class AIHubApi:
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=3)
 
+    def api_info(self, api_id: str) -> "V1DeploymentTemplate":
+        try:
+            return self._client.deployment_templates_service_get_deployment_template(api_id)
+        except Exception as e:
+            stack_trace = traceback.format_exc()
+            if "record not found" in stack_trace:
+                raise ValueError(f"api_id={api_id} not found.") from None
+            raise e
+
     @backoff.on_predicate(backoff.expo, lambda x: not x, max_tries=5)
     def list_apis(self, search_query: str) -> List[V1DeploymentTemplateGalleryResponse]:
         kwargs = {"show_globally_visible": True}

lightning_sdk/cli/ai_hub.py

@@ -10,6 +10,17 @@ class _AIHub(_StudiosMenu):
     def __init__(self) -> None:
         self._hub = AIHub()
 
+    def api_info(self, api_id: str) -> dict:
+        """Get full API template info such as input details.
+
+        Example:
+          lightning aihub api_info [API_ID]
+
+        Args:
+          api_id: The ID of the API for which information is requested.
+        """
+        return self._hub.api_info(api_id)
+
     def list_apis(self, search: Optional[str] = None) -> List[dict]:
         """List API templates available in the AI Hub.
 

lightning_sdk/cli/download.py

@@ -13,19 +13,19 @@ from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 class _Downloads(_StudiosMenu):
     """Download files and folders from Lightning AI."""
 
-    def model(self, name: str, path: Optional[str] = None) -> None:
+    def model(self, name: str, download_dir: str = ".") -> None:
         """Download a Model.
 
         Args:
           name: The name of the Model you want to download.
             This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
-          path: The path to the directory where the Model should be downloaded.
+          download_dir: The directory where the Model should be downloaded.
         """
         org_name, teamspace_name, model_name = _parse_model_name(name)
         teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
         teamspace.download_model(
             name=model_name,
-            download_dir=path or ".",
+            download_dir=download_dir,
             progress_bar=True,
         )
 

lightning_sdk/lightning_cloud/login.py

@@ -8,7 +8,7 @@ from dataclasses import dataclass
 from typing import Optional
 from urllib.parse import urlencode
 
-import click
+import webbrowser
 
 import requests
 import uvicorn
@@ -165,7 +165,10 @@ class AuthServer:
             )
 
         logger.info(f"login started for lightning.ai, opening {url}")
-        click.launch(url)
+        ok = webbrowser.open(url)
+        if not ok:
+            # can't open a browser, authentication failed
+            raise RuntimeError("Failed to authenticate to Lightning. When running without access to a browser, `LIGHTNING_USER_ID` and `LIGHTNING_API_KEY` should be exported.")
 
         @app.get("/login-complete")
         async def save_token(request: Request,

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -5610,6 +5610,7 @@ class CloudSpaceServiceApi(object):
         :param str cloud_space_id: (required)
         :param bool draft:
         :param str limit:
+        :param bool include_publications:
         :return: V1ListCloudSpaceVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -5634,12 +5635,13 @@ class CloudSpaceServiceApi(object):
         :param str cloud_space_id: (required)
         :param bool draft:
         :param str limit:
+        :param bool include_publications:
         :return: V1ListCloudSpaceVersionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cloud_space_id', 'draft', 'limit']  # noqa: E501
+        all_params = ['project_id', 'cloud_space_id', 'draft', 'limit', 'include_publications']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -5676,6 +5678,8 @@ class CloudSpaceServiceApi(object):
             query_params.append(('draft', params['draft']))  # noqa: E501
         if 'limit' in params:
             query_params.append(('limit', params['limit']))  # noqa: E501
+        if 'include_publications' in params:
+            query_params.append(('includePublications', params['include_publications']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py

@@ -42,6 +42,7 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
     """
     swagger_types = {
         'autoscaling': 'V1AutoscalingSpec',
+        'cloudspace_id': 'str',
         'cluster_id': 'str',
         'endpoint': 'V1Endpoint',
         'name': 'str',
@@ -53,6 +54,7 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
 
     attribute_map = {
         'autoscaling': 'autoscaling',
+        'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'endpoint': 'endpoint',
         'name': 'name',
@@ -62,9 +64,10 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
         'strategy': 'strategy'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None):  # noqa: E501
         """CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
+        self._cloudspace_id = None
         self._cluster_id = None
         self._endpoint = None
         self._name = None
@@ -75,6 +78,8 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
         self.discriminator = None
         if autoscaling is not None:
             self.autoscaling = autoscaling
+        if cloudspace_id is not None:
+            self.cloudspace_id = cloudspace_id
         if cluster_id is not None:
             self.cluster_id = cluster_id
         if endpoint is not None:
@@ -111,6 +116,27 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
 
         self._autoscaling = autoscaling
 
+    @property
+    def cloudspace_id(self) -> 'str':
+        """Gets the cloudspace_id of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+
+
+        :return: The cloudspace_id of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloudspace_id
+
+    @cloudspace_id.setter
+    def cloudspace_id(self, cloudspace_id: 'str'):
+        """Sets the cloudspace_id of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.
+
+
+        :param cloudspace_id: The cloudspace_id of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+        :type: str
+        """
+
+        self._cloudspace_id = cloudspace_id
+
     @property
     def cluster_id(self) -> 'str':
         """Gets the cluster_id of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py

@@ -42,6 +42,7 @@ class DeploymentsIdBody(object):
     """
     swagger_types = {
         'autoscaling': 'V1AutoscalingSpec',
+        'cloudspace_id': 'str',
         'created_at': 'datetime',
         'desired_state': 'V1DeploymentState',
         'endpoint': 'V1Endpoint',
@@ -59,6 +60,7 @@ class DeploymentsIdBody(object):
 
     attribute_map = {
         'autoscaling': 'autoscaling',
+        'cloudspace_id': 'cloudspaceId',
         'created_at': 'createdAt',
         'desired_state': 'desiredState',
         'endpoint': 'endpoint',
@@ -74,9 +76,10 @@ class DeploymentsIdBody(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """DeploymentsIdBody - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
+        self._cloudspace_id = None
         self._created_at = None
         self._desired_state = None
         self._endpoint = None
@@ -93,6 +96,8 @@ class DeploymentsIdBody(object):
         self.discriminator = None
         if autoscaling is not None:
             self.autoscaling = autoscaling
+        if cloudspace_id is not None:
+            self.cloudspace_id = cloudspace_id
         if created_at is not None:
             self.created_at = created_at
         if desired_state is not None:
@@ -141,6 +146,27 @@ class DeploymentsIdBody(object):
 
         self._autoscaling = autoscaling
 
+    @property
+    def cloudspace_id(self) -> 'str':
+        """Gets the cloudspace_id of this DeploymentsIdBody.  # noqa: E501
+
+
+        :return: The cloudspace_id of this DeploymentsIdBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloudspace_id
+
+    @cloudspace_id.setter
+    def cloudspace_id(self, cloudspace_id: 'str'):
+        """Sets the cloudspace_id of this DeploymentsIdBody.
+
+
+        :param cloudspace_id: The cloudspace_id of this DeploymentsIdBody.  # noqa: E501
+        :type: str
+        """
+
+        self._cloudspace_id = cloudspace_id
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this DeploymentsIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py

@@ -46,6 +46,7 @@ class SnowflakeExportBody(object):
         'compress': 'bool',
         'connection_id': 'str',
         'format': 'str',
+        'include_header': 'bool',
         'password': 'str',
         'query': 'str',
         'query_name': 'str',
@@ -58,19 +59,21 @@ class SnowflakeExportBody(object):
         'compress': 'compress',
         'connection_id': 'connectionId',
         'format': 'format',
+        'include_header': 'includeHeader',
         'password': 'password',
         'query': 'query',
         'query_name': 'queryName',
         'username': 'username'
     }
 
-    def __init__(self, account: 'str' =None, cluster_id: 'str' =None, compress: 'bool' =None, connection_id: 'str' =None, format: 'str' =None, password: 'str' =None, query: 'str' =None, query_name: 'str' =None, username: 'str' =None):  # noqa: E501
+    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
         """SnowflakeExportBody - a model defined in Swagger"""  # noqa: E501
         self._account = None
         self._cluster_id = None
         self._compress = None
         self._connection_id = None
         self._format = None
+        self._include_header = None
         self._password = None
         self._query = None
         self._query_name = None
@@ -86,6 +89,8 @@ class SnowflakeExportBody(object):
             self.connection_id = connection_id
         if format is not None:
             self.format = format
+        if include_header is not None:
+            self.include_header = include_header
         if password is not None:
             self.password = password
         if query is not None:
@@ -200,6 +205,27 @@ class SnowflakeExportBody(object):
 
         self._format = format
 
+    @property
+    def include_header(self) -> 'bool':
+        """Gets the include_header of this SnowflakeExportBody.  # noqa: E501
+
+
+        :return: The include_header of this SnowflakeExportBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._include_header
+
+    @include_header.setter
+    def include_header(self, include_header: 'bool'):
+        """Sets the include_header of this SnowflakeExportBody.
+
+
+        :param include_header: The include_header of this SnowflakeExportBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._include_header = include_header
+
     @property
     def password(self) -> 'str':
         """Gets the password of this SnowflakeExportBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py

@@ -48,6 +48,7 @@ class V1CapacityBlockOffering(object):
         'fee': 'float',
         'instance_count': 'int',
         'instance_type': 'str',
+        'provider_fee': 'float',
         'region': 'str',
         'start_date': 'datetime',
         'upfront_fee': 'float'
@@ -61,12 +62,13 @@ class V1CapacityBlockOffering(object):
         'fee': 'fee',
         'instance_count': 'instanceCount',
         'instance_type': 'instanceType',
+        'provider_fee': 'providerFee',
         'region': 'region',
         'start_date': 'startDate',
         'upfront_fee': 'upfrontFee'
     }
 
-    def __init__(self, capacity_block_duration_hours: 'int' =None, capacity_block_offering_id: 'str' =None, currency_code: 'str' =None, end_date: 'datetime' =None, fee: 'float' =None, instance_count: 'int' =None, instance_type: 'str' =None, region: 'str' =None, start_date: 'datetime' =None, upfront_fee: 'float' =None):  # noqa: E501
+    def __init__(self, capacity_block_duration_hours: 'int' =None, capacity_block_offering_id: 'str' =None, currency_code: 'str' =None, end_date: 'datetime' =None, fee: 'float' =None, instance_count: 'int' =None, instance_type: 'str' =None, provider_fee: 'float' =None, region: 'str' =None, start_date: 'datetime' =None, upfront_fee: 'float' =None):  # noqa: E501
         """V1CapacityBlockOffering - a model defined in Swagger"""  # noqa: E501
         self._capacity_block_duration_hours = None
         self._capacity_block_offering_id = None
@@ -75,6 +77,7 @@ class V1CapacityBlockOffering(object):
         self._fee = None
         self._instance_count = None
         self._instance_type = None
+        self._provider_fee = None
         self._region = None
         self._start_date = None
         self._upfront_fee = None
@@ -93,6 +96,8 @@ class V1CapacityBlockOffering(object):
             self.instance_count = instance_count
         if instance_type is not None:
             self.instance_type = instance_type
+        if provider_fee is not None:
+            self.provider_fee = provider_fee
         if region is not None:
             self.region = region
         if start_date is not None:
@@ -247,6 +252,27 @@ class V1CapacityBlockOffering(object):
 
         self._instance_type = instance_type
 
+    @property
+    def provider_fee(self) -> 'float':
+        """Gets the provider_fee of this V1CapacityBlockOffering.  # noqa: E501
+
+
+        :return: The provider_fee of this V1CapacityBlockOffering.  # noqa: E501
+        :rtype: float
+        """
+        return self._provider_fee
+
+    @provider_fee.setter
+    def provider_fee(self, provider_fee: 'float'):
+        """Sets the provider_fee of this V1CapacityBlockOffering.
+
+
+        :param provider_fee: The provider_fee of this V1CapacityBlockOffering.  # noqa: E501
+        :type: float
+        """
+
+        self._provider_fee = provider_fee
+
     @property
     def region(self) -> 'str':
         """Gets the region of this V1CapacityBlockOffering.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py

@@ -49,6 +49,7 @@ class V1CloudSpaceVersion(object):
         'created_at': 'datetime',
         'description': 'str',
         'draft': 'bool',
+        'globally_visible': 'bool',
         'hide_files': 'bool',
         'id': 'str',
         'license': 'str',
@@ -58,6 +59,7 @@ class V1CloudSpaceVersion(object):
         'paper_org_avatar_url': 'str',
         'paper_url': 'str',
         'project_id': 'str',
+        'unpublished': 'bool',
         'updated_at': 'datetime',
         'user_id': 'str',
         'version_name': 'str'
@@ -72,6 +74,7 @@ class V1CloudSpaceVersion(object):
         'created_at': 'createdAt',
         'description': 'description',
         'draft': 'draft',
+        'globally_visible': 'globallyVisible',
         'hide_files': 'hideFiles',
         'id': 'id',
         'license': 'license',
@@ -81,12 +84,13 @@ class V1CloudSpaceVersion(object):
         'paper_org_avatar_url': 'paperOrgAvatarUrl',
         'paper_url': 'paperUrl',
         'project_id': 'projectId',
+        'unpublished': 'unpublished',
         'updated_at': 'updatedAt',
         'user_id': 'userId',
         'version_name': 'versionName'
     }
 
-    def __init__(self, about_page_content: 'str' =None, about_page_id: 'str' =None, cloud_space_id: 'str' =None, code_url: 'str' =None, code_version: 'V1CloudSpaceCodeVersion' =None, created_at: 'datetime' =None, description: 'str' =None, draft: 'bool' =None, hide_files: 'bool' =None, id: 'str' =None, license: 'str' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, project_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None, version_name: 'str' =None):  # noqa: E501
+    def __init__(self, about_page_content: 'str' =None, about_page_id: 'str' =None, cloud_space_id: 'str' =None, code_url: 'str' =None, code_version: 'V1CloudSpaceCodeVersion' =None, created_at: 'datetime' =None, description: 'str' =None, draft: 'bool' =None, globally_visible: 'bool' =None, hide_files: 'bool' =None, id: 'str' =None, license: 'str' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, project_id: 'str' =None, unpublished: 'bool' =None, updated_at: 'datetime' =None, user_id: 'str' =None, version_name: 'str' =None):  # noqa: E501
         """V1CloudSpaceVersion - a model defined in Swagger"""  # noqa: E501
         self._about_page_content = None
         self._about_page_id = None
@@ -96,6 +100,7 @@ class V1CloudSpaceVersion(object):
         self._created_at = None
         self._description = None
         self._draft = None
+        self._globally_visible = None
         self._hide_files = None
         self._id = None
         self._license = None
@@ -105,6 +110,7 @@ class V1CloudSpaceVersion(object):
         self._paper_org_avatar_url = None
         self._paper_url = None
         self._project_id = None
+        self._unpublished = None
         self._updated_at = None
         self._user_id = None
         self._version_name = None
@@ -125,6 +131,8 @@ class V1CloudSpaceVersion(object):
             self.description = description
         if draft is not None:
             self.draft = draft
+        if globally_visible is not None:
+            self.globally_visible = globally_visible
         if hide_files is not None:
             self.hide_files = hide_files
         if id is not None:
@@ -143,6 +151,8 @@ class V1CloudSpaceVersion(object):
             self.paper_url = paper_url
         if project_id is not None:
             self.project_id = project_id
+        if unpublished is not None:
+            self.unpublished = unpublished
         if updated_at is not None:
             self.updated_at = updated_at
         if user_id is not None:
@@ -318,6 +328,27 @@ class V1CloudSpaceVersion(object):
 
         self._draft = draft
 
+    @property
+    def globally_visible(self) -> 'bool':
+        """Gets the globally_visible of this V1CloudSpaceVersion.  # noqa: E501
+
+
+        :return: The globally_visible of this V1CloudSpaceVersion.  # noqa: E501
+        :rtype: bool
+        """
+        return self._globally_visible
+
+    @globally_visible.setter
+    def globally_visible(self, globally_visible: 'bool'):
+        """Sets the globally_visible of this V1CloudSpaceVersion.
+
+
+        :param globally_visible: The globally_visible of this V1CloudSpaceVersion.  # noqa: E501
+        :type: bool
+        """
+
+        self._globally_visible = globally_visible
+
     @property
     def hide_files(self) -> 'bool':
         """Gets the hide_files of this V1CloudSpaceVersion.  # noqa: E501
@@ -507,6 +538,27 @@ class V1CloudSpaceVersion(object):
 
         self._project_id = project_id
 
+    @property
+    def unpublished(self) -> 'bool':
+        """Gets the unpublished of this V1CloudSpaceVersion.  # noqa: E501
+
+
+        :return: The unpublished of this V1CloudSpaceVersion.  # noqa: E501
+        :rtype: bool
+        """
+        return self._unpublished
+
+    @unpublished.setter
+    def unpublished(self, unpublished: 'bool'):
+        """Sets the unpublished of this V1CloudSpaceVersion.
+
+
+        :param unpublished: The unpublished of this V1CloudSpaceVersion.  # noqa: E501
+        :type: bool
+        """
+
+        self._unpublished = unpublished
+
     @property
     def updated_at(self) -> 'datetime':
         """Gets the updated_at of this V1CloudSpaceVersion.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py

@@ -56,6 +56,7 @@ class V1ClusterAccelerator(object):
         'device_card': 'str',
         'device_info': 'str',
         'display_name': 'str',
+        'dws_only': 'bool',
         'enabled': 'bool',
         'family': 'str',
         'instance_id': 'str',
@@ -95,6 +96,7 @@ class V1ClusterAccelerator(object):
         'device_card': 'deviceCard',
         'device_info': 'deviceInfo',
         'display_name': 'displayName',
+        'dws_only': 'dwsOnly',
         'enabled': 'enabled',
         'family': 'family',
         'instance_id': 'instanceId',
@@ -118,7 +120,7 @@ class V1ClusterAccelerator(object):
         'spot_price': 'spotPrice'
     }
 
-    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None):  # noqa: E501
+    def __init__(self, accelerator_type: 'str' =None, allowed_resources: 'list[str]' =None, available_in_seconds: 'str' =None, available_in_seconds_spot: 'str' =None, available_zones: 'list[str]' =None, byoc_only: 'bool' =None, capacity_block_only: 'bool' =None, capacity_block_price: 'float' =None, capacity_blocks_available: 'list[V1ClusterCapacityReservation]' =None, cluster_id: 'str' =None, cost: 'float' =None, detailed_quotas_info: 'list[V1AcceleratorQuotaInfo]' =None, device_card: 'str' =None, device_info: 'str' =None, display_name: 'str' =None, dws_only: 'bool' =None, enabled: 'bool' =None, family: 'str' =None, instance_id: 'str' =None, is_custom: 'bool' =None, is_tier_restricted: 'bool' =None, local_disk_size: 'str' =None, local_disk_supported: 'bool' =None, max_available_quota: 'str' =None, non_spot: 'bool' =None, quota_checked_at: 'datetime' =None, quota_code: 'str' =None, quota_name: 'str' =None, quota_page_url: 'str' =None, quota_service_code: 'str' =None, quota_utilization: 'str' =None, quota_value: 'str' =None, reservable: 'bool' =None, reservation_available_zones: 'list[str]' =None, resources: 'V1Resources' =None, slug: 'str' =None, spot_price: 'float' =None):  # noqa: E501
         """V1ClusterAccelerator - a model defined in Swagger"""  # noqa: E501
         self._accelerator_type = None
         self._allowed_resources = None
@@ -135,6 +137,7 @@ class V1ClusterAccelerator(object):
         self._device_card = None
         self._device_info = None
         self._display_name = None
+        self._dws_only = None
         self._enabled = None
         self._family = None
         self._instance_id = None
@@ -187,6 +190,8 @@ class V1ClusterAccelerator(object):
             self.device_info = device_info
         if display_name is not None:
             self.display_name = display_name
+        if dws_only is not None:
+            self.dws_only = dws_only
         if enabled is not None:
             self.enabled = enabled
         if family is not None:
@@ -545,6 +550,27 @@ class V1ClusterAccelerator(object):
 
         self._display_name = display_name
 
+    @property
+    def dws_only(self) -> 'bool':
+        """Gets the dws_only of this V1ClusterAccelerator.  # noqa: E501
+
+
+        :return: The dws_only of this V1ClusterAccelerator.  # noqa: E501
+        :rtype: bool
+        """
+        return self._dws_only
+
+    @dws_only.setter
+    def dws_only(self, dws_only: 'bool'):
+        """Sets the dws_only of this V1ClusterAccelerator.
+
+
+        :param dws_only: The dws_only of this V1ClusterAccelerator.  # noqa: E501
+        :type: bool
+        """
+
+        self._dws_only = dws_only
+
     @property
     def enabled(self) -> 'bool':
         """Gets the enabled of this V1ClusterAccelerator.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py

@@ -42,6 +42,7 @@ class V1Deployment(object):
     """
     swagger_types = {
         'autoscaling': 'V1AutoscalingSpec',
+        'cloudspace_id': 'str',
         'created_at': 'datetime',
         'desired_state': 'V1DeploymentState',
         'endpoint': 'V1Endpoint',
@@ -61,6 +62,7 @@ class V1Deployment(object):
 
     attribute_map = {
         'autoscaling': 'autoscaling',
+        'cloudspace_id': 'cloudspaceId',
         'created_at': 'createdAt',
         'desired_state': 'desiredState',
         'endpoint': 'endpoint',
@@ -78,9 +80,10 @@ class V1Deployment(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Deployment - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
+        self._cloudspace_id = None
         self._created_at = None
         self._desired_state = None
         self._endpoint = None
@@ -99,6 +102,8 @@ class V1Deployment(object):
         self.discriminator = None
         if autoscaling is not None:
             self.autoscaling = autoscaling
+        if cloudspace_id is not None:
+            self.cloudspace_id = cloudspace_id
         if created_at is not None:
             self.created_at = created_at
         if desired_state is not None:
@@ -151,6 +156,27 @@ class V1Deployment(object):
 
         self._autoscaling = autoscaling
 
+    @property
+    def cloudspace_id(self) -> 'str':
+        """Gets the cloudspace_id of this V1Deployment.  # noqa: E501
+
+
+        :return: The cloudspace_id of this V1Deployment.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloudspace_id
+
+    @cloudspace_id.setter
+    def cloudspace_id(self, cloudspace_id: 'str'):
+        """Sets the cloudspace_id of this V1Deployment.
+
+
+        :param cloudspace_id: The cloudspace_id of this V1Deployment.  # noqa: E501
+        :type: str
+        """
+
+        self._cloudspace_id = cloudspace_id
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this V1Deployment.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py

@@ -56,6 +56,7 @@ class V1GetUserResponse(object):
         'opted_in_marketing_emails': 'bool',
         'organization': 'str',
         'organizations': 'list[V1Organization]',
+        'phone_number': 'str',
         'picture_url': 'str',
         'preferred_color_scheme': 'str',
         'preferred_ide': 'str',
@@ -90,6 +91,7 @@ class V1GetUserResponse(object):
         'opted_in_marketing_emails': 'optedInMarketingEmails',
         'organization': 'organization',
         'organizations': 'organizations',
+        'phone_number': 'phoneNumber',
         'picture_url': 'pictureUrl',
         'preferred_color_scheme': 'preferredColorScheme',
         'preferred_ide': 'preferredIde',
@@ -108,7 +110,7 @@ class V1GetUserResponse(object):
         'website': 'website'
     }
 
-    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, email: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, email: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, phone_number: 'str' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None):  # noqa: E501
         """V1GetUserResponse - a model defined in Swagger"""  # noqa: E501
         self._agree_to_terms_and_conditions = None
         self._api_key = None
@@ -125,6 +127,7 @@ class V1GetUserResponse(object):
         self._opted_in_marketing_emails = None
         self._organization = None
         self._organizations = None
+        self._phone_number = None
         self._picture_url = None
         self._preferred_color_scheme = None
         self._preferred_ide = None
@@ -172,6 +175,8 @@ class V1GetUserResponse(object):
             self.organization = organization
         if organizations is not None:
             self.organizations = organizations
+        if phone_number is not None:
+            self.phone_number = phone_number
         if picture_url is not None:
             self.picture_url = picture_url
         if preferred_color_scheme is not None:
@@ -520,6 +525,27 @@ class V1GetUserResponse(object):
 
         self._organizations = organizations
 
+    @property
+    def phone_number(self) -> 'str':
+        """Gets the phone_number of this V1GetUserResponse.  # noqa: E501
+
+
+        :return: The phone_number of this V1GetUserResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._phone_number
+
+    @phone_number.setter
+    def phone_number(self, phone_number: 'str'):
+        """Sets the phone_number of this V1GetUserResponse.
+
+
+        :param phone_number: The phone_number of this V1GetUserResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._phone_number = phone_number
+
     @property
     def picture_url(self) -> 'str':
         """Gets the picture_url of this V1GetUserResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py

@@ -44,6 +44,7 @@ class V1GoogleCloudDirectV1(object):
         'bucket_name': 'str',
         'credentials_secret_id': 'str',
         'credentials_service_account_email': 'str',
+        'dws_enabled': 'bool',
         'primary_region': 'str',
         'project_id': 'str',
         'regions': 'list[str]',
@@ -55,6 +56,7 @@ class V1GoogleCloudDirectV1(object):
         'bucket_name': 'bucketName',
         'credentials_secret_id': 'credentialsSecretId',
         'credentials_service_account_email': 'credentialsServiceAccountEmail',
+        'dws_enabled': 'dwsEnabled',
         'primary_region': 'primaryRegion',
         'project_id': 'projectId',
         'regions': 'regions',
@@ -62,11 +64,12 @@ class V1GoogleCloudDirectV1(object):
         'source_cidr_ips': 'sourceCidrIps'
     }
 
-    def __init__(self, bucket_name: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
+    def __init__(self, bucket_name: 'str' =None, credentials_secret_id: 'str' =None, credentials_service_account_email: 'str' =None, dws_enabled: 'bool' =None, primary_region: 'str' =None, project_id: 'str' =None, regions: 'list[str]' =None, service_account_email: 'str' =None, source_cidr_ips: 'list[str]' =None):  # noqa: E501
         """V1GoogleCloudDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._bucket_name = None
         self._credentials_secret_id = None
         self._credentials_service_account_email = None
+        self._dws_enabled = None
         self._primary_region = None
         self._project_id = None
         self._regions = None
@@ -79,6 +82,8 @@ class V1GoogleCloudDirectV1(object):
             self.credentials_secret_id = credentials_secret_id
         if credentials_service_account_email is not None:
             self.credentials_service_account_email = credentials_service_account_email
+        if dws_enabled is not None:
+            self.dws_enabled = dws_enabled
         if primary_region is not None:
             self.primary_region = primary_region
         if project_id is not None:
@@ -155,6 +160,27 @@ class V1GoogleCloudDirectV1(object):
 
         self._credentials_service_account_email = credentials_service_account_email
 
+    @property
+    def dws_enabled(self) -> 'bool':
+        """Gets the dws_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+
+
+        :return: The dws_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+        :rtype: bool
+        """
+        return self._dws_enabled
+
+    @dws_enabled.setter
+    def dws_enabled(self, dws_enabled: 'bool'):
+        """Sets the dws_enabled of this V1GoogleCloudDirectV1.
+
+
+        :param dws_enabled: The dws_enabled of this V1GoogleCloudDirectV1.  # noqa: E501
+        :type: bool
+        """
+
+        self._dws_enabled = dws_enabled
+
     @property
     def primary_region(self) -> 'str':
         """Gets the primary_region of this V1GoogleCloudDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py

@@ -44,6 +44,7 @@ class V1SnowflakeDataConnection(object):
         'bucket_name': 'str',
         'compress': 'bool',
         'format': 'str',
+        'include_header': 'bool',
         'query': 'str',
         'secret_account': 'str',
         'secret_password': 'str',
@@ -54,17 +55,19 @@ class V1SnowflakeDataConnection(object):
         'bucket_name': 'bucketName',
         'compress': 'compress',
         'format': 'format',
+        'include_header': 'includeHeader',
         'query': 'query',
         'secret_account': 'secretAccount',
         'secret_password': 'secretPassword',
         'secret_username': 'secretUsername'
     }
 
-    def __init__(self, bucket_name: 'str' =None, compress: 'bool' =None, format: 'str' =None, query: 'str' =None, secret_account: 'str' =None, secret_password: 'str' =None, secret_username: 'str' =None):  # noqa: E501
+    def __init__(self, bucket_name: 'str' =None, compress: 'bool' =None, format: 'str' =None, include_header: 'bool' =None, query: 'str' =None, secret_account: 'str' =None, secret_password: 'str' =None, secret_username: 'str' =None):  # noqa: E501
         """V1SnowflakeDataConnection - a model defined in Swagger"""  # noqa: E501
         self._bucket_name = None
         self._compress = None
         self._format = None
+        self._include_header = None
         self._query = None
         self._secret_account = None
         self._secret_password = None
@@ -76,6 +79,8 @@ class V1SnowflakeDataConnection(object):
             self.compress = compress
         if format is not None:
             self.format = format
+        if include_header is not None:
+            self.include_header = include_header
         if query is not None:
             self.query = query
         if secret_account is not None:
@@ -148,6 +153,27 @@ class V1SnowflakeDataConnection(object):
 
         self._format = format
 
+    @property
+    def include_header(self) -> 'bool':
+        """Gets the include_header of this V1SnowflakeDataConnection.  # noqa: E501
+
+
+        :return: The include_header of this V1SnowflakeDataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._include_header
+
+    @include_header.setter
+    def include_header(self, include_header: 'bool'):
+        """Sets the include_header of this V1SnowflakeDataConnection.
+
+
+        :param include_header: The include_header of this V1SnowflakeDataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._include_header = include_header
+
     @property
     def query(self) -> 'str':
         """Gets the query of this V1SnowflakeDataConnection.  # noqa: E501

{lightning_sdk-0.1.34.dist-info → lightning_sdk-0.1.35.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.34
+Version: 0.1.35
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-0.1.34.dist-info → lightning_sdk-0.1.35.dist-info}/RECORD

@@ -1,7 +1,7 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=-9FknEmfW5Gc3b-j-SsKaj-QELmPoGpXvD4aNFPqp0E,925
+lightning_sdk/__init__.py,sha256=pHg4y-QNZLtwRTAaU_wY1WLB0GSyAp8iOFClabH419o,925
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
-lightning_sdk/ai_hub.py,sha256=iNlRFzTPTuzRdDIT7WhEw1Lp5Srn8g2SMKs-xE4QHDA,4231
+lightning_sdk/ai_hub.py,sha256=aaK40OBdAtWDaHaB50G0s4Ogx18kLRN__B31732vl2o,6292
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=RnQwUquc_YPotjh6YXOoJvZs8krX_QFhd7kGv4U_spQ,1844
 lightning_sdk/machine.py,sha256=VdFXStR6ilYBEYuxgGWzcAw2TtW-nEQVsh6hz-2aaEw,750
@@ -14,7 +14,7 @@ lightning_sdk/teamspace.py,sha256=T3nxbZG1HtKu40F8gtpapqn-GVqSL6nBv7d32vmB_DY,10
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
-lightning_sdk/api/ai_hub_api.py,sha256=8mDWQpXi0Sud3cRmjPvhlDRghYyJJpCDX8SvK-yY9ek,2565
+lightning_sdk/api/ai_hub_api.py,sha256=W4gF1F_L8h4B-UCg0eJ0DDQCWpnKQjGNnXaY8n-lTsE,3002
 lightning_sdk/api/deployment_api.py,sha256=9HhOHz7ElmjgX88YkPUtETPgz_jMFkK1nTO18-jUniI,21505
 lightning_sdk/api/job_api.py,sha256=pdQ3R5x04fyC49UqggWCIuPDLCmLYrnq791Ij-3vMr0,8457
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
@@ -23,8 +23,8 @@ lightning_sdk/api/teamspace_api.py,sha256=o-GBR3KLo298kDxO0myx-qlcCzSZnbR2OhZ73t
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
 lightning_sdk/api/utils.py,sha256=eHxq0dU_e9Pq5XjPlp7sNJ5_N3d44EtHKpbEiPPMkOQ,20562
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lightning_sdk/cli/ai_hub.py,sha256=AiwA-nhtP4arWqdezPRo6CNItTXOw756JmjEk2wTBXM,1306
-lightning_sdk/cli/download.py,sha256=b6dpUMLhro6p0y6CleCHxggQnZlZMxCq0bjuf8B6_eA,6191
+lightning_sdk/cli/ai_hub.py,sha256=WMNQ5WPL_KmP3T-qvHZ2O7h1oG8A_N25J7KCNCfmOa8,1616
+lightning_sdk/cli/download.py,sha256=f_pXtLyPXH4r6PVzg7-39O7dd479PTU5iLB0PgTKM7k,6185
 lightning_sdk/cli/entrypoint.py,sha256=ypNZ13FJlmlX-Osnp5eiZPmABt0xc8Rw2Yi9oDqrXGo,1332
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
@@ -42,7 +42,7 @@ lightning_sdk/job/work.py,sha256=JNuFp6TaNvO-V-YoO4tDu8e8AFAkHZj3vzt_tqlD6KE,969
 lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
 lightning_sdk/lightning_cloud/env.py,sha256=XZXpF4sD9jlB8DY0herTy_8XiUJuDVjxy5APjRD2_aU,1379
-lightning_sdk/lightning_cloud/login.py,sha256=8SVZ-RGszjjbJt4cEoeMhFG7RqBZ4xIlT2Remv3l08o,7040
+lightning_sdk/lightning_cloud/login.py,sha256=NSC53eJ24g59dySI9zRDl_-antxiflKDknYKxUNcwzQ,7307
 lightning_sdk/lightning_cloud/rest_client.py,sha256=oJG8xYf-QIIH4D21s6sq5uIVbc6SGwHtgB9eNGWIgVY,6622
 lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=YqR_GdQTDcA9DEQ1w6mRlX1cd_MrlpdEwzsBV2cZzJc,599
@@ -56,7 +56,7 @@ lightning_sdk/lightning_cloud/openapi/api/analytics_service_api.py,sha256=0BccCX
 lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=s9BG3P1bmSK5Q-uJh1iEX64VYNUO35S0QrzKxcZkiqs,95870
 lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAltWyj1wYg3HNk0cxT2HWbNhAqQSLvMDgM,30948
 lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=SDJ80TezTWKhE3LR67OI6dhHDHABc4BfG7FjcFX7Ba4,68129
-lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=oc6unkOxhBBvBujaVDIAakZgNM1nj0FZ2PMzBa9fmeM,369766
+lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=hjCLl9jQC7kVI0TrqR0oEY-yZkK_tFr2Q5siMmh4v7s,370022
 lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=m2JDI2cij2_3lqcKxcnx1UjZQaUAO_r7ZIS4CbM2kqk,145530
 lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=bvoAUiLsdPvyYMEWWvcEDe9oarIfvLzxz_ZMoADDHOM,50181
 lightning_sdk/lightning_cloud/openapi/api/dataset_service_api.py,sha256=wCrcXOMuVs5rA4PHyrVJqpInO-Vy1hulddjCJUenPJc,23508
@@ -111,10 +111,10 @@ lightning_sdk/lightning_cloud/openapi/models/command_argument_command_argument_t
 lightning_sdk/lightning_cloud/openapi/models/conversations_id_body.py,sha256=blW66WBRE-qMCkzLCOrx4EqmM1vI42O3MvP9rIlx_tE,3651
 lightning_sdk/lightning_cloud/openapi/models/create.py,sha256=xrtxd9B0VkqfdszT5GXOmfD5PLmvPrbxTd2Ktjq55eE,14373
 lightning_sdk/lightning_cloud/openapi/models/create_checkout_session_request_wallet_type.py,sha256=6mUpEXvoZxyrnOvfT9PfF34E3OGvRMzMfLFndUe-jKU,3217
-lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=tbgFc6advfcyZCmeTiGzPGdASHbOzV9jxUa3DPgaapA,11124
+lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=wF_p7KJBvtfsgtKolLYefd0Y2j8LRPLHJURS6l5kKDo,12144
 lightning_sdk/lightning_cloud/openapi/models/data_connection_mount_data_connection_mount_copy_status.py,sha256=ytC9VwBzQMvpOSZJ78uekfrMmwGRaTOclX5XC3a-dMs,3343
 lightning_sdk/lightning_cloud/openapi/models/datasets_id_body.py,sha256=OcNzTWEbLGF36eY-yImNCmTwY4ZMX2P8yPuHAvc6KSk,5533
-lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=CzAP4ulw4Q7nBFyLU3xHckqDLDKY7TRAB5ZaM64mV8o,13675
+lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=LeQdxsea0gFt2ziPSvE8kwcAbu-4LWGzZ1sefPUs04c,14475
 lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py,sha256=2TNCtt0YINExIsa6GYNNps5ton8QN-PPWSo6juG-oRo,14223
 lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py,sha256=-Nv94Q8XoAT3tfYVhcjutpihfoNS2b2STbTHJr1gomo,12309
 lightning_sdk/lightning_cloud/openapi/models/experiment_name_variant_name_body.py,sha256=f-pBOnWAd03BreivwAA0D1P1wEBGq2jaC1nK-GtGxVA,3910
@@ -200,7 +200,7 @@ lightning_sdk/lightning_cloud/openapi/models/service_artifact_artifact_kind.py,s
 lightning_sdk/lightning_cloud/openapi/models/service_health_service_status.py,sha256=AvLUZ5iHK9ZuqMbJ6O0N-sMSRfryKDU8djk4xaK9g5A,3196
 lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py,sha256=_-jU6n4JExbtPpHQwqiLc3iXeSQ7alk-GKUJiVK2dqs,10757
 lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py,sha256=bhq4Txf2UALD-AabsJpoxgLWXkj8InzgtySKb2hCgOg,9334
-lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py,sha256=ZUyB7BpOnFYfmUHL48FYz0iTSZLqE4jnWVl9aFpXrfc,9446
+lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py,sha256=AJPZ5MwsthL1IXljUbpBbaHVoVoba1umhrUibMJlmxg,10280
 lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py,sha256=6IT60t9IvhG21gl5bptnWhd5pj63_zgb0D4LNr57Nf0,6540
 lightning_sdk/lightning_cloud/openapi/models/spec_lightningapp_instance_id_works_body.py,sha256=dEI8pgLEkjFn1qy85Wb8cssJLWY6q0ERVbWlU4RvoU4,5432
 lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py,sha256=3OL3VqMUCNhwIvI7Q51-xwpBs4C6pyZYVUPGsaPuiPU,6824
@@ -248,7 +248,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py,sha256=yYc1ioTtL
 lightning_sdk/lightning_cloud/openapi/models/v1_build_spec.py,sha256=hqAjFwjZZeW-Jk3xWknetyoBA0J7xSrcUwl8rlkVUCA,6706
 lightning_sdk/lightning_cloud/openapi/models/v1_cancel_cloud_space_instance_switch_response.py,sha256=b7511arCBmJy8Lp7oWblcifG2r1xyV44q9nSW7yQzzM,3130
 lightning_sdk/lightning_cloud/openapi/models/v1_cancellation_metadata.py,sha256=JjtaI0vryHiLsmvMVbH4KZ6MHA8GzcWhQg0fkuNx-gM,4402
-lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py,sha256=QO9vjLcu5NMTPyXHGEyWmb3sDXV02LLgKPhfwm3iih4,11408
+lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py,sha256=PUlCYKrhcW_5ornWCughL92z_GYx0r1bSbCj4usZfWo,12224
 lightning_sdk/lightning_cloud/openapi/models/v1_check_external_service_status_response.py,sha256=k1cRcRhrmqavxVtAX7w1ZCU4AtkngJJin65JjBhEHHk,4026
 lightning_sdk/lightning_cloud/openapi/models/v1_check_snowflake_connection_response.py,sha256=ZwjPdhw9O8bja24on7zgUGOltKKzp6hP3Ym0j_aJ_yU,3887
 lightning_sdk/lightning_cloud/openapi/models/v1_checkbox.py,sha256=YcD2bdGu_5_9X3pqoSskvOzaVx1_xn8Zkg46vGQV56g,5110
@@ -270,9 +270,9 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_publication_type.py,
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_seed_file.py,sha256=cKaZ-EgDOrh9SS-6RQwMxPBfrwjHlAW4BBOKdpjSREw,4354
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py,sha256=rIn7s4wJs7m_lsuTkJfHsnYsAk9CyT_ccaMkUWW_Lnw,4909
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_state.py,sha256=yLpcNIGMVO5LGk10PenG4uGNxee34tecmSkP7dw_JfY,3239
-lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py,sha256=IEP20Wlmd3UcAqdtJEf0hY47z7cAbqXC7G6ndeZsabw,18433
+lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py,sha256=ZQSAgV4qEeByenlyFqX_kZBy51_yVPVqNGhrCUvZ3_g,20082
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version_publication.py,sha256=n1oHRq_UjYSP_kY2lmMjtqhHUu0H-wHQ5XiDJUEvB6s,6386
-lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py,sha256=rbjyuy6IuDSujcxZ76fZhUkY32lIH3zDE2olLZLi_V4,33537
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py,sha256=s3yLHyU3h0okcUqs2NwJt-7Kj8Bb2HYtzZ6IE02z6n4,34255
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_availability.py,sha256=WrS0UpcTqqCu79ny5yOBJ16RNs5B4AQSXbniWE7iez0,7103
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py,sha256=22jd_N2SYqN_ZZlBltNGdS6jp9zfkIC33qN_IQHY2yg,10949
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py,sha256=GyhAXDpVtfXfaUybBE0aRRvJ6d3DfD7VV3xH_fgQOlE,4859
@@ -386,7 +386,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_studio_job_response.py,sh
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_user_slurm_job_response.py,sha256=kmy5RJKy63bRzBcWdMk00-vT3m2lVs05cic7jfU1JyQ,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_cache_state.py,sha256=VL2MY_XvHWAd8toG-myqZlhuoEwjAIX1cemNSXxqk-k,3210
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_file_info.py,sha256=Fyh7eRcxFJ8Y4LHA3uJe_D5F9RhOxT4ChllHph_ryfQ,4571
-lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=gcZLLsjCX21_IanX_pmHC2TkDhFoH8feGKAd9sXhbwo,14646
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=qnxuWlrNFvtVrB7cQ2Yi_OV9sXqJCX8aK6VIyyptPJ0,15426
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event.py,sha256=XDpq8DM8sN2UKCIlBWGEk-vzjGQGNoc0KjyRcCzDqZ4,9994
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event_type.py,sha256=uuogrGRymVVSz-OcYoJ7eZyFj3POXY22EAQJkXpTVm4,3197
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py,sha256=xHhh0-aPxmyyH3g1Ks2h2oFnEy-CtZjDWH68jjetPEQ,3611
@@ -475,10 +475,10 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_snowflake_query_response.py,
 lightning_sdk/lightning_cloud/openapi/models/v1_get_usage_details_response.py,sha256=FBuN_OgpDuk0uvF4hre_1RbfOkH8Nefv7FRwudif6mQ,3853
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_balance_response.py,sha256=c38qYUoYTtR5droDQ7oaesHXhFPR0S0lUE26xmO0YP4,6226
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_notification_preferences_response.py,sha256=9LzkmmAYYCcMHZX2a55wiwE87QitC_-MiKeyX3uJeDQ,4402
-lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=_hYYNjkKYWXIawdh8nOzwUocl5xDxkXPqZTM48NBUsM,28800
+lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=3jq4PJGOLFzlf_VGdWocmiL48XSmSkzlyRVQKUWiOvE,29580
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py,sha256=igCbjaeZXEbiPAa9cUjwG4cPUivt-7MiiWYikELHwsw,9983
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py,sha256=0k1scTY1IAeKr3p0r7eV_xWRE7Nx54doTGygiQQPkbM,6879
-lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=kZAFa1KG2H323-IgRNa0aYBjkOh0Cej4GXdn4m9mW4E,10906
+lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=f2pe3MlcdouyKH6lDbBO4AKUh-gHcxv90VqMEpN865o,11688
 lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py,sha256=Md6rMiUCMY4oDVE2IYaALD_I0o2PQIKF-9628IGmTyY,3791
 lightning_sdk/lightning_cloud/openapi/models/v1_gpu_system_metrics.py,sha256=io_MkTCTYi6qmYw8KqFZf2zHx8jgBsPKsEwqjr2BG0U,7722
 lightning_sdk/lightning_cloud/openapi/models/v1_health_check_exec.py,sha256=CGhU0XZD-Pbog5aH0rxGzCDwaIvBchRhQPlRTgz8udg,3671
@@ -717,7 +717,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py,sha256=6sTWksnYtWb
 lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py,sha256=fCvSuH3PV0F1Bvy2ySQomwcML3tFKFYsvwP0oFk-2wk,3610
 lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1_job_status.py,sha256=2hr16NFxKl3FVlfThHhG3nsRibUfcZmziS-St1c9Qng,7461
 lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1_status.py,sha256=Is3hBQyd2mW899qZj5jUJaktT58iwZxF5maG_b9PeOM,4368
-lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py,sha256=q1PvOXFXTkEqBdBO9aCkZzCjGorEyhwa5qmFjNNzDCE,8531
+lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py,sha256=NnSOlOXsL7BqtDsTgpny9Jt2zkHIEZNCcTUBfkDNWh0,9389
 lightning_sdk/lightning_cloud/openapi/models/v1_source_type.py,sha256=vfr8Y771DVzERWAp2BaPDid4aH-P7Gx-ak0sWRb-pMk,3073
 lightning_sdk/lightning_cloud/openapi/models/v1_ssh_key_pair.py,sha256=efa6VNtM3UaSdm2yhpcax9Cxk1JToGSe4e7JLrC23Zg,5582
 lightning_sdk/lightning_cloud/openapi/models/v1_ssh_public_key.py,sha256=fRXbHIpFYcM5Bif1o0emSBNY6bsnGUbiiW97yaIOdW8,7876
@@ -810,9 +810,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=gU3MSko9Y7rE4jcnVwstNBaW83OFnSgvM-N44Ibrc_A,5148
-lightning_sdk-0.1.34.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.34.dist-info/METADATA,sha256=y8KZbcLuafKkE6InZWOws9-jJnrSdonywvPdy8qtuts,3941
-lightning_sdk-0.1.34.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.34.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.34.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.34.dist-info/RECORD,,
+lightning_sdk-0.1.35.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.35.dist-info/METADATA,sha256=UzHOowX1I7-XsXw_ir--s-ELJJ7nV2OnEkauvEWGDbY,3941
+lightning_sdk-0.1.35.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.35.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.35.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.35.dist-info/RECORD,,