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

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.19.post0"
+__version__ = "2025.08.21"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/llm_api.py

@@ -146,7 +146,6 @@ class LLMApi:
                     {"contentType": "text", "parts": [prompt]},
                 ],
             },
-            "max_tokens": max_completion_tokens,
             "conversation_id": conversation_id,
             "billing_project_id": billing_project_id,
             "name": name,
@@ -159,6 +158,9 @@ class LLMApi:
             "parent_message_id": kwargs.get("parent_message_id", ""),
             "tools": tools,
         }
+        if max_completion_tokens is not None:
+            body["max_completion_tokens"] = max_completion_tokens
+
         if images:
             for image in images:
                 url = image
@@ -203,7 +205,6 @@ class LLMApi:
                     {"contentType": "text", "parts": [prompt]},
                 ],
             },
-            "max_completion_tokens": max_completion_tokens,
             "conversation_id": conversation_id,
             "billing_project_id": billing_project_id,
             "name": name,
@@ -216,6 +217,9 @@ class LLMApi:
             "parent_message_id": kwargs.get("parent_message_id", ""),
             "sent_at": datetime.datetime.now(datetime.timezone.utc).isoformat(timespec="microseconds"),
         }
+        if max_completion_tokens is not None:
+            body["max_completion_tokens"] = max_completion_tokens
+
         if images:
             for image in images:
                 url = image

lightning_sdk/api/studio_api.py

@@ -205,6 +205,32 @@ class StudioApi:
             instance_id = code_status.in_use.cloud_space_instance_id
             print(f"Studio started | {teamspace_id=} {studio_id=} {instance_id=}")
 
+    def start_studio_async(
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool = False,
+        max_runtime: Optional[int] = None,
+    ) -> None:
+        """Start an existing Studio without blocking."""
+        # need to go via kwargs for typing compatibility since autogenerated apis accept None but aren't typed with None
+        optional_kwargs_compute_body = {}
+
+        if max_runtime is not None:
+            optional_kwargs_compute_body["requested_run_duration_seconds"] = str(max_runtime)
+        self._client.cloud_space_service_start_cloud_space_instance(
+            IdStartBody(
+                compute_config=V1UserRequestedComputeConfig(
+                    name=_machine_to_compute_name(machine),
+                    spot=interruptible,
+                    **optional_kwargs_compute_body,
+                )
+            ),
+            teamspace_id,
+            studio_id,
+        )
+
     def stop_studio(self, studio_id: str, teamspace_id: str) -> None:
         """Stop an existing Studio."""
         self.stop_keeping_alive(teamspace_id=teamspace_id, studio_id=studio_id)
@@ -289,6 +315,79 @@ class StudioApi:
                 break
             time.sleep(1)
 
