lightning-sdk 2025.12.16__py3-none-any.whl → 2026.1.22__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 (141) hide show
  1. lightning_sdk/__version__.py +1 -1
  2. lightning_sdk/api/k8s_api.py +75 -29
  3. lightning_sdk/api/org_api.py +7 -0
  4. lightning_sdk/api/studio_api.py +223 -36
  5. lightning_sdk/api/teamspace_api.py +28 -9
  6. lightning_sdk/api/user_api.py +5 -0
  7. lightning_sdk/cli/cp/__init__.py +64 -0
  8. lightning_sdk/cli/entrypoint.py +2 -0
  9. lightning_sdk/cli/groups.py +22 -0
  10. lightning_sdk/cli/legacy/clusters_menu.py +2 -2
  11. lightning_sdk/cli/legacy/deploy/_auth.py +7 -6
  12. lightning_sdk/cli/legacy/download.py +2 -1
  13. lightning_sdk/cli/legacy/run.py +13 -2
  14. lightning_sdk/cli/legacy/studios_menu.py +8 -1
  15. lightning_sdk/cli/studio/__init__.py +4 -0
  16. lightning_sdk/cli/studio/cp.py +20 -64
  17. lightning_sdk/cli/studio/ls.py +57 -0
  18. lightning_sdk/cli/studio/rm.py +71 -0
  19. lightning_sdk/cli/utils/logging.py +2 -1
  20. lightning_sdk/cli/utils/studio_filesystem.py +65 -0
  21. lightning_sdk/cli/utils/teamspace_selection.py +5 -0
  22. lightning_sdk/exceptions.py +4 -0
  23. lightning_sdk/job/base.py +24 -5
  24. lightning_sdk/job/job.py +12 -5
  25. lightning_sdk/job/v1.py +5 -2
  26. lightning_sdk/job/v2.py +9 -1
  27. lightning_sdk/k8s_cluster.py +9 -10
  28. lightning_sdk/lightning_cloud/__version__.py +1 -1
  29. lightning_sdk/lightning_cloud/openapi/__init__.py +46 -3
  30. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  31. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +118 -1
  32. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +787 -125
  33. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +5 -1
  34. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +11 -11
  35. lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py +246 -19
  36. lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py +116 -11
  37. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +588 -2
  38. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +9 -1
  39. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +5 -1
  40. lightning_sdk/lightning_cloud/openapi/api/virtual_machine_service_api.py +557 -0
  41. lightning_sdk/lightning_cloud/openapi/models/__init__.py +45 -3
  42. lightning_sdk/lightning_cloud/openapi/models/cloud_space_environment_template_service_update_cloud_space_environment_template_body.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/cluster_service_add_container_registry_body.py +123 -0
  44. lightning_sdk/lightning_cloud/openapi/models/cluster_service_create_cluster_capacity_reservation_body.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/cluster_service_create_machine_body.py +79 -1
  46. lightning_sdk/lightning_cloud/openapi/models/cluster_service_create_org_cluster_capacity_reservation_body.py +409 -0
  47. lightning_sdk/lightning_cloud/openapi/models/cluster_service_refresh_container_registry_credentials_body.py +97 -0
  48. lightning_sdk/lightning_cloud/openapi/models/cluster_service_report_machine_system_metrics_body.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/cluster_service_validate_container_registry_body.py +97 -0
  50. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/lit_logger_service_create_lit_logger_media_body.py +305 -0
  52. lightning_sdk/lightning_cloud/openapi/models/lit_logger_service_update_lit_logger_media_body.py +149 -0
  53. lightning_sdk/lightning_cloud/openapi/models/lit_logger_service_update_metrics_stream_body.py +53 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_add_container_registry_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_capacity_reservation_used_by.py +227 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +27 -1
  58. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +1 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +53 -27
  63. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +0 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry.py +253 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry_info.py +281 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry_integration.py +123 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry_status.py +105 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_logger_media_response.py +149 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_create_org_cluster_capacity_reservation_response.py +201 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_create_sdk_command_history_request.py +29 -3
  72. lightning_sdk/lightning_cloud/openapi/models/v1_cudo_direct_v1.py +175 -0
  73. lightning_sdk/lightning_cloud/openapi/models/v1_delete_container_registry_response.py +97 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_logger_media_response.py +97 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_delete_virtual_machine_response.py +97 -0
  76. lightning_sdk/lightning_cloud/openapi/models/{v1_ai_pod_v1.py → v1_ecr_registry_config.py} +49 -49
  77. lightning_sdk/lightning_cloud/openapi/models/v1_ecr_registry_config_input.py +123 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_ecr_registry_details.py +201 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -27
  80. lightning_sdk/lightning_cloud/openapi/models/v1_external_search_user.py +27 -1
  81. lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py +29 -3
  82. lightning_sdk/lightning_cloud/openapi/models/v1_get_kubernetes_pod_logs_response.py +149 -0
  83. lightning_sdk/lightning_cloud/openapi/models/{v1_get_machine_response.py → v1_get_kubernetes_pod_response.py} +23 -23
  84. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  85. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  86. lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py +27 -1
  87. lightning_sdk/lightning_cloud/openapi/models/v1_k8s_incident_setting.py +149 -0
  88. lightning_sdk/lightning_cloud/openapi/models/v1_k8s_incident_type.py +108 -0
  89. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_settings_v1.py +53 -1
  90. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +105 -1
  91. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_pod.py +27 -1
  92. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_pod_logs_page.py +227 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_kubevirt_config.py +357 -0
  94. lightning_sdk/lightning_cloud/openapi/models/v1_list_container_registries_response.py +123 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_list_kubernetes_pods_response.py +43 -17
  96. lightning_sdk/lightning_cloud/openapi/models/v1_list_kubernetes_pods_sort_order.py +104 -0
  97. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_logger_media_response.py +149 -0
  98. lightning_sdk/lightning_cloud/openapi/models/v1_list_models_response.py +55 -3
  99. lightning_sdk/lightning_cloud/openapi/models/v1_list_virtual_machines_response.py +123 -0
  100. lightning_sdk/lightning_cloud/openapi/models/v1_lit_logger_media.py +513 -0
  101. lightning_sdk/lightning_cloud/openapi/models/v1_machine.py +53 -53
  102. lightning_sdk/lightning_cloud/openapi/models/v1_machine_direct_v1.py +107 -3
  103. lightning_sdk/lightning_cloud/openapi/models/v1_media_type.py +104 -0
  104. lightning_sdk/lightning_cloud/openapi/models/v1_nebius_direct_v1.py +29 -3
  105. lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py +27 -1
  106. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  107. lightning_sdk/lightning_cloud/openapi/models/v1_refresh_container_registry_credentials_response.py +123 -0
  108. lightning_sdk/lightning_cloud/openapi/models/v1_report_cloud_space_instance_idle_state_response.py +97 -0
  109. lightning_sdk/lightning_cloud/openapi/models/v1_report_machine_system_metrics_response.py +97 -0
  110. lightning_sdk/lightning_cloud/openapi/models/v1_search_user.py +27 -1
  111. lightning_sdk/lightning_cloud/openapi/models/v1_tenant_credentials.py +201 -0
  112. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +217 -113
  113. lightning_sdk/lightning_cloud/openapi/models/v1_validate_container_registry_response.py +149 -0
  114. lightning_sdk/lightning_cloud/openapi/models/v1_virtual_machine.py +409 -0
  115. lightning_sdk/lightning_cloud/openapi/models/v1_vm_configuration.py +149 -0
  116. lightning_sdk/lightning_cloud/openapi/models/v1_vm_provider_configuration.py +227 -0
  117. lightning_sdk/lightning_cloud/openapi/models/virtual_machine_service_create_virtual_machine_body.py +565 -0
  118. lightning_sdk/lightning_cloud/openapi/models/virtual_machine_service_update_virtual_machine_body.py +97 -0
  119. lightning_sdk/lightning_cloud/openapi/rest.py +2 -2
  120. lightning_sdk/lightning_cloud/rest_client.py +0 -2
  121. lightning_sdk/machine.py +3 -3
  122. lightning_sdk/mmt/base.py +26 -7
  123. lightning_sdk/mmt/mmt.py +11 -6
  124. lightning_sdk/mmt/v1.py +5 -2
  125. lightning_sdk/mmt/v2.py +5 -2
  126. lightning_sdk/organization.py +10 -1
  127. lightning_sdk/owner.py +4 -0
  128. lightning_sdk/plugin.py +2 -2
  129. lightning_sdk/studio.py +47 -6
  130. lightning_sdk/user.py +22 -2
  131. lightning_sdk/utils/logging.py +2 -1
  132. lightning_sdk/utils/resolve.py +9 -7
  133. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/METADATA +1 -5
  134. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/RECORD +139 -94
  135. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/WHEEL +1 -1
  136. lightning_sdk/lightning_cloud/cli/__main__.py +0 -29
  137. lightning_sdk/lightning_cloud/source_code/logs_socket_api.py +0 -103
  138. /lightning_sdk/lightning_cloud/openapi/models/{v1_list_filesystem_mm_ts_response.py → v1_list_filesystem_mmts_response.py} +0 -0
  139. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/LICENSE +0 -0
  140. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/entry_points.txt +0 -0
  141. {lightning_sdk-2025.12.16.dist-info → lightning_sdk-2026.1.22.dist-info}/top_level.txt +0 -0
