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

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.34"
+__version__ = "0.1.36"
 _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)
@@ -73,22 +127,26 @@ class AIHub:
         name: Optional[str] = None,
         teamspace: Optional[Union[str, "Teamspace"]] = None,
         org: Optional[Union[str, "Organization"]] = None,
-        **kwargs: Dict[str, Any],
+        api_arguments: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Union[str, bool]]:
         """Deploy an API from the AI Hub.
 
         Example:
             from lightning_sdk import AIHub
-            ai_hub = AIHub()
-            deployment = ai_hub.deploy("temp_01jc37n6qpqkdptjpyep0z06hy", batch_size=10)
+            hub = AIHub()
+            deployment = hub.deploy("temp_01jc37n6qpqkdptjpyep0z06hy")
+
+            # Using API arguments
+            api_arugments = {"batch_size" 10, "batch_timeout": 0.001, "env_token": "lit_xxxx"}
+            deployment = hub.deploy("temp_01jc37n6qpqkdptjpyep0z06hy", api_arugments=api_arugments)
 
         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.
-            **kwargs: Additional keyword arguments for deployment.
+            api_arguments: Additional API argument, such as model name, or batch size.
 
         Returns:
             A dictionary containing the name of the deployed API,
@@ -101,8 +159,9 @@ class AIHub:
         teamspace = self._authenticate(teamspace, org)
         teamspace_id = teamspace.id
 
+        api_arguments = api_arguments or {}
         deployment = self._api.deploy_api(
-            template_id=api_id, cluster_id=cluster, project_id=teamspace_id, name=name, **kwargs
+            template_id=api_id, cluster_id=cluster, project_id=teamspace_id, name=name, api_arguments=api_arguments
         )
         url = quote(f"{LIGHTNING_CLOUD_URL}/{teamspace._org.name}/{teamspace.name}/jobs/{deployment.name}", safe=":/()")
         print("Deployment available at:", url)

lightning_sdk/api/ai_hub_api.py

@@ -1,12 +1,16 @@
-import re
-from typing import List, Optional
+import traceback
+from typing import Dict, List, Optional
 
 import backoff
 
 from lightning_sdk.lightning_cloud.openapi.models import (
     CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
     V1Deployment,
-    V1ParameterizationSpec,
+    V1DeploymentTemplate,
+    V1DeploymentTemplateParameter,
+    V1DeploymentTemplateParameterPlacement,
+    V1DeploymentTemplateParameterType,
+    V1JobSpec,
 )
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import (
     V1DeploymentTemplateGalleryResponse,
@@ -18,6 +22,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}
@@ -26,31 +39,59 @@ class AIHubApi:
         ).templates
 
     @staticmethod
-    def _parse_and_update_args(cmd: str, **kwargs: dict) -> list:
-        """Parse the command and update the arguments with the provided kwargs.
+    def _update_parameters(
+        job: V1JobSpec, placements: List[V1DeploymentTemplateParameterPlacement], pattern: str, value: str
+    ) -> None:
+        for placement in placements:
+            if placement == V1DeploymentTemplateParameterPlacement.COMMAND:
+                job.command = job.command.replace(pattern, str(value))
+            if placement == V1DeploymentTemplateParameterPlacement.ENTRYPOINT:
+                job.entrypoint = job.entrypoint.replace(pattern, str(value))
 
