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

lightning_sdk/__init__.py

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

lightning_sdk/ai_hub.py

@@ -127,14 +127,18 @@ 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.
@@ -142,7 +146,7 @@ class AIHub:
             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,
@@ -155,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,6 +1,5 @@
-import re
 import traceback
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 import backoff
 
@@ -8,7 +7,10 @@ from lightning_sdk.lightning_cloud.openapi.models import (
     CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
     V1Deployment,
     V1DeploymentTemplate,
-    V1ParameterizationSpec,
+    V1DeploymentTemplateParameter,
+    V1DeploymentTemplateParameterPlacement,
+    V1DeploymentTemplateParameterType,
+    V1JobSpec,
 )
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import (
     V1DeploymentTemplateGalleryResponse,
@@ -37,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/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/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 = {}
 

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/externalv1_cloud_space_instance_status.py

@@ -54,6 +54,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
         'instance_url': 'str',
         'jupyterlab_url': 'str',
         'phase': 'V1CloudSpaceInstanceState',
+        'price': 'float',
         'ssh_host': 'str',
         'ssh_port': 'int',
         'ssh_username': 'str',
@@ -84,6 +85,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
         'instance_url': 'instanceUrl',
         'jupyterlab_url': 'jupyterlabUrl',
         'phase': 'phase',
+        'price': 'price',
         'ssh_host': 'sshHost',
         'ssh_port': 'sshPort',
         'ssh_username': 'sshUsername',
@@ -100,7 +102,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
         'vscode_url': 'vscodeUrl'
     }
 
-    def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, vscode_url: 'str' =None):  # noqa: E501
+    def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, vscode_url: 'str' =None):  # noqa: E501
         """Externalv1CloudSpaceInstanceStatus - a model defined in Swagger"""  # noqa: E501
         self._app_url = None
         self._cloud_space_id = None
@@ -115,6 +117,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
         self._instance_url = None
         self._jupyterlab_url = None
         self._phase = None
+        self._price = None
         self._ssh_host = None
         self._ssh_port = None
         self._ssh_username = None
@@ -156,6 +159,8 @@ class Externalv1CloudSpaceInstanceStatus(object):
             self.jupyterlab_url = jupyterlab_url
         if phase is not None:
             self.phase = phase
+        if price is not None:
+            self.price = price
         if ssh_host is not None:
             self.ssh_host = ssh_host
         if ssh_port is not None:
@@ -458,6 +463,27 @@ class Externalv1CloudSpaceInstanceStatus(object):
 
         self._phase = phase
 
+    @property
+    def price(self) -> 'float':
+        """Gets the price of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+
+
+        :return: The price of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+        :rtype: float
+        """
+        return self._price
+
+    @price.setter
+    def price(self, price: 'float'):
+        """Sets the price of this Externalv1CloudSpaceInstanceStatus.
+
+
+        :param price: The price of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501
+        :type: float
+        """
+
+        self._price = price
+
     @property
     def ssh_host(self) -> 'str':
         """Gets the ssh_host of this Externalv1CloudSpaceInstanceStatus.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py

@@ -53,6 +53,7 @@ class V1Assistant(object):
         'knowledge': 'str',
         'knowledge_configuration': 'V1KnowledgeConfiguration',
         'model': 'str',
+        'model_provider': 'str',
         'name': 'str',
         'org_id': 'str',
         'project_id': 'str',
@@ -78,6 +79,7 @@ class V1Assistant(object):
         'knowledge': 'knowledge',
         'knowledge_configuration': 'knowledgeConfiguration',
         'model': 'model',
+        'model_provider': 'modelProvider',
         'name': 'name',
         'org_id': 'orgId',
         'project_id': 'projectId',
@@ -90,7 +92,7 @@ class V1Assistant(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, id: 'str' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, name: 'str' =None, org_id: 'str' =None, project_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, id: 'str' =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, project_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
         """V1Assistant - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._cluster_id = None
@@ -104,6 +106,7 @@ class V1Assistant(object):
         self._knowledge = None
         self._knowledge_configuration = None
         self._model = None
+        self._model_provider = None
         self._name = None
         self._org_id = None
         self._project_id = None
@@ -139,6 +142,8 @@ class V1Assistant(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:
@@ -412,6 +417,27 @@ class V1Assistant(object):
 
         self._model = model
 
+    @property
+    def model_provider(self) -> 'str':
+        """Gets the model_provider of this V1Assistant.  # noqa: E501
+
+
+        :return: The model_provider of this V1Assistant.  # noqa: E501
+        :rtype: str
+        """
+        return self._model_provider
+
+    @model_provider.setter
+    def model_provider(self, model_provider: 'str'):
+        """Sets the model_provider of this V1Assistant.
+
+
+        :param model_provider: The model_provider of this V1Assistant.  # noqa: E501
+        :type: str
+        """
+
+        self._model_provider = model_provider
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1Assistant.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_release.py

@@ -43,6 +43,7 @@ class V1DeploymentRelease(object):
     swagger_types = {
         'archived': 'bool',
         'created_at': 'datetime',
+        'first_job_state': 'str',
         'id': 'str',
         'project_id': 'str',
         'replicas': 'int',
@@ -54,6 +55,7 @@ class V1DeploymentRelease(object):
     attribute_map = {
         'archived': 'archived',
         'created_at': 'createdAt',
+        'first_job_state': 'firstJobState',
         'id': 'id',
         'project_id': 'projectId',
         'replicas': 'replicas',
@@ -62,10 +64,11 @@ class V1DeploymentRelease(object):
         'version': 'version'
     }
 
-    def __init__(self, archived: 'bool' =None, created_at: 'datetime' =None, id: 'str' =None, project_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, user_id: 'str' =None, version: 'int' =None):  # noqa: E501
+    def __init__(self, archived: 'bool' =None, created_at: 'datetime' =None, first_job_state: 'str' =None, id: 'str' =None, project_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, user_id: 'str' =None, version: 'int' =None):  # noqa: E501
         """V1DeploymentRelease - a model defined in Swagger"""  # noqa: E501
         self._archived = None
         self._created_at = None
+        self._first_job_state = None
         self._id = None
         self._project_id = None
         self._replicas = None
@@ -77,6 +80,8 @@ class V1DeploymentRelease(object):
             self.archived = archived
         if created_at is not None:
             self.created_at = created_at
+        if first_job_state is not None:
+            self.first_job_state = first_job_state
         if id is not None:
             self.id = id
         if project_id is not None:
@@ -132,6 +137,27 @@ class V1DeploymentRelease(object):
 
         self._created_at = created_at
 
+    @property
+    def first_job_state(self) -> 'str':
+        """Gets the first_job_state of this V1DeploymentRelease.  # noqa: E501
+
+
+        :return: The first_job_state of this V1DeploymentRelease.  # noqa: E501
+        :rtype: str
+        """
+        return self._first_job_state
+
+    @first_job_state.setter
+    def first_job_state(self, first_job_state: 'str'):
+        """Sets the first_job_state of this V1DeploymentRelease.
+
+
+        :param first_job_state: The first_job_state of this V1DeploymentRelease.  # noqa: E501
+        :type: str
+        """
+
+        self._first_job_state = first_job_state
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1DeploymentRelease.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py

@@ -40,6 +40,7 @@ class V1DeploymentState(object):
     UNSPECIFIED = "DEPLOYMENT_STATE_UNSPECIFIED"
     RUNNING = "DEPLOYMENT_STATE_RUNNING"
     DELETED = "DEPLOYMENT_STATE_DELETED"
+    FAILED = "DEPLOYMENT_STATE_FAILED"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py

@@ -43,6 +43,7 @@ class V1DeploymentStatus(object):
     swagger_types = {
         'deleting_replicas': 'str',
         'failing_replicas': 'str',
+        'first_job_state_current_release': 'str',
         'message': 'str',
         'pending_replicas': 'str',
         'ready_replicas': 'str',
@@ -52,16 +53,18 @@ class V1DeploymentStatus(object):
     attribute_map = {
         'deleting_replicas': 'deletingReplicas',
         'failing_replicas': 'failingReplicas',
+        'first_job_state_current_release': 'firstJobStateCurrentRelease',
         'message': 'message',
         'pending_replicas': 'pendingReplicas',
         'ready_replicas': 'readyReplicas',
         'urls': 'urls'
     }
 
-    def __init__(self, deleting_replicas: 'str' =None, failing_replicas: 'str' =None, message: 'str' =None, pending_replicas: 'str' =None, ready_replicas: 'str' =None, urls: 'list[str]' =None):  # noqa: E501
+    def __init__(self, deleting_replicas: 'str' =None, failing_replicas: 'str' =None, first_job_state_current_release: 'str' =None, message: 'str' =None, pending_replicas: 'str' =None, ready_replicas: 'str' =None, urls: 'list[str]' =None):  # noqa: E501
         """V1DeploymentStatus - a model defined in Swagger"""  # noqa: E501
         self._deleting_replicas = None
         self._failing_replicas = None
+        self._first_job_state_current_release = None
         self._message = None
         self._pending_replicas = None
         self._ready_replicas = None
@@ -71,6 +74,8 @@ class V1DeploymentStatus(object):
             self.deleting_replicas = deleting_replicas
         if failing_replicas is not None:
             self.failing_replicas = failing_replicas
+        if first_job_state_current_release is not None:
+            self.first_job_state_current_release = first_job_state_current_release
         if message is not None:
             self.message = message
         if pending_replicas is not None:
@@ -122,6 +127,27 @@ class V1DeploymentStatus(object):
 
         self._failing_replicas = failing_replicas
 
+    @property
+    def first_job_state_current_release(self) -> 'str':
+        """Gets the first_job_state_current_release of this V1DeploymentStatus.  # noqa: E501
+
+
+        :return: The first_job_state_current_release of this V1DeploymentStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._first_job_state_current_release
+
+    @first_job_state_current_release.setter
+    def first_job_state_current_release(self, first_job_state_current_release: 'str'):
+        """Sets the first_job_state_current_release of this V1DeploymentStatus.
+
+
+        :param first_job_state_current_release: The first_job_state_current_release of this V1DeploymentStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._first_job_state_current_release = first_job_state_current_release
+
     @property
     def message(self) -> 'str':
         """Gets the message of this V1DeploymentStatus.  # noqa: E501