lightning-sdk 0.2.14__py3-none-any.whl → 0.2.16__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 (122) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +79 -0
  3. lightning_sdk/api/cluster_api.py +83 -1
  4. lightning_sdk/api/license_api.py +48 -0
  5. lightning_sdk/api/llm_api.py +73 -12
  6. lightning_sdk/api/studio_api.py +50 -1
  7. lightning_sdk/api/teamspace_api.py +127 -1
  8. lightning_sdk/api/utils.py +4 -0
  9. lightning_sdk/base_studio.py +83 -0
  10. lightning_sdk/cli/create.py +21 -1
  11. lightning_sdk/cli/delete.py +6 -8
  12. lightning_sdk/cli/deploy/__init__.py +0 -0
  13. lightning_sdk/cli/deploy/_auth.py +189 -0
  14. lightning_sdk/cli/deploy/devbox.py +157 -0
  15. lightning_sdk/cli/{serve.py → deploy/serve.py} +22 -281
  16. lightning_sdk/cli/download.py +69 -16
  17. lightning_sdk/cli/entrypoint.py +1 -1
  18. lightning_sdk/cli/open.py +21 -2
  19. lightning_sdk/cli/start.py +12 -3
  20. lightning_sdk/cli/teamspace_menu.py +9 -1
  21. lightning_sdk/cli/upload.py +2 -5
  22. lightning_sdk/lightning_cloud/openapi/__init__.py +29 -0
  23. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  24. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +121 -0
  25. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +9 -1
  26. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +226 -0
  27. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  28. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +178 -0
  29. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +984 -101
  30. lightning_sdk/lightning_cloud/openapi/api/product_license_service_api.py +525 -0
  31. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +93 -0
  32. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  33. lightning_sdk/lightning_cloud/openapi/models/__init__.py +28 -0
  34. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  35. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body.py +331 -0
  37. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body1.py +305 -0
  38. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +53 -1
  39. lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/models_id_body.py +123 -0
  42. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +183 -1
  43. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +107 -3
  47. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/update.py +79 -1
  49. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py +55 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +53 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +3 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +123 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +79 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +104 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_source_type.py +103 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +66 -66
  58. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_upload.py +149 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload.py +55 -3
  62. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +79 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_alerting_policy_response.py +175 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_event.py +487 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy.py +383 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_frequency.py +105 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_operation.py +105 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_severity.py +106 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_type.py +111 -0
  72. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +27 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py +4 -4
  74. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_external_search_user.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_ge_list_deployment_routing_telemetry_response.py +123 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_open_ports_response.py +123 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_content_response.py +123 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadata_response.py +331 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +1 -27
  82. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +27 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_job_type.py +1 -0
  84. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_events_response.py +123 -0
  85. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_policies_response.py +175 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_list_product_licenses_response.py +123 -0
  87. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  88. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +43 -17
  89. lightning_sdk/lightning_cloud/openapi/models/v1_modify_filesystem_volume_response.py +97 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +183 -1
  91. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +6 -6
  92. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_state.py +111 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py +53 -1
  94. lightning_sdk/lightning_cloud/openapi/models/v1_product_license.py +409 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_product_license_check_response.py +123 -0
  96. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  97. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +43 -17
  98. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +107 -3
  99. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +53 -1
  100. lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py +53 -1
  101. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +253 -0
  102. lightning_sdk/lightning_cloud/openapi/models/v1_secret_type.py +1 -0
  103. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +2 -0
  104. lightning_sdk/lightning_cloud/openapi/models/v1_sleep_server_response.py +97 -0
  105. lightning_sdk/lightning_cloud/openapi/models/v1_trigger_filesystem_upgrade_response.py +123 -0
  106. lightning_sdk/lightning_cloud/openapi/models/v1_upload_project_artifact_response.py +27 -1
  107. lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py +79 -1
  108. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +347 -113
  109. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  110. lightning_sdk/lightning_cloud/rest_client.py +4 -0
  111. lightning_sdk/llm/llm.py +132 -40
  112. lightning_sdk/services/__init__.py +1 -1
  113. lightning_sdk/services/license.py +236 -0
  114. lightning_sdk/studio.py +62 -1
  115. lightning_sdk/teamspace.py +68 -0
  116. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/METADATA +1 -1
  117. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/RECORD +122 -86
  118. /lightning_sdk/services/{finetune/__init__.py → finetune_llm.py} +0 -0
  119. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/LICENSE +0 -0
  120. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/WHEEL +0 -0
  121. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/entry_points.txt +0 -0
  122. {lightning_sdk-0.2.14.dist-info → lightning_sdk-0.2.16.dist-info}/top_level.txt +0 -0
