lightning-sdk 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/ai_hub.py

@@ -3,6 +3,7 @@ from urllib.parse import quote
 
 from lightning_sdk.api import AIHubApi
 from lightning_sdk.api.utils import _get_cloud_url
+from lightning_sdk.lightning_cloud.openapi.models import V1Deployment
 from lightning_sdk.user import User
 from lightning_sdk.utils.resolve import _resolve_teamspace
 
@@ -109,7 +110,7 @@ class AIHub:
         org: Optional[Union[str, "Organization"]] = None,
         user: Optional[Union[str, "User"]] = None,
         machine: Optional[Union[str, "Machine"]] = None,
-    ) -> Dict[str, Union[str, bool]]:
+    ) -> V1Deployment:
         """Deploy an API from the AI Hub.
 
         Example:
@@ -133,7 +134,7 @@ class AIHub:
             machine: The machine to run the deployment on. Defaults to the first option set in the AI Hub template.
 
         Returns:
-            A dictionary containing the name of the deployed API,
+            A V1Deployment object containing the name of the deployed API,
             the URL to access it, and whether it is interruptible.
 
         Raises:
@@ -169,16 +170,9 @@ class AIHub:
 
         print("Deployment available at:", url)
 
-        return {
-            "id": deployment.id,
-            "name": deployment.name,
-            "deployment_url": url,
-            "api_endpoint": deployment.status.urls[0],
-            "interruptible": deployment.spec.spot,
-            "teamspace id": teamspace_id,
-        }
+        return deployment
 
-    def delete_deployment(self, deployment: Dict[str, Union[str, bool]]) -> None:
+    def delete_deployment(self, deployment: V1Deployment) -> None:
         """Delete a deployment from the AI Hub.
 
         Example:
@@ -188,9 +182,6 @@ class AIHub:
             hub.delete_deployment(deployment)
 
         Args:
-            deployment: The deployment dictionary returned by the run method.
+            deployment: The deployment object returned by the run method.
         """
-        if "teamspace id" not in deployment or "id" not in deployment:
-            raise ValueError("Deployment dictionary must contain 'teamspace id' and 'id' keys.")
-
-        self._api.delete_api(deployment["id"], deployment["teamspace id"])
+        self._api.delete_api(deployment.id, deployment.project_id)

lightning_sdk/api/ai_hub_api.py

@@ -95,15 +95,17 @@ class AIHubApi:
 
                 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")
+                        (p.checkbox.true_value or "") if p.checkbox.is_checked else (p.checkbox.false_value or "")
                     )
 
         for p in parameters:
             name = p.name
             pattern = f"${{{name}}}"
             if name in api_arguments:
+                if p.type == V1DeploymentTemplateParameterType.CHECKBOX and p.checkbox:
+                    api_arguments[p.name] = (
+                        (p.checkbox.true_value or "") if api_arguments[name] is True else (p.checkbox.false_value or "")
+                    )
                 AIHubApi._update_parameters(job, p.placements, pattern, api_arguments[name])
             elif not p.required:
                 AIHubApi._update_parameters(job, p.placements, pattern, "")

lightning_sdk/cli/serve.py

@@ -4,17 +4,19 @@ from pathlib import Path
 from typing import Optional, Union
 
 import click
-import docker
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm
 
+from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.deployment_api import DeploymentApi
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.serve import _LitServeDeployer
 
+_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+
 
 @click.group("serve")
 def serve() -> None:
@@ -24,11 +26,10 @@ def serve() -> None:
         lightning serve api server.py  # serve locally
 
     Example:
-        lightning serve api server.py --cloud  # deploy to the cloud
+        lightning serve api server.py --cloud --name litserve-api  # deploy to the cloud
 
     You can deploy the API to the cloud by running `lightning serve api server.py --cloud`.
-    This will generate a Dockerfile, build the image, and push it to the image registry.
-    Deploying to the cloud requires pre-login to the docker registry.
+    This will build a docker container for the server.py script and deploy it to the Lightning AI platform.
     """
 
 
@@ -48,7 +49,6 @@ def serve() -> None:
     flag_value=True,
     help="Deploy the model to the Lightning AI platform",
 )
