lightning-sdk 2025.8.6rc1__py3-none-any.whl → 2025.8.7__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.06rc1"
+__version__ = "2025.08.07"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/llm_api.py

@@ -4,9 +4,15 @@ import datetime
 import json
 import os
 import threading
+import time
 import warnings
 from typing import Any, AsyncGenerator, Dict, Generator, List, Optional, Union
+from urllib.parse import urlencode
 
+from rich.prompt import Confirm
+
+from lightning_sdk.lightning_cloud import env
+from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi.models import (
     StreamResultOfV1ConversationResponseChunk,
     V1ConversationResponseChunk,
@@ -15,6 +21,48 @@ from lightning_sdk.lightning_cloud.openapi.models import (
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
+LITAI_CODE = os.environ.get("LITAI_CODE", "x334uv8t7v")
+
+
+class _AuthServer(AuthServer):
+    def __init__(self, model: str, *args: Any, **kwargs: Any) -> None:
+        self._model = model
+        super().__init__(*args, **kwargs)
+
+    def get_auth_url(self, port: int) -> str:
+        redirect_uri = f"http://localhost:{port}/login-complete"
+        params = urlencode({"redirectTo": redirect_uri, "okbhrt": LITAI_CODE, "litaimodel": self._model})
+        return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
+
+
+class _AuthLitAI(Auth):
+    def __init__(self, model: str, shall_confirm: bool = False) -> None:
+        super().__init__()
+        self._model = model
+        self._shall_confirm = shall_confirm
+
+    def _run_server(self) -> None:
+        if self._shall_confirm:
+            proceed = Confirm.ask(
+                "[bold yellow]LitAI needs to authenticate with Lightning AI.[/bold yellow]\n"
+                "This will open a browser window for login.\n"
+                "Do you want to continue?",
+                default=True,
+            )
+            if not proceed:
+                raise RuntimeError(
+                    "Login cancelled. Please login at https://lightning.ai/sign-up. Or run `lightning login` to login."
+                )
+        print("Opening browser for authentication...")
+        print("Please come back to the terminal after logging in.")
+        time.sleep(1)
+        _AuthServer(self._model).login_with_browser(self)
+
+
+def authenticate(model: str, shall_confirm: bool = True) -> None:
+    auth = _AuthLitAI(model, shall_confirm)
+    auth.authenticate()
+
 
 class LLMApi:
     def __init__(self) -> None:

lightning_sdk/cli/clusters_menu.py

@@ -1,12 +1,12 @@
 import sys
-from typing import List
+from typing import List, Optional
 
 from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk import Teamspace
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
-from lightning_sdk.lightning_cloud.openapi import Externalv1Cluster, V1ProjectClusterBinding
+from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1ProjectClusterBinding
 
 
 class _ClustersMenu:
@@ -22,18 +22,21 @@ class _ClustersMenu:
 
         return TerminalMenu(cluster_ids, title=title, clear_menu_on_exit=True)
 
-    def _resolve_cluster(self, teamspace: Teamspace) -> Externalv1Cluster:
+    def _resolve_cluster(self, teamspace: Teamspace) -> Optional[str]:
         selected_cluster_id = None
         console = Console()
         try:
-            selected_cluster_id = self._get_cluster_from_interactive_menu(
+            selected_cluster_id = teamspace.default_cloud_account or self._get_cluster_from_interactive_menu(
                 possible_clusters=teamspace.cloud_account_objs
             )
+
             cloud_account_api = CloudAccountApi()
 
-            return cloud_account_api.get_cloud_account(
-                cluster_id=selected_cluster_id, org_id=teamspace.owner.id, project_id=teamspace.id
+            resolved_cluster_obj = cloud_account_api.get_cloud_account(
+                cloud_account_id=selected_cluster_id, org_id=teamspace.owner.id, teamspace_id=teamspace.id
             )
+
+            return None if resolved_cluster_obj.spec.cluster_type == V1ClusterType.GLOBAL else resolved_cluster_obj.id
         except KeyboardInterrupt:
             console.print("Operation cancelled by user")
             sys.exit(0)

lightning_sdk/cli/deploy/serve.py

@@ -15,6 +15,7 @@ from rich.prompt import Confirm
 from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_registry_url
+from lightning_sdk.cli.clusters_menu import _ClustersMenu
 from lightning_sdk.cli.deploy._auth import (
     _AuthMode,
     _Onboarding,
@@ -377,18 +378,24 @@ def _handle_cloud(
     else:
         resolved_teamspace = select_teamspace(teamspace, org, user)
 
+    lightning_containers_cloud_account = cloud_account
+    if not cloud_account:
+        clusters_menu = _ClustersMenu()
+        lightning_containers_cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
+        cloud_account = resolved_teamspace.default_cloud_account
+
     # list containers to create the project if it doesn't exist
     lit_cr = LitContainerApi()
-    lit_cr.list_containers(resolved_teamspace.id, cloud_account=cloud_account)
+    lit_cr.list_containers(resolved_teamspace.id, cloud_account=lightning_containers_cloud_account)
 
     registry_url = _get_registry_url()
     container_basename = repository.split("/")[-1]
     image = (
-        f"{registry_url}/lit-container{f'-{cloud_account}' if cloud_account is not None else ''}/"
-        f"{resolved_teamspace.owner.name}/{resolved_teamspace.name}/{container_basename}"
+        f"{registry_url}/lit-container"
+        + (f"-{lightning_containers_cloud_account}" if lightning_containers_cloud_account is not None else "")
+        + f"/{resolved_teamspace.owner.name}/{resolved_teamspace.name}/{container_basename}"
     )
 
-    cloud_account = cloud_account or resolved_teamspace.default_cloud_account
     if from_onboarding:
         thread = Thread(
             target=ls_deployer.run_on_cloud,
@@ -411,13 +418,17 @@ def _handle_cloud(
         )
         thread.start()
         console.print("🚀 Deployment started")
-        if not _upload_container(console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, cloud_account):
+        if not _upload_container(
+            console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, lightning_containers_cloud_account
+        ):
             thread.join()
             return
         thread.join()
         return
 
-    if not _upload_container(console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, cloud_account):
+    if not _upload_container(
+        console, ls_deployer, repository, tag, resolved_teamspace, lit_cr, lightning_containers_cloud_account
+    ):
         return
 
     deployment_status = ls_deployer.run_on_cloud(

lightning_sdk/cli/download.py

@@ -136,7 +136,7 @@ def folder(
     "--studio",
     default=None,
     help=(
-        "The name of the studio to upload to. "
+        "The name of the studio to download from. "
         "Will show a menu with user's owned studios for selection if not specified. "
         "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
         "The teamspace and studio names can be regular expressions to match, "
@@ -256,7 +256,7 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
     # give user friendlier error message
     except Exception as e:
         raise StudioCliError(
-            f"Could not find the given Studio {studio} to upload files to. "
+            f"Could not find the given Studio {studio} to download files from. "
             "Please contact Lightning AI directly to resolve this issue."
         ) from e
 

lightning_sdk/cli/list.py

@@ -9,7 +9,7 @@ from typing_extensions import Literal
 from lightning_sdk import Job, Machine, Studio, Teamspace
 from lightning_sdk.cli.clusters_menu import _ClustersMenu
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1MultiMachineJob
+from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -247,8 +247,7 @@ def containers(teamspace: Optional[str] = None, cloud_account: Optional[str] = N
     resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
 
     if not cloud_account:
-        cloud_account_obj = clusters_menu._resolve_cluster(resolved_teamspace)
-        cloud_account = "" if cloud_account_obj.spec.cluster_type == V1ClusterType.GLOBAL else cloud_account_obj.id
+        cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
 
     result = api.list_containers(
         teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name, cloud_account=cloud_account

lightning_sdk/lightning_cloud/openapi/configuration.py

@@ -1,22 +1,5 @@
 # coding: utf-8
 
-"""
-    external/v1/auth_service.proto
-
-    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
-
-    OpenAPI spec version: version not set
-    
-    Generated by: https://github.com/swagger-api/swagger-codegen.git
-
-    NOTE
-    ----
-    standard swagger-codegen-cli for this python client has been modified
-    by custom templates. The purpose of these templates is to include
-    typing information in the API and Model code. Please refer to the
-    main grid repository for more info
-"""
-
 from __future__ import absolute_import
 
 import copy
@@ -100,8 +83,9 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
         # per pool. urllib3 uses 1 connection as default value, but this is
         # not the best value when you are making a lot of possibly parallel
         # requests to the same host, which is often the case here.
-        # cpu_count * 5 is used as default value to increase performance.
-        self.connection_pool_maxsize = multiprocessing.cpu_count() * 5
+        # the max of cpu_count * 5 and 500 is used as default value to increase performance.
+        # each re-used connection can use memory somewhere in the range of 50–100 KB by default.
+        self.connection_pool_maxsize = max(multiprocessing.cpu_count() * 5, 500)
 
         # Proxy URL
         self.proxy = None

lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py

@@ -44,6 +44,8 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'float',
         'cpu_util': 'float',
         'filesystem': 'dict(str, V1FilesystemMetrics)',
+        'mem_total': 'float',
+        'mem_util': 'float',
         'node_name': 'str',
         'per_gpu_mem_free': 'dict(str, float)',
         'per_gpu_mem_used': 'dict(str, float)',
@@ -57,6 +59,8 @@ class V1NodeMetrics(object):
         'cpu_capacity_cores': 'cpuCapacityCores',
         'cpu_util': 'cpuUtil',
         'filesystem': 'filesystem',
+        'mem_total': 'memTotal',
+        'mem_util': 'memUtil',
         'node_name': 'nodeName',
         'per_gpu_mem_free': 'perGpuMemFree',
         'per_gpu_mem_used': 'perGpuMemUsed',
@@ -66,11 +70,13 @@ class V1NodeMetrics(object):
         'timestamp': 'timestamp'
     }
 
-    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
+    def __init__(self, cpu_capacity_cores: 'float' =None, cpu_util: 'float' =None, filesystem: 'dict(str, V1FilesystemMetrics)' =None, mem_total: 'float' =None, mem_util: 'float' =None, node_name: 'str' =None, per_gpu_mem_free: 'dict(str, float)' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, timestamp: 'datetime' =None):  # noqa: E501
         """V1NodeMetrics - a model defined in Swagger"""  # noqa: E501
         self._cpu_capacity_cores = None
         self._cpu_util = None
         self._filesystem = None
+        self._mem_total = None
+        self._mem_util = None
         self._node_name = None
         self._per_gpu_mem_free = None
         self._per_gpu_mem_used = None
@@ -85,6 +91,10 @@ class V1NodeMetrics(object):
             self.cpu_util = cpu_util
         if filesystem is not None:
             self.filesystem = filesystem
+        if mem_total is not None:
+            self.mem_total = mem_total
+        if mem_util is not None:
+            self.mem_util = mem_util
         if node_name is not None:
             self.node_name = node_name
         if per_gpu_mem_free is not None:
@@ -163,6 +173,48 @@ class V1NodeMetrics(object):
 
         self._filesystem = filesystem
 
+    @property
+    def mem_total(self) -> 'float':
+        """Gets the mem_total of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The mem_total of this V1NodeMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._mem_total
+
+    @mem_total.setter
+    def mem_total(self, mem_total: 'float'):
+        """Sets the mem_total of this V1NodeMetrics.
+
+
+        :param mem_total: The mem_total of this V1NodeMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._mem_total = mem_total
+
+    @property
+    def mem_util(self) -> 'float':
+        """Gets the mem_util of this V1NodeMetrics.  # noqa: E501
+
+
+        :return: The mem_util of this V1NodeMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._mem_util
+
+    @mem_util.setter
+    def mem_util(self, mem_util: 'float'):
+        """Sets the mem_util of this V1NodeMetrics.
+
+
+        :param mem_util: The mem_util of this V1NodeMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._mem_util = mem_util
+
     @property
     def node_name(self) -> 'str':
         """Gets the node_name of this V1NodeMetrics.  # noqa: E501

lightning_sdk/llm/llm.py

@@ -2,7 +2,7 @@ import os
 from typing import Any, AsyncGenerator, ClassVar, Dict, Generator, List, Optional, Tuple, Union
 
 from lightning_sdk.api import TeamspaceApi, UserApi
-from lightning_sdk.api.llm_api import LLMApi
+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
 
@@ -56,9 +56,9 @@ class LLM:
                     "(e.g., 'my-org/my-teamspace')."
                 ) from e
 
+        self._model_provider, self._model_name = self._parse_model_name(name)
         self._get_auth_info(teamspace_name)
 
-        self._model_provider, self._model_name = self._parse_model_name(name)
         self._enable_async = enable_async
 
         # Reuse LLMApi per teamspace (as billing is based on teamspace)
@@ -98,6 +98,7 @@ class LLM:
             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()
@@ -137,16 +138,21 @@ class LLM:
         self._cloud_url = LLM._cached_auth_info["cloud_url"]
         self._org = None
 
-    def _parse_model_name(self, name: str) -> Tuple[str, str]:
-        parts = name.split("/")
-        if len(parts) == 1:
-            # a user model or a org model
-            return None, parts[0]
-        if len(parts) == 2:
-            return parts[0].lower(), parts[1]
-        raise ValueError(
-            f"Model name must be in the format `organization/model_name` or `model_name`, but got '{name}'."
-        )
+    @staticmethod
+    def _parse_model_name(name: str) -> Tuple[str, str]:
+        """Parses the model name into provider and model name.
+
+        >>> LLM._parse_model_name("openai/v1/gpt-3.5-turbo")
+        ('openai', 'v1/gpt-3.5-turbo')
+        """
+        if "/" not in name:
+            raise ValueError(
+                f"Invalid model name format: '{name}'. "
+                "Model name must be in the format `provider/model_name`."
+                "(e.g., 'lightning-ai/gpt-oss-20b')"
+            )
+        provider, model_name = name.split("/", maxsplit=1)
+        return provider.lower(), model_name
 
     # returns the assistant ID
     def _get_model_id(self) -> str:

{lightning_sdk-2025.8.6rc1.dist-info → lightning_sdk-2025.8.7.dist-info}/METADATA

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

{lightning_sdk-2025.8.6rc1.dist-info → lightning_sdk-2025.8.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=qnwB64Jaqr_ltm_0_b24mmEhBByB07nWnm-woXvM0GI,1148
+lightning_sdk/__init__.py,sha256=aRSIYgnr41LLH9M5afQAlOTU4TBikwxsKWIefv98Tg8,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
@@ -26,7 +26,7 @@ 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=oc3OB1l0Jca5Nc6MAkfI-Nq-42hXJkgmkSqn22J4TeA,9261
+lightning_sdk/api/llm_api.py,sha256=HSCpTyPknKAv6j0UYIzasXK2mFYt7FE3KF5TGUYvo2w,11049
 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
@@ -36,21 +36,21 @@ lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0
 lightning_sdk/api/utils.py,sha256=zUvTxQ01rGwZX2PNuA2V2l2esXLPSoGM_dl4AvfKTAE,24090
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/ai_hub.py,sha256=8MxGQC7SZALAsIj0tZ9HMZCeVHESUcve0cNazRzIFbE,1812
-lightning_sdk/cli/clusters_menu.py,sha256=AVlaWggNYU_-RZu9egagHB1JElDArx8WZMVH1DKPm2A,1801
+lightning_sdk/cli/clusters_menu.py,sha256=sj5FD_GDHOAyJZjltiS9_C3shPpu9fGQ8lcSsuhFQlg,1983
 lightning_sdk/cli/coloring.py,sha256=YEb1LiYb6CekVfCRb83-npAmNgd-7c1LzXuNDo4GLSc,2415
 lightning_sdk/cli/configure.py,sha256=ZZkCYmnxrRSiZdpbNMLbnxLGy4i4ce7lEcV3NVvYfp8,4457
 lightning_sdk/cli/connect.py,sha256=LjCi_0ZJkTm8IbIzLyIL0SATEQ2bWdVb6aARXnYXcVw,988
 lightning_sdk/cli/create.py,sha256=hpts8ZjyoDEj6gG3nST7qWlZBPtN1-y4EUAfp7R_BDs,4030
 lightning_sdk/cli/delete.py,sha256=zHMAkLzt6RgF1cmM-XC6mz8TyBgvCul0FeQFK3cIOwY,3812
 lightning_sdk/cli/docker_cli.py,sha256=EozLC4qnvHhgukmnu35itfI5n7v-abCpwKPkPy3eOV4,997
-lightning_sdk/cli/download.py,sha256=v7IECH3boKcs-_02YvbeHfYXr7KsB-YrtJP4vJH3-rU,11385
+lightning_sdk/cli/download.py,sha256=eJmOZcGVDGFCH8gTdCe_XEoUOvQA1SKbFq9sNBB5vjw,11393
 lightning_sdk/cli/entrypoint.py,sha256=aOPjtZioylgORVWJk76fSh4EYMFeyqSsdI0-i5YIx7s,3652
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/generate.py,sha256=8mjm5tMYIXMoN2X_eEKpwBIUcCWaARiQAmreh-9LIBg,1575
 lightning_sdk/cli/inspection.py,sha256=GA3XDC3DSkqJ4B5NQR9qmGGvkP0K8Sk90xZhzTGzC1o,1614
 lightning_sdk/cli/job_and_mmt_action.py,sha256=OV_i5K3g-NnMUF8Sq6QliQdltnhGVRdtoyZqJVXu4Ss,1659
 lightning_sdk/cli/jobs_menu.py,sha256=_2qsOBTdIKT5NCjwrBo9-AAn3fIvdN9IcVUVHF_KGQk,2162
-lightning_sdk/cli/list.py,sha256=Jc793aG0LZ9iFttv0fQZOqdh51Fgi53hEAyNkpwRsLg,10992
+lightning_sdk/cli/list.py,sha256=VEFLG1_kQjpcbmENoWbzhR-Xqn-3GiHv7H_Mh1S6vxc,10857
 lightning_sdk/cli/mmts_menu.py,sha256=HUXo3ZoZ3fWOCNWTQWoJgUlFXYq5uVm_6uFjAq7BDe8,2219
 lightning_sdk/cli/open.py,sha256=zdQ3E9Wbi5f5wOoJqZ3yFL45kWysGaza-rCPUinrOnA,2747
 lightning_sdk/cli/run.py,sha256=CG5efVA_lMx5J31aomPE3k5JTvcYXrsGo9uXO6Swn94,13963
@@ -63,7 +63,7 @@ lightning_sdk/cli/upload.py,sha256=b_GbSJB9UP7E4w3QE0laNh6urbcAs6wJtHJJ1iVqVcM,1
 lightning_sdk/cli/deploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/deploy/_auth.py,sha256=ccIzl0W9hJtFmhdEAn1xcBGIHQfO4d_k8grTbq4ufIs,7711
 lightning_sdk/cli/deploy/devbox.py,sha256=HOQMWehwaveawVTyBDaZhNZs_NL5g4RCWs_lf0hIW5U,6800
-lightning_sdk/cli/deploy/serve.py,sha256=K2e7gz1gg_UTr0j5FuI1_EKRa0t6hRaoHGpwFAXRSso,15040
+lightning_sdk/cli/deploy/serve.py,sha256=-UzN3SWoMddfQu2Yu0viVDfj8woTTZuMW5MddkdXNko,15457
 lightning_sdk/deployment/__init__.py,sha256=dXsa4psDzFYFklsq3JC-2V_L4FQjGZnQAf-ZiVlqG9c,545
 lightning_sdk/deployment/deployment.py,sha256=7Sk4ORKmgMDk7Owpk_q-89ws8-Y9orX0jRZF99tNv1E,21775
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
@@ -81,7 +81,7 @@ lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
 lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=TIHcypKt2NwQeUiiWp8rH6bDzZ4jlarRzEDR2b2OIyU,109792
 lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
-lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=8KpFKEm-IKe3t0_o5H1Y0EJX06r94Gk5cIQerA0TNk0,8292
+lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=SkBPJ3WZ9SF1LxxWeGKa4UwbGRsk9p3-yL_Kn7l5CSM,7866
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
 lightning_sdk/lightning_cloud/openapi/api/__init__.py,sha256=JRILuZsGz6w_N3114hY5C1JuqXZmum3zRmWkv0ZtM20,4327
 lightning_sdk/lightning_cloud/openapi/api/agent_service_api.py,sha256=Va0QcK6cO3rgHxV16xI-fnq08mgZe_eFQl8mgyaZols,56603
@@ -843,7 +843,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_named_get_logger_metrics.py,sha2
 lightning_sdk/lightning_cloud/openapi/models/v1_nebius_direct_v1.py,sha256=FXjBRxuNMVlNopuedGI5DV8y6Boy3CKoen6Odg9CnVk,4513
 lightning_sdk/lightning_cloud/openapi/models/v1_network_config.py,sha256=bHDZVIsqgiSi7eBf72RunvrLR-d-4b9S7oDMAZFBvCM,5476
 lightning_sdk/lightning_cloud/openapi/models/v1_new_feature.py,sha256=FMsA5E3T_94-IoZ7Dv-8N3agoZwmP7wTeFO443L_2Jw,10716
-lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py,sha256=setc9e61vS9LxKlfR6BcLM77ecTP_GF_OGt1xWLmc00,11683
+lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py,sha256=47bu85lUHuGgLYuQwmHeh8yUfhG0K_btRA6u9ppfe-Q,13095
 lightning_sdk/lightning_cloud/openapi/models/v1_notification_preference.py,sha256=W46XDkg7j9X8V9uRu1LHcPK1jSPTSzBVVEmbJ4FVofY,6930
 lightning_sdk/lightning_cloud/openapi/models/v1_notification_preferences_request.py,sha256=CHgLj8xQ5S-v5fTqNP6d0YrJbWEvlAQ6HZ3gxsXQ428,6497
 lightning_sdk/lightning_cloud/openapi/models/v1_notification_type.py,sha256=sTNMa_VKfBaZq6_itBewEYNTqdmBC2F9KIgHDJFai44,3251
@@ -1081,7 +1081,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=doRKMKceSxR0qN7uvwMc0qEl-wCd-Dn66EKQOwTk2V8,15574
+lightning_sdk/llm/llm.py,sha256=2pBLt2_WZDTkgVJaBz-mfY4P6PFuvwLubY1X9uqECxc,15851
 lightning_sdk/llm/public_assistants.py,sha256=ZhzHGJtg_XhPKj4w9xtjHDuBW_iSUUoqGfXqvM8kL8c,955
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=iqVudKDxazomuezj6l7pa-m9I1EQnM82OKs4ZQbo4Ck,16434
@@ -1103,9 +1103,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=JjHwWNQq375d1JBcVkJrZWkcTehvuTxF2h6awvSUDoo,8094
-lightning_sdk-2025.8.6rc1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.8.6rc1.dist-info/METADATA,sha256=JEU93V1stWNTBz13PGuOwpa4nIs_7UVMhreZt93DH2M,4110
-lightning_sdk-2025.8.6rc1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.8.6rc1.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-2025.8.6rc1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.8.6rc1.dist-info/RECORD,,
+lightning_sdk-2025.8.7.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.7.dist-info/METADATA,sha256=6RvCT65xX5k0-DlSRXMpQPbQnfrvmiksZHU-gZliv1A,4107
+lightning_sdk-2025.8.7.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.7.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.7.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.7.dist-info/RECORD,,