lightning-sdk 2025.9.16__py3-none-any.whl → 2025.9.29__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 (78) hide show
  1. lightning_sdk/__init__.py +4 -3
  2. lightning_sdk/api/cloud_account_api.py +12 -1
  3. lightning_sdk/api/job_api.py +12 -11
  4. lightning_sdk/api/mmt_api.py +1 -1
  5. lightning_sdk/api/studio_api.py +1 -1
  6. lightning_sdk/api/teamspace_api.py +18 -0
  7. lightning_sdk/api/user_api.py +8 -2
  8. lightning_sdk/cli/entrypoint.py +3 -1
  9. lightning_sdk/cli/groups.py +8 -1
  10. lightning_sdk/cli/legacy/entrypoint.py +1 -1
  11. lightning_sdk/cli/studio/create.py +19 -5
  12. lightning_sdk/cli/studio/delete.py +9 -5
  13. lightning_sdk/cli/studio/list.py +5 -1
  14. lightning_sdk/cli/studio/ssh.py +9 -3
  15. lightning_sdk/cli/studio/start.py +26 -3
  16. lightning_sdk/cli/studio/stop.py +7 -3
  17. lightning_sdk/cli/studio/switch.py +21 -5
  18. lightning_sdk/cli/utils/owner_selection.py +110 -0
  19. lightning_sdk/cli/utils/studio_selection.py +22 -15
  20. lightning_sdk/cli/utils/teamspace_selection.py +63 -62
  21. lightning_sdk/cli/vm/__init__.py +20 -0
  22. lightning_sdk/cli/vm/create.py +33 -0
  23. lightning_sdk/cli/vm/delete.py +25 -0
  24. lightning_sdk/cli/vm/list.py +30 -0
  25. lightning_sdk/cli/vm/ssh.py +31 -0
  26. lightning_sdk/cli/vm/start.py +60 -0
  27. lightning_sdk/cli/vm/stop.py +25 -0
  28. lightning_sdk/cli/vm/switch.py +38 -0
  29. lightning_sdk/lightning_cloud/openapi/__init__.py +20 -1
  30. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +2 -95
  31. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +24 -8
  32. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +420 -0
  33. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +121 -0
  34. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +655 -0
  35. lightning_sdk/lightning_cloud/openapi/models/__init__.py +20 -1
  36. lightning_sdk/lightning_cloud/openapi/models/create_machine_request_represents_the_request_to_create_a_machine.py +435 -0
  37. lightning_sdk/lightning_cloud/openapi/models/job_id_reportroutingtelemetry_body.py +123 -0
  38. lightning_sdk/lightning_cloud/openapi/models/project_id_storagetransfers_body.py +149 -0
  39. lightning_sdk/lightning_cloud/openapi/models/user_id_affiliatelinks_body.py +107 -3
  40. lightning_sdk/lightning_cloud/openapi/models/v1_abort_storage_transfer_response.py +97 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_session_daily_aggregated.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +2 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +1 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_machine_response.py +123 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_delete_machine_response.py +97 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_get_machine_response.py +123 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_elastic_cluster_v1.py +97 -0
  54. lightning_sdk/lightning_cloud/openapi/models/{v1_get_model_total_usage_metrics_response.py → v1_list_machines_response.py} +37 -37
  55. lightning_sdk/lightning_cloud/openapi/models/v1_list_storage_transfers_response.py +123 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_machine.py +539 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_machine_direct_v1.py +123 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_pause_storage_transfer_response.py +97 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_purchase_annual_upsell_request.py +123 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_report_deployment_routing_telemetry_response.py +97 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_resume_storage_transfer_response.py +97 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +79 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_slack_notifier.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_slack_notifier_type.py +105 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_storage_transfer.py +435 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_storage_transfer_status.py +108 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +105 -79
  70. lightning_sdk/machine.py +16 -1
  71. lightning_sdk/studio.py +55 -11
  72. lightning_sdk/teamspace.py +65 -2
  73. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/METADATA +1 -1
  74. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/RECORD +78 -50
  75. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/LICENSE +0 -0
  76. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/WHEEL +0 -0
  77. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/entry_points.txt +0 -0
  78. {lightning_sdk-2025.9.16.dist-info → lightning_sdk-2025.9.29.dist-info}/top_level.txt +0 -0