lightning_sdk/job/job.py CHANGED
@@ -97,7 +97,7 @@ class Job(_BaseJob):
97
97
  interruptible: bool = False,
98
98
  image_credentials: Optional[str] = None,
99
99
  cloud_account_auth: bool = False,
100
- entrypoint: str = "sh -c",
100
+ entrypoint: Optional[str] = None,
101
101
  path_mappings: Optional[Dict[str, str]] = None,
102
102
  max_runtime: Optional[int] = None,
103
103
  artifacts_local: Optional[str] = None, # deprecated in terms of path_mappings
@@ -132,8 +132,10 @@ class Job(_BaseJob):
132
132
  cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
133
133
  Required if the registry is part of a cloud provider (e.g. ECR).
134
134
  entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
135
- just runs the provided command in a standard shell.
136
- To use the pre-defined entrypoint of the provided image, set this to an empty string.
135
+ just runs the provided command in a standard shell if a command is provided.
136
+ If no command is provided, it will run the pre-defined entrypoint of the provided image.
137
+ To use the pre-defined entrypoint of the provided image with a specified command,
138
+ set this to an empty string.
137
139
  Only applicable when submitting docker jobs.
138
140
  path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
139
141
  represents the data-connection name and the path inside that connection.
@@ -193,7 +195,7 @@ class Job(_BaseJob):
193
195
  cloud_provider: Optional[Union["CloudProvider", str]] = None,
