lightning-sdk 2025.8.8__py3-none-any.whl → 2025.8.14__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -77,6 +77,7 @@ class V1UserFeatures(object):
         'f242': 'bool',
         'f243': 'bool',
         'f244': 'bool',
+        'f245': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
         'gcp_overprovisioning': 'bool',
@@ -92,8 +93,6 @@ class V1UserFeatures(object):
         'mmt_fault_tolerance': 'bool',
         'mmt_strategy_selector': 'bool',
         'model_api_dashboard': 'bool',
-        'model_api_dashboard_clickhouse': 'bool',
-        'multicloud_folders': 'bool',
         'multiple_studio_versions': 'bool',
         'nerf_fs_nonpaying': 'bool',
         'onboarding_v2': 'bool',
@@ -105,7 +104,6 @@ class V1UserFeatures(object):
         'plugin_label_studio': 'bool',
         'plugin_langflow': 'bool',
         'plugin_python_profiler': 'bool',
-        'plugin_service': 'bool',
         'plugin_sweeps': 'bool',
         'pricing_updates': 'bool',
         'product_generator': 'bool',
@@ -170,6 +168,7 @@ class V1UserFeatures(object):
         'f242': 'f242',
         'f243': 'f243',
         'f244': 'f244',
+        'f245': 'f245',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
         'gcp_overprovisioning': 'gcpOverprovisioning',
@@ -185,8 +184,6 @@ class V1UserFeatures(object):
         'mmt_fault_tolerance': 'mmtFaultTolerance',
         'mmt_strategy_selector': 'mmtStrategySelector',
         'model_api_dashboard': 'modelApiDashboard',
-        'model_api_dashboard_clickhouse': 'modelApiDashboardClickhouse',
-        'multicloud_folders': 'multicloudFolders',
         'multiple_studio_versions': 'multipleStudioVersions',
         'nerf_fs_nonpaying': 'nerfFsNonpaying',
         'onboarding_v2': 'onboardingV2',
@@ -198,7 +195,6 @@ class V1UserFeatures(object):
         'plugin_label_studio': 'pluginLabelStudio',
         'plugin_langflow': 'pluginLangflow',
         'plugin_python_profiler': 'pluginPythonProfiler',
-        'plugin_service': 'pluginService',
         'plugin_sweeps': 'pluginSweeps',
         'pricing_updates': 'pricingUpdates',
         'product_generator': 'productGenerator',
@@ -226,7 +222,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, academic_tier: 'bool' =None, add_data_v2: 'bool' =None, 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, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f244: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: '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, model_api_dashboard_clickhouse: 'bool' =None, multicloud_folders: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: '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_service: '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, r2_data_connections: '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, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, academic_tier: 'bool' =None, add_data_v2: 'bool' =None, 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, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f244: 'bool' =None, f245: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: '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, onboarding_v2: '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, r2_data_connections: '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, use_internal_data_connection_mounts: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._academic_tier = None
         self._add_data_v2 = None
@@ -264,6 +260,7 @@ class V1UserFeatures(object):
         self._f242 = None
         self._f243 = None
         self._f244 = None
+        self._f245 = None
         self._fair_share = None
         self._featured_studios_admin = None
         self._gcp_overprovisioning = None
@@ -279,8 +276,6 @@ class V1UserFeatures(object):
         self._mmt_fault_tolerance = None
         self._mmt_strategy_selector = None
         self._model_api_dashboard = None
-        self._model_api_dashboard_clickhouse = None
-        self._multicloud_folders = None
         self._multiple_studio_versions = None
         self._nerf_fs_nonpaying = None
         self._onboarding_v2 = None
@@ -292,7 +287,6 @@ class V1UserFeatures(object):
         self._plugin_label_studio = None
         self._plugin_langflow = None
         self._plugin_python_profiler = None
-        self._plugin_service = None
         self._plugin_sweeps = None
         self._pricing_updates = None
         self._product_generator = None
@@ -391,6 +385,8 @@ class V1UserFeatures(object):
             self.f243 = f243
         if f244 is not None:
             self.f244 = f244
+        if f245 is not None:
+            self.f245 = f245
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
@@ -421,10 +417,6 @@ class V1UserFeatures(object):
             self.mmt_strategy_selector = mmt_strategy_selector
         if model_api_dashboard is not None:
             self.model_api_dashboard = model_api_dashboard
