lightning-sdk 0.2.24rc0__py3-none-any.whl → 2025.7.9__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.24.rc0"
+__version__ = "2025.07.09"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cluster_api.py

@@ -4,6 +4,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     Externalv1Cluster,
     V1CloudProvider,
     V1ClusterType,
+    V1ExternalCluster,
     V1ListClusterAcceleratorsResponse,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
@@ -28,6 +29,20 @@ class ClusterApi:
             raise ValueError(f"Cluster {cluster_id} does not exist")
         return res
 
+    def list_clusters(self, project_id: str) -> List[V1ExternalCluster]:
+        """Lists the clusters for a given project.
+
+        Args:
+            project_id: The project to list clusters for
+
+        Returns:
+            A list of clusters
+        """
+        res = self._client.cluster_service_list_project_clusters(
+            project_id=project_id,
+        )
+        return res.clusters
+
     def list_cluster_accelerators(self, cluster_id: str, org_id: str) -> V1ListClusterAcceleratorsResponse:
         """Lists the accelerators for a given cluster.
 

lightning_sdk/api/llm_api.py

@@ -69,7 +69,7 @@ class LLMApi:
         self,
         prompt: str,
         system_prompt: Optional[str],
-        max_completion_tokens: int,
+        max_completion_tokens: Optional[int],
         assistant_id: str,
         images: Optional[List[str]] = None,
         conversation_id: Optional[str] = None,
@@ -77,8 +77,13 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
+        ephemeral = os.getenv("LIGHTNING_EPHEMERAL", "false").lower() == "true"
+        if ephemeral:
+            conversation_id = None
+            name = None
         body = {
             "message": {
                 "author": {"role": "user"},
@@ -94,6 +99,9 @@ class LLMApi:
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
             "system_prompt": system_prompt,
+            "ephemeral": ephemeral,
+            "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
+            "parent_message_id": kwargs.get("parent_message_id", ""),
         }
         if images:
             for image in images:
@@ -117,7 +125,7 @@ class LLMApi:
         self,
         prompt: str,
         system_prompt: Optional[str],
-        max_completion_tokens: int,
+        max_completion_tokens: Optional[int],
         assistant_id: str,
         images: Optional[List[str]] = None,
         conversation_id: Optional[str] = None,
@@ -125,8 +133,13 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, AsyncGenerator[V1ConversationResponseChunk, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
+        ephemeral = os.getenv("LIGHTNING_EPHEMERAL", "false").lower() == "true"
+        if ephemeral:
+            conversation_id = None
+            name = None
         body = {
             "message": {
                 "author": {"role": "user"},
@@ -142,6 +155,9 @@ class LLMApi:
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
             "system_prompt": system_prompt,
+            "ephemeral": ephemeral,
+            "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
+            "parent_message_id": kwargs.get("parent_message_id", ""),
         }
         if images:
             for image in images:

lightning_sdk/api/pipeline_api.py

@@ -55,7 +55,7 @@ class PipelineApi:
         body = ProjectIdPipelinesBody(
             name=name,
             steps=steps,
-            shared_filesystem=self._prepare_shared_filesytem(shared_filesystem, steps, teamspace),
+            shared_filesystem=self._prepare_shared_filesystem(shared_filesystem, steps, teamspace),
             parent_pipeline_id=parent_pipeline_id or "",
         )
 
@@ -89,7 +89,7 @@ class PipelineApi:
     def delete(self, project_id: str, pipeline_id: str) -> V1DeletePipelineResponse:
         return self._client.pipelines_service_delete_pipeline(project_id, pipeline_id)
 
-    def _prepare_shared_filesytem(
+    def _prepare_shared_filesystem(
         self, shared_filesystem: Union[bool, V1SharedFilesystem], steps: List["V1PipelineStep"], teamspace: Teamspace
     ) -> V1SharedFilesystem:
         if not shared_filesystem:
@@ -97,14 +97,22 @@ class PipelineApi:
 
         from lightning_sdk.pipeline.utils import _get_cloud_account
 
-        cluster = self._cluster_api.get_cluster(
-            cluster_id=_get_cloud_account(steps), project_id=teamspace.id, org_id=teamspace.owner.id
-        )
+        clusters = self._cluster_api.list_clusters(project_id=teamspace.id)
+
+        selected_cluster = None
+        selected_cluster_id = _get_cloud_account(steps)
+        for cluster in clusters:
+            if cluster.id == selected_cluster_id:
+                selected_cluster = cluster
+                break
+
+        if selected_cluster is None:
+            raise ValueError(f"Cloud Account {selected_cluster_id} not found")
 
-        if cluster.spec.aws_v1:
+        if selected_cluster.spec.aws_v1:
             return V1SharedFilesystem(enabled=True, s3_folder=True)
 
-        if cluster.spec.google_cloud_v1:
+        if selected_cluster.spec.google_cloud_v1:
             return V1SharedFilesystem(enabled=True, gcs_folder=True)
 
         raise NotImplementedError("This cluster isn't support yet")

lightning_sdk/api/studio_api.py

@@ -2,7 +2,6 @@ import json
 import os
 import tempfile
 import time
-import warnings
 import zipfile
 from threading import Event, Thread
 from typing import Any, Dict, Generator, Mapping, Optional, Tuple, Union
@@ -172,14 +171,6 @@ class StudioApi:
         self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: False
     ) -> None:
         """Start an existing Studio."""
-        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
-            warnings.warn(
-                f"{Machine.CPU_SMALL} is not a valid machine for starting a Studio. "
-                "It is reserved for running jobs only. "
-                "The Studio will be started with a CPU machine instead."
-            )
-            machine = Machine.CPU
-
         self._client.cloud_space_service_start_cloud_space_instance(
             IdStartBody(
                 compute_config=V1UserRequestedComputeConfig(name=_machine_to_compute_name(machine), spot=interruptible)
@@ -229,14 +220,6 @@ class StudioApi:
         self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
     ) -> None:
         """Switches given Studio to a new machine type."""
-        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
-            warnings.warn(
-                f"{Machine.CPU_SMALL} is not a valid machine for switching a Studio. "
-                "It is reserved for running jobs only. "
-                "The Studio will be switched to a CPU machine instead."
-            )
-            machine = Machine.CPU
-
         compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -63,6 +63,7 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi
 # import ApiClient
 from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
 from lightning_sdk.lightning_cloud.openapi.configuration import Configuration
@@ -394,6 +395,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1Da
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_mount import V1DataConnectionMount
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_state import V1DataConnectionState
+from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_tier import V1DataConnectionTier
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_path import V1DataPath
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
@@ -575,6 +577,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import V1GetVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
@@ -869,6 +872,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_action_type import V1ScheduleActionType
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
@@ -947,6 +951,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_volume_response import V1UpdateVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_artifact_response import V1UploadAgentJobArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_response import V1UploadAgentJobOutputResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
@@ -991,6 +996,7 @@ from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import V
 from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_version_body1 import VersionsVersionBody1
+from lightning_sdk.lightning_cloud.openapi.models.volumes_id_body import VolumesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec as Gridv1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody as Body

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -44,3 +44,4 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -2428,6 +2428,8 @@ class ClusterServiceApi(object):
         :param datetime start_time:
         :param datetime end_time:
         :param bool available_only:
+        :param bool from_aggregate:
+        :param str apparent_provider:
         :return: V1ListClusterCapacityReservationsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -2453,12 +2455,14 @@ class ClusterServiceApi(object):
         :param datetime start_time:
         :param datetime end_time:
         :param bool available_only:
+        :param bool from_aggregate:
+        :param str apparent_provider:
         :return: V1ListClusterCapacityReservationsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cluster_id', 'start_time', 'end_time', 'available_only']  # noqa: E501
+        all_params = ['project_id', 'cluster_id', 'start_time', 'end_time', 'available_only', 'from_aggregate', 'apparent_provider']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -2497,6 +2501,10 @@ class ClusterServiceApi(object):
             query_params.append(('endTime', params['end_time']))  # noqa: E501
         if 'available_only' in params:
             query_params.append(('availableOnly', params['available_only']))  # noqa: E501
+        if 'from_aggregate' in params:
+            query_params.append(('fromAggregate', params['from_aggregate']))  # noqa: E501
+        if 'apparent_provider' in params:
+            query_params.append(('apparentProvider', params['apparent_provider']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py

@@ -1329,6 +1329,9 @@ class LitLoggerServiceApi(object):
         :param str user_id:
         :param str cloud_space_id:
         :param str app_id:
+        :param int limit:
+        :param int offset:
+        :param str order_by:
         :return: V1ListMetricsStreamsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1353,12 +1356,15 @@ class LitLoggerServiceApi(object):
         :param str user_id:
         :param str cloud_space_id:
         :param str app_id:
+        :param int limit:
+        :param int offset:
+        :param str order_by:
         :return: V1ListMetricsStreamsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'user_id', 'cloud_space_id', 'app_id']  # noqa: E501
+        all_params = ['project_id', 'user_id', 'cloud_space_id', 'app_id', 'limit', 'offset', 'order_by']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1391,6 +1397,12 @@ class LitLoggerServiceApi(object):
             query_params.append(('cloudSpaceId', params['cloud_space_id']))  # noqa: E501
         if 'app_id' in params:
             query_params.append(('appId', params['app_id']))  # noqa: E501
+        if 'limit' in params:
+            query_params.append(('limit', params['limit']))  # noqa: E501
+        if 'offset' in params:
+            query_params.append(('offset', params['offset']))  # noqa: E501
+        if 'order_by' in params:
+            query_params.append(('orderBy', params['order_by']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py

@@ -0,0 +1,258 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+from __future__ import absolute_import
+
+import re  # noqa: F401
+from typing import TYPE_CHECKING, Any
+
+# python 2 and python 3 compatibility library
+import six
+
+from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class VolumeServiceApi(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    Ref: https://github.com/swagger-api/swagger-codegen
+    """
+
+    def __init__(self, api_client=None):
+        if api_client is None:
+            api_client = ApiClient()
+        self.api_client = api_client
+
+    def volume_service_get_volume(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetVolumeResponse':  # noqa: E501
+        """volume_service_get_volume  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.volume_service_get_volume(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1GetVolumeResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.volume_service_get_volume_with_http_info(project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.volume_service_get_volume_with_http_info(project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def volume_service_get_volume_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetVolumeResponse':  # noqa: E501
+        """volume_service_get_volume  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.volume_service_get_volume_with_http_info(project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1GetVolumeResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', '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 volume_service_get_volume" % 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 `volume_service_get_volume`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `volume_service_get_volume`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['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}/volumes/{id}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetVolumeResponse',  # 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 volume_service_update_volume(self, body: 'VolumesIdBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateVolumeResponse':  # noqa: E501
+        """volume_service_update_volume  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.volume_service_update_volume(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VolumesIdBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateVolumeResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.volume_service_update_volume_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.volume_service_update_volume_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def volume_service_update_volume_with_http_info(self, body: 'VolumesIdBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateVolumeResponse':  # noqa: E501
+        """volume_service_update_volume  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.volume_service_update_volume_with_http_info(body, project_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param VolumesIdBody body: (required)
+        :param str project_id: (required)
+        :param str id: (required)
+        :return: V1UpdateVolumeResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', '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 volume_service_update_volume" % 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 `volume_service_update_volume`")  # noqa: E501
+        # 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 `volume_service_update_volume`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `volume_service_update_volume`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        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/projects/{projectId}/volumes/{id}', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateVolumeResponse',  # 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)

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -348,6 +348,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1Da
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_mount import V1DataConnectionMount
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_state import V1DataConnectionState
+from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_tier import V1DataConnectionTier
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_path import V1DataPath
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
@@ -529,6 +530,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import V1GetVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
@@ -823,6 +825,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_action_type import V1ScheduleActionType
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
@@ -901,6 +904,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_volume_response import V1UpdateVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_artifact_response import V1UploadAgentJobArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_response import V1UploadAgentJobOutputResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
@@ -945,4 +949,5 @@ from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import V
 from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_version_body1 import VersionsVersionBody1
+from lightning_sdk.lightning_cloud.openapi.models.volumes_id_body import VolumesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody