lightning-sdk 0.1.30__py3-none-any.whl → 0.1.32__py3-none-any.whl

lightning_sdk/__init__.py

@@ -1,4 +1,5 @@
 from lightning_sdk.agents import Agent
+from lightning_sdk.ai_hub import AIHub
 from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__  # noqa: F401
 from lightning_sdk.helpers import _check_version_and_prompt_upgrade
 from lightning_sdk.job import Job
@@ -23,7 +24,8 @@ __all__ = [
     "Teamspace",
     "User",
     "Agent",
+    "AIHub",
 ]
 
-__version__ = "0.1.30"
+__version__ = "0.1.32"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/ai_hub.py

@@ -0,0 +1,29 @@
+from typing import Dict, List
+
+from lightning_sdk.api.ai_hub_api import AIHubApi
+
+
+class AIHub:
+    """An interface to interact with the AI Hub.
+
+    Example:
+        ai_hub = AIHub()
+        api_list = ai_hub.list_apis()
+    """
+
+    def __init__(self) -> None:
+        self._api = AIHubApi()
+
+    def list_apis(self) -> List[Dict[str, str]]:
+        """Get a list of AI Hub API templates."""
+        api_templates = self._api.list_apis()
+        results = []
+        for template in api_templates:
+            result = {
+                "id": template.id,
+                "name": template.name,
+                "description": template.description,
+                "creator_username": template.creator_username,
+            }
+            results.append(result)
+        return results

lightning_sdk/api/__init__.py

@@ -1,4 +1,5 @@
 from lightning_sdk.api.agents_api import AgentApi
+from lightning_sdk.api.ai_hub_api import AIHubApi
 from lightning_sdk.api.org_api import OrgApi
 from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.api.teamspace_api import TeamspaceApi
@@ -10,4 +11,5 @@ __all__ = [
     "TeamspaceApi",
     "UserApi",
     "AgentApi",
+    "AIHubApi",
 ]

lightning_sdk/api/ai_hub_api.py

@@ -0,0 +1,15 @@
+from typing import List
+
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import (
+    V1DeploymentTemplateGalleryResponse,
+)
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
+
+
+class AIHubApi:
+    def __init__(self) -> None:
+        self._client = LightningClient(max_tries=7)
+
+    def list_apis(self) -> List[V1DeploymentTemplateGalleryResponse]:
+        kwargs = {"show_globally_visible": True}
+        return self._client.deployment_templates_service_list_published_deployment_templates(**kwargs).templates

lightning_sdk/api/teamspace_api.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Optional
 from lightning_sdk.api.utils import _download_model_files, _DummyBody, _ModelFileUploader
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
+    ModelIdVersionsBody,
     ModelsStoreApi,
     ProjectIdAgentsBody,
     ProjectIdModelsBody,
@@ -28,6 +29,7 @@ class TeamspaceApi:
 
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=7)
+        self._models: Optional[ModelsStoreApi] = None
 
     def get_teamspace(self, name: str, owner_id: str) -> V1Project:
         """Get the current teamspace from the owner."""
@@ -165,9 +167,36 @@ class TeamspaceApi:
         teamspace_id: str,
         cluster_id: str,
     ) -> V1ModelVersionArchive:
-        api = ModelsStoreApi(self._client.api_client)
-        body = ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private)
-        return api.models_store_create_model(body, project_id=teamspace_id)
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        # ask if such model already exists by listing models with specific name
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        if len(models) == 0:
+            return self._models.models_store_create_model(
+                body=ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private),
+                project_id=teamspace_id,
+            )
+        assert len(models) == 1, "Multiple models with the same name found"
+        return self._models.models_store_create_model_version(
+            body=ModelIdVersionsBody(cluster_id=cluster_id),
+            project_id=teamspace_id,
+            model_id=models[0].id,
+        )
+
+    def delete_model(self, name: str, version: Optional[str], teamspace_id: str) -> None:
+        """Delete a model or a version from the model store."""
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        assert len(models) == 1, "Multiple models with the same name found"
+        model_id = models[0].id
+        # decide if delete only version of whole model
+        if version:
+            if version == "latest":
+                version = models[0].latest_version
+            self._models.models_store_delete_model_version(project_id=teamspace_id, model_id=model_id, version=version)
+        else:
+            self._models.models_store_delete_model(project_id=teamspace_id, model_id=model_id)
 
     def upload_model_file(
         self,
@@ -213,8 +242,9 @@ class TeamspaceApi:
             )
 
     def complete_model_upload(self, model_id: str, version: str, teamspace_id: str) -> None:
-        api = ModelsStoreApi(self._client.api_client)
-        api.models_store_complete_model_upload(
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        self._models.models_store_complete_model_upload(
             body=_DummyBody(),
             project_id=teamspace_id,
             model_id=model_id,

lightning_sdk/api/user_api.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Union
 
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
@@ -19,8 +19,14 @@ class UserApi:
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=7)
 
-    def get_user(self, name: str) -> V1SearchUser:
+    def get_user(self, name: str) -> Union[V1SearchUser, V1GetUserResponse]:
         """Gets user by name."""
+        authed_user = self._client.auth_service_get_user()
+        if authed_user.username == name:
+            return authed_user
+
+        # if it's not the authed user, lookup by name
+        # TODO: This API won't necesarily return the correct thing
         response = self._client.user_service_search_users(query=name)
 
         users = [u for u in response.users if u.username == name]

lightning_sdk/cli/upload.py

@@ -76,13 +76,14 @@ class _Uploads(_StudiosMenu):
         if remote_path is None:
             remote_path = os.path.basename(path)
 
+        if not Path(path).exists():
+            raise FileNotFoundError(f"The provided path does not exist: {path}.")
+        if not Path(path).is_dir():
+            raise StudioCliError(f"The provided path is not a folder: {path}. Use `lightning upload file` instead.")
+
         selected_studio = self._resolve_studio(studio)
 
         print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
-        if not Path(path).exists:
-            raise FileNotFoundError(f"The provided path does not exist: {path}.")
-        if not Path(path).is_dir:
-            raise StudioCliError(f"The provided path is not a folder: {path}. Use `lightning upload file` instead.")
 
         pairs = {}
         for root, _, files in os.walk(path):
@@ -130,11 +131,14 @@ class _Uploads(_StudiosMenu):
         if remote_path is None:
             remote_path = os.path.basename(path)
 
+        if Path(path).is_dir():
+            raise StudioCliError(f"The provided path is a folder: {path}. Use `lightning upload folder` instead.")
+        if not Path(path).exists():
+            raise FileNotFoundError(f"The provided path does not exist: {path}.")
+
         selected_studio = self._resolve_studio(studio)
 
         print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
-        if Path(path).is_dir:
-            raise StudioCliError(f"The provided path is a folder: {path}. Use `lightning upload folder` instead.")
 
         self._single_file_upload(selected_studio, path, remote_path, True)
 

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -34,6 +34,7 @@ from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import Datase
 from lightning_sdk.lightning_cloud.openapi.api.deployment_templates_service_api import DeploymentTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.endpoint_service_api import EndpointServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import ExperimentsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
@@ -104,6 +105,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCo
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
 from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
 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
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body1 import IdExecuteBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_fork_body import IdForkBody
@@ -127,12 +129,14 @@ 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
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
+from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.loggermetrics_id_body import LoggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifacts_body import MetricsStreamIdLoggerartifactsBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
 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.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -196,6 +200,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_app_type import V1AppType
 from lightning_sdk.lightning_cloud.openapi.models.v1_append_logger_metrics_response import V1AppendLoggerMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_approve_auto_join_domain_response import V1ApproveAutoJoinDomainResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_artifact import V1Artifact
@@ -289,6 +294,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_ssh_public_key_request import V1CreateSSHPublicKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_request import V1CreateSharedMetricsStreamRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_response import V1CreateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection_response import V1CreateSnowflakeConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_request import V1CreateSubscriptionCheckoutSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_response import V1CreateSubscriptionCheckoutSessionResponse
@@ -298,6 +305,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1Da
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_mount import V1DataConnectionMount
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_state import V1DataConnectionState
+from lightning_sdk.lightning_cloud.openapi.models.v1_data_path import V1DataPath
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_affiliate_link_response import V1DeleteAffiliateLinkResponse
@@ -346,6 +354,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_role_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_ssh_public_key_response import V1DeleteSSHPublicKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response import V1DeleteSecretResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_service_execution_response import V1DeleteServiceExecutionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_studio_job_response import V1DeleteStudioJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_response import V1DeleteUserSLURMJobResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state import V1DependencyCacheState
@@ -353,12 +362,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_release import V1DeploymentRelease
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_spec import V1DeploymentSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1DeploymentState
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_status import V1DeploymentStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_strategy import V1DeploymentStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template import V1DeploymentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_engagement_response import V1DeploymentTemplateEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import V1DeploymentTemplateGalleryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter import V1DeploymentTemplateParameter
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_placement import V1DeploymentTemplateParameterPlacement
@@ -376,6 +387,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_efs_folder_data_connection import V1EFSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_ebs import V1Ebs
 from lightning_sdk.lightning_cloud.openapi.models.v1_efs_data_connection import V1EfsDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
@@ -390,9 +402,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_experiment import V1Experim
 from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_response import V1ExportSnowflakeQueryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_app import V1FilesystemApp
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space import V1FilesystemCloudSpace
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_job import V1FilesystemJob
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import V1FilesystemSlurmJob
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
+from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
 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_gcs_folder_data_connection import V1GCSFolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
 from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_component import V1GalleryComponent
@@ -482,6 +502,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response im
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_instances_response import V1ListCloudSpaceInstancesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_publications_response import V1ListCloudSpacePublicationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_python_versions_response import V1ListCloudSpacePythonVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_sessions_response import V1ListCloudSpaceSessionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_response import V1ListCloudSpaceTagsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
@@ -505,6 +526,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployments_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_endpoints_response import V1ListEndpointsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_experiments_response import V1ListExperimentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_file_endpoints_response import V1ListFileEndpointsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_response import V1ListFilesystemAppsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_lightningapps_response import V1ListGalleryLightningappsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_files_response import V1ListJobFilesResponse
@@ -591,6 +618,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_org_role import V1OrgRole
 from lightning_sdk.lightning_cloud.openapi.models.v1_organization import V1Organization
 from lightning_sdk.lightning_cloud.openapi.models.v1_owner_type import V1OwnerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1PackageManager
+from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
@@ -643,6 +671,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_rule_action import V1RuleAc
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_condition import V1RuleCondition
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_effect import V1RuleEffect
 from lightning_sdk.lightning_cloud.openapi.models.v1_rule_resource import V1RuleResource
+from lightning_sdk.lightning_cloud.openapi.models.v1_s3_folder_data_connection import V1S3FolderDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
@@ -696,6 +725,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -15,6 +15,7 @@ from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import Datase
 from lightning_sdk.lightning_cloud.openapi.api.deployment_templates_service_api import DeploymentTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.endpoint_service_api import EndpointServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import ExperimentsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -1955,7 +1955,7 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1964,7 +1964,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse
@@ -1978,7 +1978,7 @@ class CloudSpaceServiceApi(object):
             (data) = self.cloud_space_service_engage_cloud_space_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
             return data
 
-    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1987,7 +1987,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse
@@ -5209,6 +5209,107 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_list_cloud_space_python_versions(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ListCloudSpacePythonVersionsResponse':  # noqa: E501
+        """cloud_space_service_list_cloud_space_python_versions  # 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_space_python_versions(project_id, cloudspace_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :return: V1ListCloudSpacePythonVersionsResponse
+                 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_space_python_versions_with_http_info(project_id, cloudspace_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_list_cloud_space_python_versions_with_http_info(project_id, cloudspace_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_list_cloud_space_python_versions_with_http_info(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ListCloudSpacePythonVersionsResponse':  # noqa: E501
+        """cloud_space_service_list_cloud_space_python_versions  # 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_space_python_versions_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)
+        :return: V1ListCloudSpacePythonVersionsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cloudspace_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_list_cloud_space_python_versions" % 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_list_cloud_space_python_versions`")  # 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_list_cloud_space_python_versions`")  # 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 = []
+
+        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}/python-versions', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListCloudSpacePythonVersionsResponse',  # 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_cloud_space_sessions(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ListCloudSpaceSessionsResponse':  # noqa: E501
         """Session endpoints  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1106,10 +1106,10 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1134,16 +1134,16 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'capacity_block_duration_hours', 'start_date', 'timezone', 'org_id']  # noqa: E501
+        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_days']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1180,14 +1180,14 @@ class ClusterServiceApi(object):
             query_params.append(('instanceType', params['instance_type']))  # noqa: E501
         if 'instance_count' in params:
             query_params.append(('instanceCount', params['instance_count']))  # noqa: E501
-        if 'capacity_block_duration_hours' in params:
-            query_params.append(('capacityBlockDurationHours', params['capacity_block_duration_hours']))  # noqa: E501
         if 'start_date' in params:
             query_params.append(('startDate', params['start_date']))  # noqa: E501
         if 'timezone' in params:
             query_params.append(('timezone', params['timezone']))  # noqa: E501
         if 'org_id' in params:
             query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'capacity_block_duration_days' in params:
+            query_params.append(('capacityBlockDurationDays', params['capacity_block_duration_days']))  # noqa: E501
 
         header_params = {}