-        if model_api_dashboard_clickhouse is not None:
-            self.model_api_dashboard_clickhouse = model_api_dashboard_clickhouse
-        if multicloud_folders is not None:
-            self.multicloud_folders = multicloud_folders
         if multiple_studio_versions is not None:
             self.multiple_studio_versions = multiple_studio_versions
         if nerf_fs_nonpaying is not None:
@@ -447,8 +439,6 @@ class V1UserFeatures(object):
             self.plugin_langflow = plugin_langflow
         if plugin_python_profiler is not None:
             self.plugin_python_profiler = plugin_python_profiler
-        if plugin_service is not None:
-            self.plugin_service = plugin_service
         if plugin_sweeps is not None:
             self.plugin_sweeps = plugin_sweeps
         if pricing_updates is not None:
@@ -1256,6 +1246,27 @@ class V1UserFeatures(object):
 
         self._f244 = f244
 
+    @property
+    def f245(self) -> 'bool':
+        """Gets the f245 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f245 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f245
+
+    @f245.setter
+    def f245(self, f245: 'bool'):
+        """Sets the f245 of this V1UserFeatures.
+
+
+        :param f245: The f245 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f245 = f245
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1571,48 +1582,6 @@ class V1UserFeatures(object):
 
         self._model_api_dashboard = model_api_dashboard
 
-    @property
-    def model_api_dashboard_clickhouse(self) -> 'bool':
-        """Gets the model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._model_api_dashboard_clickhouse
-
-    @model_api_dashboard_clickhouse.setter
-    def model_api_dashboard_clickhouse(self, model_api_dashboard_clickhouse: 'bool'):
-        """Sets the model_api_dashboard_clickhouse of this V1UserFeatures.
-
-
-        :param model_api_dashboard_clickhouse: The model_api_dashboard_clickhouse of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._model_api_dashboard_clickhouse = model_api_dashboard_clickhouse
-
-    @property
-    def multicloud_folders(self) -> 'bool':
-        """Gets the multicloud_folders of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The multicloud_folders of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._multicloud_folders
-
-    @multicloud_folders.setter
-    def multicloud_folders(self, multicloud_folders: 'bool'):
-        """Sets the multicloud_folders of this V1UserFeatures.
-
-
-        :param multicloud_folders: The multicloud_folders of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._multicloud_folders = multicloud_folders
-
     @property
     def multiple_studio_versions(self) -> 'bool':
         """Gets the multiple_studio_versions of this V1UserFeatures.  # noqa: E501
@@ -1844,27 +1813,6 @@ class V1UserFeatures(object):
 
         self._plugin_python_profiler = plugin_python_profiler
 
-    @property
-    def plugin_service(self) -> 'bool':
-        """Gets the plugin_service of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The plugin_service of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._plugin_service
-
-    @plugin_service.setter
-    def plugin_service(self, plugin_service: 'bool'):
-        """Sets the plugin_service of this V1UserFeatures.
-
-
-        :param plugin_service: The plugin_service of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._plugin_service = plugin_service
-
     @property
     def plugin_sweeps(self) -> 'bool':
         """Gets the plugin_sweeps of this V1UserFeatures.  # noqa: E501

lightning_sdk/llm/llm.py

@@ -5,6 +5,7 @@ from lightning_sdk.api import TeamspaceApi, UserApi
 from lightning_sdk.api.llm_api import LLMApi, authenticate
 from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
 from lightning_sdk.llm.public_assistants import PUBLIC_MODELS
+from lightning_sdk.utils.resolve import _resolve_org, _resolve_teamspace
 
 PUBLIC_MODEL_PROVIDERS: Dict[str, str] = {
     "openai": "OpenAI",
@@ -31,33 +32,33 @@ class LLM:
     ) -> None:
         """Initializes the LLM instance with teamspace information, which is required for billing purposes.
 
-        Teamspace information is resolved through the following methods:
-        1. `.lightning/credentials.json` - Attempts to retrieve the teamspace from the local credentials file.
-        2. Environment Variables - Checks for `LIGHTNING_*` environment variables.
-        3. User Authentication - Redirects the user to the login page if teamspace information is not found.
-
         Args:
             name (str): The name of the model or resource.
-            teamspace (Optional[str]): The specified teamspace for billing. If not provided, it will be resolved
-                                       through the above methods.
+            teamspace (Optional[str]): The specified teamspace for billing. If not provided, it will
+                                        use the available default teamspace.
             enable_async (Optional[bool]): Enable async requests
 
         Raises:
             ValueError: If teamspace information cannot be resolved.
         """
         teamspace_name = None
+        teamspace_owner = None
         if teamspace:
-            try:
-                owner, teamspace_name = teamspace.split("/", maxsplit=1)
-            except ValueError as e:
-                raise ValueError(
-                    f"Invalid teamspace format: '{teamspace}'. "
-                    "Teamspace should be specified as '{teamspace_owner}/{teamspace_name}' "
-                    "(e.g., 'my-org/my-teamspace')."
-                ) from e
+            if "/" in teamspace:
+                try:
+                    teamspace_owner, teamspace_name = teamspace.split("/", maxsplit=1)
+                except ValueError as e:
+                    raise ValueError(
+                        f"Invalid teamspace format: '{teamspace}'. "
+                        "Teamspace should be specified as '{org}/{teamspace_name}' or '{org}'"
+                        "(e.g., 'my-org/my-teamspace', 'my-org')."
+                    ) from e
+            else:
+                # org is given
+                teamspace_name = teamspace
 
         self._model_provider, self._model_name = self._parse_model_name(name)
-        self._get_auth_info(teamspace_name)
+        self._get_auth_info(teamspace_owner, teamspace_name)
 
         self._enable_async = enable_async
 
@@ -88,34 +89,63 @@ class LLM:
 
         return int(context_info["context_length"])
 
-    def _get_auth_info(self, teamspace_name: Optional[str] = None) -> None:
-        # TODO: Validate user input teamspace name
+    def _get_auth_info(self, teamspace_owner: Optional[str] = None, teamspace_name: Optional[str] = None) -> None:
         if not LLM._auth_info_cached:
-            if teamspace_name is None:
-                # studio users
-                teamspace_name = os.environ.get("LIGHTNING_TEAMSPACE", None)
-
-            if teamspace_name is None:
-                # local users with no given teamspace
+            if teamspace_owner and teamspace_name:
+                # org with specific teamspace
+                try:
+                    t = _resolve_teamspace(teamspace=teamspace_name, org=None, user=teamspace_owner)
+                except Exception:
+                    try:
+                        t = _resolve_teamspace(teamspace=teamspace_name, org=teamspace_owner, user=None)
+                    except Exception as err:
+                        raise ValueError(
+                            f"Teamspace {teamspace_owner}/{teamspace_name} not found."
+                            "Please verify owner name (username or organization) and the teamspace name are correct."
+                        ) from err
+
+                    os.environ["LIGHTNING_TEAMSPACE"] = t.name
+                    os.environ["LIGHTNING_CLOUD_PROJECT_ID"] = t.id
+
+            elif teamspace_name and teamspace_owner is None:
+                # if only org name is given, use the default teamspace
                 try:
-                    authenticate(model=f"{self.provider}/{self.name}")
+                    org = _resolve_org(teamspace_name)
                     teamspace_api = TeamspaceApi()
-                    user_api = UserApi()
-                    authed_user = user_api._client.auth_service_get_user()
-                    default_teamspace = teamspace_api.list_teamspaces(owner_id=authed_user.id)[0]
-                    teamspace_name = default_teamspace.name
-                    teamspace_id = default_teamspace.id
-                    os.environ["LIGHTNING_CLOUD_PROJECT_ID"] = teamspace_id
-                    os.environ["LIGHTNING_TEAMSPACE"] = teamspace_name
+                    teamspace = teamspace_api.list_teamspaces(org.id)[0]
+
                 except Exception as err:
-                    # throw an appropriate error that guides users to login through the platform
                     raise ValueError(
-                        "Teamspace information is missing. "
-                        "If this is your first time using LitAI, please log in at https://lightning.ai/sign-up "
-                        "and re-run your script, or set the environment variable LIGHTNING_TEAMSPACE=<your-teamspace>."
+                        f"Organization {teamspace_name} not found. Please verify the organization name."
                     ) from err