@@ -107,6 +107,7 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body1 import
107
107
  from lightning_sdk.lightning_cloud.openapi.models.create import Create
108
108
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
109
109
  from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
110
+ from lightning_sdk.lightning_cloud.openapi.models.create_machine_request_represents_the_request_to_create_a_machine import CreateMachineRequestRepresentsTheRequestToCreateAMachine
110
111
  from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body import CreditsAutoreplenishBody
111
112
  from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body1 import CreditsAutoreplenishBody1
112
113
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
@@ -156,6 +157,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploa
156
157
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
157
158
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
158
159
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdVisibilityBody2
160
+ from lightning_sdk.lightning_cloud.openapi.models.job_id_reportroutingtelemetry_body import JobIdReportroutingtelemetryBody
159
161
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
160
162
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
161
163
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -207,6 +209,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body impo
207
209
  from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
208
210
  from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
209
211
  from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import ProjectIdStorageBody
212
+ from lightning_sdk.lightning_cloud.openapi.models.project_id_storagetransfers_body import ProjectIdStoragetransfersBody
210
213
  from lightning_sdk.lightning_cloud.openapi.models.project_tab_management_messages import ProjectTabManagementMessages
211
214
  from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
212
215
  from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body import ProjectsProjectIdBody
@@ -246,6 +249,7 @@ from lightning_sdk.lightning_cloud.openapi.models.user_user_id_body import UserU
246
249
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_cluster_credentials import V1AWSClusterCredentials
247
250
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1 import V1AWSDirectV1
248
251
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import V1AWSDirectV1Status
252
+ from lightning_sdk.lightning_cloud.openapi.models.v1_abort_storage_transfer_response import V1AbortStorageTransferResponse
249
253
  from lightning_sdk.lightning_cloud.openapi.models.v1_accelerator_quota_info import V1AcceleratorQuotaInfo
250
254
  from lightning_sdk.lightning_cloud.openapi.models.v1_ack_user_storage_violation_response import V1AckUserStorageViolationResponse
251
255
  from lightning_sdk.lightning_cloud.openapi.models.v1_add_job_timing_response import V1AddJobTimingResponse
@@ -388,6 +392,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V
388
392
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
389
393
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
390
394
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
395
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_machine_response import V1CreateMachineResponse
391
396
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
392
397
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_model_metrics_response import V1CreateModelMetricsResponse
393
398
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
@@ -448,6 +453,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response im
448
453
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_registry_repository_image_artifact_version_by_digest_response import V1DeleteLitRegistryRepositoryImageArtifactVersionByDigestResponse
449
454
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
450
455
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
456
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_machine_response import V1DeleteMachineResponse
451
457
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_endpoint_response import V1DeleteManagedEndpointResponse
452
458
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
453
459
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
@@ -579,12 +585,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_files_url_r
579
585
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_page_response import V1GetLitPageResponse
580
586
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_logger_metrics_response import V1GetLoggerMetricsResponse
581
587
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in_cloud_space_response import V1GetLongRunningCommandInCloudSpaceResponse
588
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_machine_response import V1GetMachineResponse
582
589
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_response import V1GetModelFileUploadUrlsResponse
583
590
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
584
591
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
585
592
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
586
593
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_metrics_response import V1GetModelMetricsResponse
587
- from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_total_usage_metrics_response import V1GetModelTotalUsageMetricsResponse
588
594
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
589
595
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
590
596
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
@@ -645,6 +651,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 impor
645
651
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
646
652
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
647
653
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_basic_auth import V1LightningBasicAuth
654
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_elastic_cluster_v1 import V1LightningElasticClusterV1
648
655
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_o_auth_auth import V1LightningOAuthAuth
649
656
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_run import V1LightningRun
650
657
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightningapp_instance_artifact import V1LightningappInstanceArtifact
@@ -731,6 +738,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response
731
738
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
732
739
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
733
740
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
741
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_machines_response import V1ListMachinesResponse
734
742
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
735
743
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
736
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
@@ -768,6 +776,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_respon
768
776
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedule_runs_response import V1ListScheduleRunsResponse
769
777
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
770
778
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
779
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_storage_transfers_response import V1ListStorageTransfersResponse
771
780
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
772
781
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
773
782
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
@@ -788,6 +797,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_logout_request import V1Log
788
797
  from lightning_sdk.lightning_cloud.openapi.models.v1_logout_response import V1LogoutResponse
