lightning-sdk 2025.9.11__py3-none-any.whl → 2025.9.16__py3-none-any.whl

lightning_sdk/__init__.py

@@ -10,7 +10,7 @@ from lightning_sdk.organization import Organization
 from lightning_sdk.plugin import JobsPlugin, MultiMachineTrainingPlugin, Plugin, SlurmJobsPlugin
 from lightning_sdk.status import Status
 from lightning_sdk.studio import Studio
-from lightning_sdk.teamspace import Teamspace
+from lightning_sdk.teamspace import FolderLocation, Teamspace
 from lightning_sdk.user import User
 
 __all__ = [
@@ -29,9 +29,10 @@ __all__ = [
     "Status",
     "Studio",
     "Teamspace",
+    "FolderLocation",
     "User",
 ]
 
-__version__ = "2025.09.11"
+__version__ = "2025.09.16"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/__init__.py

@@ -1,5 +1,6 @@
 from lightning_sdk.api.agents_api import AgentApi
 from lightning_sdk.api.ai_hub_api import AIHubApi
+from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.api.org_api import OrgApi
 from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.api.teamspace_api import TeamspaceApi
@@ -12,4 +13,5 @@ __all__ = [
     "UserApi",
     "AgentApi",
     "AIHubApi",
+    "CloudAccountApi",
 ]

lightning_sdk/api/cloud_account_api.py

@@ -131,13 +131,20 @@ class CloudAccountApi:
         )
         return list(filtered_cloud_accounts)
 
-    def get_cloud_account_provider_mapping(self, teamspace_id: str) -> Dict["CloudProvider", str]:
+    def get_cloud_account_provider_mapping(self, teamspace_id: str) -> Dict["CloudProvider", V1ExternalCluster]:
         """Gets the cloud account <-> provider mapping."""
         res = self.list_global_cloud_accounts(teamspace_id=teamspace_id)
-        return {self._get_cloud_account_provider(cloud_account): cloud_account.id for cloud_account in res}
+        cloud_accounts = {cloud_account.id: cloud_account for cloud_account in res}
+        providers = {cloud_account.id: self._get_cloud_account_provider(cloud_account) for cloud_account in res}
+
+        mapping = {}
+        for cloud_account_id, provider in providers.items():
+            if provider is not None:
+                mapping[provider] = cloud_accounts[cloud_account_id]
+        return mapping
 
     @staticmethod
-    def _get_cloud_account_provider(cloud_account: Optional[V1ExternalCluster]) -> "CloudProvider":
+    def _get_cloud_account_provider(cloud_account: Optional[V1ExternalCluster]) -> Optional["CloudProvider"]:
         """Determines the cloud provider based on the cloud_account configuration.
 
         Args:
@@ -172,7 +179,7 @@ class CloudAccountApi:
             if cloud_account.spec.nebius_v1:
                 return CloudProvider.NEBIUS
 
-        return CloudProvider.AWS
+        return None
 
     def resolve_cloud_account(
         self,
@@ -201,7 +208,7 @@ class CloudAccountApi:
         if cloud_provider:
             cloud_account_mapping = self.get_cloud_account_provider_mapping(teamspace_id=teamspace_id)
             if cloud_provider and cloud_provider in cloud_account_mapping:
-                return cloud_account_mapping[cloud_provider]
+                return cloud_account_mapping[cloud_provider].id
 
         if default_cloud_account:
             return default_cloud_account

lightning_sdk/api/teamspace_api.py

@@ -17,6 +17,7 @@ from lightning_sdk.api.utils import (
 )
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
+    Create,
     Externalv1LightningappInstance,
     ModelIdVersionsBody,
     ModelsStoreApi,
@@ -28,6 +29,8 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1CloudSpace,
     V1ClusterAccelerator,
     V1Endpoint,
+    V1ExternalCluster,
+    V1GCSFolderDataConnection,
     V1Job,
     V1LoginRequest,
     V1Model,
@@ -36,6 +39,8 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1Project,
     V1ProjectClusterBinding,
     V1PromptSuggestion,
+    V1R2DataConnection,
+    V1S3FolderDataConnection,
     V1Secret,
     V1SecretType,
     V1UpstreamOpenAI,
@@ -489,3 +494,24 @@ class TeamspaceApi:
         """
         pattern = r"^[A-Za-z_][A-Za-z0-9_]*$"
         return re.match(pattern, name) is not None
