lightning-sdk 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

lightning_sdk/__init__.py

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

lightning_sdk/api/cluster_api.py

@@ -0,0 +1,22 @@
+from lightning_sdk.lightning_cloud.openapi import Externalv1Cluster
+from lightning_sdk.lightning_cloud.rest_client import LightningClient
+
+
+class ClusterApi:
+    """Internal API client for API requests to cluster endpoints."""
+
+    def __init__(self) -> None:
+        self._client = LightningClient(max_tries=7)
+
+    def get_cluster(self, cluster_id: str, project_id: str, org_id: str) -> Externalv1Cluster:
+        """Gets the cluster from given params cluster_id, project_id and owner.
+
+        :param cluster_id: cluster ID test
+        :param project_id: the project the cluster is supposed to be associated with
+        :param org_id: The owning org of this cluster
+        :return:
+        """
+        res = self._client.cluster_service_get_cluster(id=cluster_id, org_id=org_id, project_id=project_id)
+        if not res:
+            raise ValueError(f"Cluster {cluster_id} does not exist")
+        return res

lightning_sdk/api/lit_container_api.py

@@ -105,9 +105,11 @@ class LitContainerApi:
         :param cloud_account: The cluster ID of the cloud account. If None, will use the default cluster.
         :return:
         """
-        project = self._client.lit_registry_service_get_lit_project_registry(
-            project_id, cluster_id="" if cloud_account is None else cloud_account
-        )  # cloud account on the CLI is cluster_id
+        if cloud_account is None:
+            project = self._client.lit_registry_service_get_lit_project_registry(project_id)
+        else:
+            project = self._client.lit_registry_service_get_lit_project_registry(project_id, cluster_id=cloud_account)
+
         return project.repositories
 
     def delete_container(self, project_id: str, container: str) -> V1DeleteLitRepositoryResponse:

lightning_sdk/api/teamspace_api.py

@@ -177,6 +177,7 @@ class TeamspaceApi:
     def create_model(
         self,
         name: str,
+        version: Optional[str],
         metadata: Dict[str, str],
         private: bool,
         teamspace_id: str,
@@ -191,7 +192,7 @@ class TeamspaceApi:
             )
         assert len(models) == 1, "Multiple models with the same name found"
         return self.models.models_store_create_model_version(
-            body=ModelIdVersionsBody(cluster_id=cloud_account),
+            body=ModelIdVersionsBody(cluster_id=cloud_account, version=version),
             project_id=teamspace_id,
             model_id=models[0].id,
         )

lightning_sdk/cli/clusters_menu.py

@@ -0,0 +1,46 @@
+import sys
+from typing import List
+
+from rich.console import Console
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk import Teamspace
+from lightning_sdk.api.cluster_api import ClusterApi
+from lightning_sdk.lightning_cloud.openapi import Externalv1Cluster, V1ProjectClusterBinding
+
+
+class _ClustersMenu:
+    def _get_cluster_from_interactive_menu(self, possible_clusters: List[V1ProjectClusterBinding]) -> str:
+        terminal_menu = self._prepare_terminal_menu_teamspaces([cluster.cluster_id for cluster in possible_clusters])
+        terminal_menu.show()
+
+        return possible_clusters[terminal_menu.chosen_menu_index].cluster_id
+
+    @staticmethod
+    def _prepare_terminal_menu_teamspaces(cluster_ids: List[str]) -> TerminalMenu:
+        title = "Please select a cluster from the following:"
+
+        return TerminalMenu(cluster_ids, title=title, clear_menu_on_exit=True)
+
+    def _resolve_cluster(self, teamspace: Teamspace) -> Externalv1Cluster:
+        selected_cluster_id = None
+        console = Console()
+        try:
+            selected_cluster_id = self._get_cluster_from_interactive_menu(
+                possible_clusters=teamspace.cloud_account_objs
+            )
+            cluster_api = ClusterApi()
+
+            return cluster_api.get_cluster(
+                cluster_id=selected_cluster_id, org_id=teamspace.owner.id, project_id=teamspace.id
+            )
+        except KeyboardInterrupt:
+            console.print("Operation cancelled by user")
+            sys.exit(0)
+
+        except Exception:
+            console.print(
+                f"[red]Could not find the given Cluster:[/red] {selected_cluster_id}. "
+                "Please contact Lightning AI directly to resolve this issue."
+            )
+            sys.exit(1)

lightning_sdk/cli/list.py

@@ -7,8 +7,9 @@ from rich.table import Table
 from typing_extensions import Literal
 
 from lightning_sdk import Job, Machine, Studio, Teamspace
+from lightning_sdk.cli.clusters_menu import _ClustersMenu
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
+from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -232,18 +233,37 @@ def mmts(
         "If not provided, can be selected in an interactive menu."
     ),
 )
-def containers(teamspace: Optional[str] = None) -> None:
+@click.option(
+    "--cloud-account",
+    "--cloud_account",  # The UI will present the above variant, using this as a secondary to be consistent w/ models
+    default=None,
+    help="The name of the cloud account where containers are stored in.",
+)
+def containers(teamspace: Optional[str] = None, cloud_account: Optional[str] = None) -> None:
     """Display the list of available containers."""
     api = LitContainer()
     menu = _TeamspacesMenu()
+    clusters_menu = _ClustersMenu()
     resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
-    result = api.list_containers(teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name)
+
+    if not cloud_account:
+        cloud_account_obj = clusters_menu._resolve_cluster(resolved_teamspace)
+        cloud_account = "" if cloud_account_obj.spec.cluster_type == V1ClusterType.GLOBAL else cloud_account_obj.id
+
+    result = api.list_containers(
+        teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name, cloud_account=cloud_account
+    )
+
+    if not result:
+        return
+
     table = Table(pad_edge=True, box=None)
     table.add_column("REPOSITORY")
-    table.add_column("IMAGE ID")
+    table.add_column("CLOUD ACCOUNT")
+    table.add_column("LATEST TAG")
     table.add_column("CREATED")
     for repo in result:
-        table.add_row(repo["REPOSITORY"], repo["IMAGE ID"], repo["CREATED"])
+        table.add_row(repo["REPOSITORY"], repo["CLOUD ACCOUNT"], repo["LATEST TAG"], repo["CREATED"])
     Console().print(table)
 
 

lightning_sdk/cli/serve.py

@@ -1,8 +1,11 @@
 import os
 import subprocess
+import time
+import webbrowser
 from datetime import datetime
 from pathlib import Path
 from typing import Optional, Union
+from urllib.parse import urlencode
 
 import click
 from rich.console import Console
@@ -10,9 +13,13 @@ from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm
 
 from lightning_sdk import Machine, Teamspace
+from lightning_sdk.api import UserApi
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.serve import _LitServeDeployer, authenticate
+from lightning_sdk.lightning_cloud import env
+from lightning_sdk.lightning_cloud.login import Auth, AuthServer
+from lightning_sdk.serve import _LitServeDeployer
+from lightning_sdk.utils.resolve import _get_authed_user, _resolve_teamspace
 
 _MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
@@ -31,12 +38,12 @@ def serve() -> None:
     """Serve a LitServe model.
 
     Example:
