lightning-sdk 2025.8.26__py3-none-any.whl → 2025.8.28__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.26"
+__version__ = "2025.08.28"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/cli/studio/create.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 import click
 
+from lightning_sdk.cli.utils.richt_print import studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_teamspace_to_config
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
@@ -56,4 +57,4 @@ def create_studio(
             raise ValueError(f"Could not create Studio: '{name}'. Does the Studio exist?") from None
         raise ValueError(f"Could not create Studio: '{name}'. Please provide a Studio name") from None
 
-    click.echo(f"Studio '{studio.name}' created successfully")
+    click.echo(f"Studio {studio_name_link(studio)} created successfully")

lightning_sdk/cli/studio/list.py

@@ -6,21 +6,29 @@ import click
 from rich.table import Table
 
 from lightning_sdk.cli.utils.cloud_account_map import cloud_account_to_display_name
-from lightning_sdk.cli.utils.richt_print import rich_to_str
+from lightning_sdk.cli.utils.richt_print import rich_to_str, studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_teamspace_to_config
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.studio import Studio
+from lightning_sdk.utils.resolve import _get_authed_user, prevent_refetch_studio
 
 
 @click.command("list")
 @click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--all",
+    is_flag=True,
+    flag_value=True,
+    default=False,
+    help="List all studios, not just the ones belonging to the authed user",
+)
 @click.option(
     "--sort-by",
     default=None,
     type=click.Choice(["name", "teamspace", "status", "machine", "cloud-account"], case_sensitive=False),
     help="the attribute to sort the studios by.",
 )
-def list_studios(teamspace: Optional[str] = None, sort_by: Optional[str] = None) -> None:
+def list_studios(teamspace: Optional[str] = None, all: bool = False, sort_by: Optional[str] = None) -> None:  # noqa: A002
     """List Studios in a teamspace.
 
     Example:
@@ -31,6 +39,8 @@ def list_studios(teamspace: Optional[str] = None, sort_by: Optional[str] = None)
     teamspace_resolved = menu(teamspace=teamspace)
     save_teamspace_to_config(teamspace_resolved, overwrite=False)
 
+    user = _get_authed_user()
+
     studios = teamspace_resolved.studios
 
     table = Table(
@@ -42,14 +52,19 @@ def list_studios(teamspace: Optional[str] = None, sort_by: Optional[str] = None)
     table.add_column("Machine")
     table.add_column("Cloud account")
 
-    for studio in sorted(studios, key=_sort_studios_key(sort_by)):
-        table.add_row(
-            studio.name,
-            f"{studio.teamspace.owner.name}/{studio.teamspace.name}",
-            str(studio.status),
-            str(studio.machine) if studio.machine is not None else None,  # when None the cell is empty
-            str(cloud_account_to_display_name(studio.cloud_account, studio.teamspace.id)),
-        )
+    for studio in sorted(
+        filter(lambda s: all or s._studio.user_id == user.id, studios), key=_sort_studios_key(sort_by)
+    ):
+        with prevent_refetch_studio(studio):
+            table.add_row(
+                # cannot convert to ascii here, as the final rich table has to be converted to ascii
+                # otherwise the lack of support for linking in some terminals causes formatting issues.
+                studio_name_link(studio, to_ascii=False),
+                f"{studio.teamspace.owner.name}/{studio.teamspace.name}",
+                str(studio.status),
+                str(studio.machine) if studio.machine is not None else None,  # when None the cell is empty
+                str(cloud_account_to_display_name(studio.cloud_account, studio.teamspace.id)),
+            )
 
     click.echo(rich_to_str(table), color=True)
 

lightning_sdk/cli/studio/start.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 import click
 
+from lightning_sdk.cli.utils.richt_print import studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
@@ -45,7 +46,7 @@ def start_studio(
     name: Optional[str] = None,
     teamspace: Optional[str] = None,
     create: bool = False,
-    machine: Optional[str] = None,
+    machine: str = "CPU",
     interruptible: bool = False,
     cloud_provider: Optional[str] = None,
     cloud_account: Optional[str] = None,
@@ -78,4 +79,4 @@ def start_studio(
 
     Studio.show_progress = True
     studio.start(machine, interruptible=interruptible)
-    click.echo(f"Studio '{studio.name}' started successfully")
+    click.echo(f"Studio {studio_name_link(studio)} started successfully")

lightning_sdk/cli/studio/stop.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 import click
 
+from lightning_sdk.cli.utils.richt_print import studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
@@ -37,4 +38,4 @@ def stop_studio(name: Optional[str] = None, teamspace: Optional[str] = None) ->
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio '{studio.name}' stopped successfully")
+    click.echo(f"Studio {studio_name_link(studio)} stopped successfully")

lightning_sdk/cli/studio/switch.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 import click
 
+from lightning_sdk.cli.utils.richt_print import studio_name_link
 from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
@@ -46,4 +47,4 @@ def switch_studio(
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio '{studio.name}' switched to machine '{resolved_machine}' successfully")
+    click.echo(f"Studio {studio_name_link(studio)} switched to machine '{resolved_machine}' successfully")

lightning_sdk/cli/utils/richt_print.py

@@ -3,9 +3,33 @@ from typing import Any
 
 from rich.console import Console
 
+from lightning_sdk.studio import Studio
+from lightning_sdk.utils.resolve import _get_studio_url
+
 
 def rich_to_str(*renderables: Any) -> str:
     with open(os.devnull, "w") as f:
         console = Console(file=f, record=True)
         console.print(*renderables)
     return console.export_text(styles=True)
+
+
+# not supported on all terminals (e.g. older ones). in that case it's a name without a link
+# see https://gist.github.com/egmontkob/eb114294efbcd5adb1944c9f3cb5feda#hyperlinks-aka-html-like-anchors-in-terminal-emulators
+# for details and
+# https://github.com/Alhadis/OSC8-Adoption/blob/main/README.md for status of OSC8 adoption
+def studio_name_link(studio: Studio, to_ascii: bool = True) -> str:
+    """Hyperlink a studio name.
+
+    Args:
+      studio: the studio whose name to print and link to the studio url
+      to_ascii: whether return a plain ascii string with characters for linking converted to ascii as well.
+        if False, returns the rich markup directly.
+    """
+    url = _get_studio_url(studio)
+
+    studio_link_markup = f"[link={url}]{studio.name}[/link]"
+    if not to_ascii:
+        return studio_link_markup
+
+    return rich_to_str(studio_link_markup).strip("\n")

lightning_sdk/studio.py

@@ -81,6 +81,7 @@ class Studio:
         self._studio_api = StudioApi()
         self._cloud_account_api = CloudAccountApi()
 
+        self._prevent_refetch = False
         self._teamspace = None
 
         # don't resolve anything if we're skipping init

lightning_sdk/utils/resolve.py

@@ -265,6 +265,17 @@ def skip_studio_setup() -> Generator[None, None, None]:
     Studio._skip_setup.value = prev_studio_setup_state
 
 
+@contextmanager
+def prevent_refetch_studio(studio: "Studio") -> Generator[None, None, None]:
+    """Prevent refetching the studio based on current runtime."""
+    prev_prevent_refetch_state = getattr(studio, "_prevent_refetch", False)
+    studio._prevent_refetch = True
+
+    yield
+
+    studio._prevent_refetch = prev_prevent_refetch_state
+
+
 def _parse_model_and_version(name: str) -> Tuple[str, Optional[str]]:
     """Parse the model name and version from the given string.
 

