lightning-sdk 2025.9.30__py3-none-any.whl → 2025.10.14__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -75,13 +75,18 @@ class V1UserFeatures(object):
         'f247': 'bool',
         'f248': 'bool',
         'f250': 'bool',
-        'f251': 'bool',
         'f252': 'bool',
         'f253': 'bool',
         'f254': 'bool',
         'f255': 'bool',
         'f257': 'bool',
         'f258': 'bool',
+        'f259': 'bool',
+        'f260': 'bool',
+        'f261': 'bool',
+        'f262': 'bool',
+        'f263': 'bool',
+        'f264': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
         'gcs_connections_optimized': 'bool',
@@ -116,7 +121,6 @@ class V1UserFeatures(object):
         'runnable_public_studio_page': 'bool',
         'security_docs': 'bool',
         'show_dev_admin': 'bool',
-        'single_wallet': 'bool',
         'slurm': 'bool',
         'specialised_studios': 'bool',
         'storage_overuse_deletion': 'bool',
@@ -164,13 +168,18 @@ class V1UserFeatures(object):
         'f247': 'f247',
         'f248': 'f248',
         'f250': 'f250',
-        'f251': 'f251',
         'f252': 'f252',
         'f253': 'f253',
         'f254': 'f254',
         'f255': 'f255',
         'f257': 'f257',
         'f258': 'f258',
+        'f259': 'f259',
+        'f260': 'f260',
+        'f261': 'f261',
+        'f262': 'f262',
+        'f263': 'f263',
+        'f264': 'f264',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
         'gcs_connections_optimized': 'gcsConnectionsOptimized',
@@ -205,7 +214,6 @@ class V1UserFeatures(object):
         'runnable_public_studio_page': 'runnablePublicStudioPage',
         'security_docs': 'securityDocs',
         'show_dev_admin': 'showDevAdmin',
-        'single_wallet': 'singleWallet',
         'slurm': 'slurm',
         'specialised_studios': 'specialisedStudios',
         'storage_overuse_deletion': 'storageOveruseDeletion',
@@ -218,7 +226,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f251: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f264: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
@@ -254,13 +262,18 @@ class V1UserFeatures(object):
         self._f247 = None
         self._f248 = None
         self._f250 = None
-        self._f251 = None
         self._f252 = None
         self._f253 = None
         self._f254 = None
         self._f255 = None
         self._f257 = None
         self._f258 = None
+        self._f259 = None
+        self._f260 = None
+        self._f261 = None
+        self._f262 = None
+        self._f263 = None
+        self._f264 = None
         self._fair_share = None
         self._featured_studios_admin = None
         self._gcs_connections_optimized = None
@@ -295,7 +308,6 @@ class V1UserFeatures(object):
         self._runnable_public_studio_page = None
         self._security_docs = None
         self._show_dev_admin = None
-        self._single_wallet = None
         self._slurm = None
         self._specialised_studios = None
         self._storage_overuse_deletion = None
@@ -375,8 +387,6 @@ class V1UserFeatures(object):
             self.f248 = f248
         if f250 is not None:
             self.f250 = f250
-        if f251 is not None:
-            self.f251 = f251
         if f252 is not None:
             self.f252 = f252
         if f253 is not None:
@@ -389,6 +399,18 @@ class V1UserFeatures(object):
             self.f257 = f257
         if f258 is not None:
             self.f258 = f258
+        if f259 is not None:
+            self.f259 = f259
+        if f260 is not None:
+            self.f260 = f260
+        if f261 is not None:
+            self.f261 = f261
+        if f262 is not None:
+            self.f262 = f262
+        if f263 is not None:
+            self.f263 = f263
+        if f264 is not None:
+            self.f264 = f264
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
@@ -457,8 +479,6 @@ class V1UserFeatures(object):
             self.security_docs = security_docs
         if show_dev_admin is not None:
             self.show_dev_admin = show_dev_admin
-        if single_wallet is not None:
-            self.single_wallet = single_wallet
         if slurm is not None:
             self.slurm = slurm
         if specialised_studios is not None:
@@ -1194,27 +1214,6 @@ class V1UserFeatures(object):
 
         self._f250 = f250
 