-        >>> _parse_and_update_args("--arg1 1 --arg2=2", arg1=3)
-        ['--arg1 3']
-        """
-        keys = [key.lstrip("-") for key in re.findall(r"--\w+", cmd)]
-        arguments = {}
-        for key in keys:
-            if key in kwargs:
-                arguments[key] = kwargs[key]
-        return [f"--{k} {v}" for k, v in arguments.items()]
+            if placement == V1DeploymentTemplateParameterPlacement.ENV:
+                for e in job.env:
+                    if e.value == pattern:
+                        e.value = str(value)
 
     @staticmethod
-    def _resolve_api_arguments(parameter_spec: "V1ParameterizationSpec", **kwargs: dict) -> str:
-        return " ".join(AIHubApi._parse_and_update_args(parameter_spec.command, **kwargs))
+    def _set_parameters(
+        job: V1JobSpec, parameters: List[V1DeploymentTemplateParameter], api_arguments: Dict[str, str]
+    ) -> V1JobSpec:
+        for p in parameters:
+            if p.name not in api_arguments:
+                if p.type == V1DeploymentTemplateParameterType.INPUT and p.input and p.input.default_value:
+                    api_arguments[p.name] = p.input.default_value
+
+                if p.type == V1DeploymentTemplateParameterType.SELECT and p.select and len(p.select.options) > 0:
+                    api_arguments[p.name] = p.select.options[0]
+
+                if p.type == V1DeploymentTemplateParameterType.CHECKBOX and p.checkbox:
+                    api_arguments[p.name] = (
+                        (p.checkbox.true_value or "True")
+                        if p.checkbox.is_checked
+                        else (p.checkbox.false_value or "False")
+                    )
+
+        for p in parameters:
+            name = p.name
+            pattern = f"${{{name}}}"
+            if name in api_arguments:
+                AIHubApi._update_parameters(job, p.placements, pattern, api_arguments[name])
+            elif not p.required:
+                AIHubApi._update_parameters(job, p.placements, pattern, "")
+            else:
+                raise ValueError(f"API reqires argument '{p.name}' but is not provided with api_arguments.")
+
+        return job
 
     def deploy_api(
-        self, template_id: str, project_id: str, cluster_id: str, name: Optional[str], **kwargs: dict
+        self, template_id: str, project_id: str, cluster_id: str, name: Optional[str], api_arguments: Dict[str, str]
     ) -> V1Deployment:
         template = self._client.deployment_templates_service_get_deployment_template(template_id)
         name = name or template.name
         template.spec_v2.endpoint.id = None
-        command = self._resolve_api_arguments(template.parameter_spec, **kwargs)
-        template.spec_v2.job.command = command
+
+        AIHubApi._set_parameters(template.spec_v2.job, template.parameter_spec.parameters, api_arguments)
         return self._client.jobs_service_create_deployment(
             project_id=project_id,
             body=CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(

lightning_sdk/api/utils.py

@@ -509,6 +509,7 @@ def _download_model_files(
         pbar = tqdm(
             desc=f"Downloading {version}",
             unit="B",
+            total=float(response.size_bytes),
             unit_scale=True,
             unit_divisor=1000,
         )

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/cli/__main__.py

@@ -1,16 +1,9 @@
-import click
 
 
-@click.group()
-def main():
-    pass
-
-
-@main.command()
-def login():
+def login() -> None:
     """Authorize the CLI to access Grid AI resources for a particular user.
-    Use login command to force authenticate,
-    a web browser will open to complete the authentication.
+
+    Use login command to force authenticate, a web browser will open to complete the authentication.
     """
     from lightning_sdk.lightning_cloud.login import Auth  # local to avoid circular import
 
@@ -18,10 +11,19 @@ def login():
     auth.clear()
     auth._run_server()
 
-
-@main.command()
-def logout():
+def logout() -> None:
     """Logout from LightningCloud"""
     from lightning_sdk.lightning_cloud.login import Auth  # local to avoid circular import
 
     Auth.clear()
+
+
+def main() -> None:
+    """CLI entrypoint."""
+    from fire import Fire
+
+    Fire({"login": login, "logout": logout})
+
+
+if __name__ == "__main__":
+    main()

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

@@ -1753,6 +1753,7 @@ class CloudSpaceServiceApi(object):
         :param str project_id: (required)
         :param str cloud_space_id: (required)
         :param str org_id:
+        :param str cloud_space_version_id:
         :return: V1DeleteCloudSpaceVersionPublicationResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1776,12 +1777,13 @@ class CloudSpaceServiceApi(object):
         :param str project_id: (required)
         :param str cloud_space_id: (required)
         :param str org_id:
+        :param str cloud_space_version_id:
         :return: V1DeleteCloudSpaceVersionPublicationResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cloud_space_id', 'org_id']  # noqa: E501
+        all_params = ['project_id', 'cloud_space_id', 'org_id', 'cloud_space_version_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1816,6 +1818,8 @@ class CloudSpaceServiceApi(object):
         query_params = []
         if 'org_id' in params:
             query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'cloud_space_version_id' in params:
+            query_params.append(('cloudSpaceVersionId', params['cloud_space_version_id']))  # noqa: E501
 
         header_params = {}
 
@@ -5610,6 +5614,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 +5639,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 +5682,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/agents_id_body.py

@@ -52,6 +52,7 @@ class AgentsIdBody(object):
         'knowledge': 'str',
         'knowledge_configuration': 'V1KnowledgeConfiguration',
         'model': 'str',
+        'model_provider': 'str',
         'name': 'str',
         'org_id': 'str',
         'prompt_suggestions': 'list[V1PromptSuggestion]',
@@ -75,6 +76,7 @@ class AgentsIdBody(object):
         'knowledge': 'knowledge',
         'knowledge_configuration': 'knowledgeConfiguration',
         'model': 'model',
+        'model_provider': 'modelProvider',
         'name': 'name',
         'org_id': 'orgId',
         'prompt_suggestions': 'promptSuggestions',
@@ -86,7 +88,7 @@ class AgentsIdBody(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, description: 'str' =None, endpoint_id: 'str' =None, expected_cold_start_time: 'str' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, status: 'V1AssistantModelStatus' =None, thumbnail_url: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, description: 'str' =None, endpoint_id: 'str' =None, expected_cold_start_time: 'str' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, model_provider: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, status: 'V1AssistantModelStatus' =None, thumbnail_url: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """AgentsIdBody - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._cluster_id = None
@@ -99,6 +101,7 @@ class AgentsIdBody(object):
         self._knowledge = None
         self._knowledge_configuration = None
         self._model = None