789
798
  from lightning_sdk.lightning_cloud.openapi.models.v1_logs_response import V1LogsResponse
790
799
  from lightning_sdk.lightning_cloud.openapi.models.v1_lustre_data_connection import V1LustreDataConnection
800
+ from lightning_sdk.lightning_cloud.openapi.models.v1_machine import V1Machine
801
+ from lightning_sdk.lightning_cloud.openapi.models.v1_machine_direct_v1 import V1MachineDirectV1
791
802
  from lightning_sdk.lightning_cloud.openapi.models.v1_machines_selector import V1MachinesSelector
792
803
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request import V1MagicLinkLoginRequest
793
804
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
@@ -840,6 +851,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1Pa
840
851
  from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
841
852
  from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathMapping
842
853
  from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
854
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pause_storage_transfer_response import V1PauseStorageTransferResponse
843
855
  from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
844
856
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline import V1Pipeline
845
857
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter import V1PipelineParameter
@@ -875,6 +887,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_project_tab import V1Projec
875
887
  from lightning_sdk.lightning_cloud.openapi.models.v1_prompt_suggestion import V1PromptSuggestion
876
888
  from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_response import V1PublishCloudSpaceResponse
877
889
  from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
890
+ from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_request import V1PurchaseAnnualUpsellRequest
878
891
  from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_annual_upsell_response import V1PurchaseAnnualUpsellResponse
879
892
  from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
880
893
  from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
@@ -897,6 +910,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer impo
897
910
  from lightning_sdk.lightning_cloud.openapi.models.v1_render_kubernetes_template_response import V1RenderKubernetesTemplateResponse
898
911
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
899
912
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
913
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_deployment_routing_telemetry_response import V1ReportDeploymentRoutingTelemetryResponse
900
914
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
901
915
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
902
916
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
@@ -914,6 +928,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta impor
914
928
  from lightning_sdk.lightning_cloud.openapi.models.v1_restart_cloud_space_instance_response import V1RestartCloudSpaceInstanceResponse
915
929
  from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1RestartTiming
916
930
  from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
931
+ from lightning_sdk.lightning_cloud.openapi.models.v1_resume_storage_transfer_response import V1ResumeStorageTransferResponse
917
932
  from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
918
933
  from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
919
934
  from lightning_sdk.lightning_cloud.openapi.models.v1_routing_telemetry import V1RoutingTelemetry
@@ -947,6 +962,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
947
962
  from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
948
963
  from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
949
964
  from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
965
+ from lightning_sdk.lightning_cloud.openapi.models.v1_slack_notifier import V1SlackNotifier
966
+ from lightning_sdk.lightning_cloud.openapi.models.v1_slack_notifier_type import V1SlackNotifierType
950
967
  from lightning_sdk.lightning_cloud.openapi.models.v1_sleep_server_response import V1SleepServerResponse
951
968
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
952
969
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
@@ -961,6 +978,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_stop_cloud_space_instance_r
961
978
  from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset import V1StorageAsset
