lightning-sdk 0.2.4__py3-none-any.whl → 0.2.6__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 (123) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +3 -1
  3. lightning_sdk/api/ai_hub_api.py +15 -0
  4. lightning_sdk/api/studio_api.py +0 -8
  5. lightning_sdk/cli/serve.py +29 -9
  6. lightning_sdk/deployment/deployment.py +32 -4
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +57 -1
  8. lightning_sdk/lightning_cloud/openapi/api/__init__.py +4 -0
  9. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +537 -0
  10. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +115 -6
  11. lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py +1973 -0
  12. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +20 -4
  13. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +20 -20
  14. lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py +339 -0
  15. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +5 -1
  16. lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py +573 -0
  17. lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +202 -0
  18. lightning_sdk/lightning_cloud/openapi/models/__init__.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
  20. lightning_sdk/lightning_cloud/openapi/models/cluster_id_slurmusers_body.py +201 -0
  21. lightning_sdk/lightning_cloud/openapi/models/dataset_id_versions_body.py +149 -0
  22. lightning_sdk/lightning_cloud/openapi/models/dataset_id_visibility_body.py +149 -0
  23. lightning_sdk/lightning_cloud/openapi/models/environmenttemplates_id_body.py +253 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +95 -69
  25. lightning_sdk/lightning_cloud/openapi/models/externalv1_cluster.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/litdatasets_dataset_id_body.py +149 -0
  27. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +69 -17
  29. lightning_sdk/lightning_cloud/openapi/models/pipelinetemplates_id_body.py +331 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_litdatasets_body.py +227 -0
  32. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +17 -17
  33. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +201 -0
  34. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +383 -0
  35. lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py +175 -0
  36. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +15 -15
  37. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py +149 -0
  38. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body1.py +149 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +124 -20
  40. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +2 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +279 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +123 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +157 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_multi_part_upload_response.py +97 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_complete_lit_dataset_upload_response.py +97 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +175 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_create_lit_dataset_multi_part_upload_response.py +123 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py +383 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_create_server_alert_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/{v1_pipeline_schedule.py → v1_delete_cloud_space_environment_template_response.py} +32 -32
  57. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py +97 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py +97 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_delete_schedule_response.py +175 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py +79 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_file_upload_urls_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_get_lit_dataset_files_url_response.py +149 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +55 -3
  65. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +79 -27
  66. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  67. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +1 -29
  68. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_environment_templates_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_dataset_versions_response.py +123 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_datasets_response.py +123 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_registry_repository_image_artifact_versions_response.py +27 -1
  72. lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py +123 -0
  73. lightning_sdk/lightning_cloud/openapi/models/v1_list_slurm_cluster_users_response.py +123 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset.py +539 -0
  75. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_file.py +175 -0
  76. lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py +435 -0
  77. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +27 -1
  78. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  79. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +27 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +107 -3
  81. lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +69 -17
  84. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter.py +435 -0
  85. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement.py +149 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_placement_type.py +106 -0
  87. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_parameter_type.py +106 -0
  88. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template.py +513 -0
  89. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_template_visibility_type.py +105 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +27 -1
  91. lightning_sdk/lightning_cloud/openapi/models/v1_reservation_details.py +201 -0
  92. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +435 -0
  94. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +103 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert.py +175 -0
  96. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py +103 -0
  97. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +105 -0
  98. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_cluster_user.py +227 -0
  99. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +58 -6
  100. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +31 -291
  101. lightning_sdk/lightning_cloud/openapi/models/v1_subnet_spec.py +149 -0
  102. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_dataset_visibility_response.py +123 -0
  103. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  104. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +223 -171
  105. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +27 -1
  106. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +203 -0
  107. lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +1 -27
  108. lightning_sdk/lightning_cloud/openapi/models/version_default_body.py +29 -29
  109. lightning_sdk/lightning_cloud/openapi/models/version_default_body1.py +149 -0
  110. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.py +123 -0
  111. lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.py +123 -0
  112. lightning_sdk/lightning_cloud/rest_client.py +44 -40
  113. lightning_sdk/pipeline/types.py +28 -2
  114. lightning_sdk/pipeline/utils.py +1 -1
  115. lightning_sdk/plugin.py +0 -6
  116. lightning_sdk/serve.py +5 -3
  117. lightning_sdk/utils/resolve.py +1 -0
  118. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/METADATA +1 -1
  119. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/RECORD +123 -67
  120. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/WHEEL +1 -1
  121. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/LICENSE +0 -0
  122. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/entry_points.txt +0 -0
  123. {lightning_sdk-0.2.4.dist-info → lightning_sdk-0.2.6.dist-info}/top_level.txt +0 -0