-@click.option("--gpu", is_flag=True, default=False, flag_value=True, help="Use GPU for serving")
 @click.option("--name", default=None, help="Name of the deployed API (e.g., 'classification-api', 'Llama-api')")
 @click.option(
     "--non-interactive",
@@ -58,17 +58,88 @@ def serve() -> None:
     flag_value=True,
     help="Do not prompt for confirmation",
 )
+@click.option(
+    "--machine",
+    default="CPU",
+    show_default=True,
+    type=click.Choice(_MACHINE_VALUES),
+    help="The machine type to deploy the API on.",
+)
+@click.option(
+    "--interruptible",
+    is_flag=True,
+    default=False,
+    flag_value=True,
+    help="Whether the machine should be interruptible (spot) or not.",
+)
+@click.option(
+    "--teamspace",
+    default=None,
+    help="The teamspace the deployment should be associated with. Defaults to the current teamspace.",
+)
+@click.option(
+    "--org",
+    default=None,
+    help="The organization owning the teamspace (if any). Defaults to the current organization.",
+)
+@click.option("--user", default=None, help="The user owning the teamspace (if any). Defaults to the current user.")
+@click.option(
+    "--cloud-account",
+    "--cloud_account",
+    default=None,
+    help=(
+        "The cloud account to run the deployment on. "
+        "Defaults to the studio cloud account if running with studio compute env. "
+        "If not provided will fall back to the teamspaces default cloud account."
+    ),
+)
+@click.option("--port", default=8000, help="The port to expose the API on.")
+@click.option("--min_replica", "--min-replica", default=0, help="Number of replicas to start with.")
+@click.option("--max_replica", "--max-replica", default=1, help="Number of replicas to scale up to.")
+@click.option("--replicas", "--replicas", default=1, help="Deployment will start with this many replicas.")
+@click.option(
+    "--no_credentials",
+    "--no-credentials",
+    is_flag=True,
+    default=False,
+    flag_value=True,
+    help="Whether to include credentials in the deployment.",
+)
 def api(
     script_path: str,
     easy: bool,
     cloud: bool,
-    gpu: bool,
     name: str,
     non_interactive: bool,
+    machine: str,
+    interruptible: bool,
+    teamspace: Optional[str],
+    org: Optional[str],
+    user: Optional[str],
+    cloud_account: Optional[str],
+    port: Optional[int],
+    min_replica: Optional[int],
+    max_replica: Optional[int],
+    replicas: Optional[int],
+    no_credentials: Optional[bool],
 ) -> None:
     """Deploy a LitServe model script."""
     return api_impl(
-        script_path=script_path, easy=easy, cloud=cloud, gpu=gpu, repository=name, non_interactive=non_interactive
+        script_path=script_path,
+        easy=easy,
+        cloud=cloud,
+        repository=name,
+        non_interactive=non_interactive,
+        machine=machine,
+        interruptible=interruptible,
+        teamspace=teamspace,
+        org=org,
+        user=user,
+        cloud_account=cloud_account,
+        port=port,
+        min_replica=min_replica,
+        max_replica=max_replica,
+        include_credentials=not no_credentials,
     )
 
 
@@ -76,10 +147,20 @@ def api_impl(
     script_path: Union[str, Path],
     easy: bool = False,
     cloud: bool = False,
-    gpu: bool = False,
     repository: [str] = None,
     tag: Optional[str] = None,
     non_interactive: bool = False,
+    machine: str = "CPU",
+    interruptible: bool = False,
+    teamspace: Optional[str] = None,
+    org: Optional[str] = None,
+    user: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+    port: Optional[int] = 8000,
+    min_replica: Optional[int] = 0,
+    max_replica: Optional[int] = 1,
+    replicas: Optional[int] = 1,
+    include_credentials: Optional[bool] = True,
 ) -> None:
     """Deploy a LitServe model script."""
     console = Console()