+        self._model_provider = None
         self._name = None
         self._org_id = None
         self._prompt_suggestions = None
@@ -131,6 +134,8 @@ class AgentsIdBody(object):
             self.knowledge_configuration = knowledge_configuration
         if model is not None:
             self.model = model
+        if model_provider is not None:
+            self.model_provider = model_provider
         if name is not None:
             self.name = name
         if org_id is not None:
@@ -381,6 +386,27 @@ class AgentsIdBody(object):
 
         self._model = model
 
+    @property
+    def model_provider(self) -> 'str':
+        """Gets the model_provider of this AgentsIdBody.  # noqa: E501
+
+
+        :return: The model_provider of this AgentsIdBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._model_provider
+
+    @model_provider.setter
+    def model_provider(self, model_provider: 'str'):
+        """Sets the model_provider of this AgentsIdBody.
+
+
+        :param model_provider: The model_provider of this AgentsIdBody.  # noqa: E501
+        :type: str
+        """
+
+        self._model_provider = model_provider
+
     @property
     def name(self) -> 'str':
         """Gets the name of this AgentsIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_body1.py

@@ -42,21 +42,26 @@ class CloudSpaceIdVersionpublicationsBody1(object):
     """
     swagger_types = {
         'cloud_space_version_id': 'str',
+        'featured': 'bool',
         'org_id': 'str'
     }
 
     attribute_map = {
         'cloud_space_version_id': 'cloudSpaceVersionId',
+        'featured': 'featured',
         'org_id': 'orgId'
     }
 
-    def __init__(self, cloud_space_version_id: 'str' =None, org_id: 'str' =None):  # noqa: E501
+    def __init__(self, cloud_space_version_id: 'str' =None, featured: 'bool' =None, org_id: 'str' =None):  # noqa: E501
         """CloudSpaceIdVersionpublicationsBody1 - a model defined in Swagger"""  # noqa: E501
         self._cloud_space_version_id = None
+        self._featured = None
         self._org_id = None
         self.discriminator = None
         if cloud_space_version_id is not None:
             self.cloud_space_version_id = cloud_space_version_id
+        if featured is not None:
+            self.featured = featured
         if org_id is not None:
             self.org_id = org_id
 
@@ -81,6 +86,27 @@ class CloudSpaceIdVersionpublicationsBody1(object):
 
         self._cloud_space_version_id = cloud_space_version_id
 
+    @property
+    def featured(self) -> 'bool':
+        """Gets the featured of this CloudSpaceIdVersionpublicationsBody1.  # noqa: E501
+
+
+        :return: The featured of this CloudSpaceIdVersionpublicationsBody1.  # noqa: E501
+        :rtype: bool
+        """
+        return self._featured
+
+    @featured.setter
+    def featured(self, featured: 'bool'):
+        """Sets the featured of this CloudSpaceIdVersionpublicationsBody1.
+
+
+        :param featured: The featured of this CloudSpaceIdVersionpublicationsBody1.  # noqa: E501
+        :type: bool
+        """
+
+        self._featured = featured
+
     @property
     def org_id(self) -> 'str':
         """Gets the org_id of this CloudSpaceIdVersionpublicationsBody1.  # noqa: E501

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