-    @property
-    def f251(self) -> 'bool':
-        """Gets the f251 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The f251 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._f251
-
-    @f251.setter
-    def f251(self, f251: 'bool'):
-        """Sets the f251 of this V1UserFeatures.
-
-
-        :param f251: The f251 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._f251 = f251
-
     @property
     def f252(self) -> 'bool':
         """Gets the f252 of this V1UserFeatures.  # noqa: E501
@@ -1341,6 +1340,132 @@ class V1UserFeatures(object):
 
         self._f258 = f258
 
+    @property
+    def f259(self) -> 'bool':
+        """Gets the f259 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f259 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f259
+
+    @f259.setter
+    def f259(self, f259: 'bool'):
+        """Sets the f259 of this V1UserFeatures.
+
+
+        :param f259: The f259 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f259 = f259
+
+    @property
+    def f260(self) -> 'bool':
+        """Gets the f260 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f260 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f260
+
+    @f260.setter
+    def f260(self, f260: 'bool'):
+        """Sets the f260 of this V1UserFeatures.
+
+
+        :param f260: The f260 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f260 = f260
+
+    @property
+    def f261(self) -> 'bool':
+        """Gets the f261 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f261 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f261
+
+    @f261.setter
+    def f261(self, f261: 'bool'):
+        """Sets the f261 of this V1UserFeatures.
+
+
+        :param f261: The f261 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f261 = f261
+
+    @property
+    def f262(self) -> 'bool':
+        """Gets the f262 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f262 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f262
+
+    @f262.setter
+    def f262(self, f262: 'bool'):
+        """Sets the f262 of this V1UserFeatures.
+
+
+        :param f262: The f262 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f262 = f262
+
+    @property
+    def f263(self) -> 'bool':
+        """Gets the f263 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f263 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f263
+
+    @f263.setter
+    def f263(self, f263: 'bool'):
+        """Sets the f263 of this V1UserFeatures.
+
+
+        :param f263: The f263 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f263 = f263
+
+    @property
+    def f264(self) -> 'bool':
+        """Gets the f264 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f264 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f264
+
+    @f264.setter
+    def f264(self, f264: 'bool'):
+        """Sets the f264 of this V1UserFeatures.
+
+
+        :param f264: The f264 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f264 = f264
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -2055,27 +2180,6 @@ class V1UserFeatures(object):
 
         self._show_dev_admin = show_dev_admin
 
-    @property
-    def single_wallet(self) -> 'bool':
-        """Gets the single_wallet of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The single_wallet of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._single_wallet
-
-    @single_wallet.setter
-    def single_wallet(self, single_wallet: 'bool'):
-        """Sets the single_wallet of this V1UserFeatures.
-
-
-        :param single_wallet: The single_wallet of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._single_wallet = single_wallet
-
     @property
     def slurm(self) -> 'bool':
         """Gets the slurm of this V1UserFeatures.  # noqa: E501

lightning_sdk/machine.py

@@ -6,7 +6,6 @@ from typing import Any, ClassVar, Optional, Tuple
 class CloudProvider(Enum):
     AWS = "AWS"
     GCP = "GCP"
-    VULTR = "VULTR"
     LAMBDA_LABS = "LAMBDA_LABS"
     DGX = "DGX"
     VOLTAGE_PARK = "VOLTAGE_PARK"

lightning_sdk/studio.py

@@ -8,6 +8,7 @@ from tqdm.auto import tqdm
 
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.api.studio_api import StudioApi
+from lightning_sdk.base_studio import BaseStudio
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.lightning_cloud.openapi import V1ClusterType
 from lightning_sdk.machine import CloudProvider, Machine
@@ -51,6 +52,8 @@ class Studio:
             If not specified, falls backto the teamspace default cloud account.
         create_ok: whether the studio will be created if it does not yet exist. Defaults to True
         provider: the provider of the machine, the studio should be created on.
+        studio_type: Type of studio to create. Only effective during initial creation;
+            ignored for existing studios.
 
     Note:
         Since a teamspace can either be owned by an org or by a user directly,
@@ -78,6 +81,7 @@ class Studio:
         source: Optional[str] = None,
         disable_secrets: bool = False,
         provider: Optional[Union[CloudProvider, str]] = None,  # deprecated in favor of cloud_provider
+        studio_type: Optional[str] = None,  # for base studio templates
     ) -> None:
         self._studio_api = StudioApi()
         self._cloud_account_api = CloudAccountApi()
