lightning-sdk 2025.7.10__py3-none-any.whl → 2025.7.17__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.07.10"
+__version__ = "2025.07.17"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/llm_api.py

@@ -7,11 +7,12 @@ import threading
 import warnings
 from typing import Any, AsyncGenerator, Dict, Generator, List, Optional, Union
 
-from pip._vendor.urllib3 import HTTPResponse
-
-from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
-from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
-from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
+from lightning_sdk.lightning_cloud.openapi.models import (
+    StreamResultOfV1ConversationResponseChunk,
+    V1ConversationResponseChunk,
+    V1ResponseChoice,
+    V1ResponseChoiceDelta,
+)
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
 
@@ -54,7 +55,9 @@ class LLMApi:
             warnings.warn("Error decoding JSON:", decoded_line)
             return None
 
-    def _stream_chat_response(self, result: HTTPResponse) -> Generator[V1ConversationResponseChunk, None, None]:
+    def _stream_chat_response(
+        self, result: StreamResultOfV1ConversationResponseChunk
+    ) -> Generator[V1ConversationResponseChunk, None, None]:
         for line in result.stream():
             decoded_lines = line.decode("utf-8").strip()
             for decoded_line in decoded_lines.splitlines():
@@ -80,6 +83,7 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        tools: Optional[List[Dict[str, Any]]] = None,
         **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
@@ -105,6 +109,7 @@ class LLMApi:
             "ephemeral": ephemeral,
             "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
             "parent_message_id": kwargs.get("parent_message_id", ""),
+            "tools": tools,
         }
         if images:
             for image in images:

lightning_sdk/job/base.py

@@ -275,12 +275,13 @@ class _BaseJob(ABC):
         Caution: This also deletes all artifacts and snapshots associated with the job.
         """
 
-    def wait(self, interval: float = 5.0, timeout: Optional[float] = None) -> None:
+    def wait(self, interval: float = 5.0, timeout: Optional[float] = None, stop_on_timeout: bool = False) -> None:
         """Waits for the job to be either completed, manually stopped or failed.
 
         Args:
             interval: The number of seconds to spend in-between status checks.
             timeout: The maximum number of seconds to wait before raising an error. If None, waits forever.
+            stop_on_timeout: Whether to stop the job if it didn't finish within the timeout.
         """
         import time
 
@@ -292,16 +293,21 @@ class _BaseJob(ABC):
                 break
 
             if timeout is not None and time.time() - start > timeout:
+                if stop_on_timeout:
+                    self.stop()  # ensure the job is stopped if it didn't finish
                 raise TimeoutError("Job didn't finish within the provided timeout.")
 
             time.sleep(interval)
 
-    async def async_wait(self, interval: float = 5.0, timeout: Optional[float] = None) -> None:
+    async def async_wait(
+        self, interval: float = 5.0, timeout: Optional[float] = None, stop_on_timeout: bool = False
+    ) -> None:
         """Waits for the job to be either completed, manually stopped or failed.
 
         Args:
             interval: The number of seconds to spend in-between status checks.
             timeout: The maximum number of seconds to wait before raising an error. If None, waits forever.