194
196
  image_credentials: Optional[str] = None,
195
197
  cloud_account_auth: bool = False,
196
- entrypoint: str = "sh -c",
198
+ entrypoint: Optional[str] = None,
197
199
  path_mappings: Optional[Dict[str, str]] = None,
198
200
  artifacts_local: Optional[str] = None, # deprecated in terms of path_mappings
199
201
  artifacts_remote: Optional[str] = None, # deprecated in terms of path_mappings
@@ -222,7 +224,12 @@ class Job(_BaseJob):
222
224
  This should be the name of the respective credentials secret created on the Lightning AI platform.
223
225
  cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
224
226
  Required if the registry is part of a cloud provider (e.g. ECR).
225
- entrypoint: The entrypoint of your docker container. Defaults to sh -c.
227
+ entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
228
+ just runs the provided command in a standard shell if a command is provided.
229
+ If no command is provided, it will run the pre-defined entrypoint of the provided image.
230
+ To use the pre-defined entrypoint of the provided image with a specified command,
231
+ set this to an empty string.
232
+ Only applicable when submitting docker jobs.
226
233
  To use the pre-defined entrypoint of the provided image, set this to an empty string.
227
234
  Only applicable when submitting docker jobs.
228
235
  path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
lightning_sdk/job/v1.py CHANGED
@@ -108,7 +108,7 @@ class _JobV1(_BaseJob):
108
108
  cloud_account_auth: bool = False,
109
109
  artifacts_local: Optional[str] = None,
110
110
  artifacts_remote: Optional[str] = None,
111
- entrypoint: str = "sh -c",
111
+ entrypoint: Optional[str] = None,
112
112
  path_mappings: Optional[Dict[str, str]] = None,
113
113
  max_runtime: Optional[int] = None,
114
114
  reuse_snapshot: bool = True,
@@ -127,7 +127,10 @@ class _JobV1(_BaseJob):
127
127
  image_credentials: The image credentials for the job (not supported).
128
128
  cloud_account_auth: Whether to use cloud account authentication for the job (not supported).
129
129
  entrypoint: The entrypoint of your docker container (not supported).
130
- Defaults to `sh -c` which just runs the provided command in a standard shell.
130
+ Defaults to `sh -c` which just runs the provided command in a standard shell if a command is provided.
131
+ If no command is provided, it will run the pre-defined entrypoint of the provided image.
132
+ To use the pre-defined entrypoint of the provided image with a specified command,
133
+ set this to an empty string.
131
134
  To use the pre-defined entrypoint of the provided image, set this to an empty string.
132
135
  Only applicable when submitting docker jobs.
133
136
  path_mappings: The mappings from data connection inside your container (not supported)
lightning_sdk/job/v2.py CHANGED
@@ -49,7 +49,7 @@ class _JobV2(_BaseJob):
49
49
  cloud_provider: Optional[Union["CloudProvider", str]] = None,
50
50
  image_credentials: Optional[str] = None,
51
51
  cloud_account_auth: bool = False,
52
- entrypoint: str = "sh -c",
52
+ entrypoint: Optional[str] = None,
53
53
  path_mappings: Optional[Dict[str, str]] = None,
54
54
  max_runtime: Optional[int] = None,
55
55
  artifacts_local: Optional[str] = None, # deprecated in favor of path_mappings
@@ -74,6 +74,14 @@ class _JobV2(_BaseJob):
74
74
  This should be the name of the respective credentials secret created on the Lightning AI platform.
75
75
  cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
76
76
  Required if the registry is part of a cloud provider (e.g. ECR).
77
+ entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
78
+ just runs the provided command in a standard shell if a command is provided.
79
+ If no command is provided, it will run the pre-defined entrypoint of the provided image.
80
+ To use the pre-defined entrypoint of the provided image with a specified command,
81
+ set this to an empty string.
82
+ Only applicable when submitting docker jobs.
83
+ To use the pre-defined entrypoint of the provided image, set this to an empty string.
84
+ Only applicable when submitting docker jobs.
77
85
  artifacts_local: The path of inside the docker container, you want to persist images from.
78
86
  CAUTION: When setting this to "/", it will effectively erase your container.
79
87
  Only supported for jobs with a docker image compute environment.
@@ -1,8 +1,6 @@
1
1
  from dataclasses import dataclass
2
2
  from datetime import datetime
3
- from typing import List, Optional
4
-
5
- import pandas as pd
3
+ from typing import Any, Dict, List, Optional
6
4
 
7
5
  from lightning_sdk.api.k8s_api import K8sClusterApi
8
6
  from lightning_sdk.api.utils import to_iso_z
@@ -34,25 +32,26 @@ class K8sCluster:
34
32
  self._cloud_account = cloud_account
35
33
  self._k8s_cluster = K8sClusterApi(cloud_account=cloud_account)
