lightning-sdk 2025.8.6rc1__py3-none-any.whl → 2025.8.6rc2__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.06rc1"
+__version__ = "2025.08.06rc2"
 _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/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()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.8.6rc1
+Version: 2025.8.6rc2
 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.6rc2.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=RU76LC0UCj27gaVXAzO6X0pZVp3K7w14hQhPwc_7rOA,1148
 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
@@ -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=zWUUymFi2gzx4e3M-Uv3vrEPPAnBACcdQJGylIDoRbk,15659
 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.6rc2.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.6rc2.dist-info/METADATA,sha256=_2gP2vLcex82G8EH6aWUzvqJNAhkf0nCFI1FqcqvqmU,4110
+lightning_sdk-2025.8.6rc2.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.6rc2.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.6rc2.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.6rc2.dist-info/RECORD,,