+            stop_on_timeout: Whether to stop the job if it didn't finish within the timeout.
         """
         import asyncio
 
@@ -313,6 +319,8 @@ class _BaseJob(ABC):
                 break
 
             if timeout is not None and asyncio.get_event_loop().time() - start > timeout:
+                if stop_on_timeout:
+                    self.stop()  # ensure the job is stopped if it didn't finish
                 raise TimeoutError("Job didn't finish within the provided timeout.")
 
             await asyncio.sleep(interval)

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -208,6 +208,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import
 from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body import ProjectsProjectIdBody
 from lightning_sdk.lightning_cloud.openapi.models.protobuf_any import ProtobufAny
+from lightning_sdk.lightning_cloud.openapi.models.protobuf_null_value import ProtobufNullValue
 from lightning_sdk.lightning_cloud.openapi.models.query_query_id_body import QueryQueryIdBody
 from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
 from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
@@ -251,6 +252,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
 from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
@@ -263,6 +265,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_assistant import V1Assistan
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_item_status import V1AssistantKnowledgeItemStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_status import V1AssistantKnowledgeStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_model_status import V1AssistantModelStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_session_daily_aggregated import V1AssistantSessionDailyAggregated
 from lightning_sdk.lightning_cloud.openapi.models.v1_author import V1Author
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_domain_validation import V1AutoJoinDomainValidation
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_org_response import V1AutoJoinOrgResponse
@@ -526,6 +529,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
 from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_function_tool import V1FunctionTool
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
@@ -539,6 +543,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_clickhouse_assistant_session_daily_aggregated_response import V1GetClickhouseAssistantSessionDailyAggregatedResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_open_ports_response import V1GetCloudSpaceInstanceOpenPortsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
@@ -699,7 +704,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_notification_dialogs_response import V1ListNotificationDialogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_memberships_response import V1ListOrgMembershipsResponse
@@ -918,6 +923,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
+from lightning_sdk.lightning_cloud.openapi.models.v1_tool import V1Tool
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse

lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py

@@ -528,6 +528,115 @@ class BillingServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def billing_service_get_clickhouse_assistant_session_daily_aggregated(self, **kwargs) -> 'V1GetClickhouseAssistantSessionDailyAggregatedResponse':  # noqa: E501
+        """Aggregated assistant session endpoint to get credits spending information. Provides credits consumption data by particular project according to filters given.  # 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_get_clickhouse_assistant_session_daily_aggregated(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id:
+        :param datetime _from:
+        :param datetime to:
+        :param str user_id:
+        :param str org_id:
+        :param bool details:
+        :return: V1GetClickhouseAssistantSessionDailyAggregatedResponse
+                 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_get_clickhouse_assistant_session_daily_aggregated_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.billing_service_get_clickhouse_assistant_session_daily_aggregated_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def billing_service_get_clickhouse_assistant_session_daily_aggregated_with_http_info(self, **kwargs) -> 'V1GetClickhouseAssistantSessionDailyAggregatedResponse':  # noqa: E501
+        """Aggregated assistant session endpoint to get credits spending information. Provides credits consumption data by particular project according to filters given.  # 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_get_clickhouse_assistant_session_daily_aggregated_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id:
+        :param datetime _from:
+        :param datetime to:
+        :param str user_id:
+        :param str org_id:
+        :param bool details:
+        :return: V1GetClickhouseAssistantSessionDailyAggregatedResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', '_from', 'to', 'user_id', 'org_id', 'details']  # 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_get_clickhouse_assistant_session_daily_aggregated" % key
+                )
+            params[key] = val
+        del params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'project_id' in params:
+            query_params.append(('projectId', params['project_id']))  # noqa: E501
+        if '_from' in params:
+            query_params.append(('from', params['_from']))  # noqa: E501
+        if 'to' in params:
+            query_params.append(('to', params['to']))  # noqa: E501
+        if 'user_id' in params:
+            query_params.append(('userId', params['user_id']))  # noqa: E501
+        if 'org_id' in params:
+            query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'details' in params:
+            query_params.append(('details', params['details']))  # noqa: E501
+
+        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/clickhouse-assistant-session', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetClickhouseAssistantSessionDailyAggregatedResponse',  # 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_get_org_balance(self, org_id: 'str', **kwargs) -> 'V1GetProjectBalanceResponse':  # noqa: E501
         """billing_service_get_org_balance  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1642,6 +1642,7 @@ class ClusterServiceApi(object):
         :param str id: (required)
         :param str org_id:
         :param str project_id:
+        :param str auth_token:
         :return: Externalv1Cluster
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1665,12 +1666,13 @@ class ClusterServiceApi(object):
         :param str id: (required)
         :param str org_id:
         :param str project_id:
+        :param str auth_token:
         :return: Externalv1Cluster
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['id', 'org_id', 'project_id']  # noqa: E501
+        all_params = ['id', 'org_id', 'project_id', 'auth_token']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1701,6 +1703,8 @@ class ClusterServiceApi(object):
             query_params.append(('orgId', params['org_id']))  # noqa: E501
         if 'project_id' in params:
             query_params.append(('projectId', params['project_id']))  # noqa: E501
+        if 'auth_token' in params:
+            query_params.append(('authToken', params['auth_token']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/user_service_api.py

@@ -884,38 +884,38 @@ class UserServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def user_service_list_new_features_for_user(self, **kwargs) -> 'V1ListNewFeaturesForUserResponse':  # noqa: E501
+    def user_service_list_notification_dialogs(self, **kwargs) -> 'V1ListNotificationDialogsResponse':  # noqa: E501
         """we use this to show ne user unseen new features modal  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.user_service_list_new_features_for_user(async_req=True)
+        >>> thread = api.user_service_list_notification_dialogs(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
         :param bool unseen:
-        :return: V1ListNewFeaturesForUserResponse
+        :return: V1ListNotificationDialogsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
         kwargs['_return_http_data_only'] = True
         if kwargs.get('async_req'):
-            return self.user_service_list_new_features_for_user_with_http_info(**kwargs)  # noqa: E501
+            return self.user_service_list_notification_dialogs_with_http_info(**kwargs)  # noqa: E501
         else:
-            (data) = self.user_service_list_new_features_for_user_with_http_info(**kwargs)  # noqa: E501
+            (data) = self.user_service_list_notification_dialogs_with_http_info(**kwargs)  # noqa: E501
             return data
 
-    def user_service_list_new_features_for_user_with_http_info(self, **kwargs) -> 'V1ListNewFeaturesForUserResponse':  # noqa: E501
+    def user_service_list_notification_dialogs_with_http_info(self, **kwargs) -> 'V1ListNotificationDialogsResponse':  # noqa: E501
         """we use this to show ne user unseen new features modal  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.user_service_list_new_features_for_user_with_http_info(async_req=True)
+        >>> thread = api.user_service_list_notification_dialogs_with_http_info(async_req=True)
         >>> result = thread.get()
 
         :param async_req bool
         :param bool unseen:
-        :return: V1ListNewFeaturesForUserResponse
+        :return: V1ListNotificationDialogsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -931,7 +931,7 @@ class UserServiceApi(object):
             if key not in all_params:
                 raise TypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method user_service_list_new_features_for_user" % key
+                    " to method user_service_list_notification_dialogs" % key
                 )
             params[key] = val
         del params['kwargs']
@@ -958,14 +958,14 @@ class UserServiceApi(object):
         auth_settings = []  # noqa: E501
 
         return self.api_client.call_api(
-            '/v1/users/new-features', 'GET',
+            '/v1/users/notification-dialogs', 'GET',
             path_params,
             query_params,
             header_params,
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='V1ListNewFeaturesForUserResponse',  # noqa: E501
+            response_type='V1ListNotificationDialogsResponse',  # noqa: E501
             auth_settings=auth_settings,
             async_req=params.get('async_req'),
             _return_http_data_only=params.get('_return_http_data_only'),

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -161,6 +161,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import
 from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body import ProjectsProjectIdBody
 from lightning_sdk.lightning_cloud.openapi.models.protobuf_any import ProtobufAny
+from lightning_sdk.lightning_cloud.openapi.models.protobuf_null_value import ProtobufNullValue
 from lightning_sdk.lightning_cloud.openapi.models.query_query_id_body import QueryQueryIdBody
 from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
 from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
@@ -204,6 +205,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
 from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
@@ -216,6 +218,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_assistant import V1Assistan
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_item_status import V1AssistantKnowledgeItemStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_knowledge_status import V1AssistantKnowledgeStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_model_status import V1AssistantModelStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_assistant_session_daily_aggregated import V1AssistantSessionDailyAggregated
 from lightning_sdk.lightning_cloud.openapi.models.v1_author import V1Author
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_domain_validation import V1AutoJoinDomainValidation
 from lightning_sdk.lightning_cloud.openapi.models.v1_auto_join_org_response import V1AutoJoinOrgResponse
@@ -479,6 +482,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
 from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_function_tool import V1FunctionTool
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
@@ -492,6 +496,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_clickhouse_assistant_session_daily_aggregated_response import V1GetClickhouseAssistantSessionDailyAggregatedResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_open_ports_response import V1GetCloudSpaceInstanceOpenPortsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
@@ -652,7 +657,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_notification_dialogs_response import V1ListNotificationDialogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_memberships_response import V1ListOrgMembershipsResponse
@@ -871,6 +876,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
+from lightning_sdk.lightning_cloud.openapi.models.v1_tool import V1Tool
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse

lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py

@@ -56,7 +56,8 @@ class AssistantIdConversationsBody(object):
         'sent_at': 'datetime',
         'store': 'bool',
         'stream': 'bool',
-        'system_prompt': 'str'
+        'system_prompt': 'str',
+        'tools': 'list[V1Tool]'
     }
 
     attribute_map = {
@@ -75,10 +76,11 @@ class AssistantIdConversationsBody(object):
         'sent_at': 'sentAt',
         'store': 'store',
         'stream': 'stream',
-        'system_prompt': 'systemPrompt'
+        'system_prompt': 'systemPrompt',
+        '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):  # 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, 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
@@ -96,6 +98,7 @@ class AssistantIdConversationsBody(object):
         self._store = None
         self._stream = None
         self._system_prompt = None
+        self._tools = None
         self.discriminator = None
         if auto_name is not None:
             self.auto_name = auto_name
@@ -129,6 +132,8 @@ class AssistantIdConversationsBody(object):
             self.stream = stream
         if system_prompt is not None:
             self.system_prompt = system_prompt
+        if tools is not None:
+            self.tools = tools
 
     @property
     def auto_name(self) -> 'bool':
@@ -466,6 +471,27 @@ class AssistantIdConversationsBody(object):
 
         self._system_prompt = system_prompt
 
+    @property
+    def tools(self) -> 'list[V1Tool]':
+        """Gets the tools of this AssistantIdConversationsBody.  # noqa: E501
+
+
+        :return: The tools of this AssistantIdConversationsBody.  # noqa: E501
+        :rtype: list[V1Tool]
+        """
+        return self._tools
+
+    @tools.setter
+    def tools(self, tools: 'list[V1Tool]'):
+        """Sets the tools of this AssistantIdConversationsBody.
+
+
+        :param tools: The tools of this AssistantIdConversationsBody.  # noqa: E501
+        :type: list[V1Tool]
+        """
+
+        self._tools = tools
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py

@@ -41,28 +41,23 @@ class ProjectIdStorageBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'allow_multicloud': 'bool',
         'cluster_id': 'str',
         'count': 'str',
         'filename': 'str'
     }
 
     attribute_map = {
-        'allow_multicloud': 'allowMulticloud',
         'cluster_id': 'clusterId',
         'count': 'count',
         'filename': 'filename'
     }
 
-    def __init__(self, allow_multicloud: 'bool' =None, cluster_id: 'str' =None, count: 'str' =None, filename: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, count: 'str' =None, filename: 'str' =None):  # noqa: E501
         """ProjectIdStorageBody - a model defined in Swagger"""  # noqa: E501
-        self._allow_multicloud = None
         self._cluster_id = None
         self._count = None
         self._filename = None
         self.discriminator = None
-        if allow_multicloud is not None:
-            self.allow_multicloud = allow_multicloud
         if cluster_id is not None:
             self.cluster_id = cluster_id
         if count is not None:
@@ -70,27 +65,6 @@ class ProjectIdStorageBody(object):
         if filename is not None:
             self.filename = filename
 
-    @property
-    def allow_multicloud(self) -> 'bool':
-        """Gets the allow_multicloud of this ProjectIdStorageBody.  # noqa: E501
-
-
-        :return: The allow_multicloud of this ProjectIdStorageBody.  # noqa: E501
-        :rtype: bool
-        """
-        return self._allow_multicloud
-
-    @allow_multicloud.setter
-    def allow_multicloud(self, allow_multicloud: 'bool'):
-        """Sets the allow_multicloud of this ProjectIdStorageBody.
-
-
-        :param allow_multicloud: The allow_multicloud of this ProjectIdStorageBody.  # noqa: E501
-        :type: bool
-        """
-
-        self._allow_multicloud = allow_multicloud
-
     @property
     def cluster_id(self) -> 'str':
         """Gets the cluster_id of this ProjectIdStorageBody.  # noqa: E501