lightning-sdk 2025.12.1__py3-none-any.whl → 2025.12.5__py3-none-any.whl

lightning_sdk/__version__.py

@@ -1,3 +1,3 @@
 """Version information for lightning_sdk."""
 
-__version__ = "2025.12.01"
+__version__ = "2025.12.05"

lightning_sdk/api/studio_api.py

@@ -1,6 +1,7 @@
 import json
 import os
 import time
+import warnings
 from pathlib import Path
 from threading import Event, Thread
 from typing import Any, Dict, Generator, List, Mapping, Optional, Tuple, Union
@@ -659,6 +660,34 @@ class StudioApi:
         )
         return r.json()
 
+    def get_path_info(self, studio_id: str, teamspace_id: str, path: str = "") -> dict:
+        path = path.strip("/")
+
+        if "/" in path:
+            parent_path = path.rsplit("/", 1)[0]
+            target_name = path.rsplit("/", 1)[1]
+        else:
+            if path == "":
+                # root directory
+                return {"exists": True, "type": "directory", "size": None}
+            parent_path = ""
+            target_name = path
+
+        tree = self.get_tree(studio_id, teamspace_id, path=parent_path)
+        tree_items = tree.get("tree", [])
+        for item in tree_items:
+            item_name = item.get("path", "")
+            if item_name == target_name:
+                item_type = item.get("type")
+                # if type == "blob" it's a file, if "tree" it's a directory
+                return {
+                    "exists": True,
+                    "type": "file" if item_type == "blob" else "directory",
+                    "size": item.get("size", 0) if item_type == "blob" else None,
+                }
+        warnings.warn(f"If '{path}' is a directory, it may be empty and thus not detected.")
+        return {"exists": False, "type": None, "size": None}
+
     def upload_file(
         self,
         studio_id: str,

lightning_sdk/cli/entrypoint.py

@@ -21,6 +21,7 @@ from lightning_sdk.cli.groups import (
 from lightning_sdk.cli.utils import CustomHelpFormatter
 from lightning_sdk.cli.utils.logging import CommandLoggingGroup, logging_excepthook
 from lightning_sdk.lightning_cloud.login import Auth
+from lightning_sdk.utils.resolve import _get_authed_user, in_studio
 
 
 @click.group(
@@ -39,13 +40,22 @@ main_cli.context_class.formatter_class = CustomHelpFormatter
 @main_cli.command
 def login() -> None:
     """Login to Lightning AI Studios."""
-    auth = Auth()
-    auth.clear()
-
-    try:
-        auth.authenticate()
-    except ConnectionError:
-        raise RuntimeError(f"Unable to connect to {_cloud_url()}. Please check your internet connection.") from None
+    if in_studio():
+        # if inside a Studio, no need to run login command
+        try:
+            auth_user = _get_authed_user()
+        except Exception:
+            raise RuntimeError("Unable to identify user within a Studio") from None
+
+        click.echo(f'You are currently logged in as "{auth_user.name}"')
+        click.echo('"lightning login" is not required within a Studio')
+    else:
+        auth = Auth()
+        auth.clear()
+        try:
+            auth.authenticate()
+        except ConnectionError:
+            raise RuntimeError(f"Unable to connect to {_cloud_url()}. Please check your internet connection.") from None
 
 
 @main_cli.command

lightning_sdk/cli/studio/cp.py

@@ -1,5 +1,6 @@
 """Studio cp command."""
 
+import os
 from pathlib import Path
 from typing import Optional, TypedDict
 
@@ -7,7 +8,6 @@ import click
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.cli.utils.owner_selection import OwnerMenu
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
@@ -38,7 +38,7 @@ def cp_impl(source: str, destination: str) -> None:
         raise ValueError("Either source or destination must be a Studio file.")
     elif "lit://" in source:
         # Download from Studio to local
-        cp_download(studio_file_path=source, local_file_path=destination)
+        cp_download(studio_path=source, local_path=destination)
     else:
         # Upload from local to Studio
         cp_upload(local_file_path=source, studio_file_path=destination)
@@ -92,12 +92,8 @@ def cp_upload(
     studio_file_path: str,
 ) -> None:
     console = Console()
-    if Path(local_file_path).is_dir():
-        raise StudioCliError(
-            f"The provided path is a folder: {local_file_path}. Use `lightning upload folder` instead."
-        )
     if not Path(local_file_path).exists():
-        raise FileNotFoundError(f"The provided path does not exist: {local_file_path}.")
+        raise FileNotFoundError(f"The provided path does not exist: {local_file_path}")
 
     studio_path_result = parse_studio_path(studio_file_path)
 
@@ -106,7 +102,14 @@ def cp_upload(
     )
     console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
-    selected_studio.upload_file(local_file_path, studio_path_result["destination"])
+    if Path(local_file_path).is_dir():
+        selected_studio.upload_folder(local_file_path, studio_path_result["destination"])
+    else:
+        if studio_file_path.endswith(("/", "\\")):
+            # if destination ends with / or \, treat it as a directory
+            file_name = os.path.basename(local_file_path)
+            studio_path_result["destination"] = os.path.join(studio_path_result["destination"], file_name)
+        selected_studio.upload_file(local_file_path, studio_path_result["destination"])
 
     studio_url = (
         _get_cloud_url().replace(":443", "")
@@ -121,10 +124,43 @@ def cp_upload(
 
 
 def cp_download(
-    studio_file_path: str,
-    local_file_path: str,
+    studio_path: str,
+    local_path: str,
 ) -> None:
-    raise NotImplementedError("Download functionality is not implemented yet.")
+    console = Console()
+    studio_path_result = parse_studio_path(studio_path)
+
+    selected_studio = resolve_studio(
+        studio_path_result["studio"], studio_path_result["teamspace"], studio_path_result["owner"]
+    )
+
+    # check if file/folder exists
+    path_info = selected_studio._studio_api.get_path_info(
+        selected_studio._studio.id, selected_studio._teamspace.id, path=studio_path_result["destination"]
+    )
+    if not path_info["exists"]:
+        raise FileNotFoundError(
+            f"The provided path does not exist in the studio: {studio_path_result['destination']} "
+            "Note that empty folders may not be detected as existing."
+        )
+
+    console.print(f"Downloading from {selected_studio.teamspace.name}/{selected_studio.name}")
+    if path_info["type"] == "directory":
+        folder_name = os.path.basename(studio_path_result["destination"].rstrip("/"))
+        target_path = os.path.join(local_path, folder_name)
+
+        selected_studio.download_folder(studio_path_result["destination"], target_path)
+        console.print(f"See your folder at {target_path}")
+    else:
+        if os.path.isdir(local_path) or local_path.endswith(("/", "\\")):
+            # if local_path ends with / or \ or is a directory, treat it as a directory
+            file_name = os.path.basename(studio_path_result["destination"])
+            target_path = os.path.join(local_path, file_name)
+        else:
+            target_path = local_path
+        os.makedirs(os.path.dirname(target_path), exist_ok=True)
+        selected_studio.download_file(studio_path_result["destination"], target_path)
+        console.print(f"See your file at {target_path}")
 
 
 def resolve_studio(studio_name: Optional[str], teamspace: Optional[str], owner: Optional[str]) -> Studio:

lightning_sdk/studio.py

@@ -570,31 +570,6 @@ class Studio(metaclass=TrackCallsMeta):
             raise RuntimeError(output)
         return output
 
-    def get_path_info(self, path: str = "") -> dict:
-        path = path.strip("/")
-
-        if "/" in path:
-            parent_path = path.rsplit("/", 1)[0]
-            target_name = path.rsplit("/", 1)[1]
-        else:
-            parent_path = ""
-            target_name = path
-
-        tree = self._studio_api.get_tree(self._studio.id, self._teamspace.id, path=parent_path)
-        tree_items = tree.get("tree", [])
-        for item in tree_items:
-            item_name = item.get("path", "")
-            if item_name == target_name:
-                item_type = item.get("type")
-                # if type == "blob" it's a file, if "tree" it's a directory
-                return {
-                    "exists": True,
-                    "type": "file" if item_type == "blob" else "directory",
-                    "size": item.get("size", 0) if item_type == "blob" else None,
-                }
-        warnings.warn(f"If '{path}' is a directory, it may be empty and thus not detected.")
-        return {"exists": False, "type": None, "size": None}
-
     def upload_file(self, file_path: str, remote_path: Optional[str] = None, progress_bar: bool = True) -> None:
         """Uploads a given file to a remote path on the Studio."""
         if remote_path is None:

lightning_sdk/teamspace.py

@@ -498,7 +498,9 @@ class Teamspace(metaclass=TrackCallsMeta):
             )
         return self._teamspace_api.list_model_versions(teamspace_id=self.id, model_name=name)
 
-    def upload_file(self, file_path: str, remote_path: Optional[str] = None, progress_bar: bool = True) -> None:
+    def upload_file(
+        self, file_path: Union[Path, str], remote_path: Optional[str] = None, progress_bar: bool = True
+    ) -> None:
         """Uploads file to given remote path in the Teamspace drive."""
         if remote_path is None:
             remote_path = os.path.split(file_path)[1]

{lightning_sdk-2025.12.1.dist-info → lightning_sdk-2025.12.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.12.1
+Version: 2025.12.5
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -46,7 +46,7 @@ Requires-Dist: rich
 Requires-Dist: simple-term-menu
 Requires-Dist: six
 Requires-Dist: tqdm
-Requires-Dist: urllib3
+Requires-Dist: urllib3<=2.5.0
 Requires-Dist: uvicorn
 Requires-Dist: websocket-client
 Requires-Dist: wget

{lightning_sdk-2025.12.1.dist-info → lightning_sdk-2025.12.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
 lightning_sdk/__init__.py,sha256=ym1mOWljHxiO4oijOrmtcjf-eIcwYJVFXkDvLduL4FA,1368
-lightning_sdk/__version__.py,sha256=zdvYNGqKWtSI3YmWyIquULLdGllW491fyQ4wWDLEMD0,73
+lightning_sdk/__version__.py,sha256=xyg__GnThH4g-rLWswr2bg7RmNEA67sg5iBKzDfb9do,73
 lightning_sdk/agents.py,sha256=Uqsdu4PgswhpeXZzDSbp3hu30MMOhSlCnp8D56MaS_c,1658
 lightning_sdk/ai_hub.py,sha256=QzCfmX0h2CLn7r0DQyjWg6RnNw26Ey6akiP2Wffl9Ls,7006
 lightning_sdk/base_studio.py,sha256=yMvZ1dDq_avW3iu98CLEo9W9UL54mx4C6FC43IKLWAE,4691
@@ -17,8 +17,8 @@ lightning_sdk/plugin.py,sha256=yaQrIx2Tt18R0CoarJQAtB_AwaVa436f-eC5rb9yWe4,15579
 lightning_sdk/sandbox.py,sha256=_NvnWotEXW2rBiVFZZ4krKXxVjuAqfNh04qELSM0-Pg,5786
 lightning_sdk/serve.py,sha256=XbllHwmOAOHjQJ0AQukwYe3OtspEwhmf5-_HXhL-mpc,11908
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
-lightning_sdk/studio.py,sha256=yQY2q7zjqYKVCohYeWg96dDfWD1rJ-NEDUwgPZNDpi8,38660
-lightning_sdk/teamspace.py,sha256=pskDJWe3OGKFeXCKr-QZbzgor2AOQwdzgxqlS8Ahj_o,27558
+lightning_sdk/studio.py,sha256=Yx8vw_-SPmr0UzWQ7Kq_IXDFtuk-j0UhM95-Fyaexjc,37609
+lightning_sdk/teamspace.py,sha256=_6zrKLgoOjNdrDARt9aAvFG_oTZmZJclM4uTGVAj_0o,27585
 lightning_sdk/user.py,sha256=TSYh38rxoi7qKOfrK2JYh_Nknya2Kbz2ngDIY85fFOY,1778
 lightning_sdk/api/__init__.py,sha256=8boDRZ0LMl7G_IbpqHS2q3efeX_Y897_gsJ3iBF58M4,538
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
@@ -34,12 +34,12 @@ lightning_sdk/api/llm_api.py,sha256=8ouowlX-Ld0hRvueIS6fQqaEJLQMYskvhKV5-Z_A6gE,
 lightning_sdk/api/mmt_api.py,sha256=yI_iFtxGd1g0pW82xQWY80jSYVmF_xU-eT1-LgUA-KI,10815
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/pipeline_api.py,sha256=rJYp_FN7uUjC5xbc6K67l2eRSmVuOkijd5i8Nm5BF7I,4621
-lightning_sdk/api/studio_api.py,sha256=WH1wihS3FPfjsvXFdWrN893TOypprERAVdVQ9cnhMuI,41281
+lightning_sdk/api/studio_api.py,sha256=5ga510pDvxnEjc0zCLzp9Yo2GmKnA5Ppm0-Ub_YNUXs,42494
 lightning_sdk/api/teamspace_api.py,sha256=wuwW4YbhloFqI9rS_5mbgsS6Dk3zYONLv9M6OoqyDg8,19980
 lightning_sdk/api/user_api.py,sha256=o9gZmtvqNfj4kdSpo2fyyRuFAP7bM5w7mx0Oj9__Ads,4702
 lightning_sdk/api/utils.py,sha256=mB8ZPjxj8jjXMWRBtG-g0ZvR7NayZnGhMtEZfQMHtqA,29734
 lightning_sdk/cli/__init__.py,sha256=lksw08t_ZIuHOH47LCIqSVHeZ8cUXI2aJWHYhyujYHM,32
-lightning_sdk/cli/entrypoint.py,sha256=WsHbYB-rGIbXwz_Un6UiuOFdVPwXGTOs7GC5Zh4Y_B4,3467
+lightning_sdk/cli/entrypoint.py,sha256=X6yWMfC5wSvO5L8g2JVKlral9r5TlAeEPfQ3ySbyp2g,3959
 lightning_sdk/cli/groups.py,sha256=umtnKZm4xrIEtY-SWrC1KcKzYGvArr08qeP2zGPkyTk,1609
 lightning_sdk/cli/base_studio/__init__.py,sha256=npBEmXWNVuiawm98TFVNHZf038kD4aW3gzVHsjbUPNI,272
 lightning_sdk/cli/base_studio/list.py,sha256=-psOgNpJY7_x5iGRw8UeK4r3jxryVidy4zl6UrnFWx8,1157
@@ -84,7 +84,7 @@ lightning_sdk/cli/license/set.py,sha256=uGjFgq-3jkMF0IstMIIRjKn8GDzg9TIe1OZcRkDT
 lightning_sdk/cli/mmt/__init__.py,sha256=CLgr-ZHHLS6Db_JGqpxbn4G2pYrKi4Qn-uhi8e0kFNc,145
 lightning_sdk/cli/studio/__init__.py,sha256=wREPAmPJdN4b51r6RRD1o4aE212KN7xI88Z0O6HTzEU,1022
 lightning_sdk/cli/studio/connect.py,sha256=RLi5VumsVlVDlyiGbST7lDuHAntOj_-99KsNr8J3knE,5057
-lightning_sdk/cli/studio/cp.py,sha256=qQn803KmK6AR4dIjLgLuOKU8VbC8_FW7eMxAB_lwgno,4765
+lightning_sdk/cli/studio/cp.py,sha256=KZWpS_-eeQQGIvMWL1d8W-u9T1v-zWZmgTPzjguQO48,6534
 lightning_sdk/cli/studio/create.py,sha256=eVkuP1qg1aQSe2jyDgbIg-si5xPl8CwDIPcyvkUEFnQ,3179
 lightning_sdk/cli/studio/delete.py,sha256=Gi1KLCrSgjTiyoQe9NRvF4ZpsNyeM07KHnBxfdJg2FE,1483
 lightning_sdk/cli/studio/list.py,sha256=ieLiHRIfAb7E5e_r48BeTC03ib8RY6maeF7nkmrqaek,3301
@@ -1281,9 +1281,9 @@ lightning_sdk/utils/logging.py,sha256=WPyOx7KAn8OpRKqoDDlB7MkORtLDryJsj1WVXLYNqy
 lightning_sdk/utils/names.py,sha256=1EuXbIh7wldkDp1FG10oz9vIOyWrpGWeFFVy-DQBgzA,18162
 lightning_sdk/utils/progress.py,sha256=bLWw39fzq29PMWoFXaPIVfoS3Ug245950oWOFJ2ZaiU,12596
 lightning_sdk/utils/resolve.py,sha256=ukC-Zn35gNZV-fQ-ZyUgZkRPFb8nwsFh_aN7YcJ0sl8,10613
-lightning_sdk-2025.12.1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.12.1.dist-info/METADATA,sha256=PsDUWvnj0fJm-PE18cVr4cxzyQC1vboZY4Z_Z_KfENc,4152
-lightning_sdk-2025.12.1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.12.1.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
-lightning_sdk-2025.12.1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.12.1.dist-info/RECORD,,
+lightning_sdk-2025.12.5.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.12.5.dist-info/METADATA,sha256=t88SaZclJyHWhv-FJKf3XYreHtzb8iVZz91lPxMw5qk,4159
+lightning_sdk-2025.12.5.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.12.5.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
+lightning_sdk-2025.12.5.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.12.5.dist-info/RECORD,,