lightning-sdk 2025.12.17__py3-none-any.whl → 2026.1.22__py3-none-any.whl

lightning_sdk/cli/legacy/run.py

@@ -50,6 +50,15 @@ def run() -> 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-provider",
+    "--cloud_provider",
+    default=None,
+    help=(
+        "The provider to create the studio on. If set, must be in agreement with the provider from the "
+        "cloud_account (if specified)."
+    ),
+)
 @click.option(
     "--cloud-account",
     "--cloud_account",
@@ -96,7 +105,7 @@ def run() -> None:
 )
 @click.option(
     "--entrypoint",
-    default="sh -c",
+    default=None,
     show_default=True,
     help=(
         "The entrypoint of your docker container. "
@@ -141,6 +150,7 @@ def job(
     teamspace: Optional[str] = None,
     org: Optional[str] = None,
     user: Optional[str] = None,
+    cloud_provider: Optional[str] = None,
     cloud_account: Optional[str] = None,
     env: Sequence[str] = (),
     interruptible: bool = False,
@@ -184,6 +194,7 @@ def job(
         teamspace=resolved_teamspace,
         org=org,
         user=user,
+        cloud_provider=cloud_provider,
         cloud_account=cloud_account,
         env=env_dict,
         interruptible=interruptible,
@@ -289,7 +300,7 @@ def job(
 )
 @click.option(
     "--entrypoint",
-    default="sh -c",
+    default=None,
     show_default=True,
     help=(
         "The entrypoint of your docker container. "

lightning_sdk/cli/studio/__init__.py

@@ -10,6 +10,8 @@ def register_commands(group: click.Group) -> None:
     from lightning_sdk.cli.studio.create import create_studio
     from lightning_sdk.cli.studio.delete import delete_studio
     from lightning_sdk.cli.studio.list import list_studios
+    from lightning_sdk.cli.studio.ls import ls_studio
+    from lightning_sdk.cli.studio.rm import rm_studio_file
     from lightning_sdk.cli.studio.ssh import ssh_studio
     from lightning_sdk.cli.studio.start import start_studio
     from lightning_sdk.cli.studio.stop import stop_studio
@@ -24,3 +26,5 @@ def register_commands(group: click.Group) -> None:
     group.add_command(switch_studio)
     group.add_command(connect_studio)
     group.add_command(cp_studio_file)
+    group.add_command(ls_studio)
+    group.add_command(rm_studio_file)

lightning_sdk/cli/studio/cp.py

@@ -2,22 +2,20 @@
 
 import os
 from pathlib import Path
-from typing import Optional, TypedDict
+from typing import Optional
 
 import click
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_cloud_url
-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
-from lightning_sdk.studio import Studio
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
 
 
 @click.command("cp")
 @click.argument("source", nargs=1)
 @click.argument("destination", nargs=1)
-def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = None) -> None:
+@click.option("-r", "--recursive", is_flag=True, help="Copy directories recursively")
+def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = None, recursive: bool = False) -> None:
     """Copy a Studio file.
 
     SOURCE: Source file to copy from. For Studio files, use the format lit://<owner>/<my-teamspace>/studios/<my-studio>/<filepath>.
@@ -26,70 +24,29 @@ def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = Non
 
     Example:
         lightning studio cp source.txt lit://<owner>/<my-teamspace>/studios/<my-studio>/destination.txt
+        lightning studio cp -r source_folder/ lit://<owner>/<my-teamspace>/studios/<my-studio>/destination_folder/
 
     """
-    return cp_impl(source=source, destination=destination)
+    return cp_impl(source=source, destination=destination, recursive=recursive)
 
 
-def cp_impl(source: str, destination: str) -> None:
+def cp_impl(source: str, destination: str, recursive: bool = False) -> None:
     if "lit://" in source and "lit://" in destination:
         raise ValueError("Both source and destination cannot be Studio files.")
     elif "lit://" not in source and "lit://" not in destination:
         raise ValueError("Either source or destination must be a Studio file.")
     elif "lit://" in source:
         # Download from Studio to local
-        cp_download(studio_path=source, local_path=destination)
+        cp_download(studio_path=source, local_path=destination, recursive=recursive)
     else:
         # Upload from local to Studio
-        cp_upload(local_file_path=source, studio_file_path=destination)
-
-
-class StudioPathResult(TypedDict):
-    owner: Optional[str]
-    teamspace: Optional[str]
-    studio: Optional[str]
-    destination: Optional[str]
-
-
-def parse_studio_path(studio_path: str) -> StudioPathResult:
-    path_string = studio_path.removeprefix("lit://")
-    if not path_string:
-        raise ValueError("Studio path cannot be empty after prefix")
-
-    result: StudioPathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
-
-    if "/studios/" in path_string:
-        prefix_part, suffix_part = path_string.split("/studios/", 1)
-
-        # org and teamspace
-        if prefix_part:
-            org_ts_components = prefix_part.split("/")
-            if len(org_ts_components) == 2:
-                result["owner"], result["teamspace"] = org_ts_components
-            elif len(org_ts_components) == 1:
-                result["teamspace"] = org_ts_components[0]
-            else:
-                raise ValueError(f"Invalid format: '{prefix_part}'")
-
-        # studio and destination
-        path_parts = suffix_part.split("/")
-
-    else:
-        # studio and destination
-        path_parts = path_string.split("/")
-
-    if not path_parts or len(path_parts) < 2:
-        raise ValueError("Invalid: Missing studio name.")
-
-    result["studio"] = path_parts[0]
-    result["destination"] = "/".join(path_parts[1:])
-
-    return result
+        cp_upload(local_file_path=source, studio_file_path=destination, recursive=recursive)
 
 
 def cp_upload(
     local_file_path: str,
     studio_file_path: str,
+    recursive: bool = False,
 ) -> None:
     console = Console()
     if not Path(local_file_path).exists():
@@ -103,6 +60,8 @@ def cp_upload(
     console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
     if Path(local_file_path).is_dir():
+        if not recursive:
+            raise ValueError(f"'{local_file_path}' is a directory. Use -r flag to copy directories recursively.")
         selected_studio.upload_folder(local_file_path, studio_path_result["destination"])
     else:
         if studio_file_path.endswith(("/", "\\")):
@@ -126,6 +85,7 @@ def cp_upload(
 def cp_download(
     studio_path: str,
     local_path: str,
+    recursive: bool = False,
 ) -> None:
     console = Console()
     studio_path_result = parse_studio_path(studio_path)
@@ -146,7 +106,14 @@ def cp_download(
 
     console.print(f"Downloading from {selected_studio.teamspace.name}/{selected_studio.name}")
     if path_info["type"] == "directory":
+        if not recursive:
+            raise ValueError(
+                f"'{studio_path_result['destination']}' is a directory. Use -r flag to copy directories recursively."
+            )
         folder_name = os.path.basename(studio_path_result["destination"].rstrip("/"))
+        if folder_name == "" and local_path in ("./", "."):
+            # handle root directory case (e.g. lit://lightning-ai/gpt-oss/studios/manual-lime-ylu2/)
+            folder_name = selected_studio.name
         target_path = os.path.join(local_path, folder_name)
 
         selected_studio.download_folder(studio_path_result["destination"], target_path)
@@ -161,14 +128,3 @@ def cp_download(
         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:
-    owner_menu = OwnerMenu()
-    resolved_owner = owner_menu(owner=owner)
-
-    teamspace_menu = TeamspacesMenu(resolved_owner)
-    resolved_teamspace = teamspace_menu(teamspace=teamspace)
-
-    studio_menu = StudiosMenu(resolved_teamspace)
-    return studio_menu(studio=studio_name)

lightning_sdk/cli/studio/ls.py

@@ -0,0 +1,57 @@
+"""Studio ls command."""
+
+import click
+
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+
+
+@click.command("ls")
+@click.argument("path", nargs=1)
+def ls_studio(path: str) -> None:
+    """List contents of a directory in Studio.
+
+    PATH: Studio path in the format
+            lit://<owner>/<teamspace>/studios/<studio>/<directory-path>
+
+    Example:
+        lightning studio ls lit://<owner>/<my-teamspace>/studios/<my-studio>/data
+
+    """
+    return ls_impl(path=path)
+
+
+def ls_impl(path: str) -> None:
+    if not path.startswith("lit://"):
+        raise ValueError("Path must be a Studio path starting with 'lit://'.")
+
+    studio_path_result = parse_studio_path(path)
+    selected_studio = resolve_studio(
+        studio_path_result["studio"], studio_path_result["teamspace"], studio_path_result["owner"]
+    )
+
+    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."
+        )
+
+    if path_info["type"] == "file":
+        # print the file name if it's a file (bash-like behavior)
+        print(studio_path_result["destination"])
+        return
+
+    tree = selected_studio._studio_api.get_tree(
+        selected_studio._studio.id, selected_studio._teamspace.id, path=studio_path_result["destination"]
+    )
+
+    tree_items = tree.get("tree", [])
+
+    for item in tree_items:
+        name = item.get("path", "")
+        if item.get("type") == "tree":
+            name += "/"
+        print(name)

lightning_sdk/cli/studio/rm.py

@@ -0,0 +1,71 @@
+"""Studio rm command."""
+
+import click
+from rich.console import Console
+
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+from lightning_sdk.studio import Studio
+
+
+@click.command("rm")
+@click.argument("path", nargs=1)
+@click.option("-r", "--recursive", is_flag=True, help="Remove directories recursively")
+@click.option("-f", "--force", is_flag=True, help="Ignore nonexistent files, never prompt")
+def rm_studio_file(path: str, recursive: bool = False, force: bool = False) -> None:
+    """Remove a Studio file or directory.
+
+    PATH: Studio path to remove. Use the format lit://<owner>/<my-teamspace>/studios/<my-studio>/<filepath>.
+
+    Example:
+        lightning studio rm lit://<owner>/<my-teamspace>/studios/<my-studio>/file.txt
+        lightning studio rm -r lit://<owner>/<my-teamspace>/studios/<my-studio>/folder/
+
+    """
+    return rm_impl(path=path, recursive=recursive, force=force)
+
+
+def rm_impl(path: str, recursive: bool = False, force: bool = False) -> None:
+    if "lit://" not in path:
+        raise ValueError("Path must be a Studio path starting with 'lit://'.")
+
+    console = Console()
+    studio_path_result = parse_studio_path(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"]:
+        if force:
+            # silently ignore nonexistent files with -f flag
+            return
+        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"Removing from {selected_studio.teamspace.name}/{selected_studio.name}")
+
+    if path_info["type"] == "directory":
+        if not recursive:
+            raise ValueError(
+                f"'{studio_path_result['destination']}' is a directory. Use -r flag to remove directories recursively."
+            )
+        rm_folder(selected_studio=selected_studio, path=studio_path_result["destination"], console=console)
+    else:
+        rm_file(selected_studio=selected_studio, path=studio_path_result["destination"], console=console)
+
+
+def rm_file(selected_studio: Studio, path: str, console: Console) -> None:
+    selected_studio._studio_api.remove_file(selected_studio._studio.id, selected_studio._teamspace.id, path)
+    console.print(f"Removed file: {path}")
+
+
+def rm_folder(selected_studio: Studio, path: str, console: Console) -> None:
+    selected_studio._studio_api.remove_folder(selected_studio._studio.id, selected_studio._teamspace.id, path)
+    console.print(f"Removed directory: {path}")

lightning_sdk/cli/utils/logging.py

@@ -30,10 +30,11 @@ def _log_command(message: str = "", duration: int = 0, error: Optional[str] = No
     body = V1CreateSDKCommandHistoryRequest(
         command=original_command,
         duration=duration,
-        message=f"VERSION: {__version__} | {message}",
+        message=f"{message}",
         project_id=None,
         severity=V1SDKCommandHistorySeverity.INFO,
         type=V1SDKCommandHistoryType.CLI,
+        version=__version__,
     )
 
     if error:

lightning_sdk/cli/utils/studio_filesystem.py

@@ -0,0 +1,65 @@
+from typing import Optional, TypedDict
+
+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
+from lightning_sdk.studio import Studio
+
+
+class StudioPathResult(TypedDict):
+    owner: Optional[str]
+    teamspace: Optional[str]
+    studio: Optional[str]
+    destination: Optional[str]
+
+
+def parse_studio_path(studio_path: str) -> StudioPathResult:
+    path_string = studio_path.removeprefix("lit://")
+    if not path_string:
+        raise ValueError("Studio path cannot be empty after prefix")
+
+    result: StudioPathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
+
+    if "/studios/" in path_string:
+        prefix_part, suffix_part = path_string.split("/studios/", 1)
+
+        # org and teamspace
+        if prefix_part:
+            org_ts_components = prefix_part.split("/")
+            if len(org_ts_components) == 2:
+                result["owner"], result["teamspace"] = org_ts_components
+            elif len(org_ts_components) == 1:
+                result["teamspace"] = org_ts_components[0]
+            else:
+                raise ValueError(f"Invalid format: '{prefix_part}'")
+
+        # studio and destination
+        path_parts = suffix_part.split("/")
+
+    else:
+        # studio and destination
+        path_parts = path_string.split("/")
+
+    if not path_parts:
+        raise ValueError("Invalid: Missing studio name.")
+
+    if len(path_parts) == 1:
+        raise ValueError(
+            "Invalid: Invalid studio path. To refer to the studio root, add a trailing '/' (e.g., 'lit://<owner>/<my-teamspace>/studios/<my-studio>/')"
+        )
+
+    result["studio"] = path_parts[0]
+    result["destination"] = "/".join(path_parts[1:])
+
+    return result
+
+
+def resolve_studio(studio_name: Optional[str], teamspace: Optional[str], owner: Optional[str]) -> Studio:
+    owner_menu = OwnerMenu()
+    resolved_owner = owner_menu(owner=owner)
+
+    teamspace_menu = TeamspacesMenu(resolved_owner)
+    resolved_teamspace = teamspace_menu(teamspace=teamspace)
+
+    studio_menu = StudiosMenu(resolved_teamspace)
+    return studio_menu(studio=studio_name)

lightning_sdk/cli/utils/teamspace_selection.py

@@ -51,6 +51,11 @@ class TeamspacesMenu:
         return TerminalMenu(possible_teamspaces, title=title, clear_menu_on_exit=True)
 
     def _get_possible_teamspaces(self, user: User) -> Dict[str, str]:
+        if self._owner is None:
+            from lightning_sdk.cli.utils.owner_selection import OwnerMenu
+
+            menu = OwnerMenu()
+            self._owner = menu()
         user_api = user._user_api
 
         memberships = user_api._get_all_teamspace_memberships(

lightning_sdk/exceptions.py

@@ -1,2 +1,6 @@
 class OutOfCapacityError(RuntimeError):
     """Raised when the requested machine is not available in the selected cloud account."""
+
+
+class NotSupportedError(RuntimeError):
+    """Raised when the requested machine is not supported in the selected cloud account."""

lightning_sdk/job/base.py

@@ -201,7 +201,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 raise RuntimeError(
                     "image and studio are mutually exclusive as both define the environment to run the job in"
                 )
-            if cloud_account is None and in_studio():
+            if cloud_account is None and cloud_provider is None and in_studio():
                 try:
                     with skip_studio_setup():
                         resolve_studio = Studio(teamspace=teamspace, user=user, org=org)

lightning_sdk/k8s_cluster.py

@@ -1,8 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-from typing import List, Optional
-
-import pandas as pd
+from typing import Any, Dict, List, Optional
 
 from lightning_sdk.api.k8s_api import K8sClusterApi
 from lightning_sdk.api.utils import to_iso_z
@@ -34,25 +32,26 @@ class K8sCluster:
         self._cloud_account = cloud_account
         self._k8s_cluster = K8sClusterApi(cloud_account=cloud_account)
 
-    def _convert_to_k8s_usage_response(self, df: pd.DataFrame) -> K8sUsageResponse:
-        """Converts a DataFrame to K8sUsageResponse.
+    def _convert_to_k8s_usage_response(self, data: List[Dict[str, Any]]) -> K8sUsageResponse:
+        """Converts a list of usage data to K8sUsageResponse.
 
         Args:
-            df (pd.DataFrame): The DataFrame containing GPU usage data.
+            data (List[Dict[str, Any]]): The list of dictionaries containing GPU usage data.
 
         Returns:
             K8sUsageResponse: The converted response containing hourly usage and total usage.
         """
-        if df.empty:
+        if not data:
             return K8sUsageResponse(hours=[], total_usage=0.0)
-        # Convert each row of the DataFrame to HourlyUsage
+
+        # Convert each row to HourlyUsage
         hourly_usage_list: List[HourlyUsage] = [
             HourlyUsage(time=row["hour"], available_gpus=row["num_gpus"], billed_gpus=row["billed_gpus"])
-            for _, row in df.iterrows()
+            for row in data
         ]
 
         # Calculate total usage (sum of billed GPUs)
-        total_usage = df["billed_gpus"].sum()
+        total_usage = sum(row["billed_gpus"] for row in data)
 
         # Create and return the K8sUsageResponse
         return K8sUsageResponse(hours=hourly_usage_list, total_usage=total_usage)

lightning_sdk/lightning_cloud/__version__.py

@@ -1 +1 @@
-__version__ = "0.5.70"
+__version__ = "0.6.0"