+    def switch_studio_machine_with_progress(
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool,
+        progress: Any,  # StudioProgressTracker - avoid circular import
+    ) -> None:
+        """Switches given Studio to a new machine type with progress tracking."""
+        progress.update_progress(10, "Requesting machine switch...")
+
+        self._request_switch(
+            studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible
+        )
+
+        progress.update_progress(20, "Waiting for machine allocation...")
+
+        # Wait until it's time to switch
+        requested_was_found = False
+        startup_status = None
+        base_progress = 20
+        max_wait_progress = 60
+        wait_counter = 0
+
+        while True:
+            status = self.get_studio_status(studio_id, teamspace_id)
+            requested_machine = status.requested
+
+            if requested_machine is not None:
+                requested_was_found = True
+                startup_status = requested_machine.startup_status
+
+            # if the requested machine was found in the past, use the in_use status instead.
+            # it might be that it either was cancelled or it actually is ready.
+            # Either way, since we're actually blocking below for the in use startup status
+            # it's safe to switch at this point
+            elif requested_was_found:
+                in_use_machine = status.in_use
+                if in_use_machine is not None:
+                    startup_status = in_use_machine.startup_status
+
+            if startup_status and startup_status.initial_restore_finished:
+                break
+
+            # Update progress gradually while waiting
+            wait_counter += 1
+            current_progress = min(base_progress + (wait_counter * 2), max_wait_progress)
+            progress.update_progress(current_progress, "Allocating new machine...")
+            time.sleep(1)
+
+        progress.update_progress(70, "Starting machine switch...")
+        self._client.cloud_space_service_switch_cloud_space_instance(teamspace_id, studio_id)
+
+        progress.update_progress(80, "Configuring new machine...")
+
+        # Wait until the new machine is ready to use
+        switch_counter = 0
+        while True:
+            in_use = self.get_studio_status(studio_id, teamspace_id).in_use
+            if in_use is None:
+                continue
+            startup_status = in_use.startup_status
+            if startup_status and startup_status.top_up_restore_finished:
+                break
+
+            # Update progress while waiting for machine to be ready
+            switch_counter += 1
+            current_progress = min(80 + switch_counter, 95)
+            progress.update_progress(current_progress, "Finalizing machine setup...")
+            time.sleep(1)
+
+        progress.complete("Machine switch completed successfully")
+
     def get_machine(self, studio_id: str, teamspace_id: str, cloud_account_id: str, org_id: str) -> Machine:
         """Get the current machine type the given Studio is running on."""
         response: V1CloudSpaceInstanceConfig = self._client.cloud_space_service_get_cloud_space_instance_config(

lightning_sdk/cli/legacy/create.py

@@ -7,10 +7,8 @@ import click
 from rich.console import Console
 
 from lightning_sdk import Machine, Studio
-from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.cli.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.machine import CloudProvider
-from lightning_sdk.utils.resolve import _resolve_deprecated_provider
 
 _MACHINE_VALUES = tuple(
     [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
@@ -83,14 +81,6 @@ def studio(
     menu = _TeamspacesMenu()
     teamspace_resolved = menu._resolve_teamspace(teamspace)
 
-    cloud_provider = str(_resolve_deprecated_provider(cloud_provider, provider))
-
-    if cloud_provider is not None:
-        cloud_account_api = CloudAccountApi()
-        cloud_account = cloud_account_api.resolve_cloud_account(
-            teamspace_resolved.id, cloud_account, cloud_provider, teamspace_resolved.default_cloud_account
-        )
-
     # default cloud account to current studios cloud account if run from studio
     # else it will fall back to teamspace default in the backend
     if cloud_account is None:
@@ -107,11 +97,19 @@ def studio(
         console.print(f"Studio with name {name} already exists. Using {new_name} instead.")
         name = new_name
 
-    studio = Studio(name=name, teamspace=teamspace_resolved, cloud_account=cloud_account, create_ok=True)
+    studio = Studio(
+        name=name,
+        teamspace=teamspace_resolved,
+        cloud_account=cloud_account,
+        create_ok=True,
+        cloud_provider=cloud_provider,
+        provider=provider,
+    )
 
     console.print(f"Created Studio {studio.name}.")
 
     if start is not None:
         start_machine = getattr(Machine, start, start)
+        Studio.show_progress = True
         studio.start(start_machine)
         console.print(f"Started Studio {studio.name} on machine {start}")

lightning_sdk/cli/legacy/start.py

@@ -103,4 +103,5 @@ def studio(
     except KeyError:
         resolved_machine = machine
 
+    Studio.show_progress = True
     studio.start(resolved_machine)

lightning_sdk/cli/legacy/switch.py

@@ -59,4 +59,5 @@ def studio(name: str, teamspace: Optional[str] = None, machine: str = "CPU") ->
     except KeyError:
         resolved_machine = machine
 
+    Studio.show_progress = True
     studio.switch_machine(resolved_machine)

lightning_sdk/cli/studio/start.py

@@ -77,5 +77,6 @@ def start_studio(
             raise ValueError(f"Could not start Studio: '{studio_name}'. Does the Studio exist?") from None
         raise ValueError(f"Could not start Studio: '{studio_name}'. Please provide a Studio name") from None
 
+    Studio.show_progress = True
     studio.start(machine, interruptible=interruptible)
     click.echo(f"Studio '{studio.name}' started successfully")

lightning_sdk/cli/studio/switch.py

@@ -47,6 +47,7 @@ def switch_studio(
         raise ValueError(f"Could not switch Studio: '{studio_name}'. Please provide a Studio name") from None
 
     resolved_machine = Machine.from_str(machine)
+    Studio.show_progress = True
     studio.switch_machine(resolved_machine, interruptible=interruptible)
 
     click.echo(f"Studio '{studio.name}' switched to machine '{resolved_machine}' successfully")

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -853,6 +853,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,91 @@ class BillingServiceApi(object):
             _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/models/__init__.py

@@ -805,6 +805,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':

lightning_sdk/lightning_cloud/openapi/models/v1_pod_metrics.py

@@ -43,6 +43,8 @@ class V1PodMetrics(object):
     swagger_types = {
         'cpu_usage': 'float',
         'labels': 'dict(str, str)',
+        'max_gpu_temp_recorded': 'float',
+        'max_power_per_gpu': 'float',
         'mem_usage': 'float',
         'namespace': 'str',
         'node_name': 'str',
@@ -50,6 +52,10 @@ class V1PodMetrics(object):
         'num_cpus_request': 'float',
         'num_gpus': 'float',
         'per_gpu_mem_used': 'dict(str, float)',
+        'per_gpu_power_usage_watts': 'dict(str, float)',
+        'per_gpu_sm_active': 'dict(str, float)',
+        'per_gpu_sm_occupancy': 'dict(str, float)',
+        'per_gpu_temperature_c': 'dict(str, float)',
         'per_gpu_util': 'dict(str, float)',
         'phase': 'str',
         'pod_id': 'str',
@@ -61,6 +67,8 @@ class V1PodMetrics(object):
     attribute_map = {
         'cpu_usage': 'cpuUsage',
         'labels': 'labels',
+        'max_gpu_temp_recorded': 'maxGpuTempRecorded',
+        'max_power_per_gpu': 'maxPowerPerGpu',
         'mem_usage': 'memUsage',
         'namespace': 'namespace',
         'node_name': 'nodeName',
@@ -68,6 +76,10 @@ class V1PodMetrics(object):
         'num_cpus_request': 'numCpusRequest',
         'num_gpus': 'numGpus',
         'per_gpu_mem_used': 'perGpuMemUsed',
+        'per_gpu_power_usage_watts': 'perGpuPowerUsageWatts',
+        'per_gpu_sm_active': 'perGpuSmActive',
+        'per_gpu_sm_occupancy': 'perGpuSmOccupancy',
+        'per_gpu_temperature_c': 'perGpuTemperatureC',
         'per_gpu_util': 'perGpuUtil',
         'phase': 'phase',
         'pod_id': 'podId',
@@ -76,10 +88,12 @@ class V1PodMetrics(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cpu_usage: 'float' =None, labels: 'dict(str, str)' =None, mem_usage: 'float' =None, namespace: 'str' =None, node_name: 'str' =None, num_cpus_limit: 'float' =None, num_cpus_request: 'float' =None, num_gpus: 'float' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, phase: 'str' =None, pod_id: 'str' =None, pod_name: 'str' =None, timestamp: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cpu_usage: 'float' =None, labels: 'dict(str, str)' =None, max_gpu_temp_recorded: 'float' =None, max_power_per_gpu: 'float' =None, mem_usage: 'float' =None, namespace: 'str' =None, node_name: 'str' =None, num_cpus_limit: 'float' =None, num_cpus_request: 'float' =None, num_gpus: 'float' =None, per_gpu_mem_used: 'dict(str, float)' =None, per_gpu_power_usage_watts: 'dict(str, float)' =None, per_gpu_sm_active: 'dict(str, float)' =None, per_gpu_sm_occupancy: 'dict(str, float)' =None, per_gpu_temperature_c: 'dict(str, float)' =None, per_gpu_util: 'dict(str, float)' =None, phase: 'str' =None, pod_id: 'str' =None, pod_name: 'str' =None, timestamp: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1PodMetrics - a model defined in Swagger"""  # noqa: E501
         self._cpu_usage = None
         self._labels = None
+        self._max_gpu_temp_recorded = None
+        self._max_power_per_gpu = None
         self._mem_usage = None
         self._namespace = None
         self._node_name = None
@@ -87,6 +101,10 @@ class V1PodMetrics(object):
         self._num_cpus_request = None
         self._num_gpus = None
         self._per_gpu_mem_used = None
+        self._per_gpu_power_usage_watts = None
+        self._per_gpu_sm_active = None
+        self._per_gpu_sm_occupancy = None
+        self._per_gpu_temperature_c = None
         self._per_gpu_util = None
         self._phase = None
         self._pod_id = None
@@ -98,6 +116,10 @@ class V1PodMetrics(object):
             self.cpu_usage = cpu_usage
         if labels is not None:
             self.labels = labels
+        if max_gpu_temp_recorded is not None:
+            self.max_gpu_temp_recorded = max_gpu_temp_recorded
+        if max_power_per_gpu is not None:
+            self.max_power_per_gpu = max_power_per_gpu
         if mem_usage is not None:
             self.mem_usage = mem_usage
         if namespace is not None:
@@ -112,6 +134,14 @@ class V1PodMetrics(object):
             self.num_gpus = num_gpus
         if per_gpu_mem_used is not None:
             self.per_gpu_mem_used = per_gpu_mem_used
+        if per_gpu_power_usage_watts is not None:
+            self.per_gpu_power_usage_watts = per_gpu_power_usage_watts
+        if per_gpu_sm_active is not None:
+            self.per_gpu_sm_active = per_gpu_sm_active
+        if per_gpu_sm_occupancy is not None:
+            self.per_gpu_sm_occupancy = per_gpu_sm_occupancy
+        if per_gpu_temperature_c is not None:
+            self.per_gpu_temperature_c = per_gpu_temperature_c
         if per_gpu_util is not None:
             self.per_gpu_util = per_gpu_util
         if phase is not None:
@@ -167,6 +197,48 @@ class V1PodMetrics(object):
 
         self._labels = labels
 
+    @property
+    def max_gpu_temp_recorded(self) -> 'float':
+        """Gets the max_gpu_temp_recorded of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The max_gpu_temp_recorded of this V1PodMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._max_gpu_temp_recorded
+
+    @max_gpu_temp_recorded.setter
+    def max_gpu_temp_recorded(self, max_gpu_temp_recorded: 'float'):
+        """Sets the max_gpu_temp_recorded of this V1PodMetrics.
+
+
+        :param max_gpu_temp_recorded: The max_gpu_temp_recorded of this V1PodMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._max_gpu_temp_recorded = max_gpu_temp_recorded
+
+    @property
+    def max_power_per_gpu(self) -> 'float':
+        """Gets the max_power_per_gpu of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The max_power_per_gpu of this V1PodMetrics.  # noqa: E501
+        :rtype: float
+        """
+        return self._max_power_per_gpu
+
+    @max_power_per_gpu.setter
+    def max_power_per_gpu(self, max_power_per_gpu: 'float'):
+        """Sets the max_power_per_gpu of this V1PodMetrics.
+
+
+        :param max_power_per_gpu: The max_power_per_gpu of this V1PodMetrics.  # noqa: E501
+        :type: float
+        """
+
+        self._max_power_per_gpu = max_power_per_gpu
+
     @property
     def mem_usage(self) -> 'float':
         """Gets the mem_usage of this V1PodMetrics.  # noqa: E501
@@ -314,6 +386,90 @@ class V1PodMetrics(object):
 
         self._per_gpu_mem_used = per_gpu_mem_used
 
+    @property
+    def per_gpu_power_usage_watts(self) -> 'dict(str, float)':
+        """Gets the per_gpu_power_usage_watts of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_power_usage_watts of this V1PodMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_power_usage_watts
+
+    @per_gpu_power_usage_watts.setter
+    def per_gpu_power_usage_watts(self, per_gpu_power_usage_watts: 'dict(str, float)'):
+        """Sets the per_gpu_power_usage_watts of this V1PodMetrics.
+
+
+        :param per_gpu_power_usage_watts: The per_gpu_power_usage_watts of this V1PodMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_power_usage_watts = per_gpu_power_usage_watts
+
+    @property
+    def per_gpu_sm_active(self) -> 'dict(str, float)':
+        """Gets the per_gpu_sm_active of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_sm_active of this V1PodMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_sm_active
+
+    @per_gpu_sm_active.setter
+    def per_gpu_sm_active(self, per_gpu_sm_active: 'dict(str, float)'):
+        """Sets the per_gpu_sm_active of this V1PodMetrics.
+
+
+        :param per_gpu_sm_active: The per_gpu_sm_active of this V1PodMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_sm_active = per_gpu_sm_active
+
+    @property
+    def per_gpu_sm_occupancy(self) -> 'dict(str, float)':
+        """Gets the per_gpu_sm_occupancy of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_sm_occupancy of this V1PodMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_sm_occupancy
+
+    @per_gpu_sm_occupancy.setter
+    def per_gpu_sm_occupancy(self, per_gpu_sm_occupancy: 'dict(str, float)'):
+        """Sets the per_gpu_sm_occupancy of this V1PodMetrics.
+
+
+        :param per_gpu_sm_occupancy: The per_gpu_sm_occupancy of this V1PodMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_sm_occupancy = per_gpu_sm_occupancy
+
+    @property
+    def per_gpu_temperature_c(self) -> 'dict(str, float)':
+        """Gets the per_gpu_temperature_c of this V1PodMetrics.  # noqa: E501
+
+
+        :return: The per_gpu_temperature_c of this V1PodMetrics.  # noqa: E501
+        :rtype: dict(str, float)
+        """
+        return self._per_gpu_temperature_c
+
+    @per_gpu_temperature_c.setter
+    def per_gpu_temperature_c(self, per_gpu_temperature_c: 'dict(str, float)'):
+        """Sets the per_gpu_temperature_c of this V1PodMetrics.
+
+
+        :param per_gpu_temperature_c: The per_gpu_temperature_c of this V1PodMetrics.  # noqa: E501
+        :type: dict(str, float)
+        """
+
+        self._per_gpu_temperature_c = per_gpu_temperature_c
+
     @property
     def per_gpu_util(self) -> 'dict(str, float)':
         """Gets the per_gpu_util of this V1PodMetrics.  # noqa: E501