lightning-sdk 2025.11.13__py3-none-any.whl → 2025.11.21__py3-none-any.whl

lightning_sdk/__init__.py

@@ -5,6 +5,7 @@ from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__  # noq
 from lightning_sdk.deployment import Deployment
 from lightning_sdk.helpers import VersionChecker, set_tqdm_envvars_noninteractive
 from lightning_sdk.job import Job
+from lightning_sdk.k8s_cluster import K8sCluster
 from lightning_sdk.machine import CloudProvider, Machine
 from lightning_sdk.mmt import MMT
 from lightning_sdk.organization import Organization
@@ -19,6 +20,7 @@ __all__ = [
     "VM",
     "AIHub",
     "Agent",
+    "K8sCluster",
     "CloudProvider",
     "ConnectionType",
     "Deployment",

lightning_sdk/__version__.py

@@ -1,3 +1,3 @@
 """Version information for lightning_sdk."""
 
-__version__ = "2025.11.13"
+__version__ = "2025.11.21"

lightning_sdk/api/__init__.py

@@ -1,17 +1,10 @@
 from lightning_sdk.api.agents_api import AgentApi
 from lightning_sdk.api.ai_hub_api import AIHubApi
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
+from lightning_sdk.api.k8s_api import K8sClusterApi
 from lightning_sdk.api.org_api import OrgApi
 from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.api.teamspace_api import TeamspaceApi
 from lightning_sdk.api.user_api import UserApi
 
-__all__ = [
-    "OrgApi",
-    "StudioApi",
-    "TeamspaceApi",
-    "UserApi",
-    "AgentApi",
-    "AIHubApi",
-    "CloudAccountApi",
-]
+__all__ = ["OrgApi", "StudioApi", "TeamspaceApi", "UserApi", "AgentApi", "AIHubApi", "CloudAccountApi", "K8sClusterApi"]

lightning_sdk/api/k8s_api.py

@@ -0,0 +1,105 @@
+import json
+import logging
+from typing import Any, Dict, TypedDict, Union
+
+import pandas as pd
+
+from lightning_sdk.api.utils import ApiException
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
+
+logger = logging.getLogger(__name__)
+
+
+class K8sClusterApiError(Exception):
+    """Custom exception for K8sClusterApi errors."""
+
+
+class RowData(TypedDict):
+    num_allocated_gpus: int
+    num_requested_gpus: int
+    num_gpus: int
+
+
+def _calculate_billed_k8s_gpus(row: RowData) -> int:
+    """Calculate the number of GPUs to be billed based on the given row data.
+
+    The function determines the billed GPUs using the following logic:
+    1. If the number of allocated GPUs (`num_allocated_gpus`) is greater than 0,
+       it returns the allocated GPUs.
+    2. If the number of requested GPUs (`num_requested_gpus`) exceeds the available GPUs (`num_gpus`),
+       it returns the available GPUs.
+    3. Otherwise, it returns the number of requested GPUs.
+
+    Returns:
+        int: The number of GPUs to be billed.
+    """
+    if row["num_allocated_gpus"] > 0:
+        return row["num_allocated_gpus"]  # Use allocated GPUs if available
+    if row["num_requested_gpus"] > row["num_gpus"]:
+        return row["num_gpus"]  # Use available GPUs if requested exceeds available
+    return row["num_requested_gpus"]  # Otherwise, use requested GPUs
+
+
+class K8sClusterApi:
+    """Internal API client for API requests to k8s endpoints."""
+
+    def __init__(self, cloud_account: str) -> None:
+        self.cloud_account = cloud_account
+        self._client = LightningClient(max_tries=7)
+
+    def _parse_request_failure_body(self, e: ApiException) -> str:
+        """Parses the failure body from an ApiException.
+
+        Args:
+            e: The ApiException instance.
+
+        Returns:
+            The parsed failure body as a string.
+        """
+        try:
+            if e.body:
+                return json.loads(e.body)["message"]
+            return "No additional error information provided."
+        except Exception:
+            return str(e.reason)
+
+    def get_billing_usage(self, print_data: bool = False, **kwargs: Dict[str, Any]) -> Union[pd.DataFrame, pd.Series]:
+        """Gets the k8s usage metrics.
+
+        Returns:
+            The k8s usage metrics as a DataFrame or Series.
+        """
+        try:
+            response = self._client.k8_s_cluster_service_list_cluster_metrics(self.cloud_account, **kwargs)
+            cluster_metrics = [entry.to_dict() for entry in response.cluster_metrics]
+
+            df = pd.DataFrame.from_records(cluster_metrics)
+            if df.empty:
+                return df
+
+            df["hour"] = pd.to_datetime(df["timestamp"]).dt.floor("h")
+
+            # Calculate the mean of num_allocated_gpus for each hour
+            aggregated = df.groupby("hour", as_index=False)["num_allocated_gpus"].mean()
+            # Merge the aggregated values back into the original DataFrame
+            df = df.merge(aggregated, on="hour", suffixes=("", "_mean"))
+
+            # Replace the original num_allocated_gpus with the mean values
+            df["num_allocated_gpus"] = df["num_allocated_gpus_mean"]
+
+            # We group the data by hour and take the first occurrence to avoid duplicates
+            df = df.drop_duplicates(subset="hour", keep="first")
+
+            # Convert timestamp to hourly floor and rename columns
+            df["billed_gpus"] = df.apply(_calculate_billed_k8s_gpus, axis=1)
+
+            # Keep only the required columns
+            df = df[["hour", "num_gpus", "num_requested_gpus", "num_allocated_gpus", "billed_gpus"]]
+            if print_data:
+                with pd.option_context("display.max_rows", None, "display.max_columns", None):
+                    print(df)
+            return df
+        except ApiException as e:
+            msg = self._parse_request_failure_body(e)
+            logger.error(f"Failed to retrieve Kubernetes usage data: {msg}")
+            raise K8sClusterApiError(msg) from e

lightning_sdk/api/studio_api.py

@@ -411,6 +411,25 @@ class StudioApi:
 
         progress.complete("Machine switch completed successfully")
 
+    def machine_has_capacity(self, machine: Machine, teamspace_id: str, cloud_account_id: str, org_id: str) -> bool:
+        """Check capacity of the requested machine."""
+        accelerators = self._get_machines_for_cloud_account(
+            teamspace_id=teamspace_id, cloud_account_id=cloud_account_id, org_id=org_id
+        )
+
+        for accelerator in accelerators:
+            if accelerator.accelerator_type == "GPU":
+                accelerator_resources_count = accelerator.resources.gpu
+            else:
+                accelerator_resources_count = accelerator.resources.cpu
+            if (
+                machine.accelerator_count == accelerator_resources_count
+                and machine.family == accelerator.family
+                and accelerator.out_of_capacity
+            ):
+                return False
+        return True
+
     def get_machine(self, studio_id: str, teamspace_id: str, cloud_account_id: str, org_id: str) -> Machine:
         """Get the current machine type the given Studio is running on."""
         response: V1CloudSpaceInstanceConfig = self._client.cloud_space_service_get_cloud_space_instance_config(

lightning_sdk/api/utils.py

@@ -4,6 +4,7 @@ import math
 import os
 import re
 from concurrent.futures import ThreadPoolExecutor
+from datetime import datetime, timezone
 from enum import Enum
 from functools import lru_cache, partial
 from pathlib import Path
@@ -804,3 +805,21 @@ def raise_access_error_if_not_allowed(resource_type: AccessibleResource, teamspa
             f"Access to {resource_type.name} has been disabled for this teamspace. "
             "Contact a teamspace administrator to enable it."
         )
+
+
+def to_iso_z(dt: datetime) -> str:
+    """Convert a datetime object to an ISO 8601 formatted string with UTC timezone (Z).
+
+    This function takes a datetime object, converts it to UTC timezone, formats it
+    to include milliseconds, and replaces the UTC offset with 'Z' to indicate UTC.
+
+    Args:
+        dt (datetime): The datetime object to be converted.
+
+    Returns:
+        str: The ISO 8601 formatted string in UTC timezone.
+    """
+    if dt.tzinfo is None:
+        dt = dt.replace(tzinfo=timezone.utc)
+        return dt.astimezone(timezone.utc).isoformat(timespec="milliseconds")
+    return dt.isoformat(timespec="milliseconds")

lightning_sdk/exceptions.py

@@ -0,0 +1,2 @@
+class OutOfCapacityError(RuntimeError):
+    """Raised when the requested machine is not available in the selected cloud account."""

lightning_sdk/k8s_cluster.py

@@ -0,0 +1,83 @@
+from dataclasses import dataclass
+from datetime import datetime
+from typing import List, Optional
+
+import pandas as pd
+
+from lightning_sdk.api.k8s_api import K8sClusterApi
+from lightning_sdk.api.utils import to_iso_z
+
+
+@dataclass
+class HourlyUsage:
+    time: datetime
+    available_gpus: int
+    billed_gpus: int
+
+
+@dataclass
+class K8sUsageResponse:
+    hours: List[HourlyUsage]
+    total_usage: float
+
+
+class K8sCluster:
+    """A class to interact with the k8s API and retrieve usage-related information.
+
+    Methods:
+    -------
+    get_k8s_usage(account_cloud: str)
+      Retrieves Kubernetes usage information for a given cloud account.
+    """
+
+    def __init__(self, cloud_account: str) -> None:
+        self._cloud_account = cloud_account
+        self._k8s_cluster = K8sClusterApi(cloud_account=cloud_account)
+
+    def _convert_to_k8s_usage_response(self, df: pd.DataFrame) -> K8sUsageResponse:
+        """Converts a DataFrame to K8sUsageResponse.
+
+        Args:
+            df (pd.DataFrame): The DataFrame containing GPU usage data.
+
+        Returns:
+            K8sUsageResponse: The converted response containing hourly usage and total usage.
+        """
+        if df.empty:
+            return K8sUsageResponse(hours=[], total_usage=0.0)
+        # Convert each row of the DataFrame to HourlyUsage
+        hourly_usage_list: List[HourlyUsage] = [
+            HourlyUsage(time=row["hour"], available_gpus=row["num_gpus"], billed_gpus=row["billed_gpus"])
+            for _, row in df.iterrows()
+        ]
+
+        # Calculate total usage (sum of billed GPUs)
+        total_usage = df["billed_gpus"].sum()
+
+        # Create and return the K8sUsageResponse
+        return K8sUsageResponse(hours=hourly_usage_list, total_usage=total_usage)
+
+    def get_billing_usage(
+        self,
+        start_date: Optional[datetime] = None,
+        end_date: Optional[datetime] = None,
+        print_data: bool = False,
+    ) -> K8sUsageResponse:
+        """Gets the k8s usage metrics.
+
+        Args:
+            cloud_account: The cloud account to get usage for
+            start_date: The UTC start date for the usage period (optional)
+            end_date: The UTC end date for the usage period (optional)
+
+        Returns:
+            K8sUsageResponse: The Kubernetes usage response containing hourly usage and total usage.
+        """
+        k8s_args = {}
+        if start_date is not None:
+            k8s_args["start"] = to_iso_z(start_date)
+        if end_date is not None:
+            k8s_args["end"] = to_iso_z(end_date)
+        return self._convert_to_k8s_usage_response(
+            self._k8s_cluster.get_billing_usage(print_data=print_data, **k8s_args)
+        )

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -120,6 +120,7 @@ from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body imp
 from lightning_sdk.lightning_cloud.openapi.models.datasets_id_body import DatasetsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.deployment_id_alertingpolicies_body import DeploymentIdAlertingpoliciesBody
 from lightning_sdk.lightning_cloud.openapi.models.deployment_id_alertingpolicies_body1 import DeploymentIdAlertingpoliciesBody1
+from lightning_sdk.lightning_cloud.openapi.models.deployments_deployment_id_body import DeploymentsDeploymentIdBody
 from lightning_sdk.lightning_cloud.openapi.models.deployments_id_body import DeploymentsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.deploymenttemplates_id_body import DeploymenttemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.endpoints_id_body import EndpointsIdBody
@@ -135,6 +136,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_artifacts_body import IdArt
 from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCodeconfigBody
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
 from lightning_sdk.lightning_cloud.openapi.models.id_contactowner_body import IdContactownerBody
+from lightning_sdk.lightning_cloud.openapi.models.id_content_body import IdContentBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body import IdExecuteBody
@@ -164,6 +166,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdV
 from lightning_sdk.lightning_cloud.openapi.models.incident_id_messages_body import IncidentIdMessagesBody
 from lightning_sdk.lightning_cloud.openapi.models.incidents_id_body import IncidentsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.job_id_reportroutingtelemetry_body import JobIdReportroutingtelemetryBody
+from lightning_sdk.lightning_cloud.openapi.models.job_id_systemmetrics_body import JobIdSystemmetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -372,6 +375,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload import V1Co
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_project_artifact_response import V1CompleteUploadProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_temporary_artifact_request import V1CompleteUploadTemporaryArtifactRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_completed_part import V1CompletedPart
+from lightning_sdk.lightning_cloud.openapi.models.v1_compute_cluster_machines import V1ComputeClusterMachines
+from lightning_sdk.lightning_cloud.openapi.models.v1_compute_cluster_request import V1ComputeClusterRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1ComputeConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
@@ -573,6 +578,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection import
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection_setup import V1GcpDataConnectionSetup
 from lightning_sdk.lightning_cloud.openapi.models.v1_ge_list_deployment_routing_telemetry_response import V1GeListDeploymentRoutingTelemetryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_generate_ssh_key_pair_request import V1GenerateSSHKeyPairRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_generic_job import V1GenericJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response import V1GetAffiliateLinkResponse
 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
@@ -671,6 +677,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_aws_config impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_settings_v1 import V1KubernetesDirectSettingsV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod import V1KubernetesPod
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template import V1KubernetesTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
@@ -699,6 +706,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_affiliate_links_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_response import V1ListAgentJobArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_aggregated_pod_metrics_response import V1ListAggregatedPodMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_all_jobs_response import V1ListAllJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_blog_posts_response import V1ListBlogPostsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
@@ -757,8 +765,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_incidents_response imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_files_response import V1ListJobFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_resources_response import V1ListJobResourcesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_response import V1ListJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_sort_by import V1ListJobsSortBy
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_sort_order import V1ListJobsSortOrder
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations_response import V1ListJoinableOrganizationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kai_scheduler_queues_metrics_response import V1ListKaiSchedulerQueuesMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_pods_response import V1ListKubernetesPodsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_license_response import V1ListLicenseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
@@ -877,6 +888,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_notification_preferences_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_notification_type import V1NotificationType
 from lightning_sdk.lightning_cloud.openapi.models.v1_onboarding_event_request import V1OnboardingEventRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_onboarding_event_response import V1OnboardingEventResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_org_deployment_visibility import V1OrgDeploymentVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_org_member import V1OrgMember
 from lightning_sdk.lightning_cloud.openapi.models.v1_org_membership import V1OrgMembership
 from lightning_sdk.lightning_cloud.openapi.models.v1_org_membership_role_binding import V1OrgMembershipRoleBinding
@@ -904,6 +916,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_platform_notification impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_pod_metrics import V1PodMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_pod_preemption_event import V1PodPreemptionEvent
 from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_presigned_url import V1PresignedUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_capture import V1ProfilerCapture
@@ -946,6 +959,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_render_kubernetes_template_
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_deployment_routing_telemetry_response import V1ReportDeploymentRoutingTelemetryResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_job_system_metrics_response import V1ReportJobSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
@@ -1054,6 +1068,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_like_response import V1UpdateConversationLikeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_message_content_response import V1UpdateConversationMessageContentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_conversation_message_like_response import V1UpdateConversationMessageLikeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -2007,6 +2007,7 @@ class AssistantsServiceApi(object):
         :param str conversation_id: (required)
         :param str message_id: (required)
         :param str project_id:
+        :param str cloudspace_id:
         :return: V1ListConversationMessageActionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -2031,12 +2032,13 @@ class AssistantsServiceApi(object):
         :param str conversation_id: (required)
         :param str message_id: (required)
         :param str project_id:
+        :param str cloudspace_id:
         :return: V1ListConversationMessageActionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['assistant_id', 'conversation_id', 'message_id', 'project_id']  # noqa: E501
+        all_params = ['assistant_id', 'conversation_id', 'message_id', 'project_id', 'cloudspace_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -2077,6 +2079,8 @@ class AssistantsServiceApi(object):
         query_params = []
         if 'project_id' in params:
             query_params.append(('projectId', params['project_id']))  # noqa: E501
+        if 'cloudspace_id' in params:
+            query_params.append(('cloudspaceId', params['cloudspace_id']))  # noqa: E501
 
         header_params = {}
 
@@ -2983,6 +2987,135 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def assistants_service_update_conversation_message_content(self, body: 'IdContentBody', project_id: 'str', assistant_id: 'str', conversation_id: 'str', id: 'str', **kwargs) -> 'V1UpdateConversationMessageContentResponse':  # noqa: E501
+        """assistants_service_update_conversation_message_content  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.assistants_service_update_conversation_message_content(body, project_id, assistant_id, conversation_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdContentBody body: (required)
+        :param str project_id: (required)
+        :param str assistant_id: (required)
+        :param str conversation_id: (required)
+        :param str id: (required)
+        :return: V1UpdateConversationMessageContentResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.assistants_service_update_conversation_message_content_with_http_info(body, project_id, assistant_id, conversation_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.assistants_service_update_conversation_message_content_with_http_info(body, project_id, assistant_id, conversation_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def assistants_service_update_conversation_message_content_with_http_info(self, body: 'IdContentBody', project_id: 'str', assistant_id: 'str', conversation_id: 'str', id: 'str', **kwargs) -> 'V1UpdateConversationMessageContentResponse':  # noqa: E501
+        """assistants_service_update_conversation_message_content  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.assistants_service_update_conversation_message_content_with_http_info(body, project_id, assistant_id, conversation_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdContentBody body: (required)
+        :param str project_id: (required)
+        :param str assistant_id: (required)
+        :param str conversation_id: (required)
+        :param str id: (required)
+        :return: V1UpdateConversationMessageContentResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'assistant_id', 'conversation_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 assistants_service_update_conversation_message_content" % 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 `assistants_service_update_conversation_message_content`")  # 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 `assistants_service_update_conversation_message_content`")  # noqa: E501
+        # verify the required parameter 'assistant_id' is set
+        if ('assistant_id' not in params or
+                params['assistant_id'] is None):
+            raise ValueError("Missing the required parameter `assistant_id` when calling `assistants_service_update_conversation_message_content`")  # noqa: E501
+        # verify the required parameter 'conversation_id' is set
+        if ('conversation_id' not in params or
+                params['conversation_id'] is None):
+            raise ValueError("Missing the required parameter `conversation_id` when calling `assistants_service_update_conversation_message_content`")  # 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 `assistants_service_update_conversation_message_content`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'assistant_id' in params:
+            path_params['assistantId'] = params['assistant_id']  # noqa: E501
+        if 'conversation_id' in params:
+            path_params['conversationId'] = params['conversation_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}/agents/{assistantId}/conversations/{conversationId}/messages/{id}/content', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateConversationMessageContentResponse',  # 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 assistants_service_update_conversation_message_like(self, body: 'MessagesIdBody', project_id: 'str', assistant_id: 'str', conversation_id: 'str', id: 'str', **kwargs) -> 'V1UpdateConversationMessageLikeResponse':  # noqa: E501
         """assistants_service_update_conversation_message_like  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -3152,6 +3152,7 @@ class ClusterServiceApi(object):
         :param async_req bool
         :param str org_id:
         :param str project_id:
+        :param bool include_configuring_state:
         :return: V1ListClustersResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -3174,12 +3175,13 @@ class ClusterServiceApi(object):
         :param async_req bool
         :param str org_id:
         :param str project_id:
+        :param bool include_configuring_state:
         :return: V1ListClustersResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['org_id', 'project_id']  # noqa: E501
+        all_params = ['org_id', 'project_id', 'include_configuring_state']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -3204,6 +3206,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 'include_configuring_state' in params:
+            query_params.append(('includeConfiguringState', params['include_configuring_state']))  # noqa: E501
 
         header_params = {}