36
34
 
37
- def _convert_to_k8s_usage_response(self, df: pd.DataFrame) -> K8sUsageResponse:
38
- """Converts a DataFrame to K8sUsageResponse.
35
+ def _convert_to_k8s_usage_response(self, data: List[Dict[str, Any]]) -> K8sUsageResponse:
36
+ """Converts a list of usage data to K8sUsageResponse.
39
37
 
40
38
  Args:
41
- df (pd.DataFrame): The DataFrame containing GPU usage data.
39
+ data (List[Dict[str, Any]]): The list of dictionaries containing GPU usage data.
42
40
 
43
41
  Returns:
44
42
  K8sUsageResponse: The converted response containing hourly usage and total usage.
45
43
  """
46
- if df.empty:
44
+ if not data:
47
45
  return K8sUsageResponse(hours=[], total_usage=0.0)
48
- # Convert each row of the DataFrame to HourlyUsage
46
+
47
+ # Convert each row to HourlyUsage
49
48
  hourly_usage_list: List[HourlyUsage] = [
50
49
  HourlyUsage(time=row["hour"], available_gpus=row["num_gpus"], billed_gpus=row["billed_gpus"])
51
- for _, row in df.iterrows()
50
+ for row in data
52
51
  ]
53
52
 
54
53
  # Calculate total usage (sum of billed GPUs)
55
- total_usage = df["billed_gpus"].sum()
54
+ total_usage = sum(row["billed_gpus"] for row in data)
56
55
 
57
56
  # Create and return the K8sUsageResponse
58
57
  return K8sUsageResponse(hours=hourly_usage_list, total_usage=total_usage)
@@ -1 +1 @@
1
- __version__ = "0.5.70"
1
+ __version__ = "0.6.0"
@@ -67,6 +67,7 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
67
67
  from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
68
68
  from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
69
69
  from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
70
+ from lightning_sdk.lightning_cloud.openapi.api.virtual_machine_service_api import VirtualMachineServiceApi
70
71
  from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi
71
72
  # import ApiClient
72
73
  from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
@@ -132,19 +133,24 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_clo
132
133
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_version_publication_body import CloudSpaceServiceUpdateCloudSpaceVersionPublicationBody
133
134
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_visibility_body import CloudSpaceServiceUpdateCloudSpaceVisibilityBody
134
135
  from lightning_sdk.lightning_cloud.openapi.models.cloudy_service_update_user_cloudy_settings_body import CloudyServiceUpdateUserCloudySettingsBody
136
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_add_container_registry_body import ClusterServiceAddContainerRegistryBody
135
137
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_capacity_reservation_body import ClusterServiceCreateClusterCapacityReservationBody
136
138
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_proxy_body import ClusterServiceCreateClusterProxyBody
137
139
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_usage_restriction_body import ClusterServiceCreateClusterUsageRestrictionBody
138
140
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_machine_body import ClusterServiceCreateMachineBody
141
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_org_cluster_capacity_reservation_body import ClusterServiceCreateOrgClusterCapacityReservationBody
139
142
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project_cluster_body import ClusterServiceCreateProjectClusterBody
140
143
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_server_alert_body import ClusterServiceCreateServerAlertBody
141
144
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_interrupt_server_body import ClusterServiceInterruptServerBody
142
145
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_purchase_capacity_block_body import ClusterServicePurchaseCapacityBlockBody
146
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_refresh_container_registry_credentials_body import ClusterServiceRefreshContainerRegistryCredentialsBody
147
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
143
148
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_server_check_in_body import ClusterServiceServerCheckInBody
144
149
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_sleep_server_body import ClusterServiceSleepServerBody
145
150
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_body import ClusterServiceUpdateClusterBody
146
151
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_usage_restriction_body import ClusterServiceUpdateClusterUsageRestrictionBody
147
152
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_project_cluster_body import ClusterServiceUpdateProjectClusterBody
153
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_validate_container_registry_body import ClusterServiceValidateContainerRegistryBody
148
154
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
149
155
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
150
156
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_service_create_data_connection_body import DataConnectionServiceCreateDataConnectionBody
@@ -210,9 +216,11 @@ from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit
210
216
  from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit_dataset_version_body import LitDatasetServiceUpdateLitDatasetVersionBody
211
217
  from lightning_sdk.lightning_cloud.openapi.models.lit_dataset_service_update_lit_dataset_visibility_body import LitDatasetServiceUpdateLitDatasetVisibilityBody
212
218
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_append_logger_metrics_body import LitLoggerServiceAppendLoggerMetricsBody
219
+ from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_lit_logger_media_body import LitLoggerServiceCreateLitLoggerMediaBody
213
220
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_logger_artifact_body import LitLoggerServiceCreateLoggerArtifactBody
214
221
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_create_metrics_stream_body import LitLoggerServiceCreateMetricsStreamBody
215
222
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_delete_metrics_stream_body import LitLoggerServiceDeleteMetricsStreamBody
223
+ from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_lit_logger_media_body import LitLoggerServiceUpdateLitLoggerMediaBody
216
224
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_metrics_stream_body import LitLoggerServiceUpdateMetricsStreamBody
217
225
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_metrics_stream_visibility_body import LitLoggerServiceUpdateMetricsStreamVisibilityBody
218
226
  from lightning_sdk.lightning_cloud.openapi.models.lit_logger_service_update_shared_metrics_stream_body import LitLoggerServiceUpdateSharedMetricsStreamBody
@@ -295,6 +303,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import
295
303
  from lightning_sdk.lightning_cloud.openapi.models.v1_abort_storage_transfer_response import V1AbortStorageTransferResponse
296
304
  from lightning_sdk.lightning_cloud.openapi.models.v1_accelerator_quota_info import V1AcceleratorQuotaInfo
297
305
  from lightning_sdk.lightning_cloud.openapi.models.v1_ack_user_storage_violation_response import V1AckUserStorageViolationResponse
306
+ from lightning_sdk.lightning_cloud.openapi.models.v1_add_container_registry_response import V1AddContainerRegistryResponse
298
307
  from lightning_sdk.lightning_cloud.openapi.models.v1_add_job_timing_response import V1AddJobTimingResponse
299
308
  from lightning_sdk.lightning_cloud.openapi.models.v1_affiliate_link import V1AffiliateLink
300
309
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_complete_part_response import V1AgentCompletePartResponse
@@ -303,7 +312,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V
303
312
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
304
313
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
305
314
  from lightning_sdk.lightning_cloud.openapi.models.v1_aggregated_pod_metrics import V1AggregatedPodMetrics
306
- from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
307
315
  from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
308
316
  from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
309
317
  from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
@@ -336,6 +344,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance
336
344
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_running_cloud_space_instance_transfer_response import V1CancelRunningCloudSpaceInstanceTransferResponse
337
345
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
338
346
  from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
347
+ from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_reservation_used_by import V1CapacityReservationUsedBy
339
348
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
340
349
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_response import V1CheckClusterNameAvailabilityResponse
341
350
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_external_service_status_response import V1CheckExternalServiceStatusResponse
@@ -412,6 +421,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1Com
412
421
  from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
413
422
  from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
414
423
  from lightning_sdk.lightning_cloud.openapi.models.v1_container_metrics import V1ContainerMetrics
424
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry import V1ContainerRegistry
425
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_info import V1ContainerRegistryInfo
426
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_integration import V1ContainerRegistryIntegration
427
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_status import V1ContainerRegistryStatus
415
428
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
416
429
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
417
430
  from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -437,6 +450,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_incident_request imp
437
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V1CreateJobRequest
438
451
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_license_request import V1CreateLicenseRequest
439
452
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
453
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_logger_media_response import V1CreateLitLoggerMediaResponse
440
454
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
441
455
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
442
456
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_machine_response import V1CreateMachineResponse
@@ -444,6 +458,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_res
444
458
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_model_metrics_response import V1CreateModelMetricsResponse
445
459
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
446
460
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
461
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_org_cluster_capacity_reservation_response import V1CreateOrgClusterCapacityReservationResponse
447
462
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
448
463
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_pipeline_template_request import V1CreatePipelineTemplateRequest
449
464
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
@@ -457,6 +472,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection
457
472
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_request import V1CreateSubscriptionCheckoutSessionRequest
458
473
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_response import V1CreateSubscriptionCheckoutSessionResponse
459
474
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_user_secret_request import V1CreateUserSecretRequest
475
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cudo_direct_v1 import V1CudoDirectV1
460
476
  from lightning_sdk.lightning_cloud.openapi.models.v1_daily_model_metrics import V1DailyModelMetrics
461
477
  from lightning_sdk.lightning_cloud.openapi.models.v1_daily_usage import V1DailyUsage
462
478
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1DataConnection
@@ -481,6 +497,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_k
481
497
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
482
498
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
483
499
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restriction_response import V1DeleteClusterUsageRestrictionResponse
500
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_container_registry_response import V1DeleteContainerRegistryResponse
484
501
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
485
502
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
486
503
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
@@ -501,6 +518,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instanc
501
518
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
502
519
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_response import V1DeleteLitDatasetResponse
503
520
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_version_response import V1DeleteLitDatasetVersionResponse
521
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_logger_media_response import V1DeleteLitLoggerMediaResponse
504
522
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
505
523
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_registry_repository_image_artifact_version_by_digest_response import V1DeleteLitRegistryRepositoryImageArtifactVersionByDigestResponse
506
524
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
@@ -530,6 +548,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response impo
530
548
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
531
549
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_studio_job_response import V1DeleteStudioJobResponse
532
550
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_response import V1DeleteUserSLURMJobResponse
551
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_virtual_machine_response import V1DeleteVirtualMachineResponse
533
552
  from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state import V1DependencyCacheState
534
553
  from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
535
554
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
@@ -569,6 +588,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
569
588
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
570
589
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
571
590
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
591
+ from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config import V1ECRRegistryConfig
592
+ from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config_input import V1ECRRegistryConfigInput
593
+ from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_details import V1ECRRegistryDetails
572
594
  from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
573
595
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
574
596
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
@@ -632,6 +654,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_tele
632
654
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
633
655
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
634
656
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_response import V1GetJobSystemMetricsResponse
657
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_kubernetes_pod_logs_response import V1GetKubernetesPodLogsResponse
658
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_kubernetes_pod_response import V1GetKubernetesPodResponse
635
659
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_latest_model_metrics_response import V1GetLatestModelMetricsResponse
636
660
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
637
661
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
@@ -641,7 +665,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_files_url_r
641
665
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_page_response import V1GetLitPageResponse
642
666
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_logger_metrics_response import V1GetLoggerMetricsResponse
643
667
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in_cloud_space_response import V1GetLongRunningCommandInCloudSpaceResponse
644
- from lightning_sdk.lightning_cloud.openapi.models.v1_get_machine_response import V1GetMachineResponse
645
668
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_market_pricing_response import V1GetMarketPricingResponse
646
669
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
647
670
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
@@ -702,6 +725,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTimi
702
725
  from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
703
726
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
704
727
  from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_indexes import V1K8sIncidentIndexes
728
+ from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_setting import V1K8sIncidentSetting
729
+ from lightning_sdk.lightning_cloud.openapi.models.v1_k8s_incident_type import V1K8sIncidentType
705
730
  from lightning_sdk.lightning_cloud.openapi.models.v1_kai_scheduler_queue_metrics import V1KaiSchedulerQueueMetrics
706
731
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
707
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
@@ -710,8 +735,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_settings_
710
735
  from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
711
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
712
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod import V1KubernetesPod
738
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_pod_logs_page import V1KubernetesPodLogsPage
713
739
  from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template import V1KubernetesTemplate
714
740
  from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
741
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubevirt_config import V1KubevirtConfig
715
742
  from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
716
743
  from lightning_sdk.lightning_cloud.openapi.models.v1_license import V1License
717
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
@@ -764,6 +791,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_respon
764
791
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
765
792
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
766
793
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_metrics_response import V1ListContainerMetricsResponse
794
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_registries_response import V1ListContainerRegistriesResponse
767
795
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversation_message_actions_response import V1ListConversationMessageActionsResponse
768
796
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversations_response import V1ListConversationsResponse
769
797
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifacts_response import V1ListDataConnectionArtifactsResponse
@@ -783,7 +811,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_respon
783
811
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
784
812
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
785
813
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
786
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
814
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_response import V1ListFilesystemMMTsResponse
787
815
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_metrics_response import V1ListFilesystemMetricsResponse
788
816
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
789
817
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
@@ -802,6 +830,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_sort_order import
802
830
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations_response import V1ListJoinableOrganizationsResponse
803
831
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_kai_scheduler_queues_metrics_response import V1ListKaiSchedulerQueuesMetricsResponse
804
832
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_pods_response import V1ListKubernetesPodsResponse
833
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_pods_sort_order import V1ListKubernetesPodsSortOrder
805
834
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
806
835
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_license_response import V1ListLicenseResponse
807
836
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
@@ -812,6 +841,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_r
812
841
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
813
842
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_dataset_versions_response import V1ListLitDatasetVersionsResponse
814
843
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response import V1ListLitDatasetsResponse
844
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_logger_media_response import V1ListLitLoggerMediaResponse
815
845
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
816
846
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
817
847
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
@@ -856,9 +886,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response impor
856
886
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_storage_transfers_response import V1ListStorageTransfersResponse
857
887
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
858
888
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
889
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_virtual_machines_response import V1ListVirtualMachinesResponse
859
890
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
860
891
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_file import V1LitDatasetFile
861
892
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_version_archive import V1LitDatasetVersionArchive
893
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_logger_media import V1LitLoggerMedia
862
894
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
863
895
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
864
896
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
@@ -883,6 +915,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1M
883
915
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model import V1ManagedModel
884
916
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model_abilities import V1ManagedModelAbilities
885
917
  from lightning_sdk.lightning_cloud.openapi.models.v1_market_price import V1MarketPrice
918
+ from lightning_sdk.lightning_cloud.openapi.models.v1_media_type import V1MediaType
886
919
  from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Membership
887
920
  from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
888
921
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_action import V1MessageAction
@@ -981,6 +1014,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
981
1014
  from lightning_sdk.lightning_cloud.openapi.models.v1_quote_annual_upsell_response import V1QuoteAnnualUpsellResponse
982
1015
  from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
983
1016
  from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
1017
+ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_container_registry_credentials_response import V1RefreshContainerRegistryCredentialsResponse
984
1018
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
985
1019
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
986
1020
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -988,12 +1022,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
988
1022
  from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
989
1023
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
990
1024
  from lightning_sdk.lightning_cloud.openapi.models.v1_render_kubernetes_template_response import V1RenderKubernetesTemplateResponse
1025
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_idle_state_response import V1ReportCloudSpaceInstanceIdleStateResponse
991
1026
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
992
1027
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
993
1028
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_deployment_routing_telemetry_response import V1ReportDeploymentRoutingTelemetryResponse
994
1029
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_job_system_metrics_response import V1ReportJobSystemMetricsResponse
995
1030
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
996
1031
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
1032
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_machine_system_metrics_response import V1ReportMachineSystemMetricsResponse
997
1033
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
998
1034
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cloud_space_access_response import V1RequestCloudSpaceAccessResponse
999
1035
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
@@ -1074,6 +1110,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1Sys
1074
1110
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated import V1SystemMetricsAggregated
1075
1111
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
1076
1112
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
1113
+ from lightning_sdk.lightning_cloud.openapi.models.v1_tenant_credentials import V1TenantCredentials
1077
1114
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
1078
1115
  from lightning_sdk.lightning_cloud.openapi.models.v1_token_login_request import V1TokenLoginRequest
1079
1116
  from lightning_sdk.lightning_cloud.openapi.models.v1_token_login_response import V1TokenLoginResponse
@@ -1141,6 +1178,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_requested_flow_compute
1141
1178
  from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_response import V1UserSLURMJobActionResponse
1142
1179
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
1143
1180
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
1181
+ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_container_registry_response import V1ValidateContainerRegistryResponse
1144
1182
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
1145
1183
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
1146
1184
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse
@@ -1150,11 +1188,16 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_r
1150
1188
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse
1151
1189
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_storage_transfer_response import V1ValidateStorageTransferResponse
1152
1190
  from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_response import V1VerifyVerificationResponse
1191
+ from lightning_sdk.lightning_cloud.openapi.models.v1_virtual_machine import V1VirtualMachine
1192
+ from lightning_sdk.lightning_cloud.openapi.models.v1_vm_configuration import V1VmConfiguration
1193
+ from lightning_sdk.lightning_cloud.openapi.models.v1_vm_provider_configuration import V1VmProviderConfiguration
1153
1194
  from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
1154
1195
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
1155
1196
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume_state import V1VolumeState
1156
1197
  from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
1157
1198
  from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
1158
1199
  from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
1200
+ from lightning_sdk.lightning_cloud.openapi.models.virtual_machine_service_create_virtual_machine_body import VirtualMachineServiceCreateVirtualMachineBody
1201
+ from lightning_sdk.lightning_cloud.openapi.models.virtual_machine_service_update_virtual_machine_body import VirtualMachineServiceUpdateVirtualMachineBody
1159
1202
  from lightning_sdk.lightning_cloud.openapi.models.volume_service_update_volume_body import VolumeServiceUpdateVolumeBody
1160
1203
  from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec as Gridv1ImageSpec
@@ -48,4 +48,5 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
48
48
  from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
49
49
  from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
50
50
  from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
51
+ from lightning_sdk.lightning_cloud.openapi.api.virtual_machine_service_api import VirtualMachineServiceApi
51
52
  from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi
@@ -3073,6 +3073,7 @@ class CloudSpaceServiceApi(object):
3073
3073
  :param bool include_download_url:
3074
3074
  :param str cluster_id:
3075
3075
  :param bool local_index: if true, fetch from the local index
3076
+ :param bool include_folder_index: if true, include the folder index in the response
3076
3077
  :return: V1GetArtifactsPageResponse
3077
3078
  If the method is called asynchronously,
3078
3079
  returns the request thread.
@@ -3100,12 +3101,13 @@ class CloudSpaceServiceApi(object):
3100
3101
  :param bool include_download_url:
3101
3102
  :param str cluster_id:
3102
3103
  :param bool local_index: if true, fetch from the local index
3104
+ :param bool include_folder_index: if true, include the folder index in the response
3103
3105
  :return: V1GetArtifactsPageResponse
3104
3106
  If the method is called asynchronously,
3105
3107
  returns the request thread.
3106
3108
  """
