lightning-sdk 2025.10.22__py3-none-any.whl → 2025.10.23__py3-none-any.whl

lightning_sdk/__init__.py

@@ -35,7 +35,7 @@ __all__ = [
     "VM",
 ]
 
-__version__ = "2025.10.22"
+__version__ = "2025.10.23"
 
 _version_checker = VersionChecker()
 _version_checker.check_and_prompt_upgrade(__version__)

lightning_sdk/api/job_api.py

@@ -250,6 +250,7 @@ class JobApiV2:
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> V1Job:
         body = self._create_job_body(
             name=name,
@@ -267,6 +268,7 @@ class JobApiV2:
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
 
         job: V1Job = self._client.jobs_service_create_job(project_id=teamspace_id, body=body)
@@ -288,6 +290,7 @@ class JobApiV2:
         path_mappings: Optional[Dict[str, str]],
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings)
+        reuse_snapshot: bool,
         max_runtime: Optional[int] = None,
         machine_image_version: Optional[str] = None,
     ) -> ProjectIdJobsBody:
@@ -298,7 +301,7 @@ class JobApiV2:
 
         instance_name = _machine_to_compute_name(machine)
 
-        run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if studio_id is not None else ""
+        run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if (studio_id is not None and reuse_snapshot) else ""
 
         path_mappings_list = resolve_path_mappings(
             mappings=path_mappings or {},

lightning_sdk/cli/entrypoint.py

@@ -13,6 +13,7 @@ from lightning_sdk.cli.groups import (
     base_studio,
     config,
     # job,
+    license,
     # mmt,
     studio,
     vm,
@@ -61,6 +62,8 @@ main_cli.add_command(config)
 main_cli.add_command(studio)
 main_cli.add_command(vm)
 main_cli.add_command(base_studio)
+main_cli.add_command(license)
+
 if os.environ.get("LIGHTNING_EXPERIMENTAL_CLI_ONLY", "0") != "1":
     #### LEGACY COMMANDS ####
     # these commands are currently supported for backwards compatibility, but will potentially be removed in the future.

lightning_sdk/cli/groups.py

@@ -5,6 +5,7 @@ import click
 from lightning_sdk.cli.base_studio import register_commands as register_base_studio_commands
 from lightning_sdk.cli.config import register_commands as register_config_commands
 from lightning_sdk.cli.job import register_commands as register_job_commands
+from lightning_sdk.cli.license import register_commands as register_license_commands
 from lightning_sdk.cli.mmt import register_commands as register_mmt_commands
 from lightning_sdk.cli.studio import register_commands as register_studio_commands
 from lightning_sdk.cli.vm import register_commands as register_vm_commands
@@ -40,6 +41,11 @@ def base_studio() -> None:
     """Manage Lightning AI Base Studios."""
 
 
+@click.group(name="license")
+def license() -> None:  # noqa: A001
+    """Manage Lightning AI Product Licenses."""
+
+
 # Register config commands with the main config group
 register_job_commands(job)
 register_mmt_commands(mmt)
@@ -47,3 +53,4 @@ register_studio_commands(studio)
 register_config_commands(config)
 register_vm_commands(vm)
 register_base_studio_commands(base_studio)
+register_license_commands(license)

lightning_sdk/cli/license/__init__.py

@@ -0,0 +1,14 @@
+"""Base Studio CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register base studio commands with the given group."""
+    from lightning_sdk.cli.license.get import get_license
+    from lightning_sdk.cli.license.list import list_licenses
+    from lightning_sdk.cli.license.set import set_license
+
+    group.add_command(list_licenses)
+    group.add_command(get_license)
+    group.add_command(set_license)

lightning_sdk/cli/license/get.py

@@ -0,0 +1,15 @@
+import click
+
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("get")
+@click.argument("product_name")
+@click.option("--config-file", help="Path to the config file")
+def get_license(product_name: str, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """Get a license key for a given product."""
+    cfg = Config(config_file)
+    license_key = cfg.get(f"{DefaultConfigKeys.license}.{product_name}")
+    if license_key:
+        # echo the license key without any additional output to make parsing simpler
+        click.echo(license_key)

lightning_sdk/cli/license/list.py

@@ -0,0 +1,45 @@
+"""License list command."""
+
+from typing import Mapping
+
+import click
+from rich.table import Table
+
+from lightning_sdk.cli.utils.richt_print import rich_to_str
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("list")
+@click.option("--include-key", help="Print the key as well", is_flag=True)
+@click.option("--config-file", help="Path to the config file")
+def list_licenses(include_key: bool, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """List configured licenses.
+
+    Example:
+        lightning license list --include-key
+
+    """
+    return list_impl(include_key=include_key, config_path=config_file)
+
+
+def list_impl(include_key: bool, config_path: str) -> None:
+    cfg = Config(config_file=config_path)
+
+    license_cfg = cfg.get_sub_config(DefaultConfigKeys.license)
+
+    if isinstance(license_cfg, Mapping):
+        table = Table(
+            pad_edge=True,
+        )
+
+        table.add_column("Product")
+        table.add_column("License Key")
+
+        # sort by product_name
+        for product_name, license_key in sorted(license_cfg.items(), key=lambda x: x[0]):
+            table.add_row(product_name, license_key if include_key else "********")
+
+        click.echo(rich_to_str(table), color=True)
+
+    else:
+        click.echo("No licenses configured!")

lightning_sdk/cli/license/set.py

@@ -0,0 +1,13 @@
+import click
+
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("set")
+@click.argument("product_name")
+@click.argument("license_key")
+@click.option("--config-file", help="Path to the config file")
+def set_license(product_name: str, license_key: str, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """Set a license key for a given product."""
+    cfg = Config(config_file)
+    cfg.set(f"{DefaultConfigKeys.license}.{product_name}", license_key)

lightning_sdk/cli/studio/create.py

@@ -86,7 +86,7 @@ def create_impl(
             create_ok=True,
             cloud_provider=cloud_provider,
             cloud_account=cloud_account,
-            template_id=template_id,
+            studio_type=template_id,
         )
     except (RuntimeError, ValueError, ApiException):
         if name:

lightning_sdk/cli/utils/handle_machine_and_gpus_args.py

@@ -2,9 +2,7 @@ from typing import Dict, Optional, Set
 
 import click
 
-from lightning_sdk.machine import Machine
-
-DEFAULT_MACHINE = "CPU"
+from lightning_sdk.machine import DEFAULT_MACHINE, Machine
 
 
 def _split_gpus_spec(gpus: str) -> tuple[str, int]:

lightning_sdk/job/base.py

@@ -90,6 +90,7 @@ class _BaseJob(ABC):
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -134,6 +135,8 @@ class _BaseJob(ABC):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         from lightning_sdk.lightning_cloud.openapi.rest import ApiException
         from lightning_sdk.studio import Studio
@@ -231,6 +234,7 @@ class _BaseJob(ABC):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
 
     @abstractmethod
@@ -251,6 +255,7 @@ class _BaseJob(ABC):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Submit a new job to the Lightning AI platform.
 
@@ -290,6 +295,8 @@ class _BaseJob(ABC):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
 
     @abstractmethod

lightning_sdk/job/job.py

@@ -97,6 +97,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Run async workloads using a docker image or a compute environment from your studio.
 
@@ -140,6 +141,8 @@ class Job(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+        reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         ret_val = super().run(
             name=name,
@@ -162,6 +165,7 @@ class Job(_BaseJob):
             path_mappings=path_mappings,
             max_runtime=max_runtime,
             cluster=cluster,
+            reuse_snapshot=reuse_snapshot,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -186,6 +190,7 @@ class Job(_BaseJob):
         artifacts_local: Optional[str] = None,  # deprecated in terms of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in terms of path_mappings
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "Job":
         """Submit a new job to the Lightning AI platform.
 
@@ -224,6 +229,8 @@ class Job(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         self._job = self._internal_job._submit(
             machine=machine,
@@ -241,6 +248,7 @@ class Job(_BaseJob):
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         return self
 

lightning_sdk/job/v1.py

@@ -54,6 +54,7 @@ class _JobV1(_BaseJob):
         cloud_provider: Optional[str] = None,
         interruptible: bool = False,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        reuse_snapshot: bool = True,
     ) -> "_BaseJob":
         """Start a new async workload from your studio.
 
@@ -89,6 +90,7 @@ class _JobV1(_BaseJob):
             cluster=cluster,
             path_mappings=None,
             max_runtime=None,
+            reuse_snapshot=reuse_snapshot,
         )
 
     def _submit(
@@ -108,6 +110,7 @@ class _JobV1(_BaseJob):
         entrypoint: str = "sh -c",
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
+        reuse_snapshot: bool = True,
     ) -> "_JobV1":
         """Submit a job to run on a machine.
 

lightning_sdk/job/v2.py

@@ -53,6 +53,7 @@ class _JobV2(_BaseJob):
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
+        reuse_snapshot: bool = True,
     ) -> "_JobV2":
         """Submit a new job to the Lightning AI platform.
 
@@ -87,6 +88,8 @@ class _JobV2(_BaseJob):
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            reuse_snapshot: Whether the job should reuse a Studio snapshot when multiple jobs for the same Studio are
+                submitted. Turning this off may result in longer job startup times. Defaults to True.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -129,6 +132,7 @@ class _JobV2(_BaseJob):
             entrypoint=entrypoint,
             path_mappings=path_mappings,
             max_runtime=max_runtime,
+            reuse_snapshot=reuse_snapshot,
         )
         self._job = submitted
         self._name = submitted.name

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -549,6 +549,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space impo
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_job import V1FilesystemJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_mmt import V1FilesystemMMT
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_metric import V1FilesystemMetric
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_metrics import V1FilesystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import V1FilesystemSlurmJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
@@ -735,6 +736,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_space
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_metrics_response import V1ListFilesystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
@@ -780,6 +782,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_roles_response imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_encryption_keys_response import V1ListOrganizationClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organizations_response import V1ListOrganizationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_pipelines_response import V1ListPipelinesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_platform_notifications_response import V1ListPlatformNotificationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_pod_metrics_response import V1ListPodMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_profiler_captures_response import V1ListProfilerCapturesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_artifacts_response import V1ListProjectArtifactsResponse
@@ -887,6 +890,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1PipelineTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
+from lightning_sdk.lightning_cloud.openapi.models.v1_platform_notification import V1PlatformNotification
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_pod_metrics import V1PodMetrics
@@ -941,6 +945,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_required_balance_reason import V1RequiredBalanceReason
 from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
+from lightning_sdk.lightning_cloud.openapi.models.v1_reset_api_key_request import V1ResetAPIKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_reset_api_key_response import V1ResetAPIKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources

lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py

@@ -310,6 +310,91 @@ class AuthServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def auth_service_list_platform_notifications(self, **kwargs) -> 'V1ListPlatformNotificationsResponse':  # noqa: E501
+        """auth_service_list_platform_notifications  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_list_platform_notifications(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1ListPlatformNotificationsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.auth_service_list_platform_notifications_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.auth_service_list_platform_notifications_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def auth_service_list_platform_notifications_with_http_info(self, **kwargs) -> 'V1ListPlatformNotificationsResponse':  # noqa: E501
+        """auth_service_list_platform_notifications  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_list_platform_notifications_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :return: V1ListPlatformNotificationsResponse
+                 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 auth_service_list_platform_notifications" % 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/platform-notifications', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListPlatformNotificationsResponse',  # 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 auth_service_login(self, body: 'V1LoginRequest', **kwargs) -> 'V1LoginResponse':  # noqa: E501
         """auth_service_login  # noqa: E501
 
@@ -698,6 +783,103 @@ class AuthServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def auth_service_reset_api_key(self, body: 'V1ResetAPIKeyRequest', **kwargs) -> 'V1ResetAPIKeyResponse':  # noqa: E501
+        """ResetAPIKey generates a new API key for the user  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_reset_api_key(body, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param V1ResetAPIKeyRequest body: (required)
+        :return: V1ResetAPIKeyResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.auth_service_reset_api_key_with_http_info(body, **kwargs)  # noqa: E501
+        else:
+            (data) = self.auth_service_reset_api_key_with_http_info(body, **kwargs)  # noqa: E501
+            return data
+
+    def auth_service_reset_api_key_with_http_info(self, body: 'V1ResetAPIKeyRequest', **kwargs) -> 'V1ResetAPIKeyResponse':  # noqa: E501
+        """ResetAPIKey generates a new API key for the user  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.auth_service_reset_api_key_with_http_info(body, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param V1ResetAPIKeyRequest body: (required)
+        :return: V1ResetAPIKeyResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body']  # 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 auth_service_reset_api_key" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `auth_service_reset_api_key`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/auth/reset-api-key', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ResetAPIKeyResponse',  # 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 auth_service_token_login(self, body: 'V1TokenLoginRequest', **kwargs) -> 'V1TokenLoginResponse':  # noqa: E501
         """auth_service_token_login  # noqa: E501