{lightning_sdk-2025.8.26.dist-info → lightning_sdk-2025.8.28.dist-info}/METADATA

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

{lightning_sdk-2025.8.26.dist-info → lightning_sdk-2025.8.28.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=h1e_QoNOLbPkgRajxV8cZgmQ1D1P4U6P_FkGYXx3mPo,1145
+lightning_sdk/__init__.py,sha256=m4U-y4445WQpSE8meItjnzJIFiwx7THsNgH8aGqqAU4,1145
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=_Pwwl37R9GRd7t-f2kO5aQXiLNrP4sUtUNht2ZkP8LE,3678
@@ -14,7 +14,7 @@ lightning_sdk/plugin.py,sha256=f3P5-xZY6x-MX0Fs2z_Q2erSxPSiHZARO0BVkCezHw4,15192
 lightning_sdk/sandbox.py,sha256=_NvnWotEXW2rBiVFZZ4krKXxVjuAqfNh04qELSM0-Pg,5786
 lightning_sdk/serve.py,sha256=uW7zLhQ3X90ifetpxzTb8FNxifv5vIs7qZlgfEjVKzk,11794
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
-lightning_sdk/studio.py,sha256=0FjD2POIBR7VHAG5RseiwBXn9P0Tpa44U3DKomGgVTQ,30328
+lightning_sdk/studio.py,sha256=N6qxQMJXnaXJK8NXeRxdVg-4az63Tp8eMc-RvfhT8_o,30366
 lightning_sdk/teamspace.py,sha256=qYpIYgg3eyWj9vcqdfhQN-YCpS47tOc_xB3de07p4Y8,21630
 lightning_sdk/user.py,sha256=TSYh38rxoi7qKOfrK2JYh_Nknya2Kbz2ngDIY85fFOY,1778
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
@@ -73,18 +73,18 @@ lightning_sdk/cli/legacy/deploy/devbox.py,sha256=SdOtYdGefzldn-WCa6sz31zysyf2nP5
 lightning_sdk/cli/legacy/deploy/serve.py,sha256=bxN04mHlArjHtxR_QkCM0S_RPmL3WIPsgnPgFadSXwI,15478
 lightning_sdk/cli/mmt/__init__.py,sha256=CLgr-ZHHLS6Db_JGqpxbn4G2pYrKi4Qn-uhi8e0kFNc,145
 lightning_sdk/cli/studio/__init__.py,sha256=Reb5a0iHJfNMxmnP30DAmYqpJ-YDQoJcgyVrD2cYDNo,823
-lightning_sdk/cli/studio/create.py,sha256=hXLPRkcVl8bIBT0zciSQ4rFOjOmLgjdy0Lg6jmf_EV4,1995
+lightning_sdk/cli/studio/create.py,sha256=MK0OzRV81QgBSxmkQy6YVBJc1cOzq9FSTL2NG5KGI5A,2071
 lightning_sdk/cli/studio/delete.py,sha256=rzvzNviu9NVeITnCD84PzEEoizrggcq78yaOqrT6U_U,1289
-lightning_sdk/cli/studio/list.py,sha256=5cDQu6qH-fh7g48hqCzW10VBJWn93GlIzls7cS4Kitc,2375
+lightning_sdk/cli/studio/list.py,sha256=FUt1SI-uDBWh-ghPFxQZAoxkVdQspGNhy33nkVhdB_o,3079
 lightning_sdk/cli/studio/ssh.py,sha256=Qlwyl4LISCzFYedhkXArTHqipl_6Yazq7vRcrsc_2ws,3654
-lightning_sdk/cli/studio/start.py,sha256=WFdDS4tinYioZY-1s2DnfuOVgEKgAMhj9bujv9Gp2zQ,2628
-lightning_sdk/cli/studio/stop.py,sha256=ydfKJMqSrzA_mzF2c5AHRzJJP-bYV-HfO1gsN61IpGs,1159
-lightning_sdk/cli/studio/switch.py,sha256=5FdjL1pTrst58jqgx_Jkjo6RtV8vCltmsyldjn_3iNs,1669
+lightning_sdk/cli/studio/start.py,sha256=KuJDwwTnWMhhuihF9qNbcIXa_VPzPH_A8zz-Gck1sn0,2695
+lightning_sdk/cli/studio/stop.py,sha256=d-54OzDYYiY5VNlwIK4oryCH8aafJ7RRBG-ARhcFc4o,1235
+lightning_sdk/cli/studio/switch.py,sha256=qhWhLvjk-BbwU8Br3rXPQIMVQLTUTq21f0v2P-R8VfQ,1745
 lightning_sdk/cli/utils/__init__.py,sha256=0gHdWY3bqrIyiFiEh_uSBuxWpOykCjqUc8fPEV0z3no,186
 lightning_sdk/cli/utils/cloud_account_map.py,sha256=7oM7ns4ZJfkxxZhl-GMkxEkkinKbVkskx5dLcyEQiz4,414
 lightning_sdk/cli/utils/coloring.py,sha256=YEb1LiYb6CekVfCRb83-npAmNgd-7c1LzXuNDo4GLSc,2415
 lightning_sdk/cli/utils/resolve.py,sha256=RWQ8MgXEJCdsoY50y8Pa9B-A6NA_SvxOqL5pVKfvSr8,917
-lightning_sdk/cli/utils/richt_print.py,sha256=8IkTKLWqaE9_1bjJGyk3mNzHtGZZ0_YlmPGrLAtIm7U,276
+lightning_sdk/cli/utils/richt_print.py,sha256=psSY65nLAQbxK_K0w93Qq9857aUUbm77FLS1sc3Oecg,1262
 lightning_sdk/cli/utils/save_to_config.py,sha256=mMjjfA4Yv1e7MtE03iADTXcTSAuUL0Ws9OZObx6ufo0,1164
 lightning_sdk/cli/utils/studio_selection.py,sha256=tjYTm4sY3NSQaa-6Ob9R0UQ60wIXiTNT2406BUJvJbU,3927
 lightning_sdk/cli/utils/teamspace_selection.py,sha256=ZWo_aZuEuLzBWRgQEykypEM1sE_deDKLDcCyLgFWaaQ,5188
@@ -1143,10 +1143,10 @@ lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgy
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/names.py,sha256=1EuXbIh7wldkDp1FG10oz9vIOyWrpGWeFFVy-DQBgzA,18162
 lightning_sdk/utils/progress.py,sha256=IXfEcUF-rL5jIw0Hir6eSxN7VBZfR--1O2LaEhGAU70,12698
-lightning_sdk/utils/resolve.py,sha256=yqv-0NOIG_qnSvlvOaGK11PG8RuKHkstiZMV8GAFnhM,10136
-lightning_sdk-2025.8.26.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.8.26.dist-info/METADATA,sha256=wTC_RLAWY2vOMdpikf-R7owxuIxY5QYDpQzJat_7LXo,4130
-lightning_sdk-2025.8.26.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.8.26.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-2025.8.26.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.8.26.dist-info/RECORD,,
+lightning_sdk/utils/resolve.py,sha256=4TyEnIgIrvkSvYk5i5PmcIogD_5Y9pBhiphRLfLMttc,10477
+lightning_sdk-2025.8.28.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.28.dist-info/METADATA,sha256=LLjYF3GZy05IhP4gHBgNfKxuZUMtAB-ujgvYzTEuT9o,4130
+lightning_sdk-2025.8.28.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.28.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.28.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.28.dist-info/RECORD,,