lightning-sdk 0.1.31__py3-none-any.whl → 0.1.33__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.31"
+__version__ = "0.1.33"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/ai_hub.py

@@ -0,0 +1,110 @@
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
+
+from lightning_sdk.api import AIHubApi, UserApi
+from lightning_sdk.lightning_cloud import login
+from lightning_sdk.user import User
+from lightning_sdk.utils.resolve import _resolve_org, _resolve_teamspace
+
+if TYPE_CHECKING:
+    from lightning_sdk import Organization, Teamspace
+
+
+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()
+        self._auth = None
+
+    def list_apis(self, search: Optional[str] = None) -> List[Dict[str, str]]:
+        """Get a list of AI Hub API templates.
+
+        Example:
+            api_hub = AIHub()
+            api_list = api_hub.list_apis(search="Llama")
+        """
+        search_query = search or ""
+        api_templates = self._api.list_apis(search_query=search_query)
+        results = []
+        for template in api_templates:
+            result = {
+                "id": template.id,
+                "name": template.name,
+                "description": template.description,
+                "creator_username": template.creator_username,
+                "created_on": template.creation_timestamp.strftime("%Y-%m-%d %H:%M:%S")
+                if template.creation_timestamp
+                else None,
+            }
+            results.append(result)
+        return results
+
+    def _authenticate(
+        self,
+        teamspace: Optional[Union[str, "Teamspace"]] = None,
+        org: Optional[Union[str, "Organization"]] = None,
+        user: Optional[Union[str, "User"]] = None,
+    ) -> "Teamspace":
+        if self._auth is None:
+            self._auth = login.Auth()
+        try:
+            self._auth.authenticate()
+            user = User(name=UserApi()._get_user_by_id(self._auth.user_id).username)
+        except ConnectionError as e:
+            raise e
+
+        org = _resolve_org(org)
+        teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user if org is None else None)
+        if teamspace is None:
+            raise ValueError("You need to pass a teamspace or an org for your deployment.")
+        return teamspace
+
+    def deploy(
+        self,
+        api_id: str,
+        cluster: Optional[str] = None,
+        name: Optional[str] = None,
+        teamspace: Optional[Union[str, "Teamspace"]] = None,
+        org: Optional[Union[str, "Organization"]] = None,
+        **kwargs: Dict[str, Any],
+    ) -> Dict[str, Union[str, bool]]:
+        """Deploy an API from the AI Hub.
+
+        Example:
+            from lightning_sdk import AIHub
+            ai_hub = AIHub()
+            deployment = ai_hub.deploy("temp_01jc37n6qpqkdptjpyep0z06hy", batch_size=10)
+
+        Args:
+            api_id: The ID of the API you want to deploy.
+            cluster: The cluster where you want to deploy the API, such as "lightning-public-prod".
+            name: Name for the deployed API. Defaults to None.
+            teamspace: The team or group for deployment. Defaults to None.
+            org: The organization for deployment. Defaults to None.
+            **kwargs: Additional keyword arguments for deployment.
+
+        Returns:
+            A dictionary containing the name of the deployed API,
+            the URL to access it, and whether it is interruptible.
+
+        Raises:
+            ValueError: If a teamspace or organization is not provided.
+            ConnectionError: If there is an issue with logging in.
+        """
+        teamspace = self._authenticate(teamspace, org)
+        teamspace_id = teamspace.id
+
+        deployment = self._api.deploy_api(
+            template_id=api_id, cluster_id=cluster, project_id=teamspace_id, name=name, **kwargs
+        )
+        return {
+            "id": deployment.id,
+            "name": deployment.name,
+            "base_url": deployment.status.urls[0],
+            "interruptible": deployment.spec.spot,
+        }

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,61 @@
+import re
+from typing import List, Optional
+
+from lightning_sdk.lightning_cloud.openapi.models import (
+    CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
+    V1Deployment,
+    V1ParameterizationSpec,
+)
+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=3)
+
+    def list_apis(self, search_query: str) -> List[V1DeploymentTemplateGalleryResponse]:
+        kwargs = {"show_globally_visible": True}
+        return self._client.deployment_templates_service_list_published_deployment_templates(
+            search_query=search_query, **kwargs
+        ).templates
+
+    @staticmethod
+    def _parse_and_update_args(cmd: str, **kwargs: dict) -> list:
+        """Parse the command and update the arguments with the provided kwargs.
+
+        >>> _parse_and_update_args("--arg1 1 --arg2=2", arg1=3)
+        ['--arg1 3']
+        """
+        keys = [key.lstrip("-") for key in re.findall(r"--\w+", cmd)]
+        arguments = {}
+        for key in keys:
+            if key in kwargs:
+                arguments[key] = kwargs[key]
+        return [f"--{k} {v}" for k, v in arguments.items()]
+
+    @staticmethod
+    def _resolve_api_arguments(parameter_spec: "V1ParameterizationSpec", **kwargs: dict) -> str:
+        return " ".join(AIHubApi._parse_and_update_args(parameter_spec.command, **kwargs))
+
+    def deploy_api(
+        self, template_id: str, project_id: str, cluster_id: str, name: Optional[str], **kwargs: dict
+    ) -> V1Deployment:
+        template = self._client.deployment_templates_service_get_deployment_template(template_id)
+        name = name or template.name
+        template.spec_v2.endpoint.id = None
+        command = self._resolve_api_arguments(template.parameter_spec, **kwargs)
+        template.spec_v2.job.command = command
+        return self._client.jobs_service_create_deployment(
+            project_id=project_id,
+            body=CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(
+                autoscaling=template.spec_v2.autoscaling,
+                cluster_id=cluster_id,
+                endpoint=template.spec_v2.endpoint,
+                name=name,
+                replicas=0,
+                spec=template.spec_v2.job,
+            ),
+        )

lightning_sdk/api/studio_api.py

@@ -5,7 +5,7 @@ import time
 import warnings
 import zipfile
 from threading import Event, Thread
-from typing import Any, Dict, Mapping, Optional, Tuple
+from typing import Any, Dict, Mapping, Optional, Tuple, Union
 
 import backoff
 import requests
@@ -18,6 +18,7 @@ from lightning_sdk.api.utils import (
     _DummyBody,
     _DummyResponse,
     _FileUploader,
+    _machine_to_compute_name,
     _sanitize_studio_remote_path,
 )
 from lightning_sdk.api.utils import (
@@ -146,9 +147,11 @@ class StudioApi:
         """Retries checking the sync_in_progress value of the code status when there's an AttributeError."""
         return self.get_studio_status(studio_id, teamspace_id).in_use.sync_in_progress
 
-    def start_studio(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: False) -> None:
+    def start_studio(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: False
+    ) -> None:
         """Start an existing Studio."""
-        if machine == Machine.CPU_SMALL:
+        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
             warnings.warn(
                 f"{Machine.CPU_SMALL} is not a valid machine for starting a Studio. "
                 "It is reserved for running jobs only. "
@@ -158,7 +161,7 @@ class StudioApi:
 
         self._client.cloud_space_service_start_cloud_space_instance(
             IdStartBody(
-                compute_config=V1UserRequestedComputeConfig(name=_MACHINE_TO_COMPUTE_NAME[machine], spot=interruptible)
+                compute_config=V1UserRequestedComputeConfig(name=_machine_to_compute_name(machine), spot=interruptible)
             ),
             teamspace_id,
             studio_id,
@@ -204,9 +207,11 @@ class StudioApi:
     def _get_studio_instance_status_from_object(self, studio: V1CloudSpace) -> Optional[str]:
         return getattr(getattr(studio.code_status, "in_use", None), "phase", None)
 
-    def _request_switch(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: bool) -> None:
+    def _request_switch(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+    ) -> None:
         """Switches given Studio to a new machine type."""
-        if machine == Machine.CPU_SMALL:
+        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
             warnings.warn(
                 f"{Machine.CPU_SMALL} is not a valid machine for switching a Studio. "
                 "It is reserved for running jobs only. "
@@ -214,7 +219,7 @@ class StudioApi:
             )
             machine = Machine.CPU
 
-        compute_name = _MACHINE_TO_COMPUTE_NAME[machine]
+        compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))
         self._client.cloud_space_service_update_cloud_space_instance_config(
@@ -223,7 +228,9 @@ class StudioApi:
             body=body,
         )
 
-    def switch_studio_machine(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: bool) -> None:
+    def switch_studio_machine(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+    ) -> None:
         """Switches given Studio to a new machine type."""
         self._request_switch(
             studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible

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/api/utils.py

@@ -4,7 +4,7 @@ import os
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import backoff
 import requests
@@ -340,6 +340,13 @@ _MACHINE_TO_COMPUTE_NAME: Dict[Machine, str] = {
     Machine.H200_X_8: "p5e.48xlarge",
 }
 
+
+def _machine_to_compute_name(machine: Union[Machine, str]) -> str:
+    if isinstance(machine, Machine):
+        return _MACHINE_TO_COMPUTE_NAME[machine]
+    return machine
+
+
 _COMPUTE_NAME_TO_MACHINE: Dict[str, Machine] = {v: k for k, v in _MACHINE_TO_COMPUTE_NAME.items()}
 
 _DEFAULT_CLOUD_URL = "https://lightning.ai:443"

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
@@ -128,6 +129,7 @@ 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
@@ -198,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
@@ -291,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
@@ -300,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
@@ -348,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
@@ -355,6 +362,7 @@ 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
@@ -379,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
@@ -393,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
@@ -485,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
@@ -508,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
@@ -647,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
@@ -700,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

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