lightning-sdk 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_mmt.py

@@ -41,6 +41,7 @@ class V1FilesystemMMT(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'artifacts_type': 'V1JobArtifactsType',
         'cloud_space_id': 'str',
         'id': 'str',
         'jobs': 'list[V1FilesystemJob]',
@@ -48,19 +49,23 @@ class V1FilesystemMMT(object):
     }
 
     attribute_map = {
+        'artifacts_type': 'artifactsType',
         'cloud_space_id': 'cloudSpaceId',
         'id': 'id',
         'jobs': 'jobs',
         'name': 'name'
     }
 
-    def __init__(self, cloud_space_id: 'str' =None, id: 'str' =None, jobs: 'list[V1FilesystemJob]' =None, name: 'str' =None):  # noqa: E501
+    def __init__(self, artifacts_type: 'V1JobArtifactsType' =None, cloud_space_id: 'str' =None, id: 'str' =None, jobs: 'list[V1FilesystemJob]' =None, name: 'str' =None):  # noqa: E501
         """V1FilesystemMMT - a model defined in Swagger"""  # noqa: E501
+        self._artifacts_type = None
         self._cloud_space_id = None
         self._id = None
         self._jobs = None
         self._name = None
         self.discriminator = None
+        if artifacts_type is not None:
+            self.artifacts_type = artifacts_type
         if cloud_space_id is not None:
             self.cloud_space_id = cloud_space_id
         if id is not None:
@@ -70,6 +75,27 @@ class V1FilesystemMMT(object):
         if name is not None:
             self.name = name
 