@@ -93,8 +174,24 @@ def api_impl(
 
     if cloud:
         repository = repository or "litserve-model"
+        machine = Machine.from_str(machine)
         return _handle_cloud(
-            script_path, console, gpu=gpu, repository=repository, tag=tag, non_interactive=non_interactive
+            script_path,
+            console,
+            repository=repository,
+            tag=tag,
+            non_interactive=non_interactive,
+            machine=machine,
+            interruptible=interruptible,
+            teamspace=teamspace,
+            org=org,
+            user=user,
+            cloud_account=cloud_account,
+            port=port,
+            min_replica=min_replica,
+            max_replica=max_replica,
+            replicas=replicas,
+            include_credentials=include_credentials,
         )
 
     try:
@@ -111,22 +208,30 @@ def api_impl(
 def _handle_cloud(
     script_path: Union[str, Path],
     console: Console,
-    gpu: bool,
     repository: str = "litserve-model",
     tag: Optional[str] = None,
-    teamspace: Optional[str] = None,
     non_interactive: bool = False,
+    machine: Machine = "CPU",
+    interruptible: bool = False,
+    teamspace: Optional[str] = None,
+    org: Optional[str] = None,
+    user: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+    port: Optional[int] = 8000,
+    min_replica: Optional[int] = 0,
+    max_replica: Optional[int] = 1,
+    replicas: Optional[int] = 1,
+    include_credentials: Optional[bool] = True,
 ) -> None:
-    menu = _TeamspacesMenu()
-    teamspace = menu._resolve_teamspace(teamspace)
-    try:
-        client = docker.from_env()
-        client.ping()
-    except docker.errors.DockerException as e:
-        raise StudioCliError(f"Failed to connect to Docker daemon: {e!s}. Is Docker running?") from None
+    if teamspace is None:
+        menu = _TeamspacesMenu()
+        resolved_teamspace = menu._resolve_teamspace(teamspace)
+    else:
+        resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
+    port = port or 8000
     ls_deployer = _LitServeDeployer()
-    path = ls_deployer.dockerize_api(script_path, port=8000, gpu=gpu, tag=tag)
+    path = ls_deployer.dockerize_api(script_path, port=port, gpu=not machine.is_cpu(), tag=tag, print_success=False)
     console.clear()
     if non_interactive:
         console.print("[italic]non-interactive[/italic] mode enabled, skipping confirmation prompts", style="blue")
@@ -142,7 +247,8 @@ def _handle_cloud(
     lit_cr = LitContainerApi()
     deployment_name = os.path.basename(repository)
 
-    if DeploymentApi().get_deployment_by_name(deployment_name, teamspace.id):
+    ls_deployer.authenticate()
+    if DeploymentApi().get_deployment_by_name(deployment_name, resolved_teamspace.id):
         raise StudioCliError(f"Deployment {deployment_name} already exists. Please choose a different name.") from None
 
     with Progress(
@@ -154,7 +260,7 @@ def _handle_cloud(
     ) as progress:
         try:
             ls_deployer.build_container(path, repository, tag, console, progress)
-            push_status = ls_deployer.push_container(repository, tag, teamspace, lit_cr, progress)
+            push_status = ls_deployer.push_container(repository, tag, resolved_teamspace, lit_cr, progress)
         except Exception as e:
             console.print(f"❌ Deployment failed: {e}", style="red")
             return
@@ -163,6 +269,17 @@ def _handle_cloud(
     repository = push_status.get("repository")
 
     deployment_status = ls_deployer._run_on_cloud(
-        deployment_name=deployment_name, image=repository, teamspace=teamspace, ports=[8000], gpu=gpu, metric=None
+        deployment_name=deployment_name,
+        image=repository,
+        teamspace=resolved_teamspace,
+        metric=None,
+        machine=machine,
+        spot=interruptible,
+        cloud_account=cloud_account,
+        port=port,
+        min_replica=min_replica,
+        max_replica=max_replica,
+        replicas=replicas,
+        include_credentials=include_credentials,
     )
     console.print(f"🚀 Deployment started, access at [i]{deployment_status.get('url')}[/i]")

lightning_sdk/lit_container.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import Dict, List, Optional
 
 from lightning_sdk.api.lit_container_api import LitContainerApi
@@ -29,11 +30,17 @@ class LitContainer:
         repositories = self._api.list_containers(project_id)
         table = []
         for repo in repositories:
+            created_date = repo.creation_time
+            if isinstance(repo.creation_time, str):
+                created_date = datetime.fromisoformat(created_date)
+
+            created = created_date.strftime("%Y-%m-%d %H:%M:%S")
+
             table.append(
                 {
                     "REPOSITORY": repo.name,
                     "IMAGE ID": repo.id,
-                    "CREATED": repo.creation_time.strftime("%Y-%m-%d %H:%M:%S"),
+                    "CREATED": created,
                 }
             )
         return table

lightning_sdk/mmt/mmt.py

@@ -1,7 +1,5 @@
-from functools import lru_cache
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
 
-from lightning_sdk.api.user_api import UserApi
 from lightning_sdk.mmt.base import MMTMachine, _BaseMMT
 from lightning_sdk.mmt.v1 import _MMTV1
 from lightning_sdk.mmt.v2 import _MMTV2
@@ -18,21 +16,6 @@ if TYPE_CHECKING:
 _logger = _setup_logger(__name__)
 
 
-@lru_cache(maxsize=None)
-def _has_mmt_v2() -> bool:
-    api = UserApi()
-    try:
-        feature_flags = api._get_feature_flags()
-    except Exception:
-        return False
-
-    try:
-        return feature_flags.mmt_v2
-    except AttributeError:
-        # Feature flag doesn't exist anymore, so return True
-        return True
-
-
 class MMT(_BaseMMT):
     """Class to submit and manage multi-machine jobs on the Lightning AI Platform."""
 
@@ -60,7 +43,7 @@ class MMT(_BaseMMT):
         """
         from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 
-        if _has_mmt_v2() and not self._force_v1:
+        if not self._force_v1:
             # try with v2 and fall back to v1
             try:
                 mmt = _MMTV2(

lightning_sdk/mmt/v1.py

@@ -92,34 +92,7 @@ class _MMTV1(_BaseMMT):
             path_mappings: The mappings from data connection inside your container (not supported)
 
         """
-        if studio is None:
-            raise ValueError("Studio is required for submitting jobs")
-        if image is not None or image_credentials is not None or cloud_account_auth or entrypoint != "sh -c":
-            raise ValueError("Image is not supported for submitting jobs")
-
-        if artifacts_local is not None or artifacts_remote is not None:
-            raise ValueError("Specifying how to persist artifacts is not yet supported with jobs")
-
-        if env is not None:
-            raise ValueError("Environment variables are not supported for submitting jobs")
-        if command is None:
-            raise ValueError("Command is required for submitting multi-machine jobs")
-
-        _submitted = self._job_api.submit_job(
-            name=self._name,
-            num_machines=num_machines,
-            command=command,
-            studio_id=studio._studio.id,
-            teamspace_id=self._teamspace.id,
-            cloud_account=cloud_account or "",
-            machine=machine,
-            interruptible=interruptible,
-            strategy="parallel",
-        )
-
-        self._name = _submitted.name
-        self._job = _submitted
-        return self
+        raise NotImplementedError("Cannot submit new mmts with MMTV1!")
 
     def _update_internal_job(self) -> None:
         try:

lightning_sdk/models.py

@@ -72,6 +72,19 @@ def _get_teamspace(name: str, organization: str) -> "Teamspace":
     raise RuntimeError(f"Teamspace `{requested_teamspace}` not found. Available teamspaces: {os.linesep}\t{options}")
 
 
+def _extend_model_name_with_teamspace(name: str) -> str:
+    """Extend the model name with the teamspace if it can be determined from env. variables."""
+    if "/" in name:
+        return name
+    # do some magic if you run studio
+    teamspace = _resolve_teamspace(None, None, None)
+    if not teamspace:
+        raise ValueError(
+            f"Model name must be in the format `organization/teamspace/model_name` but you provided '{name}'."
+        )
+    return f"{teamspace.owner.name}/{teamspace.name}/{name}"
+
+
 def _parse_model_name_and_version(name: str) -> Tuple[str, str, str, str]:
     """Parse the name argument into its components."""
     try:
@@ -105,9 +118,7 @@ def download_model(
         download_dir: The directory where the Model should be downloaded.
         progress_bar: Whether to show a progress bar when downloading.
     """
-    if "/" not in name:  # do some magic if you run studio
-        teamspace = _resolve_teamspace(None, None, None)
-        name = f"{teamspace.owner.name}/{teamspace.name}/{name}"
+    name = _extend_model_name_with_teamspace(name)
     teamspace_owner_name, teamspace_name, model_name, version = _parse_model_name_and_version(name)
 
     download_dir = Path(download_dir)
@@ -147,9 +158,7 @@ def upload_model(
             If not provided, the default cloud account for the Teamspace will be used.
         progress_bar: Whether to show a progress bar for the upload.
     """
-    if "/" not in name:  # do some magic if you run studio
-        teamspace = _resolve_teamspace(None, None, None)
-        name = f"{teamspace.owner.name}/{teamspace.name}/{name}"
+    name = _extend_model_name_with_teamspace(name)
     org_name, teamspace_name, model_name, _ = _parse_model_name_and_version(name)
     teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
     return teamspace.upload_model(

lightning_sdk/pipeline/pipeline.py

@@ -49,8 +49,8 @@ class Pipeline:
 
         self._teamspace = _resolve_teamspace(
             teamspace=teamspace,
-            org=self._org,
-            user=self._user,
+            org=org,
+            user=user,
         )
 
         self._pipeline_api = PipelineApi()

lightning_sdk/serve.py

@@ -1,9 +1,9 @@
 import os
 import shlex
 import subprocess
-import warnings
 from pathlib import Path
-from typing import Generator, List, Optional
+from typing import Generator, Optional
+from urllib.parse import urlencode
 
 import docker
 from rich.console import Console
@@ -13,6 +13,26 @@ from lightning_sdk import Deployment, Machine, Teamspace
 from lightning_sdk.api.deployment_api import AutoScaleConfig
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_cloud_url
+from lightning_sdk.lightning_cloud import env
+from lightning_sdk.lightning_cloud.login import Auth, AuthServer
+
+_DOCKER_NOT_RUNNING_MSG = (
+    "Deploying LitServe requires Docker to be running on the machine. "
+    "If Docker is not installed, please install it from https://docs.docker.com/get-docker/ "
+    "and start the Docker daemon before running this command."
+)
+
+
+class _AuthServer(AuthServer):
+    def get_auth_url(self, port: int) -> str:
+        redirect_uri = f"http://localhost:{port}/login-complete"
+        params = urlencode({"redirectTo": redirect_uri, "inviteCode": "litserve"})
+        return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
+
+
+class _Auth(Auth):
+    def _run_server(self) -> None:
+        _AuthServer().login_with_browser(self)
 
 
 class _LitServeDeployer:
@@ -20,38 +40,49 @@ class _LitServeDeployer:
         self._console = Console()
         self._client = None
 
+    @staticmethod
+    def authenticate() -> None:
+        auth = _Auth()
+        auth.authenticate()
+
     @property
     def client(self) -> docker.DockerClient:
-        os.environ["DOCKER_BUILDKIT"] = "1"
-
         if self._client is None:
             try:
+                os.environ["DOCKER_BUILDKIT"] = "1"
                 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
+            except docker.errors.DockerException:
+                raise RuntimeError(_DOCKER_NOT_RUNNING_MSG) from None
         return self._client
 
     def dockerize_api(
-        self, server_filename: str, port: int = 8000, gpu: bool = False, tag: str = "litserve-model"
+        self,
+        server_filename: str,
+        port: int = 8000,
+        gpu: bool = False,
+        tag: str = "litserve-model",
+        print_success: bool = True,
     ) -> 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(
+            console.print(
                 f"requirements.txt not found at {os.getcwd()}. "
                 f"Make sure to install the required packages in the Dockerfile.",
-                UserWarning,
+                style="yellow",
+            )
+
+        if os.path.exists("Dockerfile"):
+            console.print(
+                "Dockerfile already exists in the current directory, we will use it for building the container."
             )
+            return os.path.abspath("Dockerfile")
         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()}")
@@ -72,7 +103,8 @@ class _LitServeDeployer:
         with open("Dockerfile", "w") as f:
             f.write(dockerfile_content)
 
-        success_msg = f"""[bold]Dockerfile created successfully[/bold]
+        if print_success:
+            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]
@@ -84,7 +116,7 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
 [bold]To push the container to a registry:[/bold]
 > [underline]docker push {tag}[/underline]
 """
-        console.print(success_msg)
+            console.print(success_msg)
         return os.path.abspath("Dockerfile")
 
     @staticmethod
@@ -187,19 +219,19 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
         deployment_name: str,
         teamspace: Teamspace,
         image: str,
-        ports: List[int],
-        gpu: bool = False,
         metric: Optional[str] = None,
         machine: Optional[Machine] = None,
-        min_replica: Optional[int] = 1,
+        min_replica: Optional[int] = 0,
         max_replica: Optional[int] = 1,
         spot: Optional[bool] = None,
-        replicas: Optional[int] = None,
+        replicas: Optional[int] = 1,
         cloud_account: Optional[str] = None,
+        port: Optional[int] = 8000,
+        include_credentials: Optional[bool] = True,
     ) -> dict:
-        machine = machine or Machine.CPU
-        metric = metric or "GPU" if gpu else "CPU"
         url = f"{_get_cloud_url()}/{teamspace.owner.name}/{teamspace.name}/jobs/{deployment_name}"
+        machine = machine or Machine.CPU
+        metric = metric or ("CPU" if machine.is_cpu() else "GPU")
         deployment = Deployment(deployment_name, teamspace)
         if deployment.is_started:
             raise RuntimeError(
@@ -211,11 +243,12 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
         deployment.start(
             machine=machine,
             image=image,
-            ports=ports,
             autoscale=autoscale,
             spot=spot,
             replicas=replicas,
             cloud_account=cloud_account,
+            ports=[port],
+            include_credentials=include_credentials,
         )
 
         return {"deployment": deployment, "url": url}

{lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.4.dist-info}/METADATA

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

{lightning_sdk-0.2.3.dist-info → lightning_sdk-0.2.4.dist-info}/RECORD

@@ -1,23 +1,23 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=YGP-QC_WV8gn8xxxESb1X6DcaOVnY-lkbM0NHBZor_s,1104
+lightning_sdk/__init__.py,sha256=0M0hCXuVUl2Lxnfphl1qbZNnMBde0xhorpbackzGcHk,1104
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
-lightning_sdk/ai_hub.py,sha256=dfw5CS93ehGmPlZCbbE2Pu3v0ANI6ZY05-ULbcHFqs4,7189
+lightning_sdk/ai_hub.py,sha256=Zsc8GhFuRQIAYk3J3znUYDfyVhe3sd0WYLk2bFcoxVI,6803
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=KWMWnORHItIIA3PGn71YPs-7RjzGi8IXa2kQ5Qo4U8M,2459
-lightning_sdk/lit_container.py,sha256=423XYYpzoGppv200kMbD9dB-iqP6xdHj7q5kejJLDKE,3778
+lightning_sdk/lit_container.py,sha256=2A_7Hc_tX2ilj6UZmdLm-us144_KIcnWB0qT96sJGQM,4000
 lightning_sdk/machine.py,sha256=EMr-ulyYEYhIkKFnGBOofnFf4asndTSeoOQxlyd3xW4,3632
-lightning_sdk/models.py,sha256=inFyRA8aLMvrAhCeIdJaTGHjp0Qtou9G_J3gXePxU4E,5975
+lightning_sdk/models.py,sha256=0aJ4OXNY0MYUCfsYI6i18fV61DU-HGKIKEHFD99r1zk,6231
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
 lightning_sdk/plugin.py,sha256=TGxCa-xsOpv_kaKiVnNBNG3jLe9MweSS5Cnz9PcxQfY,14860
-lightning_sdk/serve.py,sha256=gllkjhCzG_Pdo5iV9AF78ytwgJXxjIUmb8vebi50M1o,8334
+lightning_sdk/serve.py,sha256=3nE_0TkTOtiCpdieQ01Pkyi-rb1rIkBJiF_g_fGIos8,9456
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
 lightning_sdk/studio.py,sha256=hyvAiVhkETAtbu0RRF1Aw6F8Y__E1SSAmsB8PHfmqHo,19935
 lightning_sdk/teamspace.py,sha256=CzpQJ6Uj_at0W-HIzYd-Y3gN9qBotUbAV3YBWjNT9U8,14364
 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=Yr9VvxueIqPUUeMqExbjbWD5CE_yeWjq0dXQQv47erg,6779
+lightning_sdk/api/ai_hub_api.py,sha256=IaA57Cqu_zdj5zRwKkx9LJQIGFFo2mHmPOYekKxbZ9A,6999
 lightning_sdk/api/deployment_api.py,sha256=z_D7ZZAnsVe3Q_ZVx1azcLK2_pP5xj63tmU97qhgyOw,22583
 lightning_sdk/api/job_api.py,sha256=_mMAI_BG_48i-BLwCP_U72zgmM5zYa2KUZ7u66HWkIc,13568
 lightning_sdk/api/lit_container_api.py,sha256=m-3qZIIpZ24Z2Z8z9x8Di6cOSjkZmBFcf9R2yVMpA_4,7648
@@ -47,7 +47,7 @@ lightning_sdk/cli/list.py,sha256=dAZ94QPvE4IkH6GfL7171TMrMfcuWB53cvW6wk2eL4w,101
 lightning_sdk/cli/mmts_menu.py,sha256=HUXo3ZoZ3fWOCNWTQWoJgUlFXYq5uVm_6uFjAq7BDe8,2219
 lightning_sdk/cli/open.py,sha256=sgMLWBnkXdIq8H9XK_rph2bye3b07AKTJBQIk9fCGVc,1937
 lightning_sdk/cli/run.py,sha256=8JZiDrKwDhlaTOJd6qq2mCWJRqKm6shCWLzpbmFYIkE,13929
-lightning_sdk/cli/serve.py,sha256=9g8FygBIIgT8qsL885xa-tUt4IXGSm7pNEnx23CIdOE,5729
+lightning_sdk/cli/serve.py,sha256=IHQneUUWhyYQHP_ChQafm7vyfb5-9crIH_CmjKUy7TM,9445
 lightning_sdk/cli/start.py,sha256=jUk52lkEFC_fqiEPkwM8GwE68WMNEtzBuzjkvr3POd0,1840
 lightning_sdk/cli/stop.py,sha256=5nCrUe1BONpX1nKNhbSFqLaXXKaRhSO7PvM1BVYLgn4,2864
 lightning_sdk/cli/studios_menu.py,sha256=TA9rO6_fFHGMz0Nt4rJ6iV80X5pZE4xShrSiyXoU-oQ,4129
@@ -913,11 +913,11 @@ lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2Px
 lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=B3HC-c82bPHprEZh1mhLCPCrCE8BOKqwIhY7xCF9CXg,15152
-lightning_sdk/mmt/mmt.py,sha256=UbbK6UFEd-S2OPjezCG7cy-7kfMbyLOg3LAzs_Zq2Ds,13872
-lightning_sdk/mmt/v1.py,sha256=dROOXRWLzG2kUwGtvozB9Vywt_3QK8j4XFHVZFohWUM,9527
+lightning_sdk/mmt/mmt.py,sha256=swdGP1DOM42a_QmmY1vg3--6ZBDiC4zToAzU9Eycv4U,13446
+lightning_sdk/mmt/v1.py,sha256=TxLtL0ssDoP8eyleDaFyYr4evkOKbLJcckLVIfalOno,8429
 lightning_sdk/mmt/v2.py,sha256=Em1XBoqViqUMKm-sshzdMcSH5UTtZZwbJcsgqY6-mw0,9625
 lightning_sdk/pipeline/__init__.py,sha256=uZS2v8HPGfNz2SrCCVrZ9EhihS3kd6aGtwJLbuaI3dA,167
-lightning_sdk/pipeline/pipeline.py,sha256=L1Aj5kRffjBKLUYt6YJDerw2CwCGi8g1TTwHV3qrpfU,3774
+lightning_sdk/pipeline/pipeline.py,sha256=CCRcHUjL24yKm24NMB5xzK1sre6bQ5sY07tMifmpqI0,3762
 lightning_sdk/pipeline/types.py,sha256=dZYNrWpuytmyjBHh8OUYT8BRwMe1_5PDDSl0AWsXkyA,9662
 lightning_sdk/pipeline/utils.py,sha256=gccFmSx3VrkDpWtq26eRCnvWT8l4hKdYLntDMYlB5jw,2627
 lightning_sdk/services/__init__.py,sha256=gSWUjccEhMI9CIWL_nbrFHUK2S6TM2725mEzrLMfK1Y,225
@@ -928,9 +928,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=MALzFO5iVlkZpnEiC9QkyWcTTYksCeHsloG6MCSJl48,6461
-lightning_sdk-0.2.3.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.2.3.dist-info/METADATA,sha256=dqptSi26kADENeZWIzeE4uyJdaLagaxgvam8I_49_uQ,3991
-lightning_sdk-0.2.3.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.2.3.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.2.3.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.2.3.dist-info/RECORD,,
+lightning_sdk-0.2.4.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.2.4.dist-info/METADATA,sha256=H8Yv9rh5KKFyNlVa6zU8k8Frb6C4ZeWoaauSIO2DUnk,3991
+lightning_sdk-0.2.4.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.2.4.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.2.4.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.2.4.dist-info/RECORD,,