lightning-sdk 0.2.14__py3-none-any.whl → 0.2.16__py3-none-any.whl

lightning_sdk/cli/teamspace_menu.py

@@ -22,7 +22,15 @@ class _TeamspacesMenu:
     def _get_teamspace_from_name(
         self, teamspace: str, possible_teamspaces: Dict[str, Dict[str, str]]
     ) -> Dict[str, str]:
-        owner, name = teamspace.split("/", maxsplit=1)
+        try:
+            owner, name = teamspace.split("/", maxsplit=1)
+        except ValueError as e:
+            raise ValueError(
+                f"Invalid teamspace format: '{teamspace}'. "
+                "Teamspace should be specified as '{teamspace_owner}/{teamspace_name}' "
+                "(e.g., 'my-org/my-teamspace')."
+            ) from e
+
         for _, ts in possible_teamspaces.items():
             if ts["name"] == name and (ts["user"] == owner or ts["org"] == owner):
                 return ts

lightning_sdk/cli/upload.py

@@ -125,7 +125,6 @@ def file(path: str, studio: Optional[str] = None, remote_path: Optional[str] = N
     help="The name of the cloud account to store the Container in.",
 )
 @click.option(
-    "--platform",
     "--platform",
     default="linux/amd64",
     help="This is the platform the container pulled and push to Lightning AI will run on.",
@@ -198,8 +197,7 @@ def _folder(path: str, studio: Optional[str] = None, remote_path: Optional[str]
     if not Path(path).is_dir():
         raise StudioCliError(f"The provided path is not a folder: {path}. Use `lightning upload file` instead.")
 
-    menu = _StudiosMenu()
-    selected_studio = menu._resolve_studio(studio)
+    selected_studio = _resolve_studio(studio)
 
     console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
@@ -248,8 +246,7 @@ def _file(path: str, studio: Optional[str] = None, remote_path: Optional[str] =
     if not Path(path).exists():
         raise FileNotFoundError(f"The provided path does not exist: {path}.")
 
-    menu = _StudiosMenu()
-    selected_studio = menu._resolve_studio(studio)
+    selected_studio = _resolve_studio(studio)
 
     console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -49,6 +49,7 @@ from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsSto
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipeline_templates_service_api import PipelineTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.product_license_service_api import ProductLicenseServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
@@ -102,6 +103,8 @@ from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_con
 from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
 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_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
@@ -159,6 +162,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body impo
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
 from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.model_id_visibility_body import ModelIdVisibilityBody
+from lightning_sdk.lightning_cloud.openapi.models.models_id_body import ModelsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
@@ -281,8 +285,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_st
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics import V1CloudSpaceColdStartMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_config import V1CloudSpaceEnvironmentConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_type import V1CloudSpaceEnvironmentType
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_collab_status import V1CloudSpaceInstanceCollabStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_config import V1CloudSpaceInstanceConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_startup_status import V1CloudSpaceInstanceStartupStatus
@@ -292,6 +298,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_type import V1CloudSpacePublicationType
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
@@ -310,6 +317,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_state import V1Clus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_status import V1ClusterStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_tagging_options import V1ClusterTaggingOptions
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_type import V1ClusterType
+from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_upload import V1ClusterUpload
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_usage_restriction import V1ClusterUsageRestriction
 from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1CollabAction
 from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
@@ -389,6 +397,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restri
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_alerting_policy_response import V1DeleteDeploymentAlertingPolicyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_release_response import V1DeleteDeploymentReleaseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_response import V1DeleteDeploymentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response import V1DeleteEndpointResponse
@@ -434,6 +443,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state impo
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_api import V1DeploymentAPI
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_event import V1DeploymentAlertingEvent
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy import V1DeploymentAlertingPolicy
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_frequency import V1DeploymentAlertingPolicyFrequency
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_operation import V1DeploymentAlertingPolicyOperation
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_severity import V1DeploymentAlertingPolicySeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_type import V1DeploymentAlertingPolicyType
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_details import V1DeploymentDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
@@ -496,18 +511,21 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1Galler
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_component import V1GalleryComponent
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection import V1GcpDataConnection
 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_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
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_content_response import V1GetDeploymentRoutingTelemetryContentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
@@ -524,6 +542,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
@@ -612,6 +631,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifa
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connections_response import V1ListDataConnectionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_datasets_response import V1ListDatasetsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_default_cluster_accelerators_response import V1ListDefaultClusterAcceleratorsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_alerting_events_response import V1ListDeploymentAlertingEventsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_alerting_policies_response import V1ListDeploymentAlertingPoliciesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_events_response import V1ListDeploymentEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_releases_response import V1ListDeploymentReleasesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_tags_response import V1ListDeploymentTagsResponse
@@ -660,6 +681,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_roles_response imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_encryption_keys_response import V1ListOrganizationClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organizations_response import V1ListOrganizationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_pipelines_response import V1ListPipelinesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_product_licenses_response import V1ListProductLicensesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_profiler_captures_response import V1ListProfilerCapturesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_artifacts_response import V1ListProjectArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_cluster_accelerators_response import V1ListProjectClusterAcceleratorsResponse
@@ -716,6 +738,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1Me
 from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
+from lightning_sdk.lightning_cloud.openapi.models.v1_modify_filesystem_volume_response import V1ModifyFilesystemVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job import V1MultiMachineJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_event import V1MultiMachineJobEvent
@@ -749,6 +772,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement import V1PipelineParameterPlacement
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement_type import V1PipelineParameterPlacementType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_type import V1PipelineParameterType
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_state import V1PipelineState
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step import V1PipelineStep
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import V1PipelineStepStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
@@ -758,6 +782,8 @@ 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_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_product_license import V1ProductLicense
+from lightning_sdk.lightning_cloud.openapi.models.v1_product_license_check_response import V1ProductLicenseCheckResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_capture import V1ProfilerCapture
 from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_enabled_response import V1ProfilerEnabledResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_project import V1Project
@@ -809,6 +835,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1Res
 from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
 from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
+from lightning_sdk.lightning_cloud.openapi.models.v1_routing_telemetry import V1RoutingTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule import V1Rule
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_action import V1RuleAction
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_condition import V1RuleCondition
@@ -839,6 +866,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
 from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
+from lightning_sdk.lightning_cloud.openapi.models.v1_sleep_server_response import V1SleepServerResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1 import V1SlurmV1
@@ -867,6 +895,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_project_balance_response import V1TransferProjectBalanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_user_balance_request import V1TransferUserBalanceRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_user_balance_response import V1TransferUserBalanceResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_trigger_filesystem_upgrade_response import V1TriggerFilesystemUpgradeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_unpublish_cloud_space_response import V1UnpublishCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_request import V1UpdateAgentStatusRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_response import V1UpdateAgentStatusResponse

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -30,6 +30,7 @@ from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsSto
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipeline_templates_service_api import PipelineTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.product_license_service_api import ProductLicenseServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -1691,6 +1691,127 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def assistants_service_update_assistant_managed_endpoint_model(self, body: 'ModelsIdBody', project_id: 'str', managed_endpoint_id: 'str', id: 'str', **kwargs) -> 'V1ManagedModel':  # noqa: E501
+        """assistants_service_update_assistant_managed_endpoint_model  # 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_assistant_managed_endpoint_model(body, project_id, managed_endpoint_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelsIdBody body: (required)
+        :param str project_id: (required)
+        :param str managed_endpoint_id: (required)
+        :param str id: (required)
+        :return: V1ManagedModel
+                 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_assistant_managed_endpoint_model_with_http_info(body, project_id, managed_endpoint_id, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.assistants_service_update_assistant_managed_endpoint_model_with_http_info(body, project_id, managed_endpoint_id, id, **kwargs)  # noqa: E501
+            return data
+
+    def assistants_service_update_assistant_managed_endpoint_model_with_http_info(self, body: 'ModelsIdBody', project_id: 'str', managed_endpoint_id: 'str', id: 'str', **kwargs) -> 'V1ManagedModel':  # noqa: E501
+        """assistants_service_update_assistant_managed_endpoint_model  # 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_assistant_managed_endpoint_model_with_http_info(body, project_id, managed_endpoint_id, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelsIdBody body: (required)
+        :param str project_id: (required)
+        :param str managed_endpoint_id: (required)
+        :param str id: (required)
+        :return: V1ManagedModel
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'managed_endpoint_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_assistant_managed_endpoint_model" % 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_assistant_managed_endpoint_model`")  # 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_assistant_managed_endpoint_model`")  # noqa: E501
+        # verify the required parameter 'managed_endpoint_id' is set
+        if ('managed_endpoint_id' not in params or
+                params['managed_endpoint_id'] is None):
+            raise ValueError("Missing the required parameter `managed_endpoint_id` when calling `assistants_service_update_assistant_managed_endpoint_model`")  # 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_assistant_managed_endpoint_model`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'managed_endpoint_id' in params:
+            path_params['managedEndpointId'] = params['managed_endpoint_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}/agent-managed-endpoints/{managedEndpointId}/models/{id}', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ManagedModel',  # 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(self, body: 'ConversationsIdBody', assistant_id: 'str', id: 'str', **kwargs) -> 'V1Conversation':  # noqa: E501
         """assistants_service_update_conversation  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py

@@ -942,6 +942,8 @@ class BillingServiceApi(object):
         :param datetime to:
         :param str user_id:
         :param str time_zone:
+        :param int limit:
+        :param datetime search_after:
         :return: V1UsageReport
                  If the method is called asynchronously,
                  returns the request thread.
@@ -971,12 +973,14 @@ class BillingServiceApi(object):
         :param datetime to:
         :param str user_id:
         :param str time_zone:
+        :param int limit:
+        :param datetime search_after:
         :return: V1UsageReport
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['org_id', 'project_id', 'cluster_id', 'resource_type', 'resource_id', '_from', 'to', 'user_id', 'time_zone']  # noqa: E501
+        all_params = ['org_id', 'project_id', 'cluster_id', 'resource_type', 'resource_id', '_from', 'to', 'user_id', 'time_zone', 'limit', 'search_after']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1015,6 +1019,10 @@ class BillingServiceApi(object):
             query_params.append(('userId', params['user_id']))  # noqa: E501
         if 'time_zone' in params:
             query_params.append(('timeZone', params['time_zone']))  # noqa: E501
+        if 'limit' in params:
+            query_params.append(('limit', params['limit']))  # noqa: E501
+        if 'search_after' in params:
+            query_params.append(('searchAfter', params['search_after']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3481,6 +3481,111 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_get_cloud_space_instance_open_ports(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceOpenPortsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_instance_open_ports  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_get_cloud_space_instance_open_ports(project_id, cloudspace_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str cloudspace_instance_id:
+        :return: V1GetCloudSpaceInstanceOpenPortsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_get_cloud_space_instance_open_ports_with_http_info(project_id, cloudspace_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_get_cloud_space_instance_open_ports_with_http_info(project_id, cloudspace_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_get_cloud_space_instance_open_ports_with_http_info(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceOpenPortsResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_instance_open_ports  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_get_cloud_space_instance_open_ports_with_http_info(project_id, cloudspace_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str cloudspace_instance_id:
+        :return: V1GetCloudSpaceInstanceOpenPortsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cloudspace_id', 'cloudspace_instance_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 cloud_space_service_get_cloud_space_instance_open_ports" % 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 `cloud_space_service_get_cloud_space_instance_open_ports`")  # noqa: E501
+        # verify the required parameter 'cloudspace_id' is set
+        if ('cloudspace_id' not in params or
+                params['cloudspace_id'] is None):
+            raise ValueError("Missing the required parameter `cloudspace_id` when calling `cloud_space_service_get_cloud_space_instance_open_ports`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cloudspace_id' in params:
+            path_params['cloudspaceId'] = params['cloudspace_id']  # noqa: E501
+
+        query_params = []
+        if 'cloudspace_instance_id' in params:
+            query_params.append(('cloudspaceInstanceId', params['cloudspace_instance_id']))  # 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/projects/{projectId}/cloudspaces/{cloudspaceId}/open-ports', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetCloudSpaceInstanceOpenPortsResponse',  # 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 cloud_space_service_get_cloud_space_instance_status(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceStatusResponse':  # noqa: E501
         """cloud_space_service_get_cloud_space_instance_status  # noqa: E501
 
@@ -6122,6 +6227,127 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_list_cloud_spaces_bulk(self, **kwargs) -> 'V1ListCloudSpacesResponse':  # noqa: E501
+        """cloud_space_service_list_cloud_spaces_bulk  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_list_cloud_spaces_bulk(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id:
+        :param str cluster_id:
+        :param str page_token:
+        :param str limit:
+        :param str name:
+        :param str user_id:
+        :param bool active_only:
+        :param bool is_favorite:
+        :param bool is_locked:
+        :return: V1ListCloudSpacesResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.cloud_space_service_list_cloud_spaces_bulk_with_http_info(**kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_list_cloud_spaces_bulk_with_http_info(**kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_list_cloud_spaces_bulk_with_http_info(self, **kwargs) -> 'V1ListCloudSpacesResponse':  # noqa: E501
+        """cloud_space_service_list_cloud_spaces_bulk  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.cloud_space_service_list_cloud_spaces_bulk_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id:
+        :param str cluster_id:
+        :param str page_token:
+        :param str limit:
+        :param str name:
+        :param str user_id:
+        :param bool active_only:
+        :param bool is_favorite:
+        :param bool is_locked:
+        :return: V1ListCloudSpacesResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cluster_id', 'page_token', 'limit', 'name', 'user_id', 'active_only', 'is_favorite', 'is_locked']  # 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 cloud_space_service_list_cloud_spaces_bulk" % 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 'cluster_id' in params:
+            query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
+        if 'page_token' in params:
+            query_params.append(('pageToken', params['page_token']))  # noqa: E501
+        if 'limit' in params:
+            query_params.append(('limit', params['limit']))  # noqa: E501
+        if 'name' in params:
+            query_params.append(('name', params['name']))  # noqa: E501
+        if 'user_id' in params:
+            query_params.append(('userId', params['user_id']))  # noqa: E501
+        if 'active_only' in params:
+            query_params.append(('activeOnly', params['active_only']))  # noqa: E501
+        if 'is_favorite' in params:
+            query_params.append(('isFavorite', params['is_favorite']))  # noqa: E501
+        if 'is_locked' in params:
+            query_params.append(('isLocked', params['is_locked']))  # 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/cloudspaces', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListCloudSpacesResponse',  # 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 cloud_space_service_list_installed_plugins(self, project_id: 'str', id: 'str', **kwargs) -> 'V1PluginsListResponse':  # noqa: E501
         """cloud_space_service_list_installed_plugins  # noqa: E501