+    @property
+    def artifacts_type(self) -> 'V1JobArtifactsType':
+        """Gets the artifacts_type of this V1FilesystemMMT.  # noqa: E501
+
+
+        :return: The artifacts_type of this V1FilesystemMMT.  # noqa: E501
+        :rtype: V1JobArtifactsType
+        """
+        return self._artifacts_type
+
+    @artifacts_type.setter
+    def artifacts_type(self, artifacts_type: 'V1JobArtifactsType'):
+        """Sets the artifacts_type of this V1FilesystemMMT.
+
+
+        :param artifacts_type: The artifacts_type of this V1FilesystemMMT.  # noqa: E501
+        :type: V1JobArtifactsType
+        """
+
+        self._artifacts_type = artifacts_type
+
     @property
     def cloud_space_id(self) -> 'str':
         """Gets the cloud_space_id of this V1FilesystemMMT.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_artifacts_type.py

@@ -0,0 +1,103 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1JobArtifactsType(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    V1 = "V1"
+    V2 = "V2"
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1JobArtifactsType - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1JobArtifactsType, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1JobArtifactsType') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1JobArtifactsType):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1JobArtifactsType') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_organization.py

@@ -57,6 +57,8 @@ class V1Organization(object):
         'domain': 'str',
         'email': 'str',
         'featured_gallery': 'bool',
+        'full_story_end_date': 'datetime',
+        'full_story_start_date': 'datetime',
         'id': 'str',
         'location': 'str',
         'name': 'str',
@@ -86,6 +88,8 @@ class V1Organization(object):
         'domain': 'domain',
         'email': 'email',
         'featured_gallery': 'featuredGallery',
+        'full_story_end_date': 'fullStoryEndDate',
+        'full_story_start_date': 'fullStoryStartDate',
         'id': 'id',
         'location': 'location',
         'name': 'name',
@@ -98,7 +102,7 @@ class V1Organization(object):
         'updated_at': 'updatedAt'
     }
 
-    def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
         """V1Organization - a model defined in Swagger"""  # noqa: E501
         self._allow_aws_saas = None
         self._allow_gcp_saas = None
@@ -116,6 +120,8 @@ class V1Organization(object):
         self._domain = None
         self._email = None
         self._featured_gallery = None
+        self._full_story_end_date = None
+        self._full_story_start_date = None
         self._id = None
         self._location = None
         self._name = None
@@ -159,6 +165,10 @@ class V1Organization(object):
             self.email = email
         if featured_gallery is not None:
             self.featured_gallery = featured_gallery
+        if full_story_end_date is not None:
+            self.full_story_end_date = full_story_end_date
+        if full_story_start_date is not None:
+            self.full_story_start_date = full_story_start_date
         if id is not None:
             self.id = id
         if location is not None:
@@ -516,6 +526,48 @@ class V1Organization(object):
 
         self._featured_gallery = featured_gallery
 
+    @property
+    def full_story_end_date(self) -> 'datetime':
+        """Gets the full_story_end_date of this V1Organization.  # noqa: E501
+
+
+        :return: The full_story_end_date of this V1Organization.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._full_story_end_date
+
+    @full_story_end_date.setter
+    def full_story_end_date(self, full_story_end_date: 'datetime'):
+        """Sets the full_story_end_date of this V1Organization.
+
+
+        :param full_story_end_date: The full_story_end_date of this V1Organization.  # noqa: E501
+        :type: datetime
+        """
+
+        self._full_story_end_date = full_story_end_date
+
+    @property
+    def full_story_start_date(self) -> 'datetime':
+        """Gets the full_story_start_date of this V1Organization.  # noqa: E501
+
+
+        :return: The full_story_start_date of this V1Organization.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._full_story_start_date
+
+    @full_story_start_date.setter
+    def full_story_start_date(self, full_story_start_date: 'datetime'):
+        """Sets the full_story_start_date of this V1Organization.
+
+
+        :param full_story_start_date: The full_story_start_date of this V1Organization.  # noqa: E501
+        :type: datetime
+        """
+
+        self._full_story_start_date = full_story_start_date
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1Organization.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py

@@ -48,8 +48,9 @@ class V1Pipeline(object):
         'id': 'str',
         'message': 'str',
         'name': 'str',
-        'parent_id': 'str',
+        'parent_pipeline_id': 'str',
         'project_id': 'str',
+        'schedules': 'list[V1PipelineSchedule]',
         'shared_filesystem': 'V1SharedFilesystem',
         'state': 'str',
         'statuses': 'list[V1PipelineStepStatus]',
@@ -66,8 +67,9 @@ class V1Pipeline(object):
         'id': 'id',
         'message': 'message',
         'name': 'name',
-        'parent_id': 'parentId',
+        'parent_pipeline_id': 'parentPipelineId',
         'project_id': 'projectId',
+        'schedules': 'schedules',
         'shared_filesystem': 'sharedFilesystem',
         'state': 'state',
         'statuses': 'statuses',
@@ -76,7 +78,7 @@ class V1Pipeline(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, message: 'str' =None, name: 'str' =None, parent_id: 'str' =None, project_id: 'str' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'str' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, message: 'str' =None, name: 'str' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedules: 'list[V1PipelineSchedule]' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'str' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Pipeline - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._created_at = None
@@ -85,8 +87,9 @@ class V1Pipeline(object):
         self._id = None
         self._message = None
         self._name = None
-        self._parent_id = None
+        self._parent_pipeline_id = None
         self._project_id = None
+        self._schedules = None
         self._shared_filesystem = None
         self._state = None
         self._statuses = None
@@ -108,10 +111,12 @@ class V1Pipeline(object):
             self.message = message
         if name is not None:
             self.name = name
-        if parent_id is not None:
-            self.parent_id = parent_id
+        if parent_pipeline_id is not None:
+            self.parent_pipeline_id = parent_pipeline_id
         if project_id is not None:
             self.project_id = project_id
+        if schedules is not None:
+            self.schedules = schedules
         if shared_filesystem is not None:
             self.shared_filesystem = shared_filesystem
         if state is not None:
@@ -273,25 +278,25 @@ class V1Pipeline(object):
         self._name = name
 
     @property
-    def parent_id(self) -> 'str':
-        """Gets the parent_id of this V1Pipeline.  # noqa: E501
+    def parent_pipeline_id(self) -> 'str':
+        """Gets the parent_pipeline_id of this V1Pipeline.  # noqa: E501
 
 
-        :return: The parent_id of this V1Pipeline.  # noqa: E501
+        :return: The parent_pipeline_id of this V1Pipeline.  # noqa: E501
         :rtype: str
         """
-        return self._parent_id
+        return self._parent_pipeline_id
 
-    @parent_id.setter
-    def parent_id(self, parent_id: 'str'):
-        """Sets the parent_id of this V1Pipeline.
+    @parent_pipeline_id.setter
+    def parent_pipeline_id(self, parent_pipeline_id: 'str'):
+        """Sets the parent_pipeline_id of this V1Pipeline.
 
 
-        :param parent_id: The parent_id of this V1Pipeline.  # noqa: E501
+        :param parent_pipeline_id: The parent_pipeline_id of this V1Pipeline.  # noqa: E501
         :type: str
         """
 
-        self._parent_id = parent_id
+        self._parent_pipeline_id = parent_pipeline_id
 
     @property
     def project_id(self) -> 'str':
@@ -314,6 +319,27 @@ class V1Pipeline(object):
 
         self._project_id = project_id
 
+    @property
+    def schedules(self) -> 'list[V1PipelineSchedule]':
+        """Gets the schedules of this V1Pipeline.  # noqa: E501
+
+
+        :return: The schedules of this V1Pipeline.  # noqa: E501
+        :rtype: list[V1PipelineSchedule]
+        """
+        return self._schedules
+
+    @schedules.setter
+    def schedules(self, schedules: 'list[V1PipelineSchedule]'):
+        """Sets the schedules of this V1Pipeline.
+
+
+        :param schedules: The schedules of this V1Pipeline.  # noqa: E501
+        :type: list[V1PipelineSchedule]
+        """
+
+        self._schedules = schedules
+
     @property
     def shared_filesystem(self) -> 'V1SharedFilesystem':
         """Gets the shared_filesystem of this V1Pipeline.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_schedule.py

@@ -0,0 +1,149 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1PipelineSchedule(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'cron_expression': 'str',
+        'id': 'str'
+    }
+
+    attribute_map = {
+        'cron_expression': 'cronExpression',
+        'id': 'id'
+    }
+
+    def __init__(self, cron_expression: 'str' =None, id: 'str' =None):  # noqa: E501
+        """V1PipelineSchedule - a model defined in Swagger"""  # noqa: E501
+        self._cron_expression = None
+        self._id = None
+        self.discriminator = None
+        if cron_expression is not None:
+            self.cron_expression = cron_expression
+        if id is not None:
+            self.id = id
+
+    @property
+    def cron_expression(self) -> 'str':
+        """Gets the cron_expression of this V1PipelineSchedule.  # noqa: E501
+
+
+        :return: The cron_expression of this V1PipelineSchedule.  # noqa: E501
+        :rtype: str
+        """
+        return self._cron_expression
+
+    @cron_expression.setter
+    def cron_expression(self, cron_expression: 'str'):
+        """Sets the cron_expression of this V1PipelineSchedule.
+
+
+        :param cron_expression: The cron_expression of this V1PipelineSchedule.  # noqa: E501
+        :type: str
+        """
+
+        self._cron_expression = cron_expression
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1PipelineSchedule.  # noqa: E501
+
+
+        :return: The id of this V1PipelineSchedule.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1PipelineSchedule.
+
+
+        :param id: The id of this V1PipelineSchedule.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1PipelineSchedule, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1PipelineSchedule') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1PipelineSchedule):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1PipelineSchedule') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/serve.py

@@ -0,0 +1,134 @@
+import os
+import warnings
+from pathlib import Path
+
+import docker
+from rich.console import Console
+from rich.progress import Progress
+
+from lightning_sdk import Teamspace
+from lightning_sdk.api.lit_container_api import LitContainerApi
+
+
+class _LitServeDeployer:
+    def __init__(self) -> None:
+        self._console = Console()
+        self._client = None
+
+    @property
+    def client(self) -> docker.DockerClient:
+        if self._client is None:
+            try:
+                self._client = docker.from_env()
+                self._client.ping()
+            except docker.errors.DockerException as e:
+                raise RuntimeError(f"Failed to connect to Docker daemon: {e!s}. Is Docker running?") from None
+        return self._client
+
+    def dockerize_api(
+        self, server_filename: str, port: int = 8000, gpu: bool = False, tag: str = "litserve-model"
+    ) -> str:
+        import litserve as ls
+        from litserve import docker_builder
+
+        console = self._console
+        if os.path.exists("Dockerfile"):
+            console.print("Dockerfile already exists. Skipping generation.")
+            return os.path.abspath("Dockerfile")
+
+        requirements = ""
+        if os.path.exists("requirements.txt"):
+            requirements = "-r requirements.txt"
+        else:
+            warnings.warn(
+                f"requirements.txt not found at {os.getcwd()}. "
+                f"Make sure to install the required packages in the Dockerfile.",
+                UserWarning,
+            )
+        current_dir = Path.cwd()
+        if not (current_dir / server_filename).is_file():
+            raise FileNotFoundError(f"Server file `{server_filename}` must be in the current directory: {os.getcwd()}")
+
+        version = ls.__version__
+        if gpu:
+            run_cmd = f"docker run --gpus all -p {port}:{port} {tag}:latest"
+            docker_template = docker_builder.CUDA_DOCKER_TEMPLATE
+        else:
+            run_cmd = f"docker run -p {port}:{port} {tag}:latest"
+            docker_template = docker_builder.DOCKERFILE_TEMPLATE
+        dockerfile_content = docker_template.format(
+            server_filename=server_filename,
+            port=port,
+            version=version,
+            requirements=requirements,
+        )
+        with open("Dockerfile", "w") as f:
+            f.write(dockerfile_content)
+
+        success_msg = f"""[bold]Dockerfile created successfully[/bold]
+Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additional dependencies or commands.
+
+[bold]Build the container with:[/bold]
+> [underline]docker build -t {tag} .[/underline]
+
+[bold]To run the Docker container on the machine:[/bold]
+> [underline]{run_cmd}[/underline]
+
+[bold]To push the container to a registry:[/bold]
+> [underline]docker push {tag}[/underline]
+"""
+        console.print(success_msg)
+        return os.path.abspath("Dockerfile")
+
+    def generate_client(self) -> None:
+        console = self._console
+        try:
+            from litserve.python_client import client_template
+        except ImportError:
+            raise ImportError(
+                "litserve is not installed. Please install it with `pip install lightning_sdk[serve]`"
+            ) from None
+
+        client_path = Path("client.py")
+        if client_path.exists():
+            console.print("Skipping client generation: client.py already exists", style="blue")
+        else:
+            try:
+                client_path.write_text(client_template)
+                console.print("✅ Client generated at client.py", style="bold green")
+            except OSError as e:
+                raise OSError(f"Failed to generate client.py: {e!s}") from None
+
+    def _build_container(self, path: str, repository: str, tag: str, console: Console, progress: Progress) -> None:
+        build_task = progress.add_task("Building Docker image", total=None)
+        build_status = self.client.api.build(
+            path=os.path.dirname(path), dockerfile=path, tag=f"{repository}:{tag}", decode=True, quiet=False
+        )
+        for line in build_status:
+            if "error" in line:
+                progress.stop()
+                console.print(f"\n[red]{line}[/red]")
+                return
+            if "stream" in line and line["stream"].strip():
+                console.print(line["stream"].strip(), style="bright_black")
+                progress.update(build_task, description="Building Docker image")
+
+        progress.update(build_task, description="[green]Build completed![/green]")
+
+    def _push_container(
+        self, repository: str, tag: str, teamspace: Teamspace, lit_cr: LitContainerApi, progress: Progress
+    ) -> None:
+        console = self._console
+        push_task = progress.add_task("Pushing to registry", total=None)
+        console.print("\nPushing image...", style="bold blue")
+        lit_cr.authenticate()
+        push_status = lit_cr.upload_container(repository, teamspace, tag=tag)
+        for line in push_status:
+            if "error" in line:
+                progress.stop()
+                console.print(f"\n[red]{line}[/red]")
+                return
+            if "status" in line:
+                console.print(line["status"], style="bright_black")
+                progress.update(push_task, description="Pushing to registry")
+        progress.update(push_task, description="[green]Push completed![/green]")

{lightning_sdk-0.2.0.dist-info → lightning_sdk-0.2.1.dist-info}/METADATA

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