@@ -665,6 +665,111 @@ class ClusterServiceApi(object):
665
665
  _request_timeout=params.get('_request_timeout'),
666
666
  collection_formats=collection_formats)
667
667
 
668
+ def cluster_service_create_server_alert(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse': # noqa: E501
669
+ """cluster_service_create_server_alert # noqa: E501
670
+
671
+ This method makes a synchronous HTTP request by default. To make an
672
+ asynchronous HTTP request, please pass async_req=True
673
+ >>> thread = api.cluster_service_create_server_alert(body, server_id, async_req=True)
674
+ >>> result = thread.get()
675
+
676
+ :param async_req bool
677
+ :param ServerIdAlertsBody body: (required)
678
+ :param str server_id: (required)
679
+ :return: V1CreateServerAlertResponse
680
+ If the method is called asynchronously,
681
+ returns the request thread.
682
+ """
683
+ kwargs['_return_http_data_only'] = True
684
+ if kwargs.get('async_req'):
685
+ return self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs) # noqa: E501
686
+ else:
687
+ (data) = self.cluster_service_create_server_alert_with_http_info(body, server_id, **kwargs) # noqa: E501
688
+ return data
689
+
690
+ def cluster_service_create_server_alert_with_http_info(self, body: 'ServerIdAlertsBody', server_id: 'str', **kwargs) -> 'V1CreateServerAlertResponse': # noqa: E501
691
+ """cluster_service_create_server_alert # noqa: E501
692
+
693
+ This method makes a synchronous HTTP request by default. To make an
694
+ asynchronous HTTP request, please pass async_req=True
695
+ >>> thread = api.cluster_service_create_server_alert_with_http_info(body, server_id, async_req=True)
696
+ >>> result = thread.get()
697
+
698
+ :param async_req bool
699
+ :param ServerIdAlertsBody body: (required)
700
+ :param str server_id: (required)
701
+ :return: V1CreateServerAlertResponse
702
+ If the method is called asynchronously,
703
+ returns the request thread.
704
+ """
705
+
706
+ all_params = ['body', 'server_id'] # noqa: E501
707
+ all_params.append('async_req')
708
+ all_params.append('_return_http_data_only')
709
+ all_params.append('_preload_content')
710
+ all_params.append('_request_timeout')
711
+
712
+ params = locals()
713
+ for key, val in six.iteritems(params['kwargs']):
714
+ if key not in all_params:
715
+ raise TypeError(
716
+ "Got an unexpected keyword argument '%s'"
717
+ " to method cluster_service_create_server_alert" % key
718
+ )
719
+ params[key] = val
720
+ del params['kwargs']
721
+ # verify the required parameter 'body' is set
722
+ if ('body' not in params or
723
+ params['body'] is None):
724
+ raise ValueError("Missing the required parameter `body` when calling `cluster_service_create_server_alert`") # noqa: E501
725
+ # verify the required parameter 'server_id' is set
726
+ if ('server_id' not in params or
727
+ params['server_id'] is None):
728
+ raise ValueError("Missing the required parameter `server_id` when calling `cluster_service_create_server_alert`") # noqa: E501
729
+
730
+ collection_formats = {}
731
+
732
+ path_params = {}
733
+ if 'server_id' in params:
734
+ path_params['serverId'] = params['server_id'] # noqa: E501
735
+
736
+ query_params = []
737
+
738
+ header_params = {}
739
+
740
+ form_params = []
741
+ local_var_files = {}
742
+
743
+ body_params = None
744
+ if 'body' in params:
745
+ body_params = params['body']
746
+ # HTTP header `Accept`
747
+ header_params['Accept'] = self.api_client.select_header_accept(
748
+ ['application/json']) # noqa: E501
749
+
750
+ # HTTP header `Content-Type`
751
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
752
+ ['application/json']) # noqa: E501
753
+
754
+ # Authentication setting
755
+ auth_settings = [] # noqa: E501
756
+
757
+ return self.api_client.call_api(
758
+ '/v1/core/servers/{serverId}/alerts', 'POST',
759
+ path_params,
760
+ query_params,
761
+ header_params,
762
+ body=body_params,
763
+ post_params=form_params,
764
+ files=local_var_files,
765
+ response_type='V1CreateServerAlertResponse', # noqa: E501
766
+ auth_settings=auth_settings,
767
+ async_req=params.get('async_req'),
768
+ _return_http_data_only=params.get('_return_http_data_only'),
769
+ _preload_content=params.get('_preload_content', True),
770
+ _request_timeout=params.get('_request_timeout'),
771
+ collection_formats=collection_formats)
772
+
668
773
  def cluster_service_delete_cluster(self, id: 'str', **kwargs) -> 'V1DeleteClusterResponse': # noqa: E501