@@ -113,6 +117,25 @@ class Studio:
                 default_cloud_account=self._teamspace.default_cloud_account,
             )
 
+        self._studio_type = None
+        if studio_type:
+            self._base_studio = BaseStudio()
+            self._available_base_studios = self._base_studio.list()
+            for bst in self._available_base_studios:
+                if (
+                    bst.id == studio_type
+                    or bst.name == studio_type
+                    or bst.name.lower().replace(" ", "-") == studio_type
+                ):
+                    self._studio_type = bst.id
+
+            if not self._studio_type:
+                raise ValueError(
+                    f"Could not find studio type with ID or name '{studio_type}'. "
+                    f"Available studio types: "
+                    f"{[bst.name.lower().replace(' ', '-') for bst in self._available_base_studios]}"
+                )
+
         # Resolve studio name if not provided: explicit → env (LIGHTNING_CLOUD_SPACE_ID) → config defaults
         if name is None and not getattr(self._skip_init, "value", False):
             studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
@@ -150,6 +173,7 @@ class Studio:
                         cloud_account=_cloud_account,
                         source=source,
                         disable_secrets=self._disable_secrets,
+                        cloud_space_environment_template_id=self._studio_type,
                     )
                 else:
                     raise e
@@ -218,6 +242,14 @@ class Studio:
             _get_org_id(self._teamspace),
         )
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Returns the public IP address of the machine the Studio is running on."""
+        return self._studio_api.get_public_ip(
+            self._studio.id,
+            self._teamspace.id,
+        )
+
     @property
     def interruptible(self) -> bool:
         """Returns whether the Studio is running on a interruptible instance."""
@@ -264,10 +296,11 @@ class Studio:
             else:
                 interruptible = self.teamspace.start_studios_on_interruptible
 
+        new_machine = machine
+        if not isinstance(machine, Machine):
+            new_machine = Machine.from_str(machine)
+
         if status == Status.Running:
-            new_machine = machine
-            if not isinstance(machine, Machine):
-                new_machine = Machine.from_str(machine)
             if new_machine != self.machine:
                 raise RuntimeError(
                     f"Requested to start {self._cls_name} on {new_machine}, "
@@ -289,7 +322,11 @@ class Studio:
             with StudioProgressTracker("start", show_progress=True) as progress:
                 # Start the studio without blocking
                 self._studio_api.start_studio_async(
-                    self._studio.id, self._teamspace.id, machine, interruptible=interruptible, max_runtime=max_runtime
+                    self._studio.id,
+                    self._teamspace.id,
+                    new_machine,
+                    interruptible=interruptible,
+                    max_runtime=max_runtime,
                 )
 
                 # Track progress through completion
@@ -299,7 +336,7 @@ class Studio:
         else:
             # Use the blocking version if no progress is needed
             self._studio_api.start_studio(
-                self._studio.id, self._teamspace.id, machine, interruptible=interruptible, max_runtime=max_runtime
+                self._studio.id, self._teamspace.id, new_machine, interruptible=interruptible, max_runtime=max_runtime
             )
 
         self._setup()

lightning_sdk/utils/progress.py

@@ -17,30 +17,31 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_st
 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"
+    STARTING_STUDIO = "Starting Studio..."
+    GETTING_MACHINE = "Getting a machine..."
 
+    SWITCHING_STUDIO = "Switching Studio..."
 
-def get_switching_progress_message(percentage: int, is_base_studio: bool, is_new_cloud_space: bool) -> str:
+    SETTING_UP_MACHINE = "Setting up machine..."
+    RESTORING_STUDIO = "Restoring Studio..."
+    PREPARING_STUDIO = "Preparing Studio..."
+    RESTORING_BASE_STUDIO = "Restoring Base Studio..."
+    SETTING_UP_BASE_STUDIO = "Setting up Base Studio..."
+    DONE = "Done"
+
+
+def get_switching_progress_message(percentage: int, is_base_studio: bool) -> str:
     """Get progress message for switching studios."""
     percentage = max(0, min(100, round(percentage)))
 
     if percentage > 98:
-        message = "Done"
+        message = StartupPhase.DONE.value
     elif percentage > 80:
-        if is_new_cloud_space:
-            message = "Setting up Base Studio..." if is_base_studio else "Preparing Studio..."
-        else:
-            message = "Restoring Studio..."
+        message = StartupPhase.RESTORING_BASE_STUDIO.value if is_base_studio else StartupPhase.RESTORING_STUDIO.value
     elif percentage > 60:
-        message = "Setting up machine from cloud provider"
+        message = StartupPhase.SETTING_UP_MACHINE.value
     else:
-        message = "Allocating machine from cloud provider"
-
+        message = StartupPhase.SWITCHING_STUDIO.value
     return f"({percentage}%) {message}"
 
 
@@ -110,15 +111,13 @@ class StudioProgressTracker:
         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:
+    def update_progress(self, percentage: int, message: str = "", is_base_studio: 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)
+            display_message = get_switching_progress_message(percentage, is_base_studio)
         else:
             display_message = message or f"{self.operation_type.capitalize()}ing Studio..."
 
@@ -153,7 +152,7 @@ class StudioProgressTracker:
         message_stability_delay = 3.0  # Seconds to wait before changing message
 
         # Show initial progress immediately
-        self.update_progress(5, StartupPhase.ALLOCATING_MACHINE.value)
+        self.update_progress(5, StartupPhase.STARTING_STUDIO.value)
 
         while True:
             try:
@@ -163,7 +162,7 @@ class StudioProgressTracker:
                 # 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
+                current_message = StartupPhase.STARTING_STUDIO.value
 
                 # Check if we have detailed status information
                 if hasattr(status, "in_use") and status.in_use:
@@ -178,7 +177,7 @@ class StudioProgressTracker:
                             hasattr(startup_status, "top_up_restore_finished")
                             and startup_status.top_up_restore_finished
                         ):
-                            self.complete(StartupPhase.COMPLETED.value)
+                            self.complete(StartupPhase.DONE.value)
                             break
 
                         # Check other phases in descending priority
@@ -186,14 +185,14 @@ class StudioProgressTracker:
                             hasattr(startup_status, "initial_restore_finished")
                             and startup_status.initial_restore_finished
                         ):
-                            current_progress = max(current_progress, 85)
-                            current_message = StartupPhase.FINALIZING_SETUP.value
+                            current_progress = max(current_progress, 80)
+                            current_message = StartupPhase.PREPARING_STUDIO.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
+                            current_message = StartupPhase.SETTING_UP_MACHINE.value
+                        elif hasattr(startup_status, "machine_ready") and startup_status.machine_ready:
+                            current_progress = max(current_progress, 30)
+                            current_message = StartupPhase.GETTING_MACHINE.value
 
                     # Check general phase information
                     if hasattr(in_use, "phase") and in_use.phase:
@@ -201,7 +200,7 @@ class StudioProgressTracker:
 
                         if phase == "CLOUD_SPACE_INSTANCE_STATE_RUNNING":
                             current_progress = max(current_progress, 80)
-                            current_message = StartupPhase.STUDIO_STARTING.value
+                            current_message = StartupPhase.SETTING_UP_MACHINE.value
                         elif phase == "CLOUD_SPACE_INSTANCE_STATE_PENDING":
                             # Track time in pending phase for smoother progress
                             if "pending" not in phase_start_times:
@@ -211,7 +210,7 @@ class StudioProgressTracker:
                             # 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
+                            current_message = StartupPhase.GETTING_MACHINE.value
 
                 # Check for requested machine status (pre-allocation)
                 elif hasattr(status, "requested") and status.requested:
@@ -222,7 +221,7 @@ class StudioProgressTracker:
                     # 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
+                    current_message = StartupPhase.GETTING_MACHINE.value
 
                 # Ensure progress never decreases and moves smoothly
                 if current_progress > last_progress:
@@ -262,12 +261,12 @@ class StudioProgressTracker:
 
                 # Only update message if enough time has passed
                 current_time = time.time()
-                should_update_message = StartupPhase.ALLOCATING_MACHINE.value != self._last_message and (
+                should_update_message = StartupPhase.GETTING_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)
+                    self.update_progress(fallback_progress, StartupPhase.GETTING_MACHINE.value)
                     last_message_time = current_time
                 else:
                     # Update progress but keep existing message

{lightning_sdk-2025.9.30.dist-info → lightning_sdk-2025.10.14.dist-info}/METADATA

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