-        lightning serve server.py  # serve locally
+        lightning serve server.py  # deploy to the cloud
 
     Example:
-        lightning serve server.py --cloud  # deploy to the cloud
+        lightning serve server.py --local  # serve locally
 
-    You can deploy the API to the cloud by running `lightning serve server.py --cloud`.
+    You can deploy the API to the cloud by running `lightning serve server.py`.
     This will build a docker container for the server.py script and deploy it to the Lightning AI platform.
     """
 
@@ -218,6 +225,69 @@ def api_impl(
         raise RuntimeError(error_msg) from None
 
 
+class _AuthServer(AuthServer):
+    def get_auth_url(self, port: int) -> str:
+        redirect_uri = f"http://localhost:{port}/login-complete"
+        params = urlencode({"redirectTo": redirect_uri, "inviteCode": "litserve"})
+        return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
+
+
+class _Auth(Auth):
+    def __init__(self, shall_confirm: bool = False) -> None:
+        super().__init__()
+        self._shall_confirm = shall_confirm
+
+    def _run_server(self) -> None:
+        if self._shall_confirm:
+            proceed = Confirm.ask(
+                "Authenticating with Lightning AI. This will open a browser window. Continue?", default=True
+            )
+            if not proceed:
+                raise RuntimeError(
+                    "Login cancelled. Please login to Lightning AI to deploy the API."
+                    " Run `lightning login` to login."
+                ) from None
+        print("Opening browser for authentication...")
+        print("Please come back to the terminal after logging in.")
+        time.sleep(3)
+        _AuthServer().login_with_browser(self)
+
+
+def authenticate(shall_confirm: bool = True) -> None:
+    auth = _Auth(shall_confirm)
+    auth.authenticate()
+
+
+def select_teamspace(teamspace: Optional[str], org: Optional[str], user: Optional[str]) -> Teamspace:
+    if teamspace is None:
+        user = _get_authed_user()
+        menu = _TeamspacesMenu()
+        possible_teamspaces = menu._get_possible_teamspaces(user)
+        if len(possible_teamspaces) == 1:
+            name = next(iter(possible_teamspaces.values()))["name"]
+            return Teamspace(name=name, org=org, user=user)
+
+        return menu._resolve_teamspace(teamspace)
+
+    return _resolve_teamspace(teamspace=teamspace, org=org, user=user)
+
+
+def poll_verified_status() -> bool:
+    """Polls the verified status of the user until it is True or a timeout occurs."""
+    user_api = UserApi()
+    user = _get_authed_user()
+    start_time = datetime.now()
+    timeout = 600  # 10 minutes
+    while True:
+        user_resp = user_api.get_user(name=user.name)
+        if user_resp.status.verified:
+            return True
+        if (datetime.now() - start_time).total_seconds() > timeout:
+            break
+        time.sleep(5)
+    return False
+
+
 def _handle_cloud(
     script_path: Union[str, Path],
     console: Console,
@@ -242,16 +312,8 @@ def _handle_cloud(
     if non_interactive:
         console.print("[italic]non-interactive[/italic] mode enabled, skipping confirmation prompts", style="blue")
 
-    # Authenticate with LitServe affiliate
-    authenticate()
-    if teamspace is None:
-        menu = _TeamspacesMenu()
-        resolved_teamspace = menu._resolve_teamspace(teamspace)
-    else:
-        resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
-
     port = port or 8000
-    ls_deployer = _LitServeDeployer(name=deployment_name, teamspace=resolved_teamspace)
+    ls_deployer = _LitServeDeployer(name=deployment_name, teamspace=None)
     path = ls_deployer.dockerize_api(script_path, port=port, gpu=not machine.is_cpu(), tag=tag, print_success=False)
 
     console.print(f"\nPlease review the Dockerfile at [u]{path}[/u] and make sure it is correct.", style="bold")
@@ -260,10 +322,6 @@ def _handle_cloud(
         console.print("Please fix the Dockerfile and try again.", style="red")
         return
 
-    # list containers to create the project if it doesn't exist
-    lit_cr = LitContainerApi()
-    lit_cr.list_containers(resolved_teamspace.id, cloud_account=cloud_account)
-
     with Progress(
         SpinnerColumn(),
         TextColumn("[progress.description]{task.description}"),
@@ -280,8 +338,32 @@ def _handle_cloud(
             progress.update(build_task, description="[green]Build completed![/green]", completed=1.0)
             progress.remove_task(build_task)
 
-            # Push the container to the registry
-            console.print("\nPushing image to registry. It may take a while...", style="bold")
+        except Exception as e:
+            console.print(f"❌ Deployment failed: {e}", style="red")
+            return
+
+    # Push the container to the registry
+    console.print("\nPushing container to registry. It may take a while...", style="bold")
+    # Authenticate with LitServe affiliate
+    authenticate(shall_confirm=not non_interactive)
+    resolved_teamspace = select_teamspace(teamspace, org, user)
+    verified = poll_verified_status()
+    if not verified:
+        console.print("❌ Verify phone number to continue. Visit lightning.ai.", style="red")
+        return
+
+    # list containers to create the project if it doesn't exist
+    lit_cr = LitContainerApi()
+    lit_cr.list_containers(resolved_teamspace.id, cloud_account=cloud_account)
+
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        TimeElapsedColumn(),
+        console=console,
+        transient=True,
+    ) as progress:
+        try:
             push_task = progress.add_task("Pushing to registry", total=None)
             push_status = {}
             for line in ls_deployer.push_container(
@@ -313,3 +395,4 @@ def _handle_cloud(
         include_credentials=include_credentials,
     )
     console.print(f"🚀 Deployment started, access at [i]{deployment_status.get('url')}[/i]")
+    webbrowser.open(deployment_status.get("url"))

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -83,6 +83,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublicat
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body import CloudSpaceIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
+from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
@@ -495,6 +496,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response
 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_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
@@ -775,6 +777,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -840,6 +843,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1Subnet
 from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
@@ -903,6 +907,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
 from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
 from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
 from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -3489,6 +3489,99 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_get_cloud_space_instance_system_metrics_aggregate(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceSystemMetricsAggregateResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_instance_system_metrics_aggregate  # 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_system_metrics_aggregate(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
+                 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_system_metrics_aggregate_with_http_info(project_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(project_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceSystemMetricsAggregateResponse':  # noqa: E501
+        """cloud_space_service_get_cloud_space_instance_system_metrics_aggregate  # 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_system_metrics_aggregate_with_http_info(project_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_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_system_metrics_aggregate" % 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_system_metrics_aggregate`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_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}/system-metrics-aggregated', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1GetCloudSpaceInstanceSystemMetricsAggregateResponse',  # 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_size(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1GetCloudSpaceSizeResponse':  # noqa: E501
         """cloud_space_service_get_cloud_space_size  # noqa: E501
 
@@ -6687,6 +6780,119 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def cloud_space_service_report_cloud_space_instance_system_metrics(self, body: 'CloudspaceIdSystemmetricsBody', project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceSystemMetricsResponse':  # noqa: E501
+        """cloud_space_service_report_cloud_space_instance_system_metrics  # 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_report_cloud_space_instance_system_metrics(body, project_id, cloudspace_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param CloudspaceIdSystemmetricsBody body: (required)
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :return: V1ReportCloudSpaceInstanceSystemMetricsResponse
+                 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_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.cloud_space_service_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, **kwargs)  # noqa: E501
+            return data
+
+    def cloud_space_service_report_cloud_space_instance_system_metrics_with_http_info(self, body: 'CloudspaceIdSystemmetricsBody', project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceSystemMetricsResponse':  # noqa: E501
+        """cloud_space_service_report_cloud_space_instance_system_metrics  # 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_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param CloudspaceIdSystemmetricsBody body: (required)
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :return: V1ReportCloudSpaceInstanceSystemMetricsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', '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_report_cloud_space_instance_system_metrics" % 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 `cloud_space_service_report_cloud_space_instance_system_metrics`")  # 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 `cloud_space_service_report_cloud_space_instance_system_metrics`")  # 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_report_cloud_space_instance_system_metrics`")  # 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
+        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}/cloudspaces/{cloudspaceId}/system-metrics', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ReportCloudSpaceInstanceSystemMetricsResponse',  # 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_restart_cloud_space_instance(self, project_id: 'str', id: 'str', **kwargs) -> 'V1RestartCloudSpaceInstanceResponse':  # noqa: E501
         """cloud_space_service_restart_cloud_space_instance  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -41,6 +41,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublicat
 from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body import CloudSpaceIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
+from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
@@ -453,6 +454,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response
 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_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
@@ -733,6 +735,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -798,6 +801,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1Subnet
 from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
+from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
@@ -861,6 +865,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
 from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
 from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
 from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody