lightning-sdk 2025.8.7__py3-none-any.whl → 2025.8.8.post1__py3-none-any.whl

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

@@ -1,11 +1,32 @@
 PUBLIC_MODELS = {
-    "openai/gpt-4o": {"id": "ast_01jdjds71fs8gt47jexzed4czs", "context_length": 128000},
-    "openai/gpt-4": {"id": "ast_01jd38ze6tjbrcd4942nhz41zn", "context_length": 8192},
-    "openai/o3-mini": {"id": "ast_01jz3t13fhnjhh11t1k8b5gyp1", "context_length": 128000},
-    "anthropic/claude-3-5-sonnet-20240620": {"id": "ast_01jd3923a6p98rqwh3dpj686pq", "context_length": 200000},
-    "google/gemini-2.5-pro": {"id": "ast_01jz3tdb1fhey798k95pv61v57", "context_length": 1048576},
-    "google/gemini-2.5-flash": {"id": "ast_01jz3thxskg4fcdk4xhkjkym5a", "context_length": 8000},
-    "google/gemini-2.5-flash-lite-preview-06-17": {"id": "ast_01jz3thxskg4fcdk4xhkjkym5b", "context_length": 8000},
+    "openai/gpt-4o": {
+        "id": "ast_01jdjds71fs8gt47jexzed4czs",
+        "context_length": 128000
+    },
+    "openai/gpt-4": {
+        "id": "ast_01jd38ze6tjbrcd4942nhz41zn",
+        "context_length": 8192
+    },
+    "openai/o3-mini": {
+        "id": "ast_01jz3t13fhnjhh11t1k8b5gyp1",
+        "context_length": 128000
+    },
+    "anthropic/claude-3-5-sonnet-20240620": {
+        "id": "ast_01jd3923a6p98rqwh3dpj686pq",
+        "context_length": 200000
+    },
+    "google/gemini-2.5-pro": {
+        "id": "ast_01jz3tdb1fhey798k95pv61v57",
+        "context_length": 1048576
+    },
+    "google/gemini-2.5-flash": {
+        "id": "ast_01jz3thxskg4fcdk4xhkjkym5a",
+        "context_length": 8000
+    },
+    "google/gemini-2.5-flash-lite-preview-06-17": {
+        "id": "ast_01jz3thxskg4fcdk4xhkjkym5b",
+        "context_length": 8000
+    },
     "lightning-ai/gpt-oss-20b": {
         "id": "ast_01k1y2ywfh9zzexjzrva6y96ms",
         "context_length": 128000,
@@ -14,4 +35,16 @@ PUBLIC_MODELS = {
         "id": "ast_01k1z7ajxbz7wq62xj5bc0mrcs",
         "context_length": 128000,
     },
+    "openai/gpt-5": {
+        "id": "ast_01jdjds71fs8gt47jexzed4cza",
+        "context_length": 400000
+    },
+    "openai/gpt-5-mini": {
+        "id": "ast_01jdjds71fs8gt47jexzed4czb",
+        "context_length": 400000
+    },
+    "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 = {}

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.7.dist-info → lightning_sdk-2025.8.8.post1.dist-info}/METADATA

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

{lightning_sdk-2025.8.7.dist-info → lightning_sdk-2025.8.8.post1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=aRSIYgnr41LLH9M5afQAlOTU4TBikwxsKWIefv98Tg8,1145
+lightning_sdk/__init__.py,sha256=IIU8VWs4YQHIWb5A96dDCl6_GuKm0TmT_HMBaDe5e-0,1151
 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=1oFy2tiLJWORVgyoVOU4eQf0WO0sjbj9zbFw16D_G6k,26465
+lightning_sdk/studio.py,sha256=8oAVdEr1yW4nhkXlDZf-tI5PoG3WFAMuvi1Ow-bmnF0,26500
 lightning_sdk/teamspace.py,sha256=4KLSCxkOIpkOXwvdFMeOgGjX8Ak41NpLkLkdLz7Wu2A,20099
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
@@ -67,7 +67,7 @@ lightning_sdk/cli/deploy/serve.py,sha256=-UzN3SWoMddfQu2Yu0viVDfj8woTTZuMW5Mddkd
 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
-lightning_sdk/job/base.py,sha256=GAE7wvTH9e_F9bU8sogT88EMCP4D0d1deZMeVmBtWTE,20856
+lightning_sdk/job/base.py,sha256=BJULMWcRPxjOQ0RuNpfDUhH2S5YntouYgrxBx3OGljk,21100
 lightning_sdk/job/job.py,sha256=jPlW3u34lJ6tpszm-Q6EHzWrfMIpQfM7-z8UCFFVyaE,14780
 lightning_sdk/job/v1.py,sha256=WISVvPs_j_g1oaNcbEOAua_JpaNO_9_BueJoPgeul6E,9618
 lightning_sdk/job/v2.py,sha256=TiRkTO41VqcV-RhARG93U3182kbcqlrNnsvQ0rdXKPI,11168
@@ -79,14 +79,14 @@ 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=TIHcypKt2NwQeUiiWp8rH6bDzZ4jlarRzEDR2b2OIyU,109792
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=9PHDtg_Lk8B4lwX6iHjEx4S79uTYOFXmpEcdLTbXoKM,110087
 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
 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
 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=weZ-oM8j_Lov_z1NpjaVhjELNrIVyuc1PGgDZS0P_qM,124703
+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=MCVMWkeAn9rAF6XWE-MVouqmxrJOzvjeIaUUGUG5Um4,78117
 lightning_sdk/lightning_cloud/openapi/api/blog_posts_service_api.py,sha256=T6ogreuE2ZC3RPzwJ-LLdK3OCxxPlhRQaI9PPI1pFfQ,20832
@@ -118,7 +118,7 @@ lightning_sdk/lightning_cloud/openapi/api/product_license_service_api.py,sha256=
 lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py,sha256=hM2vTawBBuOYZrxN1SjZ6mU5ldvB_jUXRcLb0as0F4M,28286
 lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=E7PM1ZlmuXMLNnIQ2aEc_AUi2ssSOi-eWA9IS2hbRbI,84822
 lightning_sdk/lightning_cloud/openapi/api/quest_service_api.py,sha256=IQppAcHoy8J99xdnZVDfyr-yzmFIkolMN3caF-wm7D0,16318
-lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py,sha256=lRG8s1o0sCZ-SuygA3pCUQfDWLQYJ5VRtkaaqD3w72o,23902
+lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py,sha256=x2j9pSInKp3jYvQkcNn_DUrA2gvwVU0IVP3ynvlTYw8,40368
 lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py,sha256=-oTJJ_XbfGAKle82anvK2XKSYE9OxRedCmUfJFjB9O0,41339
 lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py,sha256=Xw6x9q-nUSnid9vSTKnIColQD6U_cbbi2IzIxNVu_1E,45586
 lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py,sha256=C-gEj9pz_Rp3Yo9KnSEsOqNUi2FwzKMsCaPJugpIyLM,29981
@@ -127,7 +127,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=VvxM92snJUDluscSAYUAXnxbYtmUm4vCUnTgMTyfNJA,103700
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=h6g_uyqOXMEQ6SKefWnhH8ioLhuppFYEGMX9dknfrNw,103995
 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
@@ -155,7 +155,7 @@ lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py,sh
 lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py,sha256=wZ78MQff1tn27w-dcPjFxH16pjQ729a66DDIBffU-Bk,25123
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py,sha256=L_TAOS_SSt2mZaTUHXnvRmIYsI9SkLSLIPmhrMc5rac,8129
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py,sha256=-84-T6KNNGq7Vlks4FOUtKEC84otvW-NFOfA8MK0TNA,11677
-lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py,sha256=iBFGUq9KfwN2Bd3fxQConiD5_r1e_wy__09P3e8mX1Y,5776
+lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py,sha256=ltDDb9Hng7d8cQQROul_QWu6HsXAWrJOCNLKLULz7UQ,6706
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_proxies_body.py,sha256=Pm55MKTDAEqKPZq3HI6PDdLJAyxWl2ngXUNtPwTNscw,3680
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py,sha256=Jj-x-GrDyiUaQb47I6Glp_A6n29y4zSRnHWQkzqeZjc,6163
 lightning_sdk/lightning_cloud/openapi/models/cluster_id_usagerestrictions_body.py,sha256=jNGBUguIlFnnVb5wNMsBtcW3BcTtANw9wYmpYjMdoaI,6396
@@ -384,6 +384,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cluster_availability.py,sha256=W
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py,sha256=wzig5b0HGSGuydOc_2Cx-WGEY2wntB7eedzu4J44yJI,17135
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py,sha256=68iaqDjJYOpZsvaOzGJCsBVF4ISrFe8K9-o3bwr2tDI,5793
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_encryption_key.py,sha256=Cf9AIJCmKjQO0Gp7UbXIXq2IXkxLUBgSgK3sGSYzg8U,6681
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_metrics.py,sha256=vyPYindcuyxMkBhY5oVjOJK928I0gzpxvN8wB7-bx_U,27073
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py,sha256=Xs3mhVEiuJVbkKwMarY7iB_NwKY9_sK0xGgXDMuo0cs,3637
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py,sha256=4LbHvINs2CwQnuPg_uWw03GNNRT_wdqX42m_Zlqg2dY,5547
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_resource_tag.py,sha256=U6DfNNAIBZQ8SOF-2RXp2-CgmxaT1XJrNaCnZzaqjHg,4274
@@ -787,6 +788,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_list_published_cloud_spaces_resp
 lightning_sdk/lightning_cloud/openapi/models/v1_list_published_deployment_templates_response.py,sha256=InwtlpAJVFaWVQyox6IngoUObF5cLAodNhlLRgJc6DM,6067
 lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoints_response.py,sha256=VzUaevLtGmP-wbK2Y297Cge_RV3IyxCR23PwBolMyzM,4210
 lightning_sdk/lightning_cloud/openapi/models/v1_list_quests_response.py,sha256=YOAPh5y2CidpFRG0LTMOZkpBmK_Ic-SFWkNM8FpLGgA,3741
+lightning_sdk/lightning_cloud/openapi/models/v1_list_schedule_runs_response.py,sha256=i0unWHVl4akwO4Ggsgf2vpOx2Np64zqbJvaASB4hLSE,3976
 lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py,sha256=MXa_l9Pi5hUz1M5dEBMxhv9R_huK5w6kc-hy1I51CZk,3849
 lightning_sdk/lightning_cloud/openapi/models/v1_list_secrets_response.py,sha256=jclCKTTgv7HSHW_gMMNTJhXwO6dmY28AGTdT0q9hQzw,3777
 lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py,sha256=5XDwLB_U6eozzSjkhh6zg3Jt1AiyDcBDHYqLPmbAf2E,3897
@@ -940,6 +942,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py,sha
 lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py,sha256=iVl29exmQGrMC-q2GP2qcfjlsKp5vUJd6PZNkBPqSrQ,15782
 lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py,sha256=bbAckQHtEu2Y2O2FBqoI_ap4IFYfbbzeIGCWMv24qZQ,3135
 lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py,sha256=Dd31UlMVr8GR2TobjVDoNYXa8TaXAI1V7QkCb0QXE0I,3165
+lightning_sdk/lightning_cloud/openapi/models/v1_schedule_run.py,sha256=lHs9lzWwVytiKYoF98JHDvPkTTjGspMQH9qeGjD1fHM,10032
 lightning_sdk/lightning_cloud/openapi/models/v1_search_job_logs_response.py,sha256=SIoZOP_UDvybzADuo8rCRcAtie7-GIbw-TyKJ00XKWQ,4591
 lightning_sdk/lightning_cloud/openapi/models/v1_search_user.py,sha256=L2omKao6RHEWdKmLyTWHkJJB-PrichTJ8SqSJE68MAM,11623
 lightning_sdk/lightning_cloud/openapi/models/v1_search_users_response.py,sha256=55lMWBQYE0ED28pZLVaiZbkW3NrVjLfPnW_UR-fQDRE,6355
@@ -1038,7 +1041,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=0BnolF-EjEQRkxMfJ6whJpQrKwAWdZHbrBRStkauuSI,78408
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=Xan5L3rBXqqYn_EIrS6PcxwYAqpPODQf55ZiV4YO4nY,78129
 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
@@ -1081,8 +1084,8 @@ 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=2pBLt2_WZDTkgVJaBz-mfY4P6PFuvwLubY1X9uqECxc,15851
-lightning_sdk/llm/public_assistants.py,sha256=ZhzHGJtg_XhPKj4w9xtjHDuBW_iSUUoqGfXqvM8kL8c,955
+lightning_sdk/llm/llm.py,sha256=an7VstxZdP50q4-W8bmvb624uz2MnSTnk93UYlIxopg,17305
+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
 lightning_sdk/mmt/mmt.py,sha256=CZK0KopnW8j-oez464bYMxc0UbqzsYazsNTq1vBtdlg,14982
@@ -1102,10 +1105,10 @@ lightning_sdk/services/utilities.py,sha256=WlzKDGqUrQFcT85lnLgHvexf7AB1ppxS4_RBk
 lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.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,,
+lightning_sdk/utils/resolve.py,sha256=9BHOxGtDMptkrooRZwu1FkXbYQe5VAIumWGQlz2WIy8,8405
+lightning_sdk-2025.8.8.post1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.8.post1.dist-info/METADATA,sha256=I5VArnccdkREQ3S6am1i3GRZi4T50N6HHQ79nF_hk_I,4113
+lightning_sdk-2025.8.8.post1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.8.post1.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.8.post1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.8.post1.dist-info/RECORD,,