lightning-sdk 2025.8.19.post0__py3-none-any.whl → 2025.8.26__py3-none-any.whl

lightning_sdk/cli/utils/studio_selection.py

@@ -0,0 +1,106 @@
+import os
+from contextlib import suppress
+from typing import Dict, List, Optional
+
+import click
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.studio import Studio
+from lightning_sdk.teamspace import Teamspace
+from lightning_sdk.utils.resolve import _get_authed_user
+
+
+class StudiosMenu:
+    """This class is used to select a studio from a list of possible studios within a teamspace.
+
+    It can be used to select a studio from a list of possible studios, or to resolve a studio from a name.
+    """
+
+    def __init__(self, teamspace: Teamspace) -> None:
+        """Initialize the StudiosMenu with a teamspace.
+
+        Args:
+            teamspace: The teamspace to list studios from
+        """
+        self.teamspace = teamspace
+
+    def _get_studio_from_interactive_menu(self, possible_studios: Dict[str, Studio]) -> Studio:
+        studio_names = sorted(possible_studios.keys())
+        terminal_menu = self._prepare_terminal_menu_studios(studio_names)
+        terminal_menu.show()
+
+        selected_name = studio_names[terminal_menu.chosen_menu_index]
+        return possible_studios[selected_name]
+
+    def _get_studio_from_name(self, studio: str, possible_studios: Dict[str, Studio]) -> Studio:
+        if studio in possible_studios:
+            return possible_studios[studio]
+
+        click.echo(f"Could not find Studio {studio}, please select it from the list:")
+        return self._get_studio_from_interactive_menu(possible_studios)
+
+    @staticmethod
+    def _prepare_terminal_menu_studios(studio_names: List[str], title: Optional[str] = None) -> TerminalMenu:
+        if title is None:
+            title = "Please select a Studio out of the following:"
+
+        return TerminalMenu(studio_names, title=title, clear_menu_on_exit=True)
+
+    def _get_possible_studios(self) -> Dict[str, Studio]:
+        """Get all available studios in the teamspace."""
+        studios = {}
+
+        user = _get_authed_user()
+        for studio in self.teamspace.studios:
+            if studio._studio.user_id == user.id:
+                studios[studio.name] = studio
+        return studios
+
+    def __call__(self, studio: Optional[str] = None) -> Studio:
+        """Select a studio from the teamspace.
+
+        Args:
+            studio: Optional studio name to select. If not provided, will show interactive menu.
+
+        Returns:
+            Selected Studio object
+
+        Raises:
+            StudioCliError: If studio selection fails
+        """
+        try:
+            # try to resolve the studio from the name, environment or config
+            resolved_studio = None
+
+            with suppress(Exception):
+                resolved_studio = Studio(name=studio, teamspace=self.teamspace, create_ok=False)
+
+            if resolved_studio is not None:
+                return resolved_studio
+
+            if os.environ.get("LIGHTNING_NON_INTERACTIVE", "0") == "1" and studio is None:
+                raise ValueError(
+                    "Studio selection is not supported in non-interactive mode. Please provide a studio name."
+                )
+
+            click.echo(f"Listing studios in teamspace {self.teamspace.owner.name}/{self.teamspace.name}...")
+
+            possible_studios = self._get_possible_studios()
+
+            if not possible_studios:
+                raise ValueError(f"No studios found in teamspace {self.teamspace.name}")
+
+            if studio is None:
+                return self._get_studio_from_interactive_menu(possible_studios)
+
+            return self._get_studio_from_name(studio, possible_studios)
+
+        except KeyboardInterrupt:
+            raise KeyboardInterrupt from None
+
+        except Exception as e:
+            raise StudioCliError(
+                "Could not resolve a Studio. "
+                "Please pass it as an argument or contact Lightning AI directly to resolve this issue."
+            ) from e

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
@@ -853,6 +855,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest import V1Quest
 from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1QuestStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
