lightning-sdk 2025.8.21__py3-none-any.whl → 2025.8.26__py3-none-any.whl

lightning_sdk/cli/utils/teamspace_selection.py

@@ -0,0 +1,125 @@
+import os
+from typing import Dict, List, Optional
+
+import click
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk.api import OrgApi
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.cli.utils.resolve import resolve_teamspace_owner_name_format
+from lightning_sdk.teamspace import Teamspace
+from lightning_sdk.user import User
+from lightning_sdk.utils.resolve import _get_authed_user
+
+
+class TeamspacesMenu:
+    """This class is used to select a teamspace from a list of possible teamspaces.
+
+    It can be used to select a teamspace from a list of possible teamspaces, or to resolve a teamspace from a name.
+    """
+
+    def _get_teamspace_from_interactive_menu(self, possible_teamspaces: Dict[str, Dict[str, str]]) -> Dict[str, str]:
+        teamspace_ids = sorted(possible_teamspaces.keys())
+        terminal_menu = self._prepare_terminal_menu_teamspaces([possible_teamspaces[k] for k in teamspace_ids])
+        terminal_menu.show()
+
+        selected_id = teamspace_ids[terminal_menu.chosen_menu_index]
+        return possible_teamspaces[selected_id]
+
+    def _get_teamspace_from_name(
+        self, teamspace: str, possible_teamspaces: Dict[str, Dict[str, str]]
+    ) -> Dict[str, str]:
+        try:
+            owner, 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
+
+        for _, ts in possible_teamspaces.items():
+            if ts["name"] == name and (ts["user"] == owner or ts["org"] == owner):
+                return ts
+
+        click.echo(f"Could not find Teamspace {teamspace}, please select it from the list:")
+        return self._get_teamspace_from_interactive_menu(possible_teamspaces)
+
+    @staticmethod
+    def _prepare_terminal_menu_teamspaces(
+        possible_teamspaces: List[Dict[str, str]], title: Optional[str] = None
+    ) -> TerminalMenu:
+        if title is None:
+            title = "Please select a Teamspace out of the following:"
+
+        return TerminalMenu(
+            [f"{t['user'] or t['org']}/{t['name']}" for t in possible_teamspaces], title=title, clear_menu_on_exit=True
+        )
+
+    @staticmethod
+    def _get_possible_teamspaces(user: User) -> Dict[str, Dict[str, str]]:
+        org_api = OrgApi()
+        user_api = user._user_api
+
+        user_api._get_organizations_for_authed_user()
+        memberships = user_api._get_all_teamspace_memberships(user_id=user.id)
+
+        teamspaces = {}
+        # get all teamspace memberships
+        for membership in memberships:
+            teamspace_id = membership.project_id
+            teamspace_name = membership.name
+
+            # get organization if necessary
+            if membership.owner_type == "organization":
+                org_name = org_api._get_org_by_id(membership.owner_id).name
+                user_name = None
+            else:
+                org_name = None
+
+                # don't do a request if not necessary
+                if membership.owner_id == user.id:
+                    user_name = user.name
+                else:
+                    user_name = user_api._get_user_by_id(membership.owner_id).username
+
+            teamspaces[teamspace_id] = {"user": user_name, "org": org_name, "name": teamspace_name}
+
+        return teamspaces
+
+    def __call__(self, teamspace: Optional[str] = None) -> Teamspace:
+        try:
+            # try to resolve the teamspace from the name, environment or config
+            resolved_teamspace = resolve_teamspace_owner_name_format(teamspace)
+
+            if resolved_teamspace is not None:
+                return resolved_teamspace
+
+            if os.environ.get("LIGHTNING_NON_INTERACTIVE", "0") == "1":
+                raise ValueError(
+                    "Teamspace selection is not supported in non-interactive mode. "
+                    "Please provide a teamspace name in the format of 'owner/teamspace'."
+                )
+
+            # if the teamspace is not resolved, try to get the teamspace from the interactive menu
+            # this could mean that either no teamspace was provided or the provided teamspace is not valid
+            user = _get_authed_user()
+
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            if teamspace is None:
+                teamspace = resolve_teamspace_owner_name_format(teamspace)
+                teamspace_dict = self._get_teamspace_from_interactive_menu(possible_teamspaces=possible_teamspaces)
+            else:
+                teamspace_dict = self._get_teamspace_from_name(
+                    teamspace=teamspace, possible_teamspaces=possible_teamspaces
+                )
+
+            return Teamspace(**teamspace_dict)
+        except KeyboardInterrupt:
+            raise KeyboardInterrupt from None
+
+        except Exception as e:
+            raise StudioCliError(
+                f"Could not find the given Teamspace {teamspace}. "
+                "Please contact Lightning AI directly to resolve this issue."
+            ) from e

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -656,6 +656,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_metric_timestamps_response import V1ListClusterMetricTimestampsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_metrics_response import V1ListClusterMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_metrics_response import V1ListClusterNamespaceMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_user_metrics_response import V1ListClusterNamespaceUserMetricsResponse
@@ -844,6 +845,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_project_storage import V1Pr
 from lightning_sdk.lightning_cloud.openapi.models.v1_prompt_suggestion import V1PromptSuggestion
 from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_response import V1PublishCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_response import V1PurchaseAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_query_param import V1QueryParam

lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py

@@ -1356,6 +1356,91 @@ class BillingServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def billing_service_purchase_annual_upsell(self, **kwargs) -> 'V1PurchaseAnnualUpsellResponse':  # noqa: E501
+        """billing_service_purchase_annual_upsell  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.billing_service_purchase_annual_upsell(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1PurchaseAnnualUpsellResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.billing_service_purchase_annual_upsell_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.billing_service_purchase_annual_upsell_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def billing_service_purchase_annual_upsell_with_http_info(self, **kwargs) -> 'V1PurchaseAnnualUpsellResponse':  # noqa: E501
+        """billing_service_purchase_annual_upsell  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.billing_service_purchase_annual_upsell_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1PurchaseAnnualUpsellResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = []  # 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 billing_service_purchase_annual_upsell" % key
+                )
+            params[key] = val
+        del params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        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/billing/annual-upsell', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1PurchaseAnnualUpsellResponse',  # 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 billing_service_quote_annual_upsell(self, **kwargs) -> 'V1QuoteAnnualUpsellResponse':  # noqa: E501
         """billing_service_quote_annual_upsell  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py

@@ -273,6 +273,107 @@ class K8SClusterServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def k8_s_cluster_service_list_cluster_metric_timestamps(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListClusterMetricTimestampsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_cluster_metric_timestamps  # 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_cluster_metric_timestamps(project_id, cluster_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :return: V1ListClusterMetricTimestampsResponse
+                 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_cluster_metric_timestamps_with_http_info(project_id, cluster_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.k8_s_cluster_service_list_cluster_metric_timestamps_with_http_info(project_id, cluster_id, **kwargs)  # noqa: E501
+            return data
+
+    def k8_s_cluster_service_list_cluster_metric_timestamps_with_http_info(self, project_id: 'str', cluster_id: 'str', **kwargs) -> 'V1ListClusterMetricTimestampsResponse':  # noqa: E501
+        """k8_s_cluster_service_list_cluster_metric_timestamps  # 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_cluster_metric_timestamps_with_http_info(project_id, cluster_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cluster_id: (required)
+        :return: V1ListClusterMetricTimestampsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cluster_id']  # 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_cluster_metric_timestamps" % 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_cluster_metric_timestamps`")  # 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_cluster_metric_timestamps`")  # 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
+
+        query_params = []
+
+        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}/cluster-metrics-timestamps', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListClusterMetricTimestampsResponse',  # 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/__init__.py

@@ -608,6 +608,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_metric_timestamps_response import V1ListClusterMetricTimestampsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_metrics_response import V1ListClusterMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_metrics_response import V1ListClusterNamespaceMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_namespace_user_metrics_response import V1ListClusterNamespaceUserMetricsResponse
@@ -796,6 +797,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_project_storage import V1Pr
 from lightning_sdk.lightning_cloud.openapi.models.v1_prompt_suggestion import V1PromptSuggestion
 from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_response import V1PublishCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_response import V1PurchaseAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_query_param import V1QueryParam

lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py

@@ -42,6 +42,7 @@ class Externalv1UserStatus(object):
     """
     swagger_types = {
         'acked_storage_violation': 'bool',
+        'auth_provider': 'str',
         'auto_verify_reason': 'str',
         'auto_verify_response': 'bool',
         'completed_project_onboarding': 'bool',
@@ -54,6 +55,7 @@ class Externalv1UserStatus(object):
 
     attribute_map = {
         'acked_storage_violation': 'ackedStorageViolation',
+        'auth_provider': 'authProvider',
         'auto_verify_reason': 'autoVerifyReason',
         'auto_verify_response': 'autoVerifyResponse',
         'completed_project_onboarding': 'completedProjectOnboarding',
@@ -64,9 +66,10 @@ class Externalv1UserStatus(object):
         'verified_at': 'verifiedAt'
     }
 
-    def __init__(self, acked_storage_violation: 'bool' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, acked_storage_violation: 'bool' =None, auth_provider: 'str' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None):  # noqa: E501
         """Externalv1UserStatus - a model defined in Swagger"""  # noqa: E501
         self._acked_storage_violation = None
+        self._auth_provider = None
         self._auto_verify_reason = None
         self._auto_verify_response = None
         self._completed_project_onboarding = None
@@ -78,6 +81,8 @@ class Externalv1UserStatus(object):
         self.discriminator = None
         if acked_storage_violation is not None:
             self.acked_storage_violation = acked_storage_violation
+        if auth_provider is not None:
+            self.auth_provider = auth_provider
         if auto_verify_reason is not None:
             self.auto_verify_reason = auto_verify_reason
         if auto_verify_response is not None:
@@ -116,6 +121,27 @@ class Externalv1UserStatus(object):
 
         self._acked_storage_violation = acked_storage_violation
 
+    @property
+    def auth_provider(self) -> 'str':
+        """Gets the auth_provider of this Externalv1UserStatus.  # noqa: E501
+
+
+        :return: The auth_provider of this Externalv1UserStatus.  # noqa: E501
+        :rtype: str
+        """
+        return self._auth_provider
+
+    @auth_provider.setter
+    def auth_provider(self, auth_provider: 'str'):
+        """Sets the auth_provider of this Externalv1UserStatus.
+
+
+        :param auth_provider: The auth_provider of this Externalv1UserStatus.  # noqa: E501
+        :type: str
+        """
+
+        self._auth_provider = auth_provider
+
     @property
     def auto_verify_reason(self) -> 'str':
         """Gets the auto_verify_reason of this Externalv1UserStatus.  # noqa: E501