+                os.environ["LIGHTNING_CLOUD_PROJECT_ID"] = teamspace.id
+                os.environ["LIGHTNING_TEAMSPACE"] = teamspace.name
+
+            else:
+                if teamspace_name is None:
+                    # studio users
+                    teamspace_name = os.environ.get("LIGHTNING_TEAMSPACE", None)
+
+                if teamspace_name is None:
+                    # local users with no given teamspace
+                    try:
+                        authenticate(model=f"{self.provider}/{self.name}")
+                        teamspace_api = TeamspaceApi()
+                        user_api = UserApi()
+                        authed_user = user_api._client.auth_service_get_user()
+                        default_teamspace = teamspace_api.list_teamspaces(owner_id=authed_user.id)[0]
+                        teamspace_name = default_teamspace.name
+                        teamspace_id = default_teamspace.id
+                        os.environ["LIGHTNING_CLOUD_PROJECT_ID"] = teamspace_id
+                        os.environ["LIGHTNING_TEAMSPACE"] = teamspace_name
+                    except Exception as err:
+                        # throw an appropriate error that guides users to login through the platform
+                        raise ValueError(
+                            "Teamspace information is missing. "
+                            "If this is your first time using LitAI, please log in at https://lightning.ai/sign-up "
+                            "and re-run your script, or set environment variable LIGHTNING_TEAMSPACE=<your-teamspace>."
+                        ) from err
 
-            # TODO: when teamspace_name is given, we don't know the teamspace_id yet
             # TODO: if LIGHTNING_CLOUD_PROJECT_ID does not exist, we have to get the id from the teamspace name
 
             LLM._cached_auth_info = {

lightning_sdk/llm/public_assistants.py

@@ -35,15 +35,15 @@ PUBLIC_MODELS = {
         "id": "ast_01k1z7ajxbz7wq62xj5bc0mrcs",
         "context_length": 128000,
     },
-    "openai/gpt-5-2025-08-07": {
+    "openai/gpt-5": {
         "id": "ast_01jdjds71fs8gt47jexzed4cza",
         "context_length": 400000
     },
-    "openai/gpt-5-mini-2025-08-07": {
+    "openai/gpt-5-mini": {
         "id": "ast_01jdjds71fs8gt47jexzed4czb",
         "context_length": 400000
     },
-    "openai/gpt-5-nano-2025-08-07": {
+    "openai/gpt-5-nano": {
         "id": "ast_01jdjds71fs8gt47jexzed4czc",
         "context_length": 400000
     },

lightning_sdk/studio.py

@@ -57,6 +57,7 @@ class Studio:
 
     # skips init of studio, only set when using this as a shell for names, ids etc.
     _skip_init = False
+    _skip_setup = False
 
     def __init__(
         self,
@@ -82,7 +83,7 @@ class Studio:
         self._teamspace = _teamspace
         self._cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
-        self._setup_done = False
+        self._setup_done = self._skip_setup
         self._disable_secrets = disable_secrets
 
         self._plugins = {}
@@ -258,13 +259,17 @@ class Studio:
         """Deletes the current Studio."""
         self._studio_api.delete_studio(self._studio.id, self._teamspace.id)
 
-    def duplicate(self, target_teamspace: Optional[Union["Teamspace", str]] = None) -> "Studio":
+    def duplicate(
+        self, target_teamspace: Optional[Union["Teamspace", str]] = None, machine: Machine = Machine.CPU
+    ) -> "Studio":
         """Duplicates the existing Studio.
 
         Args:
             target_teamspace: the teamspace to duplicate the studio to.
                 Must have the same owner as the source teamspace.
                 If not provided, defaults to current teamspace.
+            machine: the machine to start the duplicated studio on.
+                Defaults to CPU
         """
         if target_teamspace is None:
             target_teamspace_id = self._teamspace.id
@@ -284,7 +289,10 @@ class Studio:
             target_teamspace_id = target_teamspace.id
 
         kwargs = self._studio_api.duplicate_studio(
-            studio_id=self._studio.id, teamspace_id=self._teamspace.id, target_teamspace_id=target_teamspace_id
+            studio_id=self._studio.id,
+            teamspace_id=self._teamspace.id,
+            target_teamspace_id=target_teamspace_id,
+            machine=machine,
         )
         return Studio(**kwargs)
 

lightning_sdk/utils/resolve.py

@@ -206,6 +206,19 @@ def skip_studio_init() -> Generator[None, None, None]:
     Studio._skip_init = prev_studio_init_state
 
 
+@contextmanager
+def skip_studio_setup() -> Generator[None, None, None]:
+    """Skip studio setup based on current runtime."""
+    from lightning_sdk.studio import Studio
+
+    prev_studio_init_state = Studio._skip_setup
+    Studio._skip_setup = True
+
+    yield
+
+    Studio._skip_setup = prev_studio_init_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.8.dist-info → lightning_sdk-2025.8.14.dist-info}/METADATA

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