+from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse

lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py

@@ -1356,6 +1356,176 @@ 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
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.billing_service_quote_annual_upsell(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1QuoteAnnualUpsellResponse
+                 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_quote_annual_upsell_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.billing_service_quote_annual_upsell_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def billing_service_quote_annual_upsell_with_http_info(self, **kwargs) -> 'V1QuoteAnnualUpsellResponse':  # noqa: E501
+        """billing_service_quote_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_quote_annual_upsell_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1QuoteAnnualUpsellResponse
+                 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_quote_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', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1QuoteAnnualUpsellResponse',  # 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_subscription(self, **kwargs) -> 'V1QuoteSubscriptionResponse':  # noqa: E501
         """billing_service_quote_subscription  # 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
@@ -805,6 +807,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest import V1Quest
 from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1QuestStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
+from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse

lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py

@@ -46,7 +46,7 @@ class AssistantIdConversationsBody(object):
         'conversation_id': 'str',
         'ephemeral': 'bool',
         'internal_conversation': 'bool',
-        'max_tokens': 'str',
+        'max_completion_tokens': 'str',
         'message': 'V1Message',
         'metadata': 'dict(str, str)',
         'name': 'str',
@@ -66,7 +66,7 @@ class AssistantIdConversationsBody(object):
         'conversation_id': 'conversationId',
         'ephemeral': 'ephemeral',
         'internal_conversation': 'internalConversation',
-        'max_tokens': 'maxTokens',
+        'max_completion_tokens': 'maxCompletionTokens',
         'message': 'message',
         'metadata': 'metadata',
         'name': 'name',
@@ -80,14 +80,14 @@ class AssistantIdConversationsBody(object):
         'tools': 'tools'
     }
 
-    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None, tools: 'list[V1Tool]' =None):  # noqa: E501
+    def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_completion_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, sent_at: 'datetime' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None, tools: 'list[V1Tool]' =None):  # noqa: E501
         """AssistantIdConversationsBody - a model defined in Swagger"""  # noqa: E501
         self._auto_name = None
         self._billing_project_id = None
         self._conversation_id = None
         self._ephemeral = None
         self._internal_conversation = None
-        self._max_tokens = None
+        self._max_completion_tokens = None
         self._message = None
         self._metadata = None
         self._name = None
@@ -110,8 +110,8 @@ class AssistantIdConversationsBody(object):
             self.ephemeral = ephemeral
         if internal_conversation is not None:
             self.internal_conversation = internal_conversation
-        if max_tokens is not None:
-            self.max_tokens = max_tokens
+        if max_completion_tokens is not None:
+            self.max_completion_tokens = max_completion_tokens
         if message is not None:
             self.message = message
         if metadata is not None:
@@ -241,25 +241,25 @@ class AssistantIdConversationsBody(object):
         self._internal_conversation = internal_conversation
 
     @property
-    def max_tokens(self) -> 'str':
-        """Gets the max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+    def max_completion_tokens(self) -> 'str':
+        """Gets the max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
 
 
-        :return: The max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+        :return: The max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
         :rtype: str
         """
-        return self._max_tokens
+        return self._max_completion_tokens
 
-    @max_tokens.setter
-    def max_tokens(self, max_tokens: 'str'):
-        """Sets the max_tokens of this AssistantIdConversationsBody.
+    @max_completion_tokens.setter
+    def max_completion_tokens(self, max_completion_tokens: 'str'):
+        """Sets the max_completion_tokens of this AssistantIdConversationsBody.
 
 
-        :param max_tokens: The max_tokens of this AssistantIdConversationsBody.  # noqa: E501
+        :param max_completion_tokens: The max_completion_tokens of this AssistantIdConversationsBody.  # noqa: E501
         :type: str
         """
 
-        self._max_tokens = max_tokens
+        self._max_completion_tokens = max_completion_tokens
 
     @property
     def message(self) -> 'V1Message':