lightning-sdk 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

lightning_sdk/lit_container.py

@@ -1,6 +1,8 @@
 from datetime import datetime
 from typing import Dict, List, Optional
 
+from rich.console import Console
+
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.utils.resolve import _resolve_teamspace
 
@@ -10,7 +12,7 @@ class LitContainer:
         self._api = LitContainerApi()
 
     def list_containers(
-        self, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
+        self, teamspace: str, org: Optional[str] = None, user: Optional[str] = None, cloud_account: Optional[str] = None
     ) -> List[Dict[str, str]]:
         """List available docker repositories.
 
@@ -18,16 +20,19 @@ class LitContainer:
             teamspace: The teamspace to list containers from.
             org: The organization to list containers from.
             user: The user to list the containers from.
+            cloud_account: The cloud account to list the containers from.
 
         Returns:
             A list of dictionaries containing repository details.
         """
+        console = Console()
         try:
             teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user)
-        except Exception as e:
-            raise ValueError(f"Could not resolve teamspace: {e}") from e
+        except Exception:
+            console.print(f"[bold red]Could not resolve teamspace: {teamspace}[/bold red]")
+            return []
         project_id = teamspace.id
-        repositories = self._api.list_containers(project_id)
+        repositories = self._api.list_containers(project_id, cloud_account=cloud_account)
         table = []
         for repo in repositories:
             created_date = repo.creation_time
@@ -39,8 +44,11 @@ class LitContainer:
             table.append(
                 {
                     "REPOSITORY": repo.name,
-                    "IMAGE ID": repo.id,
+                    "CLOUD ACCOUNT": cloud_account
+                    if cloud_account != "" and cloud_account is not None
+                    else "Lightning cloud",
                     "CREATED": created,
+                    "LATEST TAG": repo.latest_artifact.tag_name,
                 }
             )
         return table

lightning_sdk/models.py

@@ -95,7 +95,7 @@ def _parse_model_name_and_version(name: str) -> Tuple[str, str, str, str]:
         ) from err
     parts = model_name.split(":")
     if len(parts) == 1:
-        return org_name, teamspace_name, parts[0], "default"
+        return org_name, teamspace_name, parts[0], None
     if len(parts) == 2:
         return org_name, teamspace_name, parts[0], parts[1]
     # The rest of the validation for name and version happens in the backend