@@ -3420,6 +3420,111 @@ class ClusterServiceApi(object):
3420
3420
  _request_timeout=params.get('_request_timeout'),
3421
3421
  collection_formats=collection_formats)
3422
3422
 
3423
+ def cluster_service_sleep_server(self, body: 'object', server_id: 'str', **kwargs) -> 'V1SleepServerResponse': # noqa: E501
3424
+ """cluster_service_sleep_server # noqa: E501
3425
+
3426
+ This method makes a synchronous HTTP request by default. To make an
3427
+ asynchronous HTTP request, please pass async_req=True
3428
+ >>> thread = api.cluster_service_sleep_server(body, server_id, async_req=True)
3429
+ >>> result = thread.get()
3430
+
3431
+ :param async_req bool
3432
+ :param object body: (required)
3433
+ :param str server_id: (required)
3434
+ :return: V1SleepServerResponse
3435
+ If the method is called asynchronously,
3436
+ returns the request thread.
3437
+ """
3438
+ kwargs['_return_http_data_only'] = True
3439
+ if kwargs.get('async_req'):
3440
+ return self.cluster_service_sleep_server_with_http_info(body, server_id, **kwargs) # noqa: E501
3441
+ else:
3442
+ (data) = self.cluster_service_sleep_server_with_http_info(body, server_id, **kwargs) # noqa: E501
3443
+ return data
3444
+
3445
+ def cluster_service_sleep_server_with_http_info(self, body: 'object', server_id: 'str', **kwargs) -> 'V1SleepServerResponse': # noqa: E501
3446
+ """cluster_service_sleep_server # noqa: E501
3447
+
3448
+ This method makes a synchronous HTTP request by default. To make an
3449
+ asynchronous HTTP request, please pass async_req=True
3450
+ >>> thread = api.cluster_service_sleep_server_with_http_info(body, server_id, async_req=True)
3451
+ >>> result = thread.get()
3452
+
3453
+ :param async_req bool
3454
+ :param object body: (required)
3455
+ :param str server_id: (required)
3456
+ :return: V1SleepServerResponse
3457
+ If the method is called asynchronously,
3458
+ returns the request thread.
3459
+ """
3460
+
3461
+ all_params = ['body', 'server_id'] # noqa: E501
3462
+ all_params.append('async_req')
3463
+ all_params.append('_return_http_data_only')
3464
+ all_params.append('_preload_content')
3465
+ all_params.append('_request_timeout')
3466
+
3467
+ params = locals()
3468
+ for key, val in six.iteritems(params['kwargs']):
3469
+ if key not in all_params:
3470
+ raise TypeError(
3471
+ "Got an unexpected keyword argument '%s'"
3472
+ " to method cluster_service_sleep_server" % key
3473
+ )
3474
+ params[key] = val
3475
+ del params['kwargs']
3476
+ # verify the required parameter 'body' is set
3477
+ if ('body' not in params or
3478
+ params['body'] is None):
3479
+ raise ValueError("Missing the required parameter `body` when calling `cluster_service_sleep_server`") # noqa: E501
3480
+ # verify the required parameter 'server_id' is set
3481
+ if ('server_id' not in params or
3482
+ params['server_id'] is None):
3483
+ raise ValueError("Missing the required parameter `server_id` when calling `cluster_service_sleep_server`") # noqa: E501
3484
+
3485
+ collection_formats = {}
3486
+
3487
+ path_params = {}
3488
+ if 'server_id' in params:
3489
+ path_params['serverId'] = params['server_id'] # noqa: E501
3490
+
3491
+ query_params = []
3492
+
3493
+ header_params = {}
3494
+
3495
+ form_params = []
3496
+ local_var_files = {}
3497
+
3498
+ body_params = None
3499
+ if 'body' in params:
3500
+ body_params = params['body']
3501
+ # HTTP header `Accept`
3502
+ header_params['Accept'] = self.api_client.select_header_accept(
3503
+ ['application/json']) # noqa: E501
3504
+
3505
+ # HTTP header `Content-Type`
3506
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
3507
+ ['application/json']) # noqa: E501
3508
+
3509
+ # Authentication setting
3510
+ auth_settings = [] # noqa: E501
3511
+
3512
+ return self.api_client.call_api(
3513
+ '/v1/core/servers/{serverId}/sleep', 'POST',
3514
+ path_params,
3515
+ query_params,
3516
+ header_params,
3517
+ body=body_params,
3518
+ post_params=form_params,
3519
+ files=local_var_files,
3520
+ response_type='V1SleepServerResponse', # noqa: E501
3521
+ auth_settings=auth_settings,
3522
+ async_req=params.get('async_req'),
3523
+ _return_http_data_only=params.get('_return_http_data_only'),
3524
+ _preload_content=params.get('_preload_content', True),
3525
+ _request_timeout=params.get('_request_timeout'),
3526
+ collection_formats=collection_formats)
3527
+
3423
3528
  def cluster_service_update_cluster(self, body: 'ClustersIdBody', id: 'str', **kwargs) -> 'Externalv1Cluster': # noqa: E501
3424
3529
  """cluster_service_update_cluster # noqa: E501
3425
3530
 
@@ -777,3 +777,181 @@ class FileSystemServiceApi(object):
777
777
  _preload_content=params.get('_preload_content', True),
778
778
  _request_timeout=params.get('_request_timeout'),
779
779
  collection_formats=collection_formats)
780
+
781
+ def file_system_service_modify_filesystem_volume(self, project_id: 'str', **kwargs) -> 'V1ModifyFilesystemVolumeResponse': # noqa: E501
782
+ """file_system_service_modify_filesystem_volume # noqa: E501
783
+
784
+ This method makes a synchronous HTTP request by default. To make an
785
+ asynchronous HTTP request, please pass async_req=True
786
+ >>> thread = api.file_system_service_modify_filesystem_volume(project_id, async_req=True)
787
+ >>> result = thread.get()
788
+
789
+ :param async_req bool
790
+ :param str project_id: (required)
791
+ :return: V1ModifyFilesystemVolumeResponse
792
+ If the method is called asynchronously,
793
+ returns the request thread.
794
+ """
795
+ kwargs['_return_http_data_only'] = True
796
+ if kwargs.get('async_req'):
797
+ return self.file_system_service_modify_filesystem_volume_with_http_info(project_id, **kwargs) # noqa: E501
798
+ else:
799
+ (data) = self.file_system_service_modify_filesystem_volume_with_http_info(project_id, **kwargs) # noqa: E501
800
+ return data
801
+
802
+ def file_system_service_modify_filesystem_volume_with_http_info(self, project_id: 'str', **kwargs) -> 'V1ModifyFilesystemVolumeResponse': # noqa: E501
803
+ """file_system_service_modify_filesystem_volume # noqa: E501
804
+
805
+ This method makes a synchronous HTTP request by default. To make an
806
+ asynchronous HTTP request, please pass async_req=True
807
+ >>> thread = api.file_system_service_modify_filesystem_volume_with_http_info(project_id, async_req=True)
808
+ >>> result = thread.get()
809
+
810
+ :param async_req bool
811
+ :param str project_id: (required)
812
+ :return: V1ModifyFilesystemVolumeResponse
813
+ If the method is called asynchronously,
814
+ returns the request thread.
815
+ """
816
+
817
+ all_params = ['project_id'] # noqa: E501
818
+ all_params.append('async_req')
819
+ all_params.append('_return_http_data_only')
820
+ all_params.append('_preload_content')
821
+ all_params.append('_request_timeout')
822
+
823
+ params = locals()
824
+ for key, val in six.iteritems(params['kwargs']):
825
+ if key not in all_params:
826
+ raise TypeError(
827
+ "Got an unexpected keyword argument '%s'"
828
+ " to method file_system_service_modify_filesystem_volume" % key
829
+ )
830
+ params[key] = val
831
+ del params['kwargs']
832
+ # verify the required parameter 'project_id' is set
833
+ if ('project_id' not in params or
834
+ params['project_id'] is None):
835
+ raise ValueError("Missing the required parameter `project_id` when calling `file_system_service_modify_filesystem_volume`") # noqa: E501
836
+
837
+ collection_formats = {}
838
+
839
+ path_params = {}
840
+ if 'project_id' in params:
841
+ path_params['projectId'] = params['project_id'] # noqa: E501
842
+
843
+ query_params = []
844
+
845
+ header_params = {}
846
+
847
+ form_params = []
848
+ local_var_files = {}
849
+
850
+ body_params = None
851
+ # HTTP header `Accept`
852
+ header_params['Accept'] = self.api_client.select_header_accept(
853
+ ['application/json']) # noqa: E501
854
+
855
+ # Authentication setting
856
+ auth_settings = [] # noqa: E501
857
+
858
+ return self.api_client.call_api(
859
+ '/v1/filesystem/{projectId}/volumes', 'PUT',
860
+ path_params,
861
+ query_params,
862
+ header_params,
863
+ body=body_params,
864
+ post_params=form_params,
865
+ files=local_var_files,
866
+ response_type='V1ModifyFilesystemVolumeResponse', # noqa: E501
867
+ auth_settings=auth_settings,
868
+ async_req=params.get('async_req'),
869
+ _return_http_data_only=params.get('_return_http_data_only'),
870
+ _preload_content=params.get('_preload_content', True),
871
+ _request_timeout=params.get('_request_timeout'),
872
+ collection_formats=collection_formats)
873
+
874
+ def file_system_service_trigger_filesystem_upgrade(self, **kwargs) -> 'V1TriggerFilesystemUpgradeResponse': # noqa: E501
875
+ """file_system_service_trigger_filesystem_upgrade # noqa: E501
876
+
877
+ This method makes a synchronous HTTP request by default. To make an
878
+ asynchronous HTTP request, please pass async_req=True
879
+ >>> thread = api.file_system_service_trigger_filesystem_upgrade(async_req=True)
880
+ >>> result = thread.get()
881
+
882
+ :param async_req bool
883
+ :return: V1TriggerFilesystemUpgradeResponse
884
+ If the method is called asynchronously,
885
+ returns the request thread.
886
+ """
887
+ kwargs['_return_http_data_only'] = True
888
+ if kwargs.get('async_req'):
889
+ return self.file_system_service_trigger_filesystem_upgrade_with_http_info(**kwargs) # noqa: E501
890
+ else:
891
+ (data) = self.file_system_service_trigger_filesystem_upgrade_with_http_info(**kwargs) # noqa: E501
892
+ return data
893
+
894
+ def file_system_service_trigger_filesystem_upgrade_with_http_info(self, **kwargs) -> 'V1TriggerFilesystemUpgradeResponse': # noqa: E501
895
+ """file_system_service_trigger_filesystem_upgrade # noqa: E501
896
+
897
+ This method makes a synchronous HTTP request by default. To make an
898
+ asynchronous HTTP request, please pass async_req=True
899
+ >>> thread = api.file_system_service_trigger_filesystem_upgrade_with_http_info(async_req=True)
900
+ >>> result = thread.get()
901
+
902
+ :param async_req bool
903
+ :return: V1TriggerFilesystemUpgradeResponse
904
+ If the method is called asynchronously,
905
+ returns the request thread.
906
+ """
907
+
908
+ all_params = [] # noqa: E501
909
+ all_params.append('async_req')
910
+ all_params.append('_return_http_data_only')
911
+ all_params.append('_preload_content')
912
+ all_params.append('_request_timeout')
913
+
914
+ params = locals()
915
+ for key, val in six.iteritems(params['kwargs']):
916
+ if key not in all_params:
917
+ raise TypeError(
918
+ "Got an unexpected keyword argument '%s'"
919
+ " to method file_system_service_trigger_filesystem_upgrade" % key
920
+ )
921
+ params[key] = val
922
+ del params['kwargs']
923
+
924
+ collection_formats = {}
925
+
926
+ path_params = {}
927
+
928
+ query_params = []
929
+
930
+ header_params = {}
931
+
932
+ form_params = []
933
+ local_var_files = {}
934
+
935
+ body_params = None
936
+ # HTTP header `Accept`
937
+ header_params['Accept'] = self.api_client.select_header_accept(
938
+ ['application/json']) # noqa: E501
939
+
940
+ # Authentication setting
941
+ auth_settings = [] # noqa: E501
942
+
943
+ return self.api_client.call_api(
944
+ '/v1/filesystem/trigger-upgrade', 'GET',
945
+ path_params,
946
+ query_params,
947
+ header_params,
948
+ body=body_params,
949
+ post_params=form_params,
950
+ files=local_var_files,
951
+ response_type='V1TriggerFilesystemUpgradeResponse', # noqa: E501
952
+ auth_settings=auth_settings,
953
+ async_req=params.get('async_req'),
954
+ _return_http_data_only=params.get('_return_http_data_only'),
955
+ _preload_content=params.get('_preload_content', True),
956
+ _request_timeout=params.get('_request_timeout'),
957
+ collection_formats=collection_formats)