+
+    def new_folder(self, teamspace_id: str, name: str, cluster: Optional[V1ExternalCluster]) -> None:
+        create_request = Create(
+            name=name,
+            create_resources=True,
+            force=True,
+            writable=True,
+        )
+
+        if cluster is None:
+            create_request.r2 = V1R2DataConnection(name=name)
+        else:
+            create_request.cluster_id = cluster.id
+            create_request.access_cluster_ids = [cluster.id]
+
+            if cluster.spec.aws_v1:
+                create_request.s3_folder = V1S3FolderDataConnection()
+            elif cluster.spec.google_cloud_v1:
+                create_request.gcs_folder = V1GCSFolderDataConnection()
+
+        self._client.data_connection_service_create_data_connection(create_request, teamspace_id)

lightning_sdk/llm/llm.py

@@ -368,7 +368,7 @@ class LLM:
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
         full_response: bool = False,
-        reasoning_effort: Optional[Literal["low", "medium", "highc"]] = None,
+        reasoning_effort: Optional[Literal["none", "low", "medium", "high"]] = None,
         **kwargs: Any,
     ) -> Union[str, AsyncGenerator[str, None]]:
         conversation_id = self._conversations.get(conversation) if conversation else None
@@ -404,13 +404,13 @@ class LLM:
         stream: bool = False,
         full_response: bool = False,
         tools: Optional[List[Dict[str, Any]]] = None,
-        reasoning_effort: Optional[Literal["low", "medium", "high"]] = None,
+        reasoning_effort: Optional[Literal["none", "low", "medium", "high"]] = None,
         **kwargs: Any,
     ) -> Union[
         V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None], str, Generator[str, None, None]
     ]:
-        if reasoning_effort is not None and reasoning_effort not in ["low", "medium", "high"]:
-            raise ValueError("reasoning_effort must be 'low', 'medium', 'high', or None")
+        if reasoning_effort is not None and reasoning_effort not in ["none", "low", "medium", "high"]:
+            raise ValueError("reasoning_effort must be 'none', 'low', 'medium', 'high', or None")
 
         if conversation and conversation not in self._conversations:
             self._get_conversations()

lightning_sdk/teamspace.py

@@ -1,6 +1,7 @@
 import glob
 import os
 import warnings
+from enum import Enum
 from pathlib import Path
 from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Union
 
@@ -8,9 +9,9 @@ from tqdm.auto import tqdm
 
 import lightning_sdk
 from lightning_sdk.agents import Agent
-from lightning_sdk.api import TeamspaceApi
-from lightning_sdk.lightning_cloud.openapi import V1Model, V1ModelVersionArchive, V1ProjectClusterBinding
-from lightning_sdk.machine import Machine
+from lightning_sdk.api import CloudAccountApi, TeamspaceApi
+from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1Model, V1ModelVersionArchive, V1ProjectClusterBinding
+from lightning_sdk.machine import CloudProvider, Machine
 from lightning_sdk.models import UploadedModelInfo
 from lightning_sdk.organization import Organization
 from lightning_sdk.owner import Owner
@@ -30,6 +31,16 @@ if TYPE_CHECKING:
     from lightning_sdk.studio import Studio
 
 
+class FolderLocation(Enum):
+    AWS = "AWS"
+    GCP = "GCP"
+    CLOUD_AGNOSTIC = "CLOUD_AGNOSTIC"
+
+    def __str__(self) -> str:
+        """Converts the FolderLocation to a str."""
+        return self.value
+
+
 class Teamspace:
     """A teamspace is a collection of Studios, Clusters, Members and an associated Budget.
 
@@ -54,6 +65,7 @@ class Teamspace:
         user: Optional[Union[str, User]] = None,
     ) -> None:
         self._teamspace_api = TeamspaceApi()
+        self._cloud_account_api = CloudAccountApi()
 
         name = _resolve_teamspace_name(name)
 
@@ -515,6 +527,53 @@ class Teamspace:
             cloud_account=self.default_cloud_account,
         )
 
+    def new_folder(
+        self, name: str, location: Optional[FolderLocation] = None, cloud_account: Optional[str] = None
+    ) -> None:
+        """Create a new folder in this Teamspace.
+
+        Args:
+            name: The name of the folder. Folders will be accesible under `/teamspace/folders/<name>`
+            location: The location of the folder. Defaults to cloud agnostic.
+            cloud_account: The cloud account to create the folder in. Not used for cloud agnostic folders.
+        """
+        if cloud_account is None:
+            cloud_account = self.default_cloud_account
+
+        cloud_accounts = self._cloud_account_api.list_cloud_accounts(self.id)
+        resolved_cloud_accounts = [
+            external_cloud for external_cloud in cloud_accounts if external_cloud.id == cloud_account
+        ]
+
+        if len(resolved_cloud_accounts) == 0:
+            raise ValueError(f"Cloud account not found: {cloud_account}")
+
+        resolved_cloud_account = resolved_cloud_accounts[0]
+
+        # if the cloud account is global, default to agnostic
+        if location is None and resolved_cloud_account.spec.cluster_type == V1ClusterType.GLOBAL:
+            location = FolderLocation.CLOUD_AGNOSTIC
+
+        # if it's global, then default to agnostic, and aws / gcp otherwise if set
+        if (
+            location is not None
+            and location != FolderLocation.CLOUD_AGNOSTIC
+            and resolved_cloud_account.spec.cluster_type == V1ClusterType.GLOBAL
+        ):
+            providers = self._cloud_account_api.get_cloud_account_provider_mapping(self.id)
+
+            if location == FolderLocation.AWS:
+                resolved_cloud_account = providers[CloudProvider.AWS]
+            elif location == FolderLocation.GCP:
+                resolved_cloud_account = providers[CloudProvider.GCP]
+
+        if location == FolderLocation.CLOUD_AGNOSTIC:
+            self._teamspace_api.new_folder(self.id, name, None)
+        else:
+            self._teamspace_api.new_folder(self.id, name, resolved_cloud_account)
+
+        return
+
 
 def _list_files(path: Union[str, Path]) -> Tuple[List[Path], List[str]]:
     """List all folders in a directory and return them as a list and relative path."""

{lightning_sdk-2025.9.11.dist-info → lightning_sdk-2025.9.16.dist-info}/METADATA

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

{lightning_sdk-2025.9.11.dist-info → lightning_sdk-2025.9.16.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=RTO9pPFR4nJ_S0mZ9BOC-0Wp3hCm4yZ1f8Lvr8RjDwI,1145
+lightning_sdk/__init__.py,sha256=1wyKGyM8HKl1zyLHQ97cOY7_jZefa5V89YaxP3-0s08,1183
 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
@@ -15,13 +15,13 @@ 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=3BE44Ya-2HXmDkF58lVQmd7KVzzzPktqX1SV7K3nkfo,31438
-lightning_sdk/teamspace.py,sha256=qYpIYgg3eyWj9vcqdfhQN-YCpS47tOc_xB3de07p4Y8,21630
+lightning_sdk/teamspace.py,sha256=fjsbMw_fmAHLldAGT03R8Byp1znR6ZYAh9PXUAzzP8w,24016
 lightning_sdk/user.py,sha256=TSYh38rxoi7qKOfrK2JYh_Nknya2Kbz2ngDIY85fFOY,1778
-lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
+lightning_sdk/api/__init__.py,sha256=xrp_RNECJGQtL5rZHF69WOzEuEIbWSLtjWAJAz4R5K4,500
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
 lightning_sdk/api/base_studio_api.py,sha256=3R8tucZX2e9yKHBcY2rWFRP4dxqLrC6H75vdBDkH0ck,3617
-lightning_sdk/api/cloud_account_api.py,sha256=b_Egg5yhDh5G39g1cqA3ZxBDmU9iuDs2PlBMmW0HVnA,8417
+lightning_sdk/api/cloud_account_api.py,sha256=pPeNUcl_7qE0BsrLfmvG0dH0rCJfHiSFexBs4UxWXBM,8732
 lightning_sdk/api/deployment_api.py,sha256=v2AfoTDkQ-1CBh75FOjFkRpf6yc3U_edDy43uYSn19I,24852
 lightning_sdk/api/job_api.py,sha256=7spFPyotlSQ0Krx6WymslnpSQaaeOmFkydvhtg6p0Ic,16410
 lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
@@ -31,7 +31,7 @@ lightning_sdk/api/mmt_api.py,sha256=hIBsGiJ2qn5UjcHDxP5WUyKGT_AIFfpSHrQVwg0afBw,
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/pipeline_api.py,sha256=rJYp_FN7uUjC5xbc6K67l2eRSmVuOkijd5i8Nm5BF7I,4621
 lightning_sdk/api/studio_api.py,sha256=Aji8lke2jGeef-1lh6wV7EQ15iuiw1Cn_IuBNnEb8SA,39492
-lightning_sdk/api/teamspace_api.py,sha256=ygQAGG5W4EqcXv2PO2cYPdAN5fGj75R-1IZkNq7Z9gU,18416
+lightning_sdk/api/teamspace_api.py,sha256=XPyUtpL5iyt9BpXw5WM-_hT2R3OzuC1GcumnI2ktVhQ,19333
 lightning_sdk/api/user_api.py,sha256=FQFgdZOopEvtFJcJT1mRRuStqKKYmg4iuVUYDSQfDVM,4514
 lightning_sdk/api/utils.py,sha256=1NJgW4HCfzMqgnAqbMA7RRr2v3iM3KTuPIUQK5klDeQ,27127
 lightning_sdk/cli/__init__.py,sha256=lksw08t_ZIuHOH47LCIqSVHeZ8cUXI2aJWHYhyujYHM,32
@@ -1152,7 +1152,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=M2gKP77SDz05flUpvnqh6vLXLdNwNzPn2jzXTIls7S0,20805
+lightning_sdk/llm/llm.py,sha256=pjBs8fbGZjs0tQRKHx-WdfTmeECKAwI_T1ShVBgdyO4,20836
 lightning_sdk/llm/public_assistants.py,sha256=k0yc41AyrKImnRa8Fv-ow9javnlrRQeP63507p2VybA,1579
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=iqVudKDxazomuezj6l7pa-m9I1EQnM82OKs4ZQbo4Ck,16434
@@ -1177,9 +1177,9 @@ lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4
 lightning_sdk/utils/names.py,sha256=1EuXbIh7wldkDp1FG10oz9vIOyWrpGWeFFVy-DQBgzA,18162
 lightning_sdk/utils/progress.py,sha256=IXfEcUF-rL5jIw0Hir6eSxN7VBZfR--1O2LaEhGAU70,12698
 lightning_sdk/utils/resolve.py,sha256=4TyEnIgIrvkSvYk5i5PmcIogD_5Y9pBhiphRLfLMttc,10477
-lightning_sdk-2025.9.11.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.9.11.dist-info/METADATA,sha256=qmaN-yNF1oQ04YxqFC1BOQJ8sdFTT8EjzzwqRihys3Q,4130
-lightning_sdk-2025.9.11.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.9.11.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
-lightning_sdk-2025.9.11.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.9.11.dist-info/RECORD,,
+lightning_sdk-2025.9.16.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.9.16.dist-info/METADATA,sha256=GR7YQsSApMYI943xTmY0qASB5yv4cUrFub3xlY0gwnI,4130
+lightning_sdk-2025.9.16.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.9.16.dist-info/entry_points.txt,sha256=OoZa4Fc8NMs6GSN0cdA1J8e6couzAcL82CbM1yo4f_M,122
+lightning_sdk-2025.9.16.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.9.16.dist-info/RECORD,,