lightning-sdk 0.2.24rc0__py3-none-any.whl → 0.2.24rc1__py3-none-any.whl

lightning_sdk/__init__.py

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

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/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

lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py

@@ -50,6 +50,7 @@ class AssistantIdConversationsBody(object):
         'message': 'V1Message',
         'metadata': 'dict(str, str)',
         'name': 'str',
+        'parent_conversation_id': 'str',
         'parent_message_id': 'str',
         'reasoning_effort': 'str',
         'store': 'bool',
@@ -67,6 +68,7 @@ class AssistantIdConversationsBody(object):
         'message': 'message',
         'metadata': 'metadata',
         'name': 'name',
+        'parent_conversation_id': 'parentConversationId',
         'parent_message_id': 'parentMessageId',
         'reasoning_effort': 'reasoningEffort',
         'store': 'store',
@@ -74,7 +76,7 @@ class AssistantIdConversationsBody(object):
         'system_prompt': 'systemPrompt'
     }
 
-    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_message_id: 'str' =None, reasoning_effort: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =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_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, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None):  # noqa: E501
         """AssistantIdConversationsBody - a model defined in Swagger"""  # noqa: E501
         self._auto_name = None
         self._billing_project_id = None
@@ -85,6 +87,7 @@ class AssistantIdConversationsBody(object):
         self._message = None
         self._metadata = None
         self._name = None
+        self._parent_conversation_id = None
         self._parent_message_id = None
         self._reasoning_effort = None
         self._store = None
@@ -109,6 +112,8 @@ class AssistantIdConversationsBody(object):
             self.metadata = metadata
         if name is not None:
             self.name = name
+        if parent_conversation_id is not None:
+            self.parent_conversation_id = parent_conversation_id
         if parent_message_id is not None:
             self.parent_message_id = parent_message_id
         if reasoning_effort is not None:
@@ -309,6 +314,27 @@ class AssistantIdConversationsBody(object):
 
         self._name = name
 
+    @property
+    def parent_conversation_id(self) -> 'str':
+        """Gets the parent_conversation_id of this AssistantIdConversationsBody.  # noqa: E501
+
+
+        :return: The parent_conversation_id of this AssistantIdConversationsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._parent_conversation_id
+
+    @parent_conversation_id.setter
+    def parent_conversation_id(self, parent_conversation_id: 'str'):
+        """Sets the parent_conversation_id of this AssistantIdConversationsBody.
+
+
+        :param parent_conversation_id: The parent_conversation_id of this AssistantIdConversationsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._parent_conversation_id = parent_conversation_id
+
     @property
     def parent_message_id(self) -> 'str':
         """Gets the parent_message_id of this AssistantIdConversationsBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py

@@ -48,7 +48,6 @@ class Externalv1UserStatus(object):
         'completed_signup': 'bool',
         'has_received_free_months': 'bool',
         'installed_grid': 'bool',
-        'should_auto_verify': 'bool',
         'verified': 'bool',
         'verified_at': 'datetime'
     }
@@ -61,12 +60,11 @@ class Externalv1UserStatus(object):
         'completed_signup': 'completedSignup',
         'has_received_free_months': 'hasReceivedFreeMonths',
         'installed_grid': 'installedGrid',
-        'should_auto_verify': 'shouldAutoVerify',
         'verified': 'verified',
         'verified_at': 'verifiedAt'
     }
 
-    def __init__(self, acked_storage_violation: 'bool' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, should_auto_verify: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, acked_storage_violation: 'bool' =None, auto_verify_reason: 'str' =None, auto_verify_response: 'bool' =None, completed_project_onboarding: 'bool' =None, completed_signup: 'bool' =None, has_received_free_months: 'bool' =None, installed_grid: 'bool' =None, verified: 'bool' =None, verified_at: 'datetime' =None):  # noqa: E501
         """Externalv1UserStatus - a model defined in Swagger"""  # noqa: E501
         self._acked_storage_violation = None
         self._auto_verify_reason = None
@@ -75,7 +73,6 @@ class Externalv1UserStatus(object):
         self._completed_signup = None
         self._has_received_free_months = None
         self._installed_grid = None
-        self._should_auto_verify = None
         self._verified = None
         self._verified_at = None
         self.discriminator = None
@@ -93,8 +90,6 @@ class Externalv1UserStatus(object):
             self.has_received_free_months = has_received_free_months
         if installed_grid is not None:
             self.installed_grid = installed_grid
-        if should_auto_verify is not None:
-            self.should_auto_verify = should_auto_verify
         if verified is not None:
             self.verified = verified
         if verified_at is not None:
@@ -247,27 +242,6 @@ class Externalv1UserStatus(object):
 
         self._installed_grid = installed_grid
 
-    @property
-    def should_auto_verify(self) -> 'bool':
-        """Gets the should_auto_verify of this Externalv1UserStatus.  # noqa: E501
-
-
-        :return: The should_auto_verify of this Externalv1UserStatus.  # noqa: E501
-        :rtype: bool
-        """
-        return self._should_auto_verify
-
-    @should_auto_verify.setter
-    def should_auto_verify(self, should_auto_verify: 'bool'):
-        """Sets the should_auto_verify of this Externalv1UserStatus.
-
-
-        :param should_auto_verify: The should_auto_verify of this Externalv1UserStatus.  # noqa: E501
-        :type: bool
-        """
-
-        self._should_auto_verify = should_auto_verify
-
     @property
     def verified(self) -> 'bool':
         """Gets the verified of this Externalv1UserStatus.  # noqa: E501