lightning-sdk 2025.8.19.post0__py3-none-any.whl → 2025.8.21__py3-none-any.whl

lightning_sdk/utils/progress.py

@@ -0,0 +1,284 @@
+"""Studio startup/switch progress bar utilities."""
+
+import time
+import types
+from enum import Enum
+from typing import Any, Callable, List, Optional, Type, Union
+
+from rich.console import Console
+from rich.progress import BarColumn, Progress, SpinnerColumn, TaskID, TextColumn, TimeElapsedColumn
+
+from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import (
+    V1GetCloudSpaceInstanceStatusResponse,
+)
+
+
+class StartupPhase(Enum):
+    """Studio startup phase messages."""
+
+    ALLOCATING_MACHINE = "Allocating machine from cloud provider..."
+    STUDIO_STARTING = "Studio is starting up..."
+    SETTING_UP_ENVIRONMENT = "Setting up Studio environment..."
+    RESTORING_STATE = "Restoring Studio state..."
+    FINALIZING_SETUP = "Finalizing Studio setup..."
+    COMPLETED = "Studio started successfully"
+
+
+def get_switching_progress_message(percentage: int, is_base_studio: bool, is_new_cloud_space: bool) -> str:
+    """Get progress message for switching studios."""
+    percentage = max(0, min(100, round(percentage)))
+
+    if percentage > 98:
+        message = "Done"
+    elif percentage > 80:
+        if is_new_cloud_space:
+            message = "Setting up Base Studio..." if is_base_studio else "Preparing Studio..."
+        else:
+            message = "Restoring Studio..."
+    elif percentage > 60:
+        message = "Setting up machine from the cloud provider"
+    else:
+        message = "Allocating machine from the cloud provider"
+
+    return f"({percentage}%) {message}"
+
+
+def estimated_studio_ready_in_seconds(
+    cloud_space: Any, cloud_space_instance_status: Any, accelerators: Optional[List[V1ClusterAccelerator]] = None
+) -> int:
+    """Calculate estimated seconds until studio is ready."""
+    # Default estimate
+    return 120
+
+
+def progress_bar_growth(default_timeout: int, counter: float) -> int:
+    """Calculate progress bar growth based on timeout and counter."""
+    if default_timeout <= 0:
+        return 100
+
+    value = ((default_timeout - counter) / default_timeout) * 100
+    if value > 100:
+        value = 100
+    if value < 0:
+        value = 0
+    return int(value)
+
+
+class StudioProgressTracker:
+    """Tracks and displays progress for studio startup/switching operations."""
+
+    def __init__(self, operation_type: str = "start", show_progress: bool = True, check_interval: float = 1.0) -> None:
+        """Initialize progress tracker.
+
+        Args:
+            operation_type: Type of operation ('start' or 'switch')
+            show_progress: Whether to display progress bar
+            check_interval: Seconds between status checks
+        """
+        self.operation_type = operation_type
+        self.show_progress = show_progress
+        self.check_interval = check_interval
+        self.progress: Optional[Progress] = None
+        self.task_id: Optional[TaskID] = None
+        self.console = Console()
+        self._last_message = ""
+
+    def __enter__(self) -> "StudioProgressTracker":
+        """Enter context manager."""
+        if self.show_progress:
+            self.progress = Progress(
+                SpinnerColumn(),
+                TextColumn("[progress.description]{task.description}"),
+                BarColumn(),
+                TextColumn("[progress.percentage]{task.percentage:>3.0f}%"),
+                TimeElapsedColumn(),
+                console=self.console,
+                transient=False,
+            )
+            self.progress.start()
+            self.task_id = self.progress.add_task(f"{self.operation_type.capitalize()}ing Studio...", total=100)
+        return self
+
+    def __exit__(
+        self,
+        exc_type: Union[Type[BaseException], None],
+        exc_val: Union[BaseException, None],
+        exc_tb: Union[types.TracebackType, None],
+    ) -> None:
+        """Exit context manager."""
+        if self.progress:
+            self.progress.stop()
+
+    def update_progress(
+        self, percentage: int, message: str = "", is_base_studio: bool = False, is_new_cloud_space: bool = False
+    ) -> None:
+        """Update progress bar with current percentage and message."""
+        if not self.progress or self.task_id is None:
+            return
+
+        if self.operation_type == "switch":
+            display_message = get_switching_progress_message(percentage, is_base_studio, is_new_cloud_space)
+        else:
+            display_message = message or f"{self.operation_type.capitalize()}ing Studio..."
+
+        # Update description if message changed
+        if display_message != self._last_message:
+            self.progress.update(self.task_id, description=display_message)
+            self._last_message = display_message
+
+        # Never show 100% until truly complete
+        completed = min(percentage, 98) if percentage < 100 else 100
+        self.progress.update(self.task_id, completed=completed)
+
+        # Force console refresh to ensure progress is visible
+        self.progress.refresh()
+
+    def complete(self, success_message: str = "") -> None:
+        """Mark operation as complete."""
+        if self.progress and self.task_id is not None:
+            self.progress.update(self.task_id, completed=100, description=success_message or "Done")
+
+    def track_startup_phases(
+        self,
+        status_getter: Callable[[], V1GetCloudSpaceInstanceStatusResponse],
+        accelerators: Optional[List[V1ClusterAccelerator]] = None,
+        timeout: int = 600,
+    ) -> None:
+        """Track startup phases and update progress accordingly."""
+        start_time = time.time()
+        last_progress = 5
+        phase_start_times = {}
+        last_message_time = 0
+        message_stability_delay = 3.0  # Seconds to wait before changing message
+
+        # Show initial progress immediately
+        self.update_progress(5, StartupPhase.ALLOCATING_MACHINE.value)
+
+        while True:
+            try:
+                status = status_getter()
+                elapsed = time.time() - start_time
+
+                # Default fallback progress based on time
+                time_based_progress = min(95, int((elapsed / timeout) * 100))
+                current_progress = max(last_progress, time_based_progress)
+                current_message = StartupPhase.ALLOCATING_MACHINE.value
+
+                # Check if we have detailed status information
+                if hasattr(status, "in_use") and status.in_use:
+                    in_use = status.in_use
+
+                    # Check startup status for detailed phases
+                    if hasattr(in_use, "startup_status") and in_use.startup_status:
+                        startup_status = in_use.startup_status
+
+                        # Check completion first
+                        if (
+                            hasattr(startup_status, "top_up_restore_finished")
+                            and startup_status.top_up_restore_finished
+                        ):
+                            self.complete(StartupPhase.COMPLETED.value)
+                            break
+
+                        # Check other phases in descending priority
+                        if (
+                            hasattr(startup_status, "initial_restore_finished")
+                            and startup_status.initial_restore_finished
+                        ):
+                            current_progress = max(current_progress, 85)
+                            current_message = StartupPhase.FINALIZING_SETUP.value
+                        elif hasattr(startup_status, "container_ready") and startup_status.container_ready:
+                            current_progress = max(current_progress, 70)
+                            current_message = StartupPhase.RESTORING_STATE.value
+                        elif hasattr(startup_status, "machine_ready") and startup_status.machine_ready:
+                            current_progress = max(current_progress, 60)
+                            current_message = StartupPhase.SETTING_UP_ENVIRONMENT.value
+
+                    # Check general phase information
+                    if hasattr(in_use, "phase") and in_use.phase:
+                        phase = in_use.phase
+
+                        if phase == "CLOUD_SPACE_INSTANCE_STATE_RUNNING":
+                            current_progress = max(current_progress, 80)
+                            current_message = StartupPhase.STUDIO_STARTING.value
+                        elif phase == "CLOUD_SPACE_INSTANCE_STATE_PENDING":
+                            # Track time in pending phase for smoother progress
+                            if "pending" not in phase_start_times:
+                                phase_start_times["pending"] = time.time()
+
+                            pending_elapsed = time.time() - phase_start_times["pending"]
+                            # Progress more smoothly through pending phase (10-60%)
+                            pending_progress = 10 + min(50, int((pending_elapsed / 60) * 50))
+                            current_progress = max(current_progress, pending_progress)
+                            current_message = StartupPhase.ALLOCATING_MACHINE.value
+
+                # Check for requested machine status (pre-allocation)
+                elif hasattr(status, "requested") and status.requested:
+                    if "allocation" not in phase_start_times:
+                        phase_start_times["allocation"] = time.time()
+
+                    allocation_elapsed = time.time() - phase_start_times["allocation"]
+                    # Progress through allocation phase (5-30%)
+                    allocation_progress = 5 + min(25, int((allocation_elapsed / 30) * 25))
+                    current_progress = max(current_progress, allocation_progress)
+                    current_message = StartupPhase.ALLOCATING_MACHINE.value
+
+                # Ensure progress never decreases and moves smoothly
+                if current_progress > last_progress:
+                    # Smooth progress increases - don't jump too much at once
+                    max_increment = 3 if current_progress - last_progress > 10 else current_progress - last_progress
+                    current_progress = last_progress + max_increment
+
+                current_progress = min(98, current_progress)  # Never show 100% until truly complete
+
+                # Only update message if enough time has passed since last message change
+                # or if this is the first message
+                current_time = time.time()
+                should_update_message = current_message != self._last_message and (
+                    current_time - last_message_time >= message_stability_delay or last_message_time == 0
+                )
+
+                if should_update_message:
+                    self.update_progress(current_progress, current_message)
+                    last_message_time = current_time
+                else:
+                    # Update progress but keep existing message
+                    if self.progress and self.task_id is not None:
+                        self.progress.update(self.task_id, completed=current_progress)
+                        self.progress.refresh()
+
+                last_progress = current_progress
+
+                # Break if we timeout
+                if elapsed > timeout:
+                    self.complete("Studio start may still be in progress...")
+                    break
+
+            except Exception:
+                # Continue on API errors but still update progress
+                elapsed = time.time() - start_time
+                fallback_progress = min(95, max(last_progress, int((elapsed / timeout) * 100)))
+
+                # Only update message if enough time has passed
+                current_time = time.time()
+                should_update_message = StartupPhase.ALLOCATING_MACHINE.value != self._last_message and (
+                    current_time - last_message_time >= message_stability_delay or last_message_time == 0
+                )
+
+                if should_update_message:
+                    self.update_progress(fallback_progress, StartupPhase.ALLOCATING_MACHINE.value)
+                    last_message_time = current_time
+                else:
+                    # Update progress but keep existing message
+                    if self.progress and self.task_id is not None:
+                        self.progress.update(self.task_id, completed=fallback_progress)
+                        self.progress.refresh()
+
+                last_progress = fallback_progress
+
+                if elapsed > timeout:
+                    self.complete("Studio start may still be in progress...")
+                    break
+
+            time.sleep(self.check_interval)

{lightning_sdk-2025.8.19.post0.dist-info → lightning_sdk-2025.8.21.dist-info}/METADATA

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

{lightning_sdk-2025.8.19.post0.dist-info → lightning_sdk-2025.8.21.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=5tWz-6f1MLRPtc-voiWA2ZawI8leB9UrVX6p1f6S8v4,1151
+lightning_sdk/__init__.py,sha256=3CTb5XWkDB36YB2MgXk4fpAT6bY06rRC9WRuES_y8a8,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=l3l-TYExmnJnAQIPA6_8R2u1ynT-ewzW-q8hV_nY7nE,27138
+lightning_sdk/studio.py,sha256=PUjKiSQNGWq8cvyHOUTw1ZftqWic-L4czz8KfB84-e8,28624
 lightning_sdk/teamspace.py,sha256=Ikg3LwGhskD_wAW3GfUOuMgiyEXOV8khleBeIMtN8wA,20751
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
@@ -26,11 +26,11 @@ lightning_sdk/api/deployment_api.py,sha256=v2AfoTDkQ-1CBh75FOjFkRpf6yc3U_edDy43u
 lightning_sdk/api/job_api.py,sha256=7spFPyotlSQ0Krx6WymslnpSQaaeOmFkydvhtg6p0Ic,16410
 lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
 lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
-lightning_sdk/api/llm_api.py,sha256=HSCpTyPknKAv6j0UYIzasXK2mFYt7FE3KF5TGUYvo2w,11049
+lightning_sdk/api/llm_api.py,sha256=K_MbVGBrcVz59Pq_UKMbFKsHp-CzKotG_fzzEu3VTb8,11166
 lightning_sdk/api/mmt_api.py,sha256=hIBsGiJ2qn5UjcHDxP5WUyKGT_AIFfpSHrQVwg0afBw,10699
 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=COKJlJuZm2XucV5sJaW19ojz9zjDKSpeZMVt3tm06tc,32481
+lightning_sdk/api/studio_api.py,sha256=_d0zCp8RZ9iLYwO4FzfzevEMtaZ0WrzI8OOI9lrTwF8,36428
 lightning_sdk/api/teamspace_api.py,sha256=CsaaxmaLmTRIRwu37wtQ3quGYql62HJT3BZJ3Q-1d9c,16854
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
 lightning_sdk/api/utils.py,sha256=SyxRa76JcWOJpslyzB-iID6OpVCJi0F0G69h1LmDCDA,27124
@@ -47,7 +47,7 @@ lightning_sdk/cli/legacy/ai_hub.py,sha256=8MxGQC7SZALAsIj0tZ9HMZCeVHESUcve0cNazR
 lightning_sdk/cli/legacy/clusters_menu.py,sha256=sj5FD_GDHOAyJZjltiS9_C3shPpu9fGQ8lcSsuhFQlg,1983
 lightning_sdk/cli/legacy/configure.py,sha256=Ielq38GhNAvUfr5a4oXidyHO1YOcyUa4BF1jsPU2Uq0,4471
 lightning_sdk/cli/legacy/connect.py,sha256=0qJ0yvykoPBkmCvvEbWKppt7wlUmRq3LNNGEs0gIgg8,1002
-lightning_sdk/cli/legacy/create.py,sha256=eAc9i7WDobvrVfkMktDNP5FoXyBLpIEKsca65XuI3l4,4037
+lightning_sdk/cli/legacy/create.py,sha256=taCk19iq8CRLbS5-A9VBTYS-gpU594vXiBByxaW2KIs,3699
 lightning_sdk/cli/legacy/delete.py,sha256=1vuiRp1Gd8aO-3MzADQgUITBuzoQ0p8DfMPgi6_2-yY,3833
 lightning_sdk/cli/legacy/docker_cli.py,sha256=EozLC4qnvHhgukmnu35itfI5n7v-abCpwKPkPy3eOV4,997
 lightning_sdk/cli/legacy/download.py,sha256=Ncnk1ZuipB7owJWa2cPyBz0_e_c3eHwvUb_YYJ3Iz_Y,11414
@@ -61,10 +61,10 @@ lightning_sdk/cli/legacy/list.py,sha256=FPfx3HZSnLRjXLcDlI8PQLE41byyLSlCfAvO5ITY
 lightning_sdk/cli/legacy/mmts_menu.py,sha256=ocVzJ_GDDQLiJgccEWqcTfXTxt2jnJ-FDNgrq7t_JNc,2226
 lightning_sdk/cli/legacy/open.py,sha256=x0S2lBUa2NPQv0R9DxOLbykZ_4PHQ493RbBqpRd5C3I,2761
 lightning_sdk/cli/legacy/run.py,sha256=CG5efVA_lMx5J31aomPE3k5JTvcYXrsGo9uXO6Swn94,13963
-lightning_sdk/cli/legacy/start.py,sha256=wWlTLFY2a7LQaJrOF4naT0G5JGHUVRZv-PAOkL8U2ok,3050
+lightning_sdk/cli/legacy/start.py,sha256=ZTqi_HqRbZVlxcvgaAE_JGXgnEzbilA1u-BsPmxsYyU,3082
 lightning_sdk/cli/legacy/stop.py,sha256=jSiuxIunDnL3A3DQ1m8GZx01Qt4Zx5vIA3u7U81C2As,2871
 lightning_sdk/cli/legacy/studios_menu.py,sha256=TA9rO6_fFHGMz0Nt4rJ6iV80X5pZE4xShrSiyXoU-oQ,4129
-lightning_sdk/cli/legacy/switch.py,sha256=v_EYw8gYfUyH273zHknt7XJlj07NbYDAUAWigc0AfzE,1913
+lightning_sdk/cli/legacy/switch.py,sha256=G3y9BmP-0d_OKJub5fMZ0ddnGZkOFBtregMkx_KWxcw,1945
 lightning_sdk/cli/legacy/teamspace_menu.py,sha256=u5_lyB1aCD8KJIsOuxi0bvPi75McU5EbluMLyD_NBUw,4104
 lightning_sdk/cli/legacy/upload.py,sha256=JNaJWlxVclF3UTuiA7OO_5tJFk9-5bCTUQ_CVKfrdSQ,13790
 lightning_sdk/cli/legacy/deploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -77,9 +77,9 @@ lightning_sdk/cli/studio/create.py,sha256=o1NkEn5H-SNdpE4yC7KyikTkQM14AVGYMRZlT3
 lightning_sdk/cli/studio/delete.py,sha256=33IrIsQC4IwsZYRrys4ZxqNbVC2CwWvy6D6ilYXrLuU,1670
 lightning_sdk/cli/studio/list.py,sha256=LxY5R1aQ4OJbo4TWepK7ugrJIju9iq3Z7m2oq3U9Zcc,2387
 lightning_sdk/cli/studio/ssh.py,sha256=0p1DqKPBkrPRNN2AFlhRK9hnuEX3-33ujiBKh24jB_c,3812
-lightning_sdk/cli/studio/start.py,sha256=h3AQE5RCgLIcT4XoPJI6-kNVr3J1D6KjY2cUAAPjMdk,2975
+lightning_sdk/cli/studio/start.py,sha256=0iArl9Uqf7PfaBLBeqfJRJqOelLTDnCmAu_SS1tiAgg,3007
 lightning_sdk/cli/studio/stop.py,sha256=pYkS1G-XYLWTxTFEIgEg7xcihzviB_gHNqgBLaW3OwQ,1635
-lightning_sdk/cli/studio/switch.py,sha256=uKuAv4J7qozS6YzPqz-45_ccfVWNkyWF5Pij_xDcG0M,1994
+lightning_sdk/cli/studio/switch.py,sha256=U49gXiTlPmDDu92YzNuNudhphvcrMCdkSn8knqdbFZQ,2026
 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
@@ -100,7 +100,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=29iN6evpk04SAkBQmjUsRbaoKUcj9ynZDX
 lightning_sdk/lightning_cloud/rest_client.py,sha256=K2J_QXihpbp5tVzs9PzffgMCQY1NeH5tRpuJBYj27vU,7125
 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=JeUr4i1rOw91V3I7-FDvcj_Yw0ZASQ8DbZgFQ8qwRnk,111027
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=Tp0ZcR_C43P6IVtKVGK4j9z8Gwi5VHgcrsb8sBjUsSA,111144
 lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
 lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=SkBPJ3WZ9SF1LxxWeGKa4UwbGRsk9p3-yL_Kn7l5CSM,7866
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
@@ -109,7 +109,7 @@ lightning_sdk/lightning_cloud/openapi/api/agent_service_api.py,sha256=Va0QcK6cO3
 lightning_sdk/lightning_cloud/openapi/api/analytics_service_api.py,sha256=0BccCXj-fpDmrWImCQzPlHz7T8Hrw-vuwPCY-ieWt8Q,5263
 lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=2uVxfTpT8C8rtRX1WJzpYLvrNJrvKxdpdiNSeWRy4tg,129671
 lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAltWyj1wYg3HNk0cxT2HWbNhAqQSLvMDgM,30948
-lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=QLi9OoVOPbpjhveo1l1EfOWcIsKElRa1FbPucTAv654,78311
+lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=BEOPZvbNQBq5PR_bBgB-4xK3cFUHaJsFUAE03NnwXJQ,81578
 lightning_sdk/lightning_cloud/openapi/api/blog_posts_service_api.py,sha256=T6ogreuE2ZC3RPzwJ-LLdK3OCxxPlhRQaI9PPI1pFfQ,20832
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py,sha256=zrLYqozyQcVCRel8yd-SZgrI1CzGymtan7FlQ2nMvUk,28428
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=L5KZ6weLgpBhGnl8qniqkotIXz6_ptOkz3d6lWZSfGM,435015
@@ -148,7 +148,7 @@ lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py,sha256=7gbiF09P
 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=bUCDZf1P16juI8-J_dpE-MvRt3x8x5HBJpluxix6Vzc,68363
 lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py,sha256=pg1aaSuYJUAlUDKLN07l8wC-poU_HFxZ_98QZHty74I,10344
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=KYLeBgLhjIi4XAv43z26p4yuNMriHC_CVGcvcyNRX6E,104935
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=_C4kII5kOjJQVIPBUnLhn_8AkiaAnm_rR4RpKJ7-h7k,105052
 lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py,sha256=X087AkCafKp2g8E2MR5ghU3pxCNNcoehZJttvuVfdC8,4274
 lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py,sha256=5HIzKC2DGqxIJIdZbs6KGh9pnBe9ik8FyxlXGUSc1HQ,10233
 lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=jKg8Q5cJ3iY9DpDHvig6rgezR_qYYPIAr5Cez2pSifs,22374
@@ -160,7 +160,7 @@ lightning_sdk/lightning_cloud/openapi/models/appinstances_id_body.py,sha256=DfwY
 lightning_sdk/lightning_cloud/openapi/models/approveautojoindomain_domain_body.py,sha256=gC2puM75rrgB36ekVmgF9rnzWE4mAp6tpaGUrlhARRU,3771
 lightning_sdk/lightning_cloud/openapi/models/apps_id_body.py,sha256=tjIpEmU6Y8nhLDT0c3JAdywxEjI-NzvgLdEhf2UXWf4,17912
 lightning_sdk/lightning_cloud/openapi/models/apps_id_body1.py,sha256=Ja_1s0PjD1hIjAvyqyp8U4WXKv2GoUGoQ-HHm8hvRbc,8874
-lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py,sha256=7mKWJC9K3myBnzF7YuIxT5YfAGqoNgvA-wrpfDeg6wE,17044
+lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py,sha256=K-WvwTVmh3ruuXf-YBdNeHYCzdn7LrYIKtBcDnYwx1w,17263
 lightning_sdk/lightning_cloud/openapi/models/billing_checkout_body.py,sha256=bMABc4XkRzK-t2RzWg1bd2YZSwZkqYZoUex9iWO1lPI,5379
 lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body.py,sha256=MaovoJiJDzOXNhCskZAN5UNwnFEFP7ONF83DPRgfhgk,4510
 lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body1.py,sha256=6zrWBAHmmKHJgqJtxahcM1mDrO3Z55WxaGK0RY10oOw,5275
@@ -904,7 +904,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py,sha256=p7fV
 lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py,sha256=fCTatqsnrI0tZOB7N1Aw3y54tZtEkivNKY4Pss2vtGg,3349
 lightning_sdk/lightning_cloud/openapi/models/v1_plugin.py,sha256=mzcUs5AnfMrqMvIf7JO0DG8DkuCqYWd3pTmUXjedR_4,6883
 lightning_sdk/lightning_cloud/openapi/models/v1_plugins_list_response.py,sha256=agkEDBSfVWdppSHPuzuJEczHJBqe5SPmjRMuI-_2Ldg,3777
-lightning_sdk/lightning_cloud/openapi/models/v1_pod_metrics.py,sha256=Zzqn33fZsSTkqG709crTV9w4HvbjczS9B0Q0UMHgZVI,13931
+lightning_sdk/lightning_cloud/openapi/models/v1_pod_metrics.py,sha256=2f1QYt40yLQpLZ3AsEs2vO6N63sI4LekAnagKT4HLJE,19794
 lightning_sdk/lightning_cloud/openapi/models/v1_post_cloud_space_artifact_events_response.py,sha256=zuDuC7QWTKRFeQxpakwVbKSM-BWKglkev6cbuLOFCIs,3118
 lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py,sha256=6XG2sQF3NL61DadYK59Qbd3KCIq50Uv5KgOdnm1_spA,4309
 lightning_sdk/lightning_cloud/openapi/models/v1_product_license.py,sha256=XSAezLE_BpUKPTt90wQ_tepQq2LYuwPyov1xz0G-yrw,12775
@@ -913,7 +913,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_profiler_capture.py,sha256=vtZLS
 lightning_sdk/lightning_cloud/openapi/models/v1_profiler_enabled_response.py,sha256=z-5iw130qv54DZFTAXE4t9VeMsME_2aoKC1zXi4koQA,3757
 lightning_sdk/lightning_cloud/openapi/models/v1_project.py,sha256=wfE_1Yy8QpUR8Qn1yBxKPrSfE5HakKzbVdWhGhza1Pg,18371
 lightning_sdk/lightning_cloud/openapi/models/v1_project_artifact.py,sha256=jMg7BHIX29bYQUK5VPHS7OXJlsZ8RvdIE9trNH5YXHU,6642
-lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py,sha256=0AIT0Yp0F-yF1ogKzNEz9o9JKXzlWFLEyZXcAiPdvXE,9403
+lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py,sha256=jv2wkiEm5u2VhpTeRcd27dur5wnyt1QTb1ae0tzSfwg,10335
 lightning_sdk/lightning_cloud/openapi/models/v1_project_compute_daily_usage.py,sha256=92qqkHCx9gfZCg7FGVn7TDMQjuHw5E0POtLC7HoQ9pE,6565
 lightning_sdk/lightning_cloud/openapi/models/v1_project_compute_usage.py,sha256=-IJmDR0j5M2S7wfkwz5qO0riW2ZolAzrLnopnBAfckU,12121
 lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py,sha256=-UQG2VvHIGLmMT27SMk7xVtSN2YDcxcExZCeXJq0mss,25380
@@ -933,6 +933,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_quest.py,sha256=Sbf-ApnCz79mRC6n
 lightning_sdk/lightning_cloud/openapi/models/v1_quest_status.py,sha256=0iyYEDeFFy-om3nOezq1hVOwf-wynGpLmcnTLQ287Zo,3146
 lightning_sdk/lightning_cloud/openapi/models/v1_queue_server_type.py,sha256=4Gl6lgZsHDp8aLgnZnhR45yBieFGktnsOt6WeIcZ6Dc,3157
 lightning_sdk/lightning_cloud/openapi/models/v1_quotas.py,sha256=XA4nv7cWZ4D9GFXzqXOdPOJ0exrOiwrpRKy8vPOnxyk,11111
+lightning_sdk/lightning_cloud/openapi/models/v1_quote_annual_upsell_response.py,sha256=EEoBSXWBSJEU4DZvK8BzjgohD2ki2ZoRipxCpfwR5MA,6477
 lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py,sha256=yp-Xbfi13-eI7dk3p-n-hBBWvlCvBgTU6I23MXabXvo,10412
 lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py,sha256=6PJpIK9tlhBVnJdhVNsO2HZkGsINDRq03iZ9vd3CLbE,8989
 lightning_sdk/lightning_cloud/openapi/models/v1_refresh_index_response.py,sha256=IU2rKiAWVuyUB3gSAlbRhSxjzucdUGt7uUS6szzl3vk,3022
@@ -1070,7 +1071,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
 lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=ozMzoGD9mfZGYy4J5j50Dps19Y6o8cn-aYW_oRMZMy8,16865
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=k9pDp9UIaOEEWz6bTNWF_KMfcNCOp-F67N-IZ9MO2Rs,8304
-lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=xhdIxKsE-ErpTmWrdefTKmVao54_gqgXFokL8z9-AIE,72492
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=Y-iImT5PxqQ_TlvH4XCxAsKkBE5bgsx7xdJE_SFazVY,71738
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=0fnZpdhxNRXGNyILHtfm3rVztfHpSF4kXGc5kTt4zls,13960
 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
@@ -1113,7 +1114,7 @@ lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgyd
 lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
 lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
 lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q,57
-lightning_sdk/llm/llm.py,sha256=an7VstxZdP50q4-W8bmvb624uz2MnSTnk93UYlIxopg,17305
+lightning_sdk/llm/llm.py,sha256=XJCIxNhvsQLZyX9oPMFZ3drdrlyzJjlwnboH2B6Sfw0,17307
 lightning_sdk/llm/public_assistants.py,sha256=4Gxbz5SD4mBKeSBSbay25XRhZ74Iq4ZF-kNUwl294aE,1449
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=iqVudKDxazomuezj6l7pa-m9I1EQnM82OKs4ZQbo4Ck,16434
@@ -1135,10 +1136,11 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/config.py,sha256=DkH5e4ljBLjOVWwEBSHsF6EMBf-akSwFjHAZKHpqVbE,5369
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
+lightning_sdk/utils/progress.py,sha256=VQYD1gi-qUWa2IlZfCqnNROqwWo4g3tQ-hL7OeT85II,12706
 lightning_sdk/utils/resolve.py,sha256=r7sh5FvVAqb8Ax_TXED0JR3QJawom6w1wgOjAQFJ-MA,10060
-lightning_sdk-2025.8.19.post0.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.8.19.post0.dist-info/METADATA,sha256=REZME-3X0MAoethplQ4xNCN5q-nzG3DTVdtMpSE2oPg,4136
-lightning_sdk-2025.8.19.post0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.8.19.post0.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-2025.8.19.post0.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.8.19.post0.dist-info/RECORD,,
+lightning_sdk-2025.8.21.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.21.dist-info/METADATA,sha256=3Oxea3JOwVDLsI_g4erufHPCxLU2KDIwReF0jCwo6UU,4130
+lightning_sdk-2025.8.21.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.21.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.21.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.21.dist-info/RECORD,,