@@ -120,6 +120,8 @@ def download_model(
     """
     name = _extend_model_name_with_teamspace(name)
     teamspace_owner_name, teamspace_name, model_name, version = _parse_model_name_and_version(name)
+    if version is None:
+        version = "default"
 
     download_dir = Path(download_dir)
 
@@ -147,6 +149,7 @@ def upload_model(
     path: Union[str, Path, List[Union[str, Path]]] = ".",
     cloud_account: Optional[str] = None,
     progress_bar: bool = True,
+    metadata: Optional[Dict[str, Any]] = None,
 ) -> UploadedModelInfo:
     """Upload a Model.
 
@@ -159,11 +162,13 @@ def upload_model(
         progress_bar: Whether to show a progress bar for the upload.
     """
     name = _extend_model_name_with_teamspace(name)
-    org_name, teamspace_name, model_name, _ = _parse_model_name_and_version(name)
+    org_name, teamspace_name, model_name, version = _parse_model_name_and_version(name)
     teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
     return teamspace.upload_model(
         path=path,
         name=model_name,
+        version=version,
         cloud_account=cloud_account,
         progress_bar=progress_bar,
+        metadata=metadata,
     )

lightning_sdk/serve.py

@@ -3,18 +3,15 @@ import shlex
 import subprocess
 from pathlib import Path
 from typing import Generator, List, Optional, Union
-from urllib.parse import urlencode
 
 import docker
 from rich.console import Console
-from rich.progress import Progress
 
 from lightning_sdk import Deployment, Machine, Teamspace
 from lightning_sdk.api.deployment_api import AutoScaleConfig, DeploymentApi, Env, Secret
 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
+from lightning_sdk.api.utils import _get_cloud_url, _get_registry_url
+from lightning_sdk.lightning_cloud.env import LIGHTNING_CLOUD_URL
 
 _DOCKER_NOT_RUNNING_MSG = (
     "Deploying LitServe requires Docker to be running on the machine. "
@@ -23,23 +20,6 @@ _DOCKER_NOT_RUNNING_MSG = (
 )
 
 
-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)
-
-
-def authenticate() -> None:
-    auth = _Auth()
-    auth.authenticate()
-
-
 class _LitServeDeployer:
     def __init__(self, name: Optional[str], teamspace: Optional[Teamspace]) -> None:
         self.name = name
@@ -182,22 +162,14 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
 
         return log_generator()
 
-    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)
+    def build_container(self, path: str, repository: str, tag: str) -> Generator[str, None, None]:
         build_logs = self._docker_build_with_logs(path, repository, tag=tag)
 
         for line in build_logs:
             if "error" in line:
-                progress.stop()
-                console.print(f"\n[red]{line}[/red]")
                 raise RuntimeError(f"Failed to build image: {line}")
             else:
-                console.print(
-                    line.strip(),
-                )
-                progress.update(build_task, description="Building Docker image")
-
-        progress.update(build_task, description="[green]Build completed![/green]")
+                yield line.strip()
 
     def push_container(
         self,
@@ -205,26 +177,31 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
         tag: str,
         teamspace: Teamspace,
         lit_cr: LitContainerApi,
-        progress: Progress,
         cloud_account: str,
-    ) -> dict:
-        console = self._console
-        push_task = progress.add_task("Pushing to registry", total=None)
-        console.print("\nPushing image...", style="bold blue")
+    ) -> Generator[dict, None, dict]:
         lit_cr.authenticate()
-        push_status = lit_cr.upload_container(repository, teamspace, tag=tag, cloud_account=cloud_account)
-        last_status = {}
+        push_status = lit_cr.upload_container(
+            repository, teamspace, tag=tag, cloud_account=cloud_account, platform=None
+        )
         for line in push_status:
-            last_status = line
             if "error" in line:
-                progress.stop()
-                console.print(f"\n[red]{line}[/red]")
                 raise RuntimeError(f"Failed to push image: {line}")
             if "status" in line:
-                console.print(line["status"].strip())
-                progress.update(push_task, description="Pushing to registry")
-        progress.update(push_task, description="[green]Push completed![/green]")
-        return last_status
+                yield {"status": line["status"].strip()}
+
+        registry_url = _get_registry_url()
+        container_basename = repository.split("/")[-1]
+        repository = (
+            f"{registry_url}/lit-container{f'-{cloud_account}' if cloud_account is not None else ''}/"
+            f"{teamspace.owner.name}/{teamspace.name}/{container_basename}"
+        )
+        yield {
+            "finish": True,
+            "status": "Container pushed successfully",
+            "url": f"{LIGHTNING_CLOUD_URL}/{teamspace.owner.name}/{teamspace.name}/containers/{container_basename}"
+            f"{f'?clusterId={cloud_account}' if cloud_account is not None else ''}",
+            "image": repository,
+        }
 
     def _update_deployment(
         self,

lightning_sdk/teamspace.py

@@ -3,8 +3,10 @@ import warnings
 from pathlib import Path
 from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 
+import lightning_sdk
 from lightning_sdk.agents import Agent
 from lightning_sdk.api import TeamspaceApi
+from lightning_sdk.lightning_cloud.openapi import V1ProjectClusterBinding
 from lightning_sdk.machine import Machine
 from lightning_sdk.models import UploadedModelInfo
 from lightning_sdk.organization import Organization
@@ -124,6 +126,11 @@ class Teamspace:
         clusters = self._teamspace_api.list_cloud_accounts(teamspace_id=self.id)
         return [cl.cluster_name for cl in clusters]
 
+    @property
+    def cloud_account_objs(self) -> List[V1ProjectClusterBinding]:
+        """All cloud accounts associated with that teamspace."""
+        return self._teamspace_api.list_cloud_accounts(teamspace_id=self.id)
+
     @property
     def clusters(self) -> List[str]:
         """All clusters associated with that teamspace."""
@@ -245,8 +252,10 @@ class Teamspace:
         self,
         path: Union[str, Path, List[Union[str, Path]]],
         name: str,
+        version: Optional[str] = None,
         cloud_account: Optional[str] = None,
         progress_bar: bool = True,
+        metadata: Optional[dict] = None,
     ) -> UploadedModelInfo:
         """Upload a local checkpoint file to the model store.
 
@@ -256,6 +265,7 @@ class Teamspace:
             cloud_account: The name of the cloud account to store the Model in.
                 If not provided, the default cloud account for the Teamspace will be used.
             progress_bar: Whether to show a progress bar for the upload.
+            metadata: Metadata to attach to the model. Can be a dictionary.
         """
         if not path:
             raise ValueError("No path provided to upload")
@@ -281,13 +291,18 @@ class Teamspace:
                 f"The listed files are: {file_paths}\nThe relative paths are: {relative_paths}"
             )
 
-        cloud_account = (
-            self._teamspace_api._determine_cloud_account(self.id) if cloud_account is None else cloud_account
-        )
+        if cloud_account is None:
+            cloud_account = self._teamspace_api._determine_cloud_account(self.id)
+        if not metadata:
+            metadata = {}
+        if not isinstance(metadata, dict):
+            raise TypeError(f"Metadata must be a dictionary, but provided {type(metadata)}")
+        metadata.update({"lightning-sdk": lightning_sdk.__version__})
 
         model = self._teamspace_api.create_model(
             name=name,
-            metadata={"filenames": ",".join(relative_paths)},
+            version=version,
+            metadata=metadata,
             private=True,
             teamspace_id=self.id,
             cloud_account=cloud_account,

{lightning_sdk-0.2.7.dist-info → lightning_sdk-0.2.9.dist-info}/METADATA

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

{lightning_sdk-0.2.7.dist-info → lightning_sdk-0.2.9.dist-info}/RECORD

@@ -1,35 +1,37 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=2YlLVkcc_oSxRiBA9ZX7-ItwTff8JwvMihcSY8xDC-Q,1104
+lightning_sdk/__init__.py,sha256=pNcU1KuZZCO0eFmHgbHUbzRn_8dipYQGo2COXVL3c0w,1104
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=KWMWnORHItIIA3PGn71YPs-7RjzGi8IXa2kQ5Qo4U8M,2459
-lightning_sdk/lit_container.py,sha256=UrHzi-s8yyU_7ALHgcq0n2GCOf1R8VnQTSxSxZWbsmI,4415
+lightning_sdk/lit_container.py,sha256=ObzkeDK0C38aLJBpptk5RtiC22v2SGwcOOxcQlYDXRA,4845
 lightning_sdk/machine.py,sha256=EMr-ulyYEYhIkKFnGBOofnFf4asndTSeoOQxlyd3xW4,3632
-lightning_sdk/models.py,sha256=5DCfv2M0b8iYKj8tAMIX7gIUXrk2CgALANX14N5OFgo,6255
+lightning_sdk/models.py,sha256=YOB3IHr64KkFQkIAah_bXwcmGIGr5eAWh-j5K3OdpzA,6407
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
 lightning_sdk/plugin.py,sha256=_K0m-PSyXE86B-SFQVQm0JEvfdIVXeARmnlATCAiMdo,14632
-lightning_sdk/serve.py,sha256=F2vBWXG4HOoTO3-sEzrghvJ7vWSFNoNpcdTPtv5Ljs4,12287
+lightning_sdk/serve.py,sha256=yV6pPFKirimvMBwJ5EKF9ZmMBaEvcaegdktfzzVwwrQ,11529
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
 lightning_sdk/studio.py,sha256=hyvAiVhkETAtbu0RRF1Aw6F8Y__E1SSAmsB8PHfmqHo,19935
-lightning_sdk/teamspace.py,sha256=EqMDDXtePCDD3JZHzx_qx7xn2tSxtH-LqwSc1JtIwEM,15073
+lightning_sdk/teamspace.py,sha256=vYQF1rvH-Ew3rucGghUDTTxomoEUZn9xgbdt8WFyIOw,15773
 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=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
+lightning_sdk/api/cluster_api.py,sha256=OxAhEdNt-9bYQ-u3MCADdVGsilXf2e7Wcxvb73Cmn8k,916
 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=lI66uqzVYuyCAQ1d9yCir67ir7t6fUKA7O_ML-Vj94A,10730
+lightning_sdk/api/lit_container_api.py,sha256=L1ZIuE4CXw3wFZxbkJ4lJmO77tlGjbQBreigj4_eTbY,10558
 lightning_sdk/api/mmt_api.py,sha256=-v7ATab-ThAM-HRClS92Ehxuu9MlBfdKWWFCGvVUHiM,8962
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/pipeline_api.py,sha256=P5P9C6qOpyBGU0t5N68h1LuFAsAKmPPgkac6uObrYKw,1676
 lightning_sdk/api/studio_api.py,sha256=BqmGmotUJrM-RoMfwYSdymAir491tBkPsIlvdJqOoUw,25858
-lightning_sdk/api/teamspace_api.py,sha256=O17f-A2O6nSGBD3WWnrpp1dzGGJ8UgOhK5tZtVFnznw,11626
+lightning_sdk/api/teamspace_api.py,sha256=bVlHNv-OIPOK-Gl6K6Ko83yBJdpIU5rDzB_O51VyehY,11675
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
 lightning_sdk/api/utils.py,sha256=6EVDZQ6OFc5hS_689J3XFzriPtr99CnDzrxIdGTPxRE,23816
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/ai_hub.py,sha256=8MxGQC7SZALAsIj0tZ9HMZCeVHESUcve0cNazRzIFbE,1812
+lightning_sdk/cli/clusters_menu.py,sha256=HX5A7yE-3JijEcF2ANeSUhE9gjq43h-5KDw0aC0op54,1767
 lightning_sdk/cli/coloring.py,sha256=YEb1LiYb6CekVfCRb83-npAmNgd-7c1LzXuNDo4GLSc,2415
 lightning_sdk/cli/configure.py,sha256=B84BtYU0ccKvc1qfK1kqn7sbnHaKW_5Po54k-m7lKHc,4174
 lightning_sdk/cli/connect.py,sha256=qsKb_NMvaA4xrESjMEApM5lyfuHJApe3eCeeI8ttrmQ,975
@@ -43,11 +45,11 @@ lightning_sdk/cli/generate.py,sha256=8mjm5tMYIXMoN2X_eEKpwBIUcCWaARiQAmreh-9LIBg
 lightning_sdk/cli/inspect.py,sha256=GA3XDC3DSkqJ4B5NQR9qmGGvkP0K8Sk90xZhzTGzC1o,1614
 lightning_sdk/cli/job_and_mmt_action.py,sha256=OV_i5K3g-NnMUF8Sq6QliQdltnhGVRdtoyZqJVXu4Ss,1659
 lightning_sdk/cli/jobs_menu.py,sha256=_2qsOBTdIKT5NCjwrBo9-AAn3fIvdN9IcVUVHF_KGQk,2162
-lightning_sdk/cli/list.py,sha256=dAZ94QPvE4IkH6GfL7171TMrMfcuWB53cvW6wk2eL4w,10156
+lightning_sdk/cli/list.py,sha256=bH_rX9J_Ubb9dkF2fpoZf0zMXH9YoEjcm63gk5v0zs4,10919
 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=BC8JsaN0QVmPT_W4x1QC93Yyur9w5czAJCcIt2UJx5Q,9654
+lightning_sdk/cli/serve.py,sha256=49T0KjWYfcQk4bfTHXJqctBPMfcnr5Odkc9Vr5qMI8k,13701
 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
@@ -69,7 +71,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=02f3itTEMJ4wKpVxOfAAfWfigVeX8G51eg
 lightning_sdk/lightning_cloud/rest_client.py,sha256=UZoHDJQ0qUu7LMGgaecBsAHzT3-auCcvSx18hQcN89E,6767
 lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
-lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=DHGW2cHqTpI_ESn3Nch6qcamjX_vTgn0PJxe8_Wi_Sc,99087
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=Ssjh8VepSAaEV8fMbbV8Bq5r0HVVksLw9ULC5wYo98w,99754
 lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
 lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=H4PTkRz1QAonxKHbx3LG-gDEtdQPAxMO-bVfMfbUfJg,8306
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
@@ -80,7 +82,7 @@ lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=EUu77
 lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAltWyj1wYg3HNk0cxT2HWbNhAqQSLvMDgM,30948
 lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=SDJ80TezTWKhE3LR67OI6dhHDHABc4BfG7FjcFX7Ba4,68129
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py,sha256=iIFr_vpFhC5EPbf95aFOYvSbMuVCAk61sDx0s_tTICg,23947
-lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=8bgg4a90xn5DtPjk7_hiAeZeDR6zCbTqjOeRmRZgLio,379685
+lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=2wf4NoJBhIc98TrrU8FWLNGK9NcbKrUYwGLCuMxOlxc,389727
 lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=hGE_GT6YrmL9HnaGzZ6t1eqJvn6hMTKTpwGDFaHBtlg,174283
 lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=ODiLVuUw09zK63GS4ilwYE4XDbTAYi7lIu5USssmyro,55071
 lightning_sdk/lightning_cloud/openapi/api/dataset_service_api.py,sha256=wCrcXOMuVs5rA4PHyrVJqpInO-Vy1hulddjCJUenPJc,23508
@@ -111,7 +113,7 @@ lightning_sdk/lightning_cloud/openapi/api/ssh_public_key_service_api.py,sha256=z
 lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py,sha256=U5atWnTx33jVy5BF1R3yAKZfxmOWyzdG14W8ZRjUhAw,65552
 lightning_sdk/lightning_cloud/openapi/api/studio_jobs_service_api.py,sha256=VJ7VcSvdmGbPkgmQZhaOSR47olcNPMzRlmT5J5uKsng,27877
 lightning_sdk/lightning_cloud/openapi/api/user_service_api.py,sha256=MR0jvZY0I-fFEPNpcSqrN3vl9XjrEn_D7_gYhwYeMrc,66796
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=fQlEbNJ6x5WIixVbzpTC6xBqivkMy0Va-en1Vz16OOk,93586
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=D7mxNSXwwyL-5sIkx3icnp2QOjBi5iF7wqE3nobOPMw,94253
 lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py,sha256=X087AkCafKp2g8E2MR5ghU3pxCNNcoehZJttvuVfdC8,4274
 lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py,sha256=NzKKkTs2PtMSIJJYxeh3aDBj-djBrI_qATGz7n5Oqdk,9234
 lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=vYRXPGrwpj8nsSVwlaf3c6d5ysuvpNOXrfeUOKZn0dM,21508
@@ -132,6 +134,7 @@ lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_
 lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versions_body.py,sha256=pnxO1R-FILLh4SFmaRUj5FwhSfUVYdZ0wJGR23GJgOg,12580
 lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_metric_body.py,sha256=0eFLDAHKdfiAHcpMZHDupjgTmt63jQqjFq4whga6EHo,6101
 lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_runs_body.py,sha256=DY_vQycymNBV3WowPhqddfNxzJdqJS0LOFyzQg_jwf4,20498
+lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_systemmetrics_body.py,sha256=eH2es4Nsm8dGZAK-8RbPzmyT1bVJVX-5EemfkEwPC_E,4926
 lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py,sha256=qw1RiMZdUkCQ5ZvuZ8T-4zk8UI_Ku9EWABKApRln8OI,22996
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py,sha256=L_TAOS_SSt2mZaTUHXnvRmIYsI9SkLSLIPmhrMc5rac,8129
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py,sha256=VcqRpRwtGTGe6ox71KlRtmEpxqbWJJUxr69PFvTN37o,9631
@@ -305,7 +308,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_batch_update_lightningwork_respo
 lightning_sdk/lightning_cloud/openapi/models/v1_billing_feature.py,sha256=vkQ7QNTECZ0ZranpKUvugXPwRv6xDUMbBzMQ8yehiUs,6512
 lightning_sdk/lightning_cloud/openapi/models/v1_billing_period.py,sha256=uTaXGz4Vincg0Z3s4MBRgqq5YTxNLuAI0lx1-v02nTU,3059
 lightning_sdk/lightning_cloud/openapi/models/v1_billing_subscription.py,sha256=WQ3MwKOTHQU6-xcCYSSYZRotxvUWwvmmcHbTz70HL3A,10290
-lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py,sha256=yYc1ioTtLgU2_OLxC_G5Yb8eJl9TuMvZuYGwtwSZIfw,3093
+lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py,sha256=Ko2-MwFZfOr9FzRw0coyZ61kjoTQhSf2mYnuNmcmGOo,3131
 lightning_sdk/lightning_cloud/openapi/models/v1_body.py,sha256=dgBZ3hPV_Q_lfc6dNoNjzBpOJZMklBtM8IiQBcvbtEA,3580
 lightning_sdk/lightning_cloud/openapi/models/v1_build_spec.py,sha256=hqAjFwjZZeW-Jk3xWknetyoBA0J7xSrcUwl8rlkVUCA,6706
 lightning_sdk/lightning_cloud/openapi/models/v1_cancel_cloud_space_instance_switch_response.py,sha256=b7511arCBmJy8Lp7oWblcifG2r1xyV44q9nSW7yQzzM,3130
@@ -324,7 +327,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_artifact_event.py,sh
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_artifact_event_type.py,sha256=DYgZQ_LK97lFaBIspvogM93kXtcDv4kqki7Z8Lw-jpI,3217
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_code_version.py,sha256=_L6L1nAtWaRN6gTx5xmIwPzxf6dyaIVfoztN3c1zwUs,11686
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_code_version_status.py,sha256=L8SPQ-vQVnDA7gwvzQWFQNFil8CWcNVGUrqeBySuSMw,3329
-lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics.py,sha256=D0wqf3EsYNfT0G_6dM7IHearZe0MawL9EkXvmXQa5ao,21017
+lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics.py,sha256=wFbDYQRNP--PtL-ovm_Yjd26_GDk9IG0G0VXBOY_ukY,22770
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_engagement_response.py,sha256=9qAqwfTU1L2GsLqIkUgW7oR8aHnsvvZL_Mb2jePKq7k,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py,sha256=IPUIKypdAVvHyHL9zc9faEuHp47uT2sKHXJUX28_pSs,8413
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py,sha256=3py6ns4-nWXHVer246gr9EVw3AfxBaotz1S3s0ag1sY,4149
@@ -348,7 +351,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cluster_encryption_key.py,sha256
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py,sha256=Xs3mhVEiuJVbkKwMarY7iB_NwKY9_sK0xGgXDMuo0cs,3637
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py,sha256=4LbHvINs2CwQnuPg_uWw03GNNRT_wdqX42m_Zlqg2dY,5547
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_resource_tag.py,sha256=U6DfNNAIBZQ8SOF-2RXp2-CgmxaT1XJrNaCnZzaqjHg,4274
-lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py,sha256=D28ntpr6hrEdUU9FZSX1gJm_Fd6SOtSu_J2-q0gXTEs,14501
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py,sha256=ryiu2Cit2n3S8ZsAJiTNIZyHmlyUCyX46oPta5ODaZM,15445
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py,sha256=DwySTanGtV6YQEZ8sESmPq2CiM9xGfPajrD8w9UJqvU,24070
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_state.py,sha256=Mvl3dEhdtb7-rU4wUsR_-YOMpd7wE8uUAyXv3MiOcOk,3331
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py,sha256=n9NhKdWg51q_7d4VM9Bsk5aakctU-o3R-0YmdzTsMW0,13267
@@ -410,7 +413,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_ses
 lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_response.py,sha256=n3jJzoW0w_tISOxcK2fbfc42LAPpkXVKbmQufltELPc,3851
 lightning_sdk/lightning_cloud/openapi/models/v1_create_user_secret_request.py,sha256=RWDzrdrlk_BxC9gE230gvtIRmV9uPUwlhRUnuo-gS9k,4364
 lightning_sdk/lightning_cloud/openapi/models/v1_daily_usage.py,sha256=gpnQCq4uBLGMOrkSzxGRh4SsLw8Rj8Z4etEB1SO3das,4303
-lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py,sha256=bLjAVpkJ8mHtwYYMPOTpPFRUqD2IEKUj7RICSrD0zIY,21524
+lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py,sha256=x4rDDpv2jdF2WiFuDXOX-cBW721r2Z3Rux4cDFf0DvI,22243
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_artifact.py,sha256=g91LNJXfc3oDV3JJEZ-Fyy-g19crePaZFnbYtj4NfkM,6824
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_mount.py,sha256=NxMM-0hO9QjCw_0bOAsLa4DFZMM_ZFqc2XpeSOMu5P0,6814
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_state.py,sha256=eSUVNEhmB-RKoFz2Aehur-IjPIBDd8Z2JPFY4-SSyog,3295
@@ -543,6 +546,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_agent_job_env_response.py,sh
 lightning_sdk/lightning_cloud/openapi/models/v1_get_agent_job_logs_metadata_response.py,sha256=c_ZsP4V1UTFAxiq6g0gsu6PKD04SEBdY_M7x7edxoTU,5538
 lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py,sha256=iC3z2-zLpKn0QB97FvIwM6-omGRgQLYNft83BuQdQLg,3879
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_status_response.py,sha256=mkmpn9psHLvC22e3HtkiL5ziWDAZ0UJDPShEfFcBmbY,7049
+lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_system_metrics_aggregate_response.py,sha256=pAw7izmA2KjytbeVuzHpnKWYJc5hkOlfID1EktXNq48,4199
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py,sha256=oTyPx69HOhPsyKaoTcTJfKMFwZahlZtRZOPNqoa0g3o,7455
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_accelerator_demand_response.py,sha256=xxRf6ry2wQM3OsUhfn9wUTUjnvJu0Srv9Dn9Mlezafo,3950
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_credentials_response.py,sha256=5ryHCeO6hkoK-5THe_dGuBk0EnHSD21u-5AzjGzvv5A,3851
@@ -576,7 +580,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_user_notification_preference
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=S5anvaSlB855YPMlXq_DevxMaFyFCWj9oxSMLbXKqAs,31405
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py,sha256=igCbjaeZXEbiPAa9cUjwG4cPUivt-7MiiWYikELHwsw,9983
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py,sha256=0k1scTY1IAeKr3p0r7eV_xWRE7Nx54doTGygiQQPkbM,6879
-lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=Pn0Ap2IfvGaDWNf4fzK5uIWj6DPxCJgatDXD--q6w3E,15826
+lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=ztfNeQMBM6CQyoKAzILARBEiCB7sA-HvdrLEm427kac,16960
 lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py,sha256=Md6rMiUCMY4oDVE2IYaALD_I0o2PQIKF-9628IGmTyY,3791
 lightning_sdk/lightning_cloud/openapi/models/v1_gpu_system_metrics.py,sha256=io_MkTCTYi6qmYw8KqFZf2zHx8jgBsPKsEwqjr2BG0U,7722
 lightning_sdk/lightning_cloud/openapi/models/v1_header.py,sha256=6GR8AL-UcMa3mH-PLtd7WWDZiG0Zi3AYKFw19AAasZc,4831
@@ -824,6 +828,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_refresh_response.py,sha256=29Pdm
 lightning_sdk/lightning_cloud/openapi/models/v1_region_state.py,sha256=c_6TjyTxEy_eM385G_Fg7Ks9rvjeBlJVmjyQ-OumqKA,3167
 lightning_sdk/lightning_cloud/openapi/models/v1_regional_load_balancer.py,sha256=oyBYtKR29jUmcXivUaCyB_omigKVT6soHVF07l8CBKY,4421
 lightning_sdk/lightning_cloud/openapi/models/v1_report_cloud_space_instance_stop_at_response.py,sha256=uF4ATd0o6tVhWWGzKsV3p1Ltfk_b8jFo1BvqIqZ294M,3130
+lightning_sdk/lightning_cloud/openapi/models/v1_report_cloud_space_instance_system_metrics_response.py,sha256=w_hru1rjgCJGdRR75wG9LJmpMZqoM9JkCZsnzlq74xo,3172
 lightning_sdk/lightning_cloud/openapi/models/v1_report_logs_activity_response.py,sha256=zA2m8-EYsmfap7Ob-fg_XaRc42pFFAB48_p5NRz7D9g,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_report_restart_timings_response.py,sha256=qVJHTSpU_6Kdm1jLPwbqDK0tJC1eiGpSoRR3XnKCltY,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_request.py,sha256=J0P8mFnD0oGn61TGJ_RH5z8IBbcYSmE6P4qzgDzLMvI,5402
@@ -855,7 +860,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_secret.py,sha256=RCaNHLqhe7fkXFI
 lightning_sdk/lightning_cloud/openapi/models/v1_secret_type.py,sha256=Iktq3g39511VdsIb6-K3VClAjTkVlh54hubAcH38JCc,3215
 lightning_sdk/lightning_cloud/openapi/models/v1_select.py,sha256=DOIP1LsjRpnheeYlYYsyc-rvCOp3Axekviz6jsOY2II,4222
 lightning_sdk/lightning_cloud/openapi/models/v1_server_alert.py,sha256=P9nGU-BdtkLB2QBSJfIc5vzjW7M5nytAeIidmfRk70A,5828
-lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_phase.py,sha256=LZmOvlNXHx5fz6cm-ZtKTaCbaqUZwnxVH-7Yqs6YNZY,3178
+lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_phase.py,sha256=aXgn2BJW5Y4IceClJSoQC0dwiQ2bVt5wfaMUqXxCEyM,3221
 lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py,sha256=wWgTEPSGoyEqa5G3jZ7bBzDVJeRWwBU9ycKeeCAki48,3145
 lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py,sha256=bPwhiTFfOAermbak6DFphQP7CdB6Qk7Anorr4CeNmH8,3241
 lightning_sdk/lightning_cloud/openapi/models/v1_server_check_in_response.py,sha256=EtdIVtxP_H89tKh6WNdwx0TQenVvwQXHpxGulhyJSZY,3028
@@ -889,6 +894,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_subnet_spec.py,sha256=BZkuFPYleu
 lightning_sdk/lightning_cloud/openapi/models/v1_switch_cloud_space_instance_response.py,sha256=D3MIVrzeQAWWwlyNWXtF1hPWJ_8nw623qG4bChZPxqE,3094
 lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py,sha256=Xd7EVD-EslEE0FhbPFg40nCjOjI6C1yLPTEyPV3LQDw,18476
 lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py,sha256=pxqJYZGogX0iTgb3Klw0eVVj0vt1Vy7wSrVOOwp2zAs,5910
+lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics_aggregated.py,sha256=DXuQIFEbVQO0tEed3FBSBMmODSyw7Sma6xHFGaN0hSM,7195
 lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics_list.py,sha256=sKU7Ue8dpeF7p-QkHADZkumfJYx00ambGqCRsiR4EV0,4613
 lightning_sdk/lightning_cloud/openapi/models/v1_telemetry.py,sha256=IH3Mdpt2NnnKpBuYSqEC_bAv2mfmD4lgE27wuPPrnQ0,9470
 lightning_sdk/lightning_cloud/openapi/models/v1_timestamp_code_telemetry.py,sha256=WCU6lxiRku2-WjWc0OyFAN0aaLXEieKBSGi09Bmu_J8,4193
@@ -936,7 +942,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
 lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=RhhnH9ygScZyExg06WhvMNPPRLSe8FYkIftqF-D9NIU,13408
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=iH67BcONBSLYzcZpGpKWSOzJTCpuqYt7FU4OUs8BJ9k,6076
-lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=pemZZWcdhWz2zdb2Yh9APTsjIevcpndVrKcf6BHiwhU,72039
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=x4SMYWB4CN5NaoXt58_W6MbYSxCy0kmAgbpNY5ZB1VE,67194
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=3jeJfpbBpYY2B2Ao2j2N93CMO2CnvmPqndE4Lw3ZfMA,13056
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py,sha256=3WpZ-lf7xPwuYyQDMdP7Uc6-dh3vf5TaaUlcMfesfMk,5208
 lightning_sdk/lightning_cloud/openapi/models/v1_user_slurm_job_action_response.py,sha256=BdNzXH8Vsf5PHjl9Rd-TVkpAgx1YC9rf8LD0js-ba20,3058
@@ -952,6 +958,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_verify_verification_response.py,
 lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py,sha256=sHERBAeKdZT21E_io9Z6AZjgv6x8o9Xd-_ZysvBK8gY,6713
 lightning_sdk/lightning_cloud/openapi/models/v1_volume.py,sha256=UIEsGm_TA6Qustavo2Umxq1No4pP-VImzsYw7CfNXMk,4702
 lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py,sha256=jq4ODNqWXJqkkyMdod6C7eiuWDPAfqMC1sEI6DSJDH0,5594
+lightning_sdk/lightning_cloud/openapi/models/v1_weka_data_connection.py,sha256=tFT345wCKUK7vrV0X9M8lrJQBelJWQGzQIHPgBsa_lw,6884
 lightning_sdk/lightning_cloud/openapi/models/v1_work.py,sha256=MwXPSG8TE_CcGiffx6OI6U7kyf1NnSsDvJEM6LuwcYA,4910
 lightning_sdk/lightning_cloud/openapi/models/validate.py,sha256=_OEykjdTe3VpAvSG1E1reBcaFLXWOKpuIC9jlDQXn-g,8903
 lightning_sdk/lightning_cloud/openapi/models/validateautojoindomain_domain_body.py,sha256=gRI7XdvbY7w09gIH6BkFai8UA-y71T2AmiR5Y2XcTe8,3781
@@ -992,9 +999,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=uYAMAZ-emtFnjdg_sXbKRo3VTe5YWx7WEgCpNlHhHG8,6462
-lightning_sdk-0.2.7.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.2.7.dist-info/METADATA,sha256=AfTdQ9ctYrQ5bNUh-yjMOxIwyHZZbLwizcAnqldNS6E,3991
-lightning_sdk-0.2.7.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-0.2.7.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.2.7.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.2.7.dist-info/RECORD,,
+lightning_sdk-0.2.9.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.2.9.dist-info/METADATA,sha256=9LU0cpELBAIAeGDcsVkC80l1dSo495Tluy_OrUzXIDY,3991
+lightning_sdk-0.2.9.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-0.2.9.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.2.9.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.2.9.dist-info/RECORD,,