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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (60) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/cluster_api.py +22 -0
  3. lightning_sdk/api/deployment_api.py +1 -0
  4. lightning_sdk/api/lit_container_api.py +24 -5
  5. lightning_sdk/api/teamspace_api.py +22 -17
  6. lightning_sdk/api/utils.py +1 -1
  7. lightning_sdk/cli/clusters_menu.py +46 -0
  8. lightning_sdk/cli/entrypoint.py +2 -2
  9. lightning_sdk/cli/list.py +25 -5
  10. lightning_sdk/cli/serve.py +232 -24
  11. lightning_sdk/cli/upload.py +4 -1
  12. lightning_sdk/deployment/deployment.py +5 -2
  13. lightning_sdk/lightning_cloud/openapi/__init__.py +10 -0
  14. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +303 -0
  15. lightning_sdk/lightning_cloud/openapi/models/__init__.py +10 -0
  16. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_systemmetrics_body.py +149 -0
  17. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +55 -3
  18. lightning_sdk/lightning_cloud/openapi/models/create.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +55 -3
  20. lightning_sdk/lightning_cloud/openapi/models/update.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  22. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  23. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics.py +53 -1
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics_stats.py +357 -0
  25. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  26. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +227 -0
  27. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +55 -3
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +79 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_capacity_reservation_response.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +53 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_direct_vpc.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_cold_start_metrics_stats_response.py +123 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_system_metrics_aggregate_response.py +123 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +53 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +43 -17
  39. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +55 -3
  40. lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py +253 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_report_cloud_space_instance_system_metrics_response.py +97 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_phase.py +1 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics_aggregated.py +227 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +104 -208
  47. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_weka_data_connection.py +201 -0
  49. lightning_sdk/lightning_cloud/openapi/models/validate.py +27 -1
  50. lightning_sdk/lit_container.py +25 -7
  51. lightning_sdk/models.py +26 -8
  52. lightning_sdk/serve.py +3 -20
  53. lightning_sdk/teamspace.py +21 -4
  54. lightning_sdk/utils/resolve.py +11 -4
  55. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/METADATA +1 -1
  56. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/RECORD +60 -48
  57. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/LICENSE +0 -0
  58. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/WHEEL +0 -0
  59. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/entry_points.txt +0 -0
  60. {lightning_sdk-0.2.8.dist-info → lightning_sdk-0.2.10.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,7 @@
1
1
  import concurrent.futures
2
2
  import json
3
3
  import os
4
+ import webbrowser
4
5
  from pathlib import Path
5
6
  from typing import Dict, Generator, List, Optional
6
7
 
@@ -162,7 +163,7 @@ def upload_container(
162
163
  # let the push with retry take control of auth moving forward
163
164
  pass
164
165
 
165
- lines = api.upload_container(container, teamspace, tag, cloud_account, platform)
166
+ lines = api.upload_container(container, teamspace, tag, cloud_account, platform, return_final_dict=True)
166
167
  _print_docker_push(lines, console, progress, push_task)
167
168
  except DockerNotRunningError as e:
168
169
  e.print_help()
@@ -303,6 +304,8 @@ def _print_docker_push(lines: Generator, console: Console, progress: Progress, p
303
304
  console.print(f"\n[red]{line}[/red]")
304
305
  return
305
306
  elif "finish" in line:
307
+ if "url" in line:
308
+ webbrowser.open(line["url"])
306
309
  console.print(f"Container available at [i]{line['url']}[/i]")
307
310
  return
308
311
  else:
@@ -117,6 +117,7 @@ class Deployment:
117
117
  cloud_account: Optional[str] = None,
118
118
  custom_domain: Optional[str] = None,
119
119
  cluster: Optional[str] = None, # deprecated in favor of cloud_account
120
+ cloudspace_id: Optional[str] = None,
120
121
  quantity: Optional[int] = None,
121
122
  include_credentials: Optional[bool] = None,
122
123
  ) -> None:
@@ -141,9 +142,10 @@ class Deployment:
141
142
  auth: The auth config to protect your services. Only Basic and Token supported.
142
143
  cloud_account: The name of the cloud account, the studio should be created on.
143
144
  Doesn't matter when the studio already exists.
144
- custom_domain: Whether your service would be referenced under a custom doamin.
145
+ custom_domain: Whether your service would be referenced under a custom domain.
146
+ cloudspace_id: Connect deployment to a Studio.
145
147
  quantity: The number of machines per replica to deploy.
146
- include_credentials: Whether to include the environement variables for the SDK to authenticate
148
+ include_credentials: Whether to include the environment variables for the SDK to authenticate
147
149
 
148
150
  Note:
149
151
  Since a teamspace can either be owned by an org or by a user directly,
@@ -166,6 +168,7 @@ class Deployment:
166
168
  name=self._name,
167
169
  project_id=self._teamspace.id,
168
170
  replicas=replicas,
171
+ cloudspace_id=cloudspace_id,
169
172
  spec=to_spec(
170
173
  cloud_account=cloud_account,
171
174
  command=command,
@@ -83,6 +83,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublicat
83
83
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body import CloudSpaceIdVersionsBody
84
84
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
85
85
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
86
+ from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
86
87
  from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
87
88
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
88
89
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
@@ -277,6 +278,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
277
278
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
278
279
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
279
280
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics import V1CloudSpaceColdStartMetrics
281
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
280
282
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
281
283
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
282
284
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
@@ -292,6 +294,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
292
294
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
293
295
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
294
296
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
297
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
295
298
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
296
299
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
297
300
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -483,6 +486,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
483
486
  from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
484
487
  from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
485
488
  from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
489
+ from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
486
490
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
487
491
  from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
488
492
  from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
@@ -494,7 +498,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
494
498
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
495
499
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
496
500
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
501
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
497
502
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
503
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
498
504
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
499
505
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
500
506
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
@@ -768,6 +774,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1Quest
768
774
  from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
769
775
  from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
770
776
  from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
777
+ from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
771
778
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
772
779
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
773
780
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -775,6 +782,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
775
782
  from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
776
783
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
777
784
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
785
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
778
786
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
779
787
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
780
788
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -840,6 +848,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1Subnet
840
848
  from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
841
849
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
842
850
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
851
+ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
843
852
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
844
853
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
845
854
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
@@ -903,6 +912,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_respons
903
912
  from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
904
913
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
905
914
  from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
915
+ from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
906
916
  from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
907
917
  from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
908
918
  from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody
@@ -3174,6 +3174,99 @@ class CloudSpaceServiceApi(object):
3174
3174
  _request_timeout=params.get('_request_timeout'),
3175
3175
  collection_formats=collection_formats)
3176
3176
 
3177
+ def cloud_space_service_get_cloud_space_cold_start_metrics_stats(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse': # noqa: E501
3178
+ """cloud_space_service_get_cloud_space_cold_start_metrics_stats # noqa: E501
3179
+
3180
+ This method makes a synchronous HTTP request by default. To make an
3181
+ asynchronous HTTP request, please pass async_req=True
3182
+ >>> thread = api.cloud_space_service_get_cloud_space_cold_start_metrics_stats(project_id, async_req=True)
3183
+ >>> result = thread.get()
3184
+
3185
+ :param async_req bool
3186
+ :param str project_id: (required)
3187
+ :return: V1GetCloudSpaceColdStartMetricsStatsResponse
3188
+ If the method is called asynchronously,
3189
+ returns the request thread.
3190
+ """
3191
+ kwargs['_return_http_data_only'] = True
3192
+ if kwargs.get('async_req'):
3193
+ return self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs) # noqa: E501
3194
+ else:
3195
+ (data) = self.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, **kwargs) # noqa: E501
3196
+ return data
3197
+
3198
+ def cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceColdStartMetricsStatsResponse': # noqa: E501
3199
+ """cloud_space_service_get_cloud_space_cold_start_metrics_stats # noqa: E501
3200
+
3201
+ This method makes a synchronous HTTP request by default. To make an
3202
+ asynchronous HTTP request, please pass async_req=True
3203
+ >>> thread = api.cloud_space_service_get_cloud_space_cold_start_metrics_stats_with_http_info(project_id, async_req=True)
3204
+ >>> result = thread.get()
3205
+
3206
+ :param async_req bool
3207
+ :param str project_id: (required)
3208
+ :return: V1GetCloudSpaceColdStartMetricsStatsResponse
3209
+ If the method is called asynchronously,
3210
+ returns the request thread.
3211
+ """
3212
+
3213
+ all_params = ['project_id'] # noqa: E501
3214
+ all_params.append('async_req')
3215
+ all_params.append('_return_http_data_only')
3216
+ all_params.append('_preload_content')
3217
+ all_params.append('_request_timeout')
3218
+
3219
+ params = locals()
3220
+ for key, val in six.iteritems(params['kwargs']):
3221
+ if key not in all_params:
3222
+ raise TypeError(
3223
+ "Got an unexpected keyword argument '%s'"
3224
+ " to method cloud_space_service_get_cloud_space_cold_start_metrics_stats" % key
3225
+ )
3226
+ params[key] = val
3227
+ del params['kwargs']
3228
+ # verify the required parameter 'project_id' is set
3229
+ if ('project_id' not in params or
3230
+ params['project_id'] is None):
3231
+ raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_get_cloud_space_cold_start_metrics_stats`") # noqa: E501
3232
+
3233
+ collection_formats = {}
3234
+
3235
+ path_params = {}
3236
+ if 'project_id' in params:
3237
+ path_params['projectId'] = params['project_id'] # noqa: E501
3238
+
3239
+ query_params = []
3240
+
3241
+ header_params = {}
3242
+
3243
+ form_params = []
3244
+ local_var_files = {}
3245
+
3246
+ body_params = None
3247
+ # HTTP header `Accept`
3248
+ header_params['Accept'] = self.api_client.select_header_accept(
3249
+ ['application/json']) # noqa: E501
3250
+
3251
+ # Authentication setting
3252
+ auth_settings = [] # noqa: E501
3253
+
3254
+ return self.api_client.call_api(
3255
+ '/v1/projects/{projectId}/cloudspaces/cold-start-stats', 'GET',
3256
+ path_params,
3257
+ query_params,
3258
+ header_params,
3259
+ body=body_params,
3260
+ post_params=form_params,
3261
+ files=local_var_files,
3262
+ response_type='V1GetCloudSpaceColdStartMetricsStatsResponse', # noqa: E501
3263
+ auth_settings=auth_settings,
3264
+ async_req=params.get('async_req'),
3265
+ _return_http_data_only=params.get('_return_http_data_only'),
3266
+ _preload_content=params.get('_preload_content', True),
3267
+ _request_timeout=params.get('_request_timeout'),
3268
+ collection_formats=collection_formats)
3269
+
3177
3270
  def cloud_space_service_get_cloud_space_folder_index(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetFolderIndexResponse': # noqa: E501
3178
3271
  """cloud_space_service_get_cloud_space_folder_index # noqa: E501
3179
3272
 
@@ -3489,6 +3582,103 @@ class CloudSpaceServiceApi(object):
3489
3582
  _request_timeout=params.get('_request_timeout'),
3490
3583
  collection_formats=collection_formats)
3491
3584
 
3585
+ def cloud_space_service_get_cloud_space_instance_system_metrics_aggregate(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceSystemMetricsAggregateResponse': # noqa: E501
3586
+ """cloud_space_service_get_cloud_space_instance_system_metrics_aggregate # noqa: E501
3587
+
3588
+ This method makes a synchronous HTTP request by default. To make an
3589
+ asynchronous HTTP request, please pass async_req=True
3590
+ >>> thread = api.cloud_space_service_get_cloud_space_instance_system_metrics_aggregate(project_id, async_req=True)
3591
+ >>> result = thread.get()
3592
+
3593
+ :param async_req bool
3594
+ :param str project_id: (required)
3595
+ :param str cloudspace_id:
3596
+ :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
3597
+ If the method is called asynchronously,
3598
+ returns the request thread.
3599
+ """
3600
+ kwargs['_return_http_data_only'] = True
3601
+ if kwargs.get('async_req'):
3602
+ return self.cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(project_id, **kwargs) # noqa: E501
3603
+ else:
3604
+ (data) = self.cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(project_id, **kwargs) # noqa: E501
3605
+ return data
3606
+
3607
+ def cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(self, project_id: 'str', **kwargs) -> 'V1GetCloudSpaceInstanceSystemMetricsAggregateResponse': # noqa: E501
3608
+ """cloud_space_service_get_cloud_space_instance_system_metrics_aggregate # noqa: E501
3609
+
3610
+ This method makes a synchronous HTTP request by default. To make an
3611
+ asynchronous HTTP request, please pass async_req=True
3612
+ >>> thread = api.cloud_space_service_get_cloud_space_instance_system_metrics_aggregate_with_http_info(project_id, async_req=True)
3613
+ >>> result = thread.get()
3614
+
3615
+ :param async_req bool
3616
+ :param str project_id: (required)
3617
+ :param str cloudspace_id:
3618
+ :return: V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
3619
+ If the method is called asynchronously,
3620
+ returns the request thread.
3621
+ """
3622
+
3623
+ all_params = ['project_id', 'cloudspace_id'] # noqa: E501
3624
+ all_params.append('async_req')
3625
+ all_params.append('_return_http_data_only')
3626
+ all_params.append('_preload_content')
3627
+ all_params.append('_request_timeout')
3628
+
3629
+ params = locals()
3630
+ for key, val in six.iteritems(params['kwargs']):
3631
+ if key not in all_params:
3632
+ raise TypeError(
3633
+ "Got an unexpected keyword argument '%s'"
3634
+ " to method cloud_space_service_get_cloud_space_instance_system_metrics_aggregate" % key
3635
+ )
3636
+ params[key] = val
3637
+ del params['kwargs']
3638
+ # verify the required parameter 'project_id' is set
3639
+ if ('project_id' not in params or
3640
+ params['project_id'] is None):
3641
+ raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_get_cloud_space_instance_system_metrics_aggregate`") # noqa: E501
3642
+
3643
+ collection_formats = {}
3644
+
3645
+ path_params = {}
3646
+ if 'project_id' in params:
3647
+ path_params['projectId'] = params['project_id'] # noqa: E501
3648
+
3649
+ query_params = []
3650
+ if 'cloudspace_id' in params:
3651
+ query_params.append(('cloudspaceId', params['cloudspace_id'])) # noqa: E501
3652
+
3653
+ header_params = {}
3654
+
3655
+ form_params = []
3656
+ local_var_files = {}
3657
+
3658
+ body_params = None
3659
+ # HTTP header `Accept`
3660
+ header_params['Accept'] = self.api_client.select_header_accept(
3661
+ ['application/json']) # noqa: E501
3662
+
3663
+ # Authentication setting
3664
+ auth_settings = [] # noqa: E501
3665
+
3666
+ return self.api_client.call_api(
3667
+ '/v1/projects/{projectId}/system-metrics-aggregated', 'GET',
3668
+ path_params,
3669
+ query_params,
3670
+ header_params,
3671
+ body=body_params,
3672
+ post_params=form_params,
3673
+ files=local_var_files,
3674
+ response_type='V1GetCloudSpaceInstanceSystemMetricsAggregateResponse', # noqa: E501
3675
+ auth_settings=auth_settings,
3676
+ async_req=params.get('async_req'),
3677
+ _return_http_data_only=params.get('_return_http_data_only'),
3678
+ _preload_content=params.get('_preload_content', True),
3679
+ _request_timeout=params.get('_request_timeout'),
3680
+ collection_formats=collection_formats)
3681
+
3492
3682
  def cloud_space_service_get_cloud_space_size(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1GetCloudSpaceSizeResponse': # noqa: E501
3493
3683
  """cloud_space_service_get_cloud_space_size # noqa: E501
3494
3684
 
@@ -6687,6 +6877,119 @@ class CloudSpaceServiceApi(object):
6687
6877
  _request_timeout=params.get('_request_timeout'),
6688
6878
  collection_formats=collection_formats)
6689
6879
 
6880
+ def cloud_space_service_report_cloud_space_instance_system_metrics(self, body: 'CloudspaceIdSystemmetricsBody', project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceSystemMetricsResponse': # noqa: E501
6881
+ """cloud_space_service_report_cloud_space_instance_system_metrics # noqa: E501
6882
+
6883
+ This method makes a synchronous HTTP request by default. To make an
6884
+ asynchronous HTTP request, please pass async_req=True
6885
+ >>> thread = api.cloud_space_service_report_cloud_space_instance_system_metrics(body, project_id, cloudspace_id, async_req=True)
6886
+ >>> result = thread.get()
6887
+
6888
+ :param async_req bool
6889
+ :param CloudspaceIdSystemmetricsBody body: (required)
6890
+ :param str project_id: (required)
6891
+ :param str cloudspace_id: (required)
6892
+ :return: V1ReportCloudSpaceInstanceSystemMetricsResponse
6893
+ If the method is called asynchronously,
6894
+ returns the request thread.
6895
+ """
6896
+ kwargs['_return_http_data_only'] = True
6897
+ if kwargs.get('async_req'):
6898
+ return self.cloud_space_service_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, **kwargs) # noqa: E501
6899
+ else:
6900
+ (data) = self.cloud_space_service_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, **kwargs) # noqa: E501
6901
+ return data
6902
+
6903
+ 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
6904
+ """cloud_space_service_report_cloud_space_instance_system_metrics # noqa: E501
6905
+
6906
+ This method makes a synchronous HTTP request by default. To make an
6907
+ asynchronous HTTP request, please pass async_req=True
6908
+ >>> thread = api.cloud_space_service_report_cloud_space_instance_system_metrics_with_http_info(body, project_id, cloudspace_id, async_req=True)
6909
+ >>> result = thread.get()
6910
+
6911
+ :param async_req bool
6912
+ :param CloudspaceIdSystemmetricsBody body: (required)
6913
+ :param str project_id: (required)
6914
+ :param str cloudspace_id: (required)
6915
+ :return: V1ReportCloudSpaceInstanceSystemMetricsResponse
6916
+ If the method is called asynchronously,
6917
+ returns the request thread.
6918
+ """
6919
+
6920
+ all_params = ['body', 'project_id', 'cloudspace_id'] # noqa: E501
6921
+ all_params.append('async_req')
6922
+ all_params.append('_return_http_data_only')
6923
+ all_params.append('_preload_content')
6924
+ all_params.append('_request_timeout')
6925
+
6926
+ params = locals()
6927
+ for key, val in six.iteritems(params['kwargs']):
6928
+ if key not in all_params:
6929
+ raise TypeError(
6930
+ "Got an unexpected keyword argument '%s'"
6931
+ " to method cloud_space_service_report_cloud_space_instance_system_metrics" % key
6932
+ )
6933
+ params[key] = val
6934
+ del params['kwargs']
6935
+ # verify the required parameter 'body' is set
6936
+ if ('body' not in params or
6937
+ params['body'] is None):
6938
+ raise ValueError("Missing the required parameter `body` when calling `cloud_space_service_report_cloud_space_instance_system_metrics`") # noqa: E501
6939
+ # verify the required parameter 'project_id' is set
6940
+ if ('project_id' not in params or
6941
+ params['project_id'] is None):
6942
+ raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_report_cloud_space_instance_system_metrics`") # noqa: E501
6943
+ # verify the required parameter 'cloudspace_id' is set
6944
+ if ('cloudspace_id' not in params or
6945
+ params['cloudspace_id'] is None):
6946
+ raise ValueError("Missing the required parameter `cloudspace_id` when calling `cloud_space_service_report_cloud_space_instance_system_metrics`") # noqa: E501
6947
+
6948
+ collection_formats = {}
6949
+
6950
+ path_params = {}
6951
+ if 'project_id' in params:
6952
+ path_params['projectId'] = params['project_id'] # noqa: E501
6953
+ if 'cloudspace_id' in params:
6954
+ path_params['cloudspaceId'] = params['cloudspace_id'] # noqa: E501
6955
+
6956
+ query_params = []
6957
+
6958
+ header_params = {}
6959
+
6960
+ form_params = []
6961
+ local_var_files = {}
6962
+
6963
+ body_params = None
6964
+ if 'body' in params:
6965
+ body_params = params['body']
6966
+ # HTTP header `Accept`
6967
+ header_params['Accept'] = self.api_client.select_header_accept(
6968
+ ['application/json']) # noqa: E501
6969
+
6970
+ # HTTP header `Content-Type`
6971
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
6972
+ ['application/json']) # noqa: E501
6973
+
6974
+ # Authentication setting
6975
+ auth_settings = [] # noqa: E501
6976
+
6977
+ return self.api_client.call_api(
6978
+ '/v1/projects/{projectId}/cloudspaces/{cloudspaceId}/system-metrics', 'POST',
6979
+ path_params,
6980
+ query_params,
6981
+ header_params,
6982
+ body=body_params,
6983
+ post_params=form_params,
6984
+ files=local_var_files,
6985
+ response_type='V1ReportCloudSpaceInstanceSystemMetricsResponse', # noqa: E501
6986
+ auth_settings=auth_settings,
6987
+ async_req=params.get('async_req'),
6988
+ _return_http_data_only=params.get('_return_http_data_only'),
6989
+ _preload_content=params.get('_preload_content', True),
6990
+ _request_timeout=params.get('_request_timeout'),
6991
+ collection_formats=collection_formats)
6992
+
6690
6993
  def cloud_space_service_restart_cloud_space_instance(self, project_id: 'str', id: 'str', **kwargs) -> 'V1RestartCloudSpaceInstanceResponse': # noqa: E501
6691
6994
  """cloud_space_service_restart_cloud_space_instance # noqa: E501
6692
6995
 
@@ -41,6 +41,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versionpublicat
41
41
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body import CloudSpaceIdVersionsBody
42
42
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
43
43
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
44
+ from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
44
45
  from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
45
46
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
46
47
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
@@ -235,6 +236,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
235
236
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
236
237
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
237
238
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics import V1CloudSpaceColdStartMetrics
239
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
238
240
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
239
241
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
240
242
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
@@ -250,6 +252,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
250
252
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
251
253
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
252
254
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
255
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
253
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
254
257
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
255
258
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -441,6 +444,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
441
444
  from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
442
445
  from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
443
446
  from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
447
+ from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
444
448
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
445
449
  from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
446
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
@@ -452,7 +456,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
452
456
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
453
457
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
454
458
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
459
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
455
460
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
461
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
456
462
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
457
463
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
458
464
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
@@ -726,6 +732,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1Quest
726
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
727
733
  from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
728
734
  from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
735
+ from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
729
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
730
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
731
738
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -733,6 +740,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
733
740
  from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
734
741
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
735
742
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
743
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
736
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
737
745
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
738
746
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -798,6 +806,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1Subnet
798
806
  from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
799
807
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
800
808
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
809
+ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
801
810
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
802
811
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
803
812
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
@@ -861,6 +870,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_respons
861
870
  from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
862
871
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
863
872
  from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
873
+ from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
864
874
  from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
865
875
  from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
866
876
  from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody