lightning-sdk 0.1.47__py3-none-any.whl → 0.1.48__py3-none-any.whl

lightning_sdk/{lit_registry.py → lit_container.py}

@@ -1,17 +1,17 @@
 from typing import Dict, List, Optional
 
-from lightning_sdk.api.lit_registry_api import LitRegistryApi
+from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.utils.resolve import _resolve_teamspace
 
 
-class LitRegistry:
+class LitContainer:
     def __init__(self) -> None:
-        self._api = LitRegistryApi()
+        self._api = LitContainerApi()
 
     def list_containers(
         self, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
     ) -> List[Dict[str, str]]:
-        """List available containers.
+        """List available docker repositories.
 
         Args:
             teamspace: The teamspace to list containers from.
@@ -37,3 +37,21 @@ class LitRegistry:
                 }
             )
         return table
+
+    def delete_container(
+        self, container: str, teamspace: str, org: Optional[str] = None, user: Optional[str] = None
+    ) -> None:
+        """Delete a docker container.
+
+        Args:
+            container: Name of the container to delete.
+            teamspace: The teamspace which contains the container.
+            org: The organization which contains the container.
+            user: The user which contains the container.
+        """
+        try:
+            teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user)
+        except Exception as e:
+            raise ValueError("Could not resolve teamspace") from e
+        project_id = teamspace.id
+        return self._api.delete_container(project_id, container)

lightning_sdk/mmt/mmt.py

@@ -1,4 +1,3 @@
-from contextlib import suppress
 from functools import lru_cache
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
 