962
979
  from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V1StorageAssetType
963
980
  from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
981
+ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_transfer import V1StorageTransfer
982
+ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_transfer_status import V1StorageTransferStatus
964
983
  from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
965
984
  from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
966
985
  from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec
@@ -1058,7 +1058,7 @@ class AssistantsServiceApi(object):
1058
1058
  collection_formats=collection_formats)
1059
1059
 
1060
1060
  def assistants_service_get_latest_model_metrics(self, **kwargs) -> 'V1GetLatestModelMetricsResponse': # noqa: E501
1061
- """assistants_service_get_latest_model_metrics # noqa: E501
1061
+ """model metrics # noqa: E501
1062
1062
 
1063
1063
  This method makes a synchronous HTTP request by default. To make an
1064
1064
  asynchronous HTTP request, please pass async_req=True
@@ -1079,7 +1079,7 @@ class AssistantsServiceApi(object):
1079
1079
  return data
1080
1080
 
1081
1081
  def assistants_service_get_latest_model_metrics_with_http_info(self, **kwargs) -> 'V1GetLatestModelMetricsResponse': # noqa: E501
1082
- """assistants_service_get_latest_model_metrics # noqa: E501
1082
+ """model metrics # noqa: E501
1083
1083
 
1084
1084
  This method makes a synchronous HTTP request by default. To make an
1085
1085
  asynchronous HTTP request, please pass async_req=True
@@ -1570,99 +1570,6 @@ class AssistantsServiceApi(object):
1570
1570
  _request_timeout=params.get('_request_timeout'),
1571
1571
  collection_formats=collection_formats)
1572
1572
 
1573
- def assistants_service_get_model_total_usage_metrics(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse': # noqa: E501
1574
- """model metrics # noqa: E501
1575
-
1576
- This method makes a synchronous HTTP request by default. To make an
1577
- asynchronous HTTP request, please pass async_req=True
1578
- >>> thread = api.assistants_service_get_model_total_usage_metrics(model_id, async_req=True)
1579
- >>> result = thread.get()
1580
-
1581
- :param async_req bool
1582
- :param str model_id: (required)
1583
- :return: V1GetModelTotalUsageMetricsResponse
1584
- If the method is called asynchronously,
1585
- returns the request thread.
1586
- """
1587
- kwargs['_return_http_data_only'] = True
1588
- if kwargs.get('async_req'):
1589
- return self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1590
- else:
1591
- (data) = self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1592
- return data
1593
-
1594
- def assistants_service_get_model_total_usage_metrics_with_http_info(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse': # noqa: E501
1595
- """model metrics # noqa: E501
1596
-
1597
- This method makes a synchronous HTTP request by default. To make an
1598
- asynchronous HTTP request, please pass async_req=True
1599
- >>> thread = api.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, async_req=True)
1600
- >>> result = thread.get()
1601
-
1602
- :param async_req bool
1603
- :param str model_id: (required)
1604
- :return: V1GetModelTotalUsageMetricsResponse
1605
- If the method is called asynchronously,
1606
- returns the request thread.
1607
- """
1608
-
1609
- all_params = ['model_id'] # noqa: E501
1610
- all_params.append('async_req')
1611
- all_params.append('_return_http_data_only')
1612
- all_params.append('_preload_content')
1613
- all_params.append('_request_timeout')
1614
-
1615
- params = locals()
1616
- for key, val in six.iteritems(params['kwargs']):
1617
- if key not in all_params:
1618
- raise TypeError(
1619
- "Got an unexpected keyword argument '%s'"
1620
- " to method assistants_service_get_model_total_usage_metrics" % key
1621
- )
1622
- params[key] = val
1623
- del params['kwargs']
1624
- # verify the required parameter 'model_id' is set
1625
- if ('model_id' not in params or
1626
- params['model_id'] is None):
1627
- raise ValueError("Missing the required parameter `model_id` when calling `assistants_service_get_model_total_usage_metrics`") # noqa: E501
1628
-
1629
- collection_formats = {}
1630
-
1631
- path_params = {}
1632
- if 'model_id' in params:
1633
- path_params['modelId'] = params['model_id'] # noqa: E501
1634
-
1635
- query_params = []
1636
-
1637
- header_params = {}
1638
-
1639
- form_params = []
1640
- local_var_files = {}
1641
-
1642
- body_params = None
1643
- # HTTP header `Accept`
1644
- header_params['Accept'] = self.api_client.select_header_accept(
1645
- ['application/json']) # noqa: E501
1646
-
1647
- # Authentication setting
1648
- auth_settings = [] # noqa: E501
1649
-
1650
- return self.api_client.call_api(
1651
- '/v1/agents/{modelId}/total-usage', 'GET',
1652
- path_params,
1653
- query_params,
1654
- header_params,
1655
- body=body_params,
1656
- post_params=form_params,
1657
- files=local_var_files,
1658
- response_type='V1GetModelTotalUsageMetricsResponse', # noqa: E501
1659
- auth_settings=auth_settings,
1660
- async_req=params.get('async_req'),
1661
- _return_http_data_only=params.get('_return_http_data_only'),
1662
- _preload_content=params.get('_preload_content', True),
1663
- _request_timeout=params.get('_request_timeout'),
1664
- collection_formats=collection_formats)
1665
-
1666
1573
  def assistants_service_get_published_managed_endpoint_by_model_id(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint': # noqa: E501
1667
1574
  """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata # noqa: E501
1668
1575
 
@@ -1356,41 +1356,43 @@ class BillingServiceApi(object):
1356
1356
  _request_timeout=params.get('_request_timeout'),
1357
1357
  collection_formats=collection_formats)
1358
1358
 
1359
- def billing_service_purchase_annual_upsell(self, **kwargs) -> 'V1PurchaseAnnualUpsellResponse': # noqa: E501
1359
+ def billing_service_purchase_annual_upsell(self, body: 'V1PurchaseAnnualUpsellRequest', **kwargs) -> 'V1PurchaseAnnualUpsellResponse': # noqa: E501
1360
1360
  """billing_service_purchase_annual_upsell # noqa: E501
1361
1361
 
1362
1362
  This method makes a synchronous HTTP request by default. To make an
1363
1363
  asynchronous HTTP request, please pass async_req=True
1364
- >>> thread = api.billing_service_purchase_annual_upsell(async_req=True)
1364
+ >>> thread = api.billing_service_purchase_annual_upsell(body, async_req=True)
1365
1365
  >>> result = thread.get()
1366
1366
 
1367
1367
  :param async_req bool
1368
+ :param V1PurchaseAnnualUpsellRequest body: (required)
1368
1369
  :return: V1PurchaseAnnualUpsellResponse
1369
1370
  If the method is called asynchronously,
1370
1371
  returns the request thread.
1371
1372
  """
1372
1373
  kwargs['_return_http_data_only'] = True
1373
1374
  if kwargs.get('async_req'):
1374
- return self.billing_service_purchase_annual_upsell_with_http_info(**kwargs) # noqa: E501
1375
+ return self.billing_service_purchase_annual_upsell_with_http_info(body, **kwargs) # noqa: E501
1375
1376
  else:
1376
- (data) = self.billing_service_purchase_annual_upsell_with_http_info(**kwargs) # noqa: E501
1377
+ (data) = self.billing_service_purchase_annual_upsell_with_http_info(body, **kwargs) # noqa: E501
1377
1378
  return data
1378
1379
 
1379
- def billing_service_purchase_annual_upsell_with_http_info(self, **kwargs) -> 'V1PurchaseAnnualUpsellResponse': # noqa: E501
1380
+ def billing_service_purchase_annual_upsell_with_http_info(self, body: 'V1PurchaseAnnualUpsellRequest', **kwargs) -> 'V1PurchaseAnnualUpsellResponse': # noqa: E501
1380
1381
  """billing_service_purchase_annual_upsell # noqa: E501
1381
1382
 
1382
1383
  This method makes a synchronous HTTP request by default. To make an
1383
1384
  asynchronous HTTP request, please pass async_req=True
1384
- >>> thread = api.billing_service_purchase_annual_upsell_with_http_info(async_req=True)
1385
+ >>> thread = api.billing_service_purchase_annual_upsell_with_http_info(body, async_req=True)
1385
1386
  >>> result = thread.get()
1386
1387
 
1387
1388
  :param async_req bool
1389
+ :param V1PurchaseAnnualUpsellRequest body: (required)
1388
1390
  :return: V1PurchaseAnnualUpsellResponse
1389
1391
  If the method is called asynchronously,
1390
1392
  returns the request thread.
1391
1393
  """
1392
1394
 
1393
- all_params = [] # noqa: E501
1395
+ all_params = ['body'] # noqa: E501
1394
1396
  all_params.append('async_req')
1395
1397
  all_params.append('_return_http_data_only')
1396
1398
  all_params.append('_preload_content')
@@ -1405,6 +1407,10 @@ class BillingServiceApi(object):
1405
1407
  )
1406
1408
  params[key] = val
1407
1409
  del params['kwargs']
1410
+ # verify the required parameter 'body' is set
1411
+ if ('body' not in params or
1412
+ params['body'] is None):
1413
+ raise ValueError("Missing the required parameter `body` when calling `billing_service_purchase_annual_upsell`") # noqa: E501
1408
1414
 
1409
1415
  collection_formats = {}
1410
1416
 
@@ -1418,10 +1424,16 @@ class BillingServiceApi(object):
1418
1424
  local_var_files = {}
1419
1425
 
1420
1426
  body_params = None
1427
+ if 'body' in params:
1428
+ body_params = params['body']
1421
1429
  # HTTP header `Accept`
1422
1430
  header_params['Accept'] = self.api_client.select_header_accept(
1423
1431
  ['application/json']) # noqa: E501
1424
1432
 
1433
+ # HTTP header `Content-Type`
1434
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1435
+ ['application/json']) # noqa: E501
1436
+
1425
1437
  # Authentication setting
1426
1438
  auth_settings = [] # noqa: E501
1427
1439
 
@@ -1450,6 +1462,7 @@ class BillingServiceApi(object):
1450
1462
  >>> result = thread.get()
1451
1463
 
1452
1464
  :param async_req bool
1465
+ :param str org_id:
1453
1466
  :return: V1QuoteAnnualUpsellResponse
1454
1467
  If the method is called asynchronously,
1455
1468
  returns the request thread.
@@ -1470,12 +1483,13 @@ class BillingServiceApi(object):
1470
1483
  >>> result = thread.get()
1471
1484
 
1472
1485
  :param async_req bool
1486
+ :param str org_id:
1473
1487
  :return: V1QuoteAnnualUpsellResponse
1474
1488
  If the method is called asynchronously,
1475
1489
  returns the request thread.
1476
1490
  """
1477
1491
 
1478
- all_params = [] # noqa: E501
1492
+ all_params = ['org_id'] # noqa: E501
1479
1493
  all_params.append('async_req')
1480
1494
  all_params.append('_return_http_data_only')
1481
1495
  all_params.append('_preload_content')
@@ -1496,6 +1510,8 @@ class BillingServiceApi(object):
1496
1510
  path_params = {}
1497
1511
 
1498
1512
  query_params = []
1513
+ if 'org_id' in params:
1514
+ query_params.append(('orgId', params['org_id'])) # noqa: E501
1499
1515
 
1500
1516
  header_params = {}
1501
1517