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

lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py

@@ -43,6 +43,123 @@ class K8SClusterServiceApi(object):
             api_client = ApiClient()
         self.api_client = api_client
 
+    def k8_s_cluster_service_list_aggregated_node_metrics(self, project_id: 'str', cluster_id: 'str', node_name: 'str', **kwargs) -> 'V1ListNodeMetricsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_aggregated_node_metrics  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.k8_s_cluster_service_list_aggregated_node_metrics(project_id, cluster_id, node_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str node_name: (required)
+        :param datetime start: Date range.
+        :param datetime end:
+        :return: V1ListNodeMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.k8_s_cluster_service_list_aggregated_node_metrics_with_http_info(project_id, cluster_id, node_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.k8_s_cluster_service_list_aggregated_node_metrics_with_http_info(project_id, cluster_id, node_name, **kwargs)  # noqa: E501
+            return data
+
+    def k8_s_cluster_service_list_aggregated_node_metrics_with_http_info(self, project_id: 'str', cluster_id: 'str', node_name: 'str', **kwargs) -> 'V1ListNodeMetricsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_aggregated_node_metrics  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.k8_s_cluster_service_list_aggregated_node_metrics_with_http_info(project_id, cluster_id, node_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :param str node_name: (required)
+        :param datetime start: Date range.
+        :param datetime end:
+        :return: V1ListNodeMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cluster_id', 'node_name', 'start', 'end']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method k8_s_cluster_service_list_aggregated_node_metrics" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `k8_s_cluster_service_list_aggregated_node_metrics`")  # noqa: E501
+        # verify the required parameter 'cluster_id' is set
+        if ('cluster_id' not in params or
+                params['cluster_id'] is None):
+            raise ValueError("Missing the required parameter `cluster_id` when calling `k8_s_cluster_service_list_aggregated_node_metrics`")  # noqa: E501
+        # verify the required parameter 'node_name' is set
+        if ('node_name' not in params or
+                params['node_name'] is None):
+            raise ValueError("Missing the required parameter `node_name` when calling `k8_s_cluster_service_list_aggregated_node_metrics`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cluster_id' in params:
+            path_params['clusterId'] = params['cluster_id']  # noqa: E501
+        if 'node_name' in params:
+            path_params['nodeName'] = params['node_name']  # noqa: E501
+
+        query_params = []
+        if 'start' in params:
+            query_params.append(('start', params['start']))  # noqa: E501
+        if 'end' in params:
+            query_params.append(('end', params['end']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/clusters/{clusterId}/aggregated-metrics/nodes/{nodeName}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListNodeMetricsResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def k8_s_cluster_service_list_cluster_metrics(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListClusterMetricsResponse':  # noqa: E501
         """k8_s_cluster_service_list_cluster_metrics  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -60,7 +60,8 @@ class V1ManagedModel(object):
         'throughput': 'float',
         'time_to_first_token': 'float',
         'token_threshold': 'str',
-        'top_k': 'str'
+        'top_k': 'str',
+        'user_id': 'str'
     }
 
     attribute_map = {
@@ -83,10 +84,11 @@ class V1ManagedModel(object):
         'throughput': 'throughput',
         'time_to_first_token': 'timeToFirstToken',
         'token_threshold': 'tokenThreshold',
-        'top_k': 'topK'
+        'top_k': 'topK',
+        'user_id': 'userId'
     }
 
-    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, assistant_id: 'str' =None, completion_token_price: 'float' =None, completion_token_price_above_threshold: 'float' =None, context_length: 'str' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, display_name: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, max_completion_tokens: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, prompt_token_price_above_threshold: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, throughput: 'float' =None, time_to_first_token: 'float' =None, token_threshold: 'str' =None, top_k: 'str' =None):  # noqa: E501
+    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, assistant_id: 'str' =None, completion_token_price: 'float' =None, completion_token_price_above_threshold: 'float' =None, context_length: 'str' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, display_name: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, max_completion_tokens: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, prompt_token_price_above_threshold: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, throughput: 'float' =None, time_to_first_token: 'float' =None, token_threshold: 'str' =None, top_k: 'str' =None, user_id: 'str' =None):  # noqa: E501
         """V1ManagedModel - a model defined in Swagger"""  # noqa: E501
         self._abilities = None
         self._assistant_id = None
@@ -108,6 +110,7 @@ class V1ManagedModel(object):
         self._time_to_first_token = None
         self._token_threshold = None
         self._top_k = None
+        self._user_id = None
         self.discriminator = None
         if abilities is not None:
             self.abilities = abilities
@@ -149,6 +152,8 @@ class V1ManagedModel(object):
             self.token_threshold = token_threshold
         if top_k is not None:
             self.top_k = top_k
+        if user_id is not None:
+            self.user_id = user_id
 
     @property
     def abilities(self) -> 'V1ManagedModelAbilities':
@@ -570,6 +575,27 @@ class V1ManagedModel(object):
 
         self._top_k = top_k
 
+    @property
+    def user_id(self) -> 'str':
+        """Gets the user_id of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The user_id of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, user_id: 'str'):
+        """Sets the user_id of this V1ManagedModel.
+
+
+        :param user_id: The user_id of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._user_id = user_id
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_notification_type.py

@@ -41,6 +41,7 @@ class V1NotificationType(object):
     LOW_FUNDS = "NOTIFICATION_TYPE_LOW_FUNDS"
     LONG_WORKLOADS = "NOTIFICATION_TYPE_LONG_WORKLOADS"
     DEPLOYMENT_ERROR = "NOTIFICATION_TYPE_DEPLOYMENT_ERROR"
+    REQUESTED_ACCESS = "NOTIFICATION_TYPE_REQUESTED_ACCESS"
     """
     Attributes:
       swagger_types (dict): The key is attribute name

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -41,7 +41,6 @@ class V1UserFeatures(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'academic_tier': 'bool',
         'add_data_v2': 'bool',
         'affiliate_links': 'bool',
         'agents_v2': 'bool',
@@ -132,7 +131,6 @@ class V1UserFeatures(object):
     }
 
     attribute_map = {
-        'academic_tier': 'academicTier',
         'add_data_v2': 'addDataV2',
         'affiliate_links': 'affiliateLinks',
         'agents_v2': 'agentsV2',
@@ -222,9 +220,8 @@ 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, 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
+    def __init__(self, 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
         self._affiliate_links = None
         self._agents_v2 = None
@@ -313,8 +310,6 @@ class V1UserFeatures(object):
         self._weka = None
         self._writable_s3_connections = None
         self.discriminator = None
-        if academic_tier is not None:
-            self.academic_tier = academic_tier
         if add_data_v2 is not None:
             self.add_data_v2 = add_data_v2
         if affiliate_links is not None:
@@ -490,27 +485,6 @@ class V1UserFeatures(object):
         if writable_s3_connections is not None:
             self.writable_s3_connections = writable_s3_connections
 
-    @property
-    def academic_tier(self) -> 'bool':
-        """Gets the academic_tier of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The academic_tier of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._academic_tier
-
-    @academic_tier.setter
-    def academic_tier(self, academic_tier: 'bool'):
-        """Sets the academic_tier of this V1UserFeatures.
-
-
-        :param academic_tier: The academic_tier of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._academic_tier = academic_tier
-
     @property
     def add_data_v2(self) -> 'bool':
         """Gets the add_data_v2 of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/utils/data_connection.py

@@ -1,7 +1,7 @@
 import os
 from time import sleep, time
 from lightning_sdk.lightning_cloud import rest_client
-from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig, V1GcpDataConnection, V1FilestoreDataConnection, V1DataConnectionTier
+from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig, V1GcpDataConnection, V1FilestoreDataConnection, V1DataConnectionTier, V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 import urllib3
 
@@ -327,6 +327,56 @@ def create_filestore_folder(folder_name: str, region: str, capacity_gb: int = 10
         sleep(1)
 
 
+def create_cloud_agnostic_folder(folder_name: str, create_timeout: int = 30) -> None:
+    """
+    Utility function to create a Cloud-Agnostic (R2) folder.
+
+    Args:
+        folder_name: The name of the folder to create.
+        create_timeout (int): The timeout for the folder creation.
+    """
+    client = rest_client.LightningClient(retry=False)
+
+    project_id = os.getenv("LIGHTNING_CLOUD_PROJECT_ID")
+    cluster_id = os.getenv("LIGHTNING_CLUSTER_ID")
+
+    # Get existing data connections
+    data_connections = client.data_connection_service_list_data_connections(project_id).data_connections
+
+    for connection in data_connections:
+        existing_folder_name = getattr(connection, 'name', None)
+        isCloudAgnostic = getattr(connection, 'r2_folder', None) is not None
+
+        if existing_folder_name == folder_name and isCloudAgnostic:
+            return
+
+    # If we get here, no matching folder was found, proceed with creation
+    body = Create(
+        name=folder_name,
+        create_resources=True,
+        r2=V1R2DataConnection(name=folder_name),
+    )
+    try:
+        connection = client.data_connection_service_create_data_connection(body, project_id)
+    except ApiException as e:
+        # Note: This function can be called in a distributed way.
+        # There is a race condition where one machine might create the entry before another machine
+        # and this request would fail with duplicated key
+        # In this case, it is fine not to raise
+        if'duplicate key value violates unique constraint' in str(e.body):
+            return
+        raise e from None
+
+    except urllib3.exceptions.HTTPError as e:
+        raise e from None
+
+    # Wait for the filesystem picks up the newly added Cloud-Agnostic folder
+    start = time()
+
+    while not os.path.isdir(f"/teamspace/lightning_storage/{folder_name}") and (time() - start) < create_timeout:
+        sleep(1)
+
+
 def delete_data_connection(name: str):
     """Utility to delete a data connection
 

lightning_sdk/studio.py

@@ -98,12 +98,25 @@ class Studio:
             default_cloud_account=self._teamspace.default_cloud_account,
         )
 
+        # Resolve studio name if not provided: explicit → env (LIGHTNING_CLOUD_SPACE_ID) → config defaults
         if name is None:
             studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
-            if studio_id is None:
-                raise ValueError("Cannot autodetect Studio. Either use the SDK from within a Studio or pass a name!")
-            self._studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
-        else:
+            if studio_id is not None:
+                # We're inside a studio, get it by ID
+                self._studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
+            else:
+                # Try config defaults
+                from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+                config = Config()
+                name = config.get_value(DefaultConfigKeys.studio)
+                if name is None:
+                    raise ValueError(
+                        "Cannot autodetect Studio. Either use the SDK from within a Studio or pass a name!"
+                    )
+
+        # If we have a name (explicit or from config), get studio by name
+        if name is not None:
             try:
                 self._studio = self._studio_api.get_studio(name, self._teamspace.id)
             except ValueError as e:
@@ -529,7 +542,7 @@ class Studio:
     def auto_sleep(self, value: bool) -> None:
         if not value and self.machine == Machine.CPU:
             warnings.warn("Disabling auto-sleep will convert the Studio from free to paid!")
-        self._studio_api.update_autoshutdown(self._studio.id, self._teamspace.id, enabled=value, studio=self._studio)
+        self._studio_api.update_autoshutdown(self._studio.id, self._teamspace.id, enabled=value)
         self._update_studio_reference()
 
     @property
@@ -540,9 +553,7 @@ class Studio:
     @auto_sleep_time.setter
     def auto_sleep_time(self, value: int) -> None:
         warnings.warn("Setting auto-sleep time will convert the Studio from free to paid!")
-        self._studio_api.update_autoshutdown(
-            self._studio.id, self._teamspace.id, idle_shutdown_seconds=value, studio=self._studio
-        )
+        self._studio_api.update_autoshutdown(self._studio.id, self._teamspace.id, idle_shutdown_seconds=value)
         self._update_studio_reference()
 
     @property

lightning_sdk/teamspace.py

@@ -70,6 +70,20 @@ class Teamspace:
             self._user = _resolve_user(user)
             self._org = _resolve_org(org)
 
+        # If still no user or org resolved, try config defaults
+        if self._user is None and self._org is None:
+            from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+            config = Config()
+            owner_type = config.get_value(DefaultConfigKeys.teamspace_owner_type)
+            owner_name = config.get_value(DefaultConfigKeys.teamspace_owner)
+
+            if owner_type and owner_name:
+                if owner_type.lower() == "organization":
+                    self._org = _resolve_org(owner_name)
+                elif owner_type.lower() == "user":
+                    self._user = _resolve_user(owner_name)
+
         self._owner: Owner
         if self._user is None and self._org is None:
             raise RuntimeError(

lightning_sdk/utils/config.py

@@ -0,0 +1,155 @@
+import os
+from collections.abc import Mapping
+from dataclasses import dataclass
+from typing import Any, Dict, Optional, Sequence
+
+import yaml
+
+_DEFAULT_CONFIG_FILE_PATH = "~/.lightning/config.yaml"
+
+
+@dataclass(frozen=True)
+class DefaultConfigKeys:
+    """Default configuration keys for the Lightning SDK."""
+
+    organization: str = "organization.name"
+    user: str = "user.name"
+
+    teamspace_name: str = "teamspace.name"
+    teamspace_owner: str = "teamspace.owner"
+    teamspace_owner_type: str = "teamspace.owner_type"
+
+    machine: str = "machine.name"
+
+    studio: str = "studio.name"
+
+
+class ConfigProxy:
+    def __init__(self, root: "Config", *path: str) -> None:
+        self._root = root
+        self._path = path  # list of keys from root
+
+    def __getattr__(self, name: str) -> "ConfigProxy":
+        """Returns a reference to a nested ConfigProxy object a level deeper in the config hierarchy.
+
+        Args:
+            name: the name of the attribute to access, which corresponds to a key in the config.
+
+        Returns:
+            ConfigProxy: the next ConfigProxy object for the attribute.
+        """
+        # Build a deeper path and return a new proxy
+        return ConfigProxy(self._root, *self._path, name)
+
+    def __setattr__(self, name: str, value: Any) -> None:
+        """Sets the name attribute to value at the current hierarchy level.
+
+        Args:
+            name: the attribute name to set, which corresponds to a key in the config.
+            value: the value to set for the given attribute name in the config.
+        """
+        if name in ("_root", "_path"):  # internal attributes
+            super().__setattr__(name, value)
+        else:
+            # Assign a nested value in the root config
+            self._root._set_nested([*self._path, name], value)
+
+
+class Config:
+    def __init__(self, config_file: Optional[str] = None) -> None:
+        """Config class to manage configuration settings for the lightning SDK and CLI.
+
+        Args:
+            config_file: the file path where the configuration is stored.
+            If None, defaults to "~/.lightning/config.yaml".
+        """
+        if config_file is None:
+            config_file = _DEFAULT_CONFIG_FILE_PATH
+        self._config_file = os.path.expanduser(config_file)
+
+    def _load_config(self) -> Dict[str, Any]:
+        if not os.path.exists(self._config_file):
+            return {}  # Return empty dict if config doesn't exist
+        with open(self._config_file) as f:
+            return yaml.safe_load(f) or {}
+
+    def _save_config(self, config: Dict[str, Any]) -> None:
+        os.makedirs(os.path.dirname(self._config_file), exist_ok=True)
+        config = _unflatten_dict(config)
+        with open(self._config_file, "w") as f:
+            yaml.safe_dump(config, f, default_flow_style=False, sort_keys=True)
+
+    def _set_nested(self, keys: Sequence[str], value: str) -> None:
+        config = self._load_config()
+        curr = config
+        for k in keys[:-1]:
+            if k not in curr or not isinstance(curr[k], Dict):
+                curr[k] = {}
+            curr = curr[k]
+        curr[keys[-1]] = value
+        self._save_config(config)
+
+    def get_value(self, key_path: str) -> Optional[str]:
+        """Gets a value from the config using dot notation.
+
+        Args:
+            key_path: the dot-separated path to the config value (e.g. "teamspace.name")
+
+        Returns:
+            The config value if found, None otherwise
+        """
+        config = self._load_config()
+        if not isinstance(config, Mapping):
+            return None
+
+        keys = key_path.split(".")
+        curr = config
+        for k in keys:
+            if not isinstance(curr, dict) or k not in curr:
+                return None
+            curr = curr[k]
+        return curr if isinstance(curr, str) else None
+
+    def __getattr__(self, name: str) -> ConfigProxy:
+        """Returns a proxy to the actual values to allow for nested access.
+
+        Args:
+            name: the name of the value to retrieve.
+
+        Returns:
+            ConfigProxy: a proxy object that allows nested access to the configuration.
+        """
+        return ConfigProxy(self, name)
+
+    def __setattr__(self, name: str, value: Any) -> None:
+        """Sets the name attribute to value at the root level."""
+        if name in ("_config_file",):  # internal attributes
+            super().__setattr__(name, value)
+        else:
+            # Assign a value at the root level
+            self._set_nested([name], value)
+
+    def __repr__(self) -> str:
+        """Returns a string representation of the config."""
+        return str(self)
+
+    def __str__(self) -> str:
+        """Returns a string representation of the config."""
+        return yaml.dump(
+            {"Config": {"config_file": self._config_file, **self._load_config()}},
+            indent=4,
+            sort_keys=True,
+        )
+
+
+def _unflatten_dict(flat_dict: Dict[str, Any]) -> Dict[str, Any]:
+    unflattened_dict = {}
+    for key, value in flat_dict.items():
+        keys = key.split(".")
+        curr = unflattened_dict
+        for k in keys[:-1]:
+            if k not in curr:
+                curr[k] = {}
+            curr = curr[k]
+        curr[keys[-1]] = value
+    return unflattened_dict

lightning_sdk/utils/resolve.py

@@ -90,6 +90,11 @@ def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[
 def _resolve_org_name(name: Optional[str]) -> Optional[str]:
     if name is None:
         name = os.environ.get("LIGHTNING_ORG", "") or None
+    if name is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        name = config.get_value(DefaultConfigKeys.organization)
     return name
 
 
@@ -118,6 +123,11 @@ def _resolve_org(org: Optional[Union[str, "Organization"]]) -> Optional["Organiz
 def _resolve_user_name(name: Optional[str]) -> Optional[str]:
     if name is None:
         name = os.environ.get("LIGHTNING_USERNAME", "") or None
+    if name is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        name = config.get_value(DefaultConfigKeys.user)
     return name
 
 
@@ -137,6 +147,11 @@ def _resolve_user(user: Optional[Union[str, "User"]]) -> Optional["User"]:
 def _resolve_teamspace_name(name: Optional[str]) -> Optional[str]:
     if name is None:
         name = os.environ.get("LIGHTNING_TEAMSPACE", "") or None
+    if name is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        name = config.get_value(DefaultConfigKeys.teamspace_name)
     return name
 
 
@@ -165,10 +180,29 @@ def _resolve_teamspace(
         return Teamspace(name=teamspace, org=org)
 
     user = _resolve_user(user)
-    if user is None:
-        raise RuntimeError("Neither user nor org provided, but one of them needs to be provided")
 
-    return Teamspace(name=teamspace, user=user)
+    # If still no user or org resolved, try config defaults
+    if user is None and org is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        owner_type = config.get_value(DefaultConfigKeys.teamspace_owner_type)
+        owner_name = config.get_value(DefaultConfigKeys.teamspace_owner)
+
+        if owner_type and owner_name:
+            if owner_type.lower() == "organization":
+                org = _resolve_org(owner_name)
+            elif owner_type.lower() == "user":
+                user = _resolve_user(owner_name)
+
+    # Final resolution check
+    if org is not None:
+        return Teamspace(name=teamspace, org=org)
+
+    if user is not None:
+        return Teamspace(name=teamspace, user=user)
+
+    raise RuntimeError("Neither user nor org provided, but one of them needs to be provided")
 
 
 def _get_organizations_for_authed_user() -> List["Organization"]:

{lightning_sdk-2025.8.14.dist-info → lightning_sdk-2025.8.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.8.14
+Version: 2025.8.18
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -39,6 +39,7 @@ Requires-Dist: docker
 Requires-Dist: fastapi
 Requires-Dist: packaging
 Requires-Dist: pyjwt
+Requires-Dist: pyyaml
 Requires-Dist: requests
 Requires-Dist: rich
 Requires-Dist: simple-term-menu