@@ -58,15 +57,40 @@ class MMT(_BaseMMT):
             user: the name of the user owning the :param`teamspace`
                 in case it is owned directly by a user instead of an org.
         """
-        internal_mmt_cls = _MMTV2 if _has_mmt_v2() and not self._force_v1 else _MMTV1
-
-        self._internal_mmt = internal_mmt_cls(
-            name=name,
-            teamspace=teamspace,
-            org=org,
-            user=user,
-            _fetch_job=_fetch_job,
-        )
+        from lightning_sdk.lightning_cloud.openapi.rest import ApiException
+
+        if _has_mmt_v2() and not self._force_v1:
+            # try with v2 and fall back to v1
+            try:
+                mmt = _MMTV2(
+                    name=name,
+                    teamspace=teamspace,
+                    org=org,
+                    user=user,
+                    _fetch_job=_fetch_job,
+                )
+            except ApiException as e:
+                try:
+                    mmt = _MMTV1(
+                        name=name,
+                        teamspace=teamspace,
+                        org=org,
+                        user=user,
+                        _fetch_job=_fetch_job,
+                    )
+                except ApiException:
+                    raise e from e
+
+        else:
+            mmt = _MMTV1(
+                name=name,
+                teamspace=teamspace,
+                org=org,
+                user=user,
+                _fetch_job=_fetch_job,
+            )
+
+        self._internal_mmt = mmt
 
     @classmethod
     def run(
@@ -148,10 +172,7 @@ class MMT(_BaseMMT):
         # required for typing with "MMT"
         assert isinstance(ret_val, cls)
 
-        msg = "Multi-Machine Job was successfully launched."
-
-        with suppress(NotImplementedError):
-            msg += f" View it at {ret_val.link}"
+        msg = f"Multi-Machine Job was successfully launched. View it at {ret_val.link}"
 
         _logger.info(msg)
         return ret_val

lightning_sdk/mmt/v1.py

@@ -1,6 +1,7 @@
 from typing import TYPE_CHECKING, Dict, Optional, Tuple, Union
 
 from lightning_sdk.api.mmt_api import MMTApiV1
+from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.job.v1 import _internal_status_to_external_status
 from lightning_sdk.job.work import Work
 
@@ -179,7 +180,10 @@ class _MMTV1(_BaseMMT):
 
     @property
     def link(self) -> str:
-        return f"https://lightning.ai/{self.teamspace.owner.name}/{self.teamspace.name}/studios/{self._job_api.get_studio_name(self._guaranteed_job)}/app?app_id=mmt&app_tab=Runs&job_name={self.name}"
+        return (
+            f"{_get_cloud_url()}/{self.teamspace.owner.name}/{self.teamspace.name}/studios/{self._job_api.get_studio_name(self._guaranteed_job)}/"
+            f"app?app_id=mmt&app_tab=Runs&job_name={self.name}"
+        )
 
     @property
     def image(self) -> Optional[str]:

lightning_sdk/mmt/v2.py

@@ -1,6 +1,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
 
 from lightning_sdk.api.mmt_api import MMTApiV2
+from lightning_sdk.api.utils import _get_cloud_url
 
 if TYPE_CHECKING:
     from lightning_sdk.job.job import Job
@@ -195,8 +196,8 @@ class _MMTV2(_BaseMMT):
 
     @property
     def link(self) -> str:
-        # TODO: Since we don't have a UI for this yet, we can't have a link
-        raise NotImplementedError
+        # TODO: MMT env with studio -> go to studio plugin
+        return f"{_get_cloud_url()}/{self.teamspace.owner.name}/{self.teamspace.name}/jobs/{self.name}?app_id=mmt"
 
     @property
     def image(self) -> Optional[str]:

{lightning_sdk-0.1.47.dist-info → lightning_sdk-0.1.48.dist-info}/METADATA

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

{lightning_sdk-0.1.47.dist-info → lightning_sdk-0.1.48.dist-info}/RECORD

@@ -1,10 +1,10 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=8lax4DfCs2aWQNTxZwlBWy3Rp6jI3-IY89WASJupwZw,925
+lightning_sdk/__init__.py,sha256=e-z42Ud9d3en8_MesDM8ZmP-uq7DC8lW5Y0dMWI9IUM,925
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=kBjtmrzVHPCgqtV_TrSNkuf4oT2DLm8SYRTz4iTQmmY,6624
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=RnQwUquc_YPotjh6YXOoJvZs8krX_QFhd7kGv4U_spQ,1844
-lightning_sdk/lit_registry.py,sha256=0zRFH-l61gHOslR6m0Rtetm-sWoUtN7240cvbxAx03I,1332
+lightning_sdk/lit_container.py,sha256=MLdhLtKjRICDikdXfFUMAW39kC-oC9IONGUv2yh8gRU,2090
 lightning_sdk/machine.py,sha256=qutfVwQJHC02c2ygB_O5wrFI-Eq9zbfA0E1OPEYcCO0,861
 lightning_sdk/models.py,sha256=d27VAYUcbWKd4kuL_CqwCi3IguyjmKUR9EVWfXWTwmc,5606
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
@@ -18,33 +18,39 @@ lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=CYQLFLA89m3xQ-6Ss3UX4TDK6ZWRwmPGA5DjyJqW3RM,5578
 lightning_sdk/api/deployment_api.py,sha256=T480Nej7LqmtkAx8SBkPGQ5JxeyQ-GVIDqUCc7Z1yfk,21448
-lightning_sdk/api/job_api.py,sha256=L7O7zt3hy-s1UYDzO2FbLS7wtGXU3x8ZBDrtklQFZFc,11648
-lightning_sdk/api/lit_registry_api.py,sha256=peAZsJKVLvon7bVHq6aTYdcTVERiGoAZRQ5DzNean3Y,385
+lightning_sdk/api/job_api.py,sha256=ykok7Fg3iNOKysgO601gt8gidXc9vO8IRqpjJel9lS8,11980
+lightning_sdk/api/lit_container_api.py,sha256=BF5AOvHmF8i2eYBIRgIKcS1R_y3wn2PL_LhZruQyEAo,821
 lightning_sdk/api/mmt_api.py,sha256=rgMVDdlC3h-KqGI2FGp7Im99gnrHg5xx4FLwQqYWqxI,7136
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/studio_api.py,sha256=Cfsq8HFc4uUsj8hncnhnD_TLhw0cg-ryclGowj8S6Y0,26374
 lightning_sdk/api/teamspace_api.py,sha256=KYNyfx3aUYJyPeluM9iYphcIogBc--Bt3cV4IAgY7A4,11236
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=GbfTdqW1yV-fjyEwVebsbAdKQQjRmwT_fXUamrcnwAY,22534
+lightning_sdk/api/utils.py,sha256=pvDK_qUv8kgDTGumN2f7YcCy-HNnn_VUwZ5SbZTDgxA,22530
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/ai_hub.py,sha256=8oy6TogDiWnHuLT3cv33XEW7vPqXPA0dDMds8kX3Z4g,1649
+lightning_sdk/cli/delete.py,sha256=TePfbCUKR2Bzq3D-avAX6AATQAR0GfFIxj_Qw3qSczo,2431
 lightning_sdk/cli/download.py,sha256=nyQN3q1vZ0fg4_cfit8cKaokQ9VUd46l_TNcAQWkLwU,5996
-lightning_sdk/cli/entrypoint.py,sha256=dR27MRdqTGnC_rQcP9RHV4Zq4GL0DpjHUlkD418-HZk,1576
+lightning_sdk/cli/entrypoint.py,sha256=8CRA_jAg6CkHmFWUbrDQRRYd-oJccE6A7DPvNoAdGtk,1803
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
+lightning_sdk/cli/inspect.py,sha256=VeiazH-9DNn0SrQmP-ZM5JDdy8veQy5yxVe_Dj-KQWI,1349
+lightning_sdk/cli/job_and_mmt_action.py,sha256=OV_i5K3g-NnMUF8Sq6QliQdltnhGVRdtoyZqJVXu4Ss,1659
+lightning_sdk/cli/jobs_menu.py,sha256=6VqJecnyybYjdozj7MXH1gJehskJlwP8SoewQEXDz8s,2119
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
-lightning_sdk/cli/list.py,sha256=7xKZI3xnJmZo_7cWWdC7_n3XHq2sOb0EwgXNvFkp3vI,2106
-lightning_sdk/cli/run.py,sha256=B6ttd9SKg373ngug-lj74CEcuEoxwz-P6nUBVnQeijI,10836
+lightning_sdk/cli/list.py,sha256=sQzzwP6CxiL5YO4JZMgZ-THE4NII4h7mMl_9tmPZ_-8,2109
+lightning_sdk/cli/mmts_menu.py,sha256=OAwWywYeFSdmXBLR7raZlL7tpAhZld4r71LUHvmnZdA,2175
+lightning_sdk/cli/run.py,sha256=0jfijJCzUYON-OGgCljk6nhzN4o0pS2f8Il1LEOEPKk,11620
 lightning_sdk/cli/serve.py,sha256=UaXhGHU6nbAzrnVigSKOTrMjLwSs-sjyhuJCdVUBwzc,8722
+lightning_sdk/cli/stop.py,sha256=qBlVaYAJGxiNSwskxqB7zizTVPIpBGlJ4c7kIXII7iw,1373
 lightning_sdk/cli/studios_menu.py,sha256=0kQGqGel8gAbpdJtjOM1a6NEat_TnIqRNprNn8QiK58,3236
 lightning_sdk/cli/teamspace_menu.py,sha256=GVTrMAqxxL3INX83alo5wybnrl6rBBKENo01ZNkWPik,3753
 lightning_sdk/cli/upload.py,sha256=H9OyipYTYAQ9Mzy2e8jtoaa-B34-uXHbTQTzY2Vmhv4,9078
 lightning_sdk/deployment/__init__.py,sha256=BLu7_cVLp97TYxe6qe-J1zKUSZXAVcvCjgcA7plV2k4,497
 lightning_sdk/deployment/deployment.py,sha256=Dp15pn8rFAfMfaDhKn0v3bphFuvLgkPFs3KSNxW6eyc,15472
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
-lightning_sdk/job/base.py,sha256=bEhnFHrl3Ct5RNVqUCJRzzoF8NgXR6CMLyQiXBMIcSE,15526
-lightning_sdk/job/job.py,sha256=ZEyzlYtVCFeQJSiv0u6T5ShH5DVQZ3AGYnZ8dKoVAUo,12666
+lightning_sdk/job/base.py,sha256=CaysK2JiWTlTp7WJfC1Ebht2auCrMncgqn6EKVf-c_Q,15615
+lightning_sdk/job/job.py,sha256=JSd2UUvBJof_ZVaO-FWWbv-Jml7othL8ZSuP6F9Sp7Y,13425
 lightning_sdk/job/v1.py,sha256=pmu-adVC0fh4jKwXvjBQb9iIDKPuC5eOo0a2BiBq1fw,10184
-lightning_sdk/job/v2.py,sha256=SakYCmTMnG_uP35Fb-zwWoiO0LHNqnCKd-pSvKsg564,9860
+lightning_sdk/job/v2.py,sha256=5DhGK9KQhkRH-M3qzALhO83T1L0w9LEziaX4AFjL1Jo,10285
 lightning_sdk/job/work.py,sha256=pe1kWU3_GPuN5QlKIDOx_DKyzP1ARhJqY-ACKmklo8U,2313
 lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
@@ -53,7 +59,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=NSC53eJ24g59dySI9zRDl_-antxiflKDkn
 lightning_sdk/lightning_cloud/rest_client.py,sha256=iXuigyt3ZD9fYiCPAbQbV6Yzfs4eIDQAq6gsq82q3LI,6657
 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=vM_5AhUer_Oylf_q1Z6JrWGMsi8Gq_YiJ_lqHwZ9B5g,85952
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=vLo-D5ix-ujPnwDyi69GY82YOTc5k3CLj4gqD2epNiw,86157
 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
@@ -77,8 +83,8 @@ lightning_sdk/lightning_cloud/openapi/api/lightningapp_v2_service_api.py,sha256=
 lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py,sha256=dXKzbr4I7TDW-PFsqgiaaw4RtYrzB1X6NZB_hxgMnts,43747
 lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py,sha256=lpEo-LECmEMMM5EHFqFRFUtVRBmZgMhc0Zt3vZ5tmuE,76631
 lightning_sdk/lightning_cloud/openapi/api/lit_page_service_api.py,sha256=xf48UmWpzjs3woKVfFeyyqCLN13NA_qDepIwMw2SRUg,20457
-lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=OzlfVFyDe0-K87mTtaCGQZbdtePKVvgNfvQUqTViOCk,14562
-lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=JlBPgAPnP_4uzN3orcp21VdRxMYkL7m4IbSnLqGZEsw,93092
+lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=XJSRbTdtPDkq5XB0FDxK5Mbbf6KpSv7sRn1Bd4T1FVY,18996
+lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=J-rIl7uWoi6-eZR_tg9J2N0FktRGgQ3TltfSs-ByG9I,104643
 lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py,sha256=8xYTFSC4yDs32CTcEoWR1ycXX8chLQWRCsKAXJlOCls,88781
 lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py,sha256=hM2vTawBBuOYZrxN1SjZ6mU5ldvB_jUXRcLb0as0F4M,28286
 lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=a0x8UcufASV-CZzmIu-37iJiwhTlzalB07uctELD48Q,85065
@@ -90,7 +96,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=4SCpucxSLULv26jHjFJWoLXM7bySp2YeJnQayXZLWDU,47660
 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=wjdWKwf3zPIyaV7RpwB6QSqDHFX_sMh4ZD2DpfSujV8,66572
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=llwvhKbCskiT1VVXVxCfOIfJ0cw15qHFh3TLtr_TWw0,80994
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=Q6vEgJLooe1vC6Cp0TG5Olu58As0Wzz6l4dDCAFXgcs,81199
 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=iQsLbzMMvWhWMShSCA-pTIqZvfjJkUyb_mVnvVqgcXg,19633
@@ -251,7 +257,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_spec.py,sha256=abLbl
 lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_target_metric.py,sha256=4bjysD69mGA-V3oBTZcm9etEHHOUN0O9M7toGlAZGDI,4384
 lightning_sdk/lightning_cloud/openapi/models/v1_aws_cluster_credentials.py,sha256=I3WZlCoh1cd785BpHHi7CStQIhKJbkfNpD_STzFj1K8,5060
 lightning_sdk/lightning_cloud/openapi/models/v1_aws_data_connection.py,sha256=a5iQvOc2A7PZ2WMhxBJXEtByZhzcmNHy8ub4eZYYPCA,11811
-lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py,sha256=pb4RYhXIbnARqlr2mZpmP6q3OwWZR15jcsoIRKKrigg,12623
+lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py,sha256=1ED0R_Brkzo9Al_KuEmzuPKTp2dSzWKuF7OdXmt221g,13386
 lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1_status.py,sha256=3x5fFdoFa4Bjy5h1wBpphJ69PXJX7XRv8gG2rW3UzXM,5049
 lightning_sdk/lightning_cloud/openapi/models/v1_batch_update_lightningwork_response.py,sha256=uaAbQIuxzAtJCS0HgZfXrRyWn5L8-eMXydbFAhouxlg,3094
 lightning_sdk/lightning_cloud/openapi/models/v1_billing_feature.py,sha256=vkQ7QNTECZ0ZranpKUvugXPwRv6xDUMbBzMQ8yehiUs,6512
@@ -360,7 +366,6 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_capacity_reservat
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_encryption_key_response.py,sha256=F6QSFYZM8j9rDtOpwzbVEH17NTCx5qSlvT_UjcdeIAE,3106
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_proxy_response.py,sha256=0YtYHebFVutcWQMuY9To1Eh7zfRe9wFxb9RkLL8Nav0,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_response.py,sha256=CCQd_OpnjAPaTP0Q3Uxw8nba58ZTnmkF-o2CEfwa9ec,3028
-lightning_sdk/lightning_cloud/openapi/models/v1_delete_container_response.py,sha256=6lYTA3fbtx1cTFjAtagGYa_42t1-wAwspT2djmZCbhY,3040
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_conversation_response.py,sha256=9X3MUKE0C8sqYP48lm-f0aN0AGe3I4H7IDUSbHprfa8,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_data_connection_response.py,sha256=DsGzX9XH4d9Zaz8MPqrfN90oYzYjxK3swK4Styu7SbE,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_dataset_response.py,sha256=3e9VCGV7FjV8qzvsXU3-SZi7Lj72Bc-qJ07x-69Ek7U,3028
@@ -375,6 +380,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_lightningapp_instance_art
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lightningapp_instance_response.py,sha256=eS11kIEnPNW5KvG-MmMMFR7ynxqpV7j5_gVPfVgMEbA,3106
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lightningwork_response.py,sha256=jAkzaCenELp4Lg-wtBKGHmacF0JPN2YSBYV8_B46tHQ,3064
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_page_response.py,sha256=dPBEV8PAH6ar5spg3cCa_hPMlajJcJ9ourbdoCku4XQ,3028
+lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_repository_response.py,sha256=iYXqZlUrb-Xljvl4awlPvOmYusGZGdFJDXg56ec30i4,3064
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_logger_artifact_response.py,sha256=bUf_tyaYQjyhQ2Wiq8D0dZWU6yl9knq_uQDJcHQ0ImM,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_managed_endpoint_response.py,sha256=McRms2Ag2FZOUa39fJErUihEQXSia8RYIpPrGm5mZMw,3076
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_metrics_stream_response.py,sha256=s3dpSMmaI5pd5FDCYG_LhO6Idmk-ZRkbbvIqTppY4jw,3064
@@ -495,7 +501,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_user_notification_preference
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=z-3_OOHg830ETJPAnIS69yHF1LJcIRgjzVs2qForT2A,30480
 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=gYnzf2hVpl7Kc5Fe8EBlSx2fJRBwDnsvkg4bCMkFfek,11861
+lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=ITH50lGG-Xx2kMrgrAZxgqbErzhMTwJfABId4gXVAdU,13286
 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
@@ -652,7 +658,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_metrics.py,sha256=dFyPDXpq71deEC
 lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py,sha256=JYdwxzFmOfNQ26WhY9CiUNA4j0V2lhtzBx2VTQ7JCyA,22831
 lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tags.py,sha256=pqSxM8PtvGzL8YShurjbOaSVSr45XuhrpINWom40i5Y,5557
 lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tracker.py,sha256=b6ihJRbKPrIuN8SUnTRqNwlBbz45vxlIOtMEmWnl2cg,10996
-lightning_sdk/lightning_cloud/openapi/models/v1_model.py,sha256=eCB-tr1-a6pj8q2MCUEa2nGriRE93Z_lEZty_xpaTDY,13433
+lightning_sdk/lightning_cloud/openapi/models/v1_model.py,sha256=P29zYfYzAqNONnCo211Dx8LzZjaqmDV5pxuZixU5ZEE,14341
 lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py,sha256=MWdpC7lfwFWioi2yL2Ajdnz4yYwWBWxdiQiXYADcfqg,4805
 lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py,sha256=dZ_SM_IKbyRxjRSAmfdBWkLEMIWAGLQxYxBmwIjEUcM,12857
 lightning_sdk/lightning_cloud/openapi/models/v1_mount_target.py,sha256=7hWm0_z3weM2V6j900nUpCMceF3q-hXH1tVI1rama2U,5823
@@ -809,7 +815,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=Dch5r7uQWI98sBKronVzlelWQnBeU2utrh8PoYIn6k0,75514
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=2KlY1j_YxyEb228z8YLiDssXc1FFfwUFKk62X7T6RFE,73647
 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
@@ -823,12 +829,14 @@ lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_respon
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_model_response.py,sha256=LshfheXdKDqRYgg5Sa5dV0L9CW3BVfFts_f2qqmnH7k,3767
 lightning_sdk/lightning_cloud/openapi/models/v1_verify_verification_response.py,sha256=VQrBJk2pQQJuu-1Om-kvKr5N9ioxEPeV2l70udHEWSw,3058
 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=ZVKoIqzl0q4BdG5Ksf6iuJFzjwcdVZU16nGdHGS5rQ4,4073
+lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py,sha256=jq4ODNqWXJqkkyMdod6C7eiuWDPAfqMC1sEI6DSJDH0,5594
 lightning_sdk/lightning_cloud/openapi/models/v1_work.py,sha256=MwXPSG8TE_CcGiffx6OI6U7kyf1NnSsDvJEM6LuwcYA,4910
 lightning_sdk/lightning_cloud/openapi/models/validate.py,sha256=wrCNhZpBR2PiA_a56C01w5fuxOSSGkbEh3Y8iMwUcYs,8086
 lightning_sdk/lightning_cloud/openapi/models/validateautojoindomain_domain_body.py,sha256=gRI7XdvbY7w09gIH6BkFai8UA-y71T2AmiR5Y2XcTe8,3781
+lightning_sdk/lightning_cloud/openapi/models/version_default_body.py,sha256=cioI_KsxCMJncr_5Y98odN_28EP7aOOiR0lPHpZtKx8,4385
 lightning_sdk/lightning_cloud/openapi/models/version_uploads_body.py,sha256=tj_gdZ7gBd0CPuT8eZv-p5taH7XfVohUUhEFqj4xA3w,3701
 lightning_sdk/lightning_cloud/openapi/models/versions_id_body.py,sha256=z4tAS_ziEH7gbpdQaUt0BB-R25BX2zoz0UFtHY5mRwc,10637
+lightning_sdk/lightning_cloud/openapi/models/versions_version_body.py,sha256=o8q2jgCYM2gW3D2IwbbubHpEjKFiVfgOqoMZ4s0mnZg,3631
 lightning_sdk/lightning_cloud/openapi/models/works_id_body.py,sha256=6xpkaHkN9PgYh0Skp-2m2XLNzD4WaELIK8zCAPS3ams,3647
 lightning_sdk/lightning_cloud/source_code/__init__.py,sha256=NaXJ1UtzKePzkLLE7d6Yg4bsIafYZPsV-noObTnHAno,113
 lightning_sdk/lightning_cloud/source_code/copytree.py,sha256=vF5TLd-MBsPlbulu7l1JAHCqUpnxI1vNrQRqG_ErexQ,6272
@@ -838,15 +846,15 @@ lightning_sdk/lightning_cloud/source_code/logs_socket_api.py,sha256=ngpv70IayG8M
 lightning_sdk/lightning_cloud/source_code/tar.py,sha256=mSKRWEX1SIAbXFcESNrQAqWSVULv4G0jgcvR_o2KJ5g,5813
 lightning_sdk/lightning_cloud/source_code/uploader.py,sha256=CQrWV6JNDiglyRfhc42ht4GWFtgRz_fBxw4tOCERzco,3207
 lightning_sdk/lightning_cloud/utils/__init__.py,sha256=kSNKoybELDJHs9WeCf279V5JtzOf1VqfKODlKYr9uRo,115
-lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=YpyQlJFGtgZsuCdjv_0iuyKh-6NzU-6rlzSEBBXALAs,6751
+lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=D5ZmvFbfzNSvmlX9MQmn1ILYcd5ExbS5kIVf7hWESyU,6774
 lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgydm78kJzhOIx3C9qKls,2011
 lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
 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=yXmussF0qEh7S9KeYcKo7xzBlIAg1DCjgdf1iWc-L6M,14489
-lightning_sdk/mmt/mmt.py,sha256=fJL_cxXnoN10NZXMqwbAOlN7f_TULdZjmz7PGoM7GXo,13369
-lightning_sdk/mmt/v1.py,sha256=1hMJVXMcmG1k4zM8sN44uOddHkuJnX1Wjl_WVz-Ox5Q,9166
-lightning_sdk/mmt/v2.py,sha256=RY5W172aqVqcKbXJCdY5mdk7r201IAsxqtwG1KuEhP4,9474
+lightning_sdk/mmt/mmt.py,sha256=v-dOMS1bqUGoGYIe45YZJc0aXPUDsUGNdRoeu8DMw60,14029
+lightning_sdk/mmt/v1.py,sha256=3c28zivn2ZVhp4uXsyNwTpa4nN3vU021AyR6PCCw5_g,9255
+lightning_sdk/mmt/v2.py,sha256=LUwNSxeJVorCkJllRhFGtDgjGRBAsx81lLTaJWSFXHQ,9589
 lightning_sdk/services/__init__.py,sha256=gSWUjccEhMI9CIWL_nbrFHUK2S6TM2725mEzrLMfK1Y,225
 lightning_sdk/services/file_endpoint.py,sha256=we5HC_o74J4Y6fSP_31jIizi_I_1FO_Rb2qblspD9eE,7855
 lightning_sdk/services/utilities.py,sha256=IeOx8hc3F8ZevHeKBysh08BXhJliTNzvKp1gwpEfdik,4087
@@ -855,9 +863,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=RWvlOWLHjaHhR0W0zT3mN719cbzhFfYCKBss38zfv3k,5783
-lightning_sdk-0.1.47.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.47.dist-info/METADATA,sha256=mgPc8AxO4hg3mbNN21lwLtjkrF9JtSjzgUNU2oTKRns,4031
-lightning_sdk-0.1.47.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.47.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.47.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.47.dist-info/RECORD,,
+lightning_sdk-0.1.48.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.48.dist-info/METADATA,sha256=8eqGkORoyWrF3suijKWe2UnwH71uucciC5ajYlGemzk,4031
+lightning_sdk-0.1.48.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.48.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.48.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.48.dist-info/RECORD,,

lightning_sdk/api/lit_registry_api.py

@@ -1,12 +0,0 @@
-from typing import List
-
-from lightning_sdk.lightning_cloud.rest_client import LightningClient
-
-
-class LitRegistryApi:
-    def __init__(self) -> None:
-        self._client = LightningClient(max_tries=3)
-
-    def list_containers(self, project_id: str) -> List:
-        project = self._client.lit_registry_service_get_lit_project_registry(project_id)
-        return project.repositories