669
774
  """cluster_service_delete_cluster # noqa: E501
670
775
 
@@ -1319,7 +1424,7 @@ class ClusterServiceApi(object):
1319
1424
  :param datetime start_date:
1320
1425
  :param str timezone:
1321
1426
  :param str org_id:
1322
- :param int capacity_block_duration_days:
1427
+ :param int capacity_block_duration_hours:
1323
1428
  :return: V1FindCapacityBlockOfferingResponse
1324
1429
  If the method is called asynchronously,
1325
1430
  returns the request thread.
@@ -1347,13 +1452,13 @@ class ClusterServiceApi(object):
1347
1452
  :param datetime start_date:
1348
1453
  :param str timezone:
1349
1454
  :param str org_id:
1350
- :param int capacity_block_duration_days:
1455
+ :param int capacity_block_duration_hours:
1351
1456
  :return: V1FindCapacityBlockOfferingResponse
1352
1457
  If the method is called asynchronously,
1353
1458
  returns the request thread.
1354
1459
  """
1355
1460
 
1356
- all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_days'] # noqa: E501
1461
+ all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_hours'] # noqa: E501
1357
1462
  all_params.append('async_req')
1358
1463
  all_params.append('_return_http_data_only')
1359
1464
  all_params.append('_preload_content')
@@ -1396,8 +1501,8 @@ class ClusterServiceApi(object):
1396
1501
  query_params.append(('timezone', params['timezone'])) # noqa: E501
1397
1502
  if 'org_id' in params:
1398
1503
  query_params.append(('orgId', params['org_id'])) # noqa: E501
1399
- if 'capacity_block_duration_days' in params:
1400
- query_params.append(('capacityBlockDurationDays', params['capacity_block_duration_days'])) # noqa: E501
1504
+ if 'capacity_block_duration_hours' in params:
1505
+ query_params.append(('capacityBlockDurationHours', params['capacity_block_duration_hours'])) # noqa: E501
1401
1506
 
1402
1507
  header_params = {}
1403
1508
 
@@ -2611,6 +2716,7 @@ class ClusterServiceApi(object):
2611
2716
  :param async_req bool
2612
2717
  :param bool include_pricing:
2613
2718
  :param str cloud_provider:
2719
+ :param str project_id:
2614
2720
  :return: V1ListDefaultClusterAcceleratorsResponse
2615
2721
  If the method is called asynchronously,
2616
2722
  returns the request thread.
@@ -2633,12 +2739,13 @@ class ClusterServiceApi(object):
2633
2739
  :param async_req bool
2634
2740
  :param bool include_pricing:
2635
2741
  :param str cloud_provider:
2742
+ :param str project_id:
2636
2743
  :return: V1ListDefaultClusterAcceleratorsResponse
2637
2744
  If the method is called asynchronously,
2638
2745
  returns the request thread.
2639
2746
  """
2640
2747
 
2641
- all_params = ['include_pricing', 'cloud_provider'] # noqa: E501
2748
+ all_params = ['include_pricing', 'cloud_provider', 'project_id'] # noqa: E501
2642
2749
  all_params.append('async_req')
2643
2750
  all_params.append('_return_http_data_only')
2644
2751
  all_params.append('_preload_content')
@@ -2663,6 +2770,8 @@ class ClusterServiceApi(object):
2663
2770
  query_params.append(('includePricing', params['include_pricing'])) # noqa: E501
2664
2771
  if 'cloud_provider' in params:
2665
2772
  query_params.append(('cloudProvider', params['cloud_provider'])) # noqa: E501
2773
+ if 'project_id' in params:
2774
+ query_params.append(('projectId', params['project_id'])) # noqa: E501
2666
2775
 
2667
2776
  header_params = {}
2668
2777