3107
3109
 
3108
- all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index'] # noqa: E501
3110
+ all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index'] # noqa: E501
3109
3111
  all_params.append('async_req')
3110
3112
  all_params.append('_return_http_data_only')
3111
3113
  all_params.append('_preload_content')
@@ -3148,6 +3150,8 @@ class CloudSpaceServiceApi(object):
3148
3150
  query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
3149
3151
  if 'local_index' in params:
3150
3152
  query_params.append(('localIndex', params['local_index'])) # noqa: E501
3153
+ if 'include_folder_index' in params:
3154
+ query_params.append(('includeFolderIndex', params['include_folder_index'])) # noqa: E501
3151
3155
 
3152
3156
  header_params = {}
3153
3157
 
@@ -7538,6 +7542,119 @@ class CloudSpaceServiceApi(object):
7538
7542
  _request_timeout=params.get('_request_timeout'),
7539
7543
  collection_formats=collection_formats)
7540
7544
 
7545
+ def cloud_space_service_report_cloud_space_instance_idle_state(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceIdleStateResponse': # noqa: E501
7546
+ """cloud_space_service_report_cloud_space_instance_idle_state # noqa: E501
7547
+
7548
+ This method makes a synchronous HTTP request by default. To make an
7549
+ asynchronous HTTP request, please pass async_req=True
7550
+ >>> thread = api.cloud_space_service_report_cloud_space_instance_idle_state(project_id, cloudspace_id, async_req=True)
7551
+ >>> result = thread.get()
7552
+
7553
+ :param async_req bool
7554
+ :param str project_id: (required)
7555
+ :param str cloudspace_id: (required)
7556
+ :param str cloudspace_instance_id:
7557
+ :param bool is_idle:
7558
+ :param datetime timestamp:
7559
+ :return: V1ReportCloudSpaceInstanceIdleStateResponse
7560
+ If the method is called asynchronously,
7561
+ returns the request thread.
7562
+ """
7563
+ kwargs['_return_http_data_only'] = True
7564
+ if kwargs.get('async_req'):
7565
+ return self.cloud_space_service_report_cloud_space_instance_idle_state_with_http_info(project_id, cloudspace_id, **kwargs) # noqa: E501
7566
+ else:
7567
+ (data) = self.cloud_space_service_report_cloud_space_instance_idle_state_with_http_info(project_id, cloudspace_id, **kwargs) # noqa: E501
7568
+ return data
7569
+
7570
+ def cloud_space_service_report_cloud_space_instance_idle_state_with_http_info(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceIdleStateResponse': # noqa: E501
7571
+ """cloud_space_service_report_cloud_space_instance_idle_state # noqa: E501
7572
+
7573
+ This method makes a synchronous HTTP request by default. To make an
7574
+ asynchronous HTTP request, please pass async_req=True
7575
+ >>> thread = api.cloud_space_service_report_cloud_space_instance_idle_state_with_http_info(project_id, cloudspace_id, async_req=True)
7576
+ >>> result = thread.get()
7577
+
7578
+ :param async_req bool
7579
+ :param str project_id: (required)
7580
+ :param str cloudspace_id: (required)
7581
+ :param str cloudspace_instance_id:
7582
+ :param bool is_idle:
7583
+ :param datetime timestamp:
7584
+ :return: V1ReportCloudSpaceInstanceIdleStateResponse
7585
+ If the method is called asynchronously,
7586
+ returns the request thread.
7587
+ """
7588
+
7589
+ all_params = ['project_id', 'cloudspace_id', 'cloudspace_instance_id', 'is_idle', 'timestamp'] # noqa: E501
7590
+ all_params.append('async_req')
7591
+ all_params.append('_return_http_data_only')
7592
+ all_params.append('_preload_content')
7593
+ all_params.append('_request_timeout')
7594
+
7595
+ params = locals()
7596
+ for key, val in six.iteritems(params['kwargs']):
7597
+ if key not in all_params:
7598
+ raise TypeError(
7599
+ "Got an unexpected keyword argument '%s'"
7600
+ " to method cloud_space_service_report_cloud_space_instance_idle_state" % key
7601
+ )
7602
+ params[key] = val
7603
+ del params['kwargs']
7604
+ # verify the required parameter 'project_id' is set
7605
+ if ('project_id' not in params or
7606
+ params['project_id'] is None):
7607
+ raise ValueError("Missing the required parameter `project_id` when calling `cloud_space_service_report_cloud_space_instance_idle_state`") # noqa: E501
7608
+ # verify the required parameter 'cloudspace_id' is set
7609
+ if ('cloudspace_id' not in params or
7610
+ params['cloudspace_id'] is None):
7611
+ raise ValueError("Missing the required parameter `cloudspace_id` when calling `cloud_space_service_report_cloud_space_instance_idle_state`") # noqa: E501
7612
+
7613
+ collection_formats = {}
7614
+
7615
+ path_params = {}
7616
+ if 'project_id' in params:
7617
+ path_params['projectId'] = params['project_id'] # noqa: E501
7618
+ if 'cloudspace_id' in params:
7619
+ path_params['cloudspaceId'] = params['cloudspace_id'] # noqa: E501
7620
+
7621
+ query_params = []
7622
+ if 'cloudspace_instance_id' in params:
7623
+ query_params.append(('cloudspaceInstanceId', params['cloudspace_instance_id'])) # noqa: E501
7624
+ if 'is_idle' in params:
7625
+ query_params.append(('isIdle', params['is_idle'])) # noqa: E501
7626
+ if 'timestamp' in params:
7627
+ query_params.append(('timestamp', params['timestamp'])) # noqa: E501
7628
+
7629
+ header_params = {}
7630
+
7631
+ form_params = []
7632
+ local_var_files = {}
7633
+
7634
+ body_params = None
7635
+ # HTTP header `Accept`
7636
+ header_params['Accept'] = self.api_client.select_header_accept(
7637
+ ['application/json']) # noqa: E501
7638
+
7639
+ # Authentication setting
7640
+ auth_settings = [] # noqa: E501
7641
+
7642
+ return self.api_client.call_api(
7643
+ '/v1/projects/{projectId}/cloudspaces/{cloudspaceId}/report-idle-state', 'GET',
7644
+ path_params,
7645
+ query_params,
7646
+ header_params,
7647
+ body=body_params,
7648
+ post_params=form_params,
7649
+ files=local_var_files,
7650
+ response_type='V1ReportCloudSpaceInstanceIdleStateResponse', # noqa: E501
7651
+ auth_settings=auth_settings,
7652
+ async_req=params.get('async_req'),
7653
+ _return_http_data_only=params.get('_return_http_data_only'),
7654
+ _preload_content=params.get('_preload_content', True),
7655
+ _request_timeout=params.get('_request_timeout'),
7656
+ collection_formats=collection_formats)
7657
+
7541
7658
  def cloud_space_service_report_cloud_space_instance_stop_at(self, project_id: 'str', cloudspace_id: 'str', **kwargs) -> 'V1ReportCloudSpaceInstanceStopAtResponse': # noqa: E501
7542
7659
  """cloud_space_service_report_cloud_space_instance_stop_at # noqa: E501
7543
7660