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
@@ -159,6 +159,7 @@ class LitRegistryServiceApi(object):
159
159
  :param async_req bool
160
160
  :param str project_id: (required)
161
161
  :param str lit_repo_name: (required)
162
+ :param str cluster_id:
162
163
  :return: V1DeleteLitRepositoryResponse
163
164
  If the method is called asynchronously,
164
165
  returns the request thread.
@@ -181,12 +182,13 @@ class LitRegistryServiceApi(object):
181
182
  :param async_req bool
182
183
  :param str project_id: (required)
183
184
  :param str lit_repo_name: (required)
185
+ :param str cluster_id:
184
186
  :return: V1DeleteLitRepositoryResponse
185
187
  If the method is called asynchronously,
186
188
  returns the request thread.
187
189
  """
188
190
 
189
- all_params = ['project_id', 'lit_repo_name'] # noqa: E501
191
+ all_params = ['project_id', 'lit_repo_name', 'cluster_id'] # noqa: E501
190
192
  all_params.append('async_req')
191
193
  all_params.append('_return_http_data_only')
192
194
  all_params.append('_preload_content')
@@ -219,6 +221,8 @@ class LitRegistryServiceApi(object):
219
221
  path_params['litRepoName'] = params['lit_repo_name'] # noqa: E501
220
222
 
221
223
  query_params = []
224
+ if 'cluster_id' in params:
225
+ query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
222
226
 
223
227
  header_params = {}
224
228
 
@@ -261,6 +265,7 @@ class LitRegistryServiceApi(object):
261
265
  :param str project_id: (required)
262
266
  :param bool latest_only:
263
267
  :param str filter_for_repo: Use this to filter for a single repository. Useful when selecting a subset of the same data from various context.
268
+ :param str cluster_id:
264
269
  :return: V1LitRegistryProject
265
270
  If the method is called asynchronously,
266
271
  returns the request thread.
@@ -284,12 +289,13 @@ class LitRegistryServiceApi(object):
284
289
  :param str project_id: (required)
285
290
  :param bool latest_only:
286
291
  :param str filter_for_repo: Use this to filter for a single repository. Useful when selecting a subset of the same data from various context.
292
+ :param str cluster_id:
287
293
  :return: V1LitRegistryProject
288
294
  If the method is called asynchronously,
289
295
  returns the request thread.
290
296
  """
291
297
 
292
- all_params = ['project_id', 'latest_only', 'filter_for_repo'] # noqa: E501
298
+ all_params = ['project_id', 'latest_only', 'filter_for_repo', 'cluster_id'] # noqa: E501
293
299
  all_params.append('async_req')
294
300
  all_params.append('_return_http_data_only')
295
301
  all_params.append('_preload_content')
@@ -320,6 +326,8 @@ class LitRegistryServiceApi(object):
320
326
  query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
321
327
  if 'filter_for_repo' in params:
322
328
  query_params.append(('filterForRepo', params['filter_for_repo'])) # noqa: E501
329
+ if 'cluster_id' in params:
330
+ query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
323
331
 
324
332
  header_params = {}
325
333
 
@@ -471,6 +479,7 @@ class LitRegistryServiceApi(object):
471
479
  :param str project_id: (required)
472
480
  :param str lit_repo_name: (required)
473
481
  :param bool latest_only:
482
+ :param str cluster_id:
474
483
  :return: V1LitRepository
475
484
  If the method is called asynchronously,
476
485
  returns the request thread.
@@ -494,12 +503,13 @@ class LitRegistryServiceApi(object):
494
503
  :param str project_id: (required)
495
504
  :param str lit_repo_name: (required)
496
505
  :param bool latest_only:
506
+ :param str cluster_id:
497
507
  :return: V1LitRepository
498
508
  If the method is called asynchronously,
499
509
  returns the request thread.
500
510
  """
501
511
 
502
- all_params = ['project_id', 'lit_repo_name', 'latest_only'] # noqa: E501
512
+ all_params = ['project_id', 'lit_repo_name', 'latest_only', 'cluster_id'] # noqa: E501
503
513
  all_params.append('async_req')
504
514
  all_params.append('_return_http_data_only')
505
515
  all_params.append('_preload_content')
@@ -534,6 +544,8 @@ class LitRegistryServiceApi(object):
534
544
  query_params = []
535
545
  if 'latest_only' in params:
536
546
  query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
547
+ if 'cluster_id' in params:
548
+ query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
537
549
 
538
550
  header_params = {}
539
551
 
@@ -576,6 +588,7 @@ class LitRegistryServiceApi(object):
576
588
  :param str project_id: (required)
577
589
  :param str lit_repo_name: (required)
578
590
  :param bool latest_only:
591
+ :param str cluster_id:
579
592
  :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
580
593
  If the method is called asynchronously,
581
594
  returns the request thread.
@@ -599,12 +612,13 @@ class LitRegistryServiceApi(object):
599
612
  :param str project_id: (required)
600
613
  :param str lit_repo_name: (required)
601
614
  :param bool latest_only:
615
+ :param str cluster_id:
602
616
  :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
603
617
  If the method is called asynchronously,
604
618
  returns the request thread.
605
619
  """
606
620
 
607
- all_params = ['project_id', 'lit_repo_name', 'latest_only'] # noqa: E501
621
+ all_params = ['project_id', 'lit_repo_name', 'latest_only', 'cluster_id'] # noqa: E501
608
622
  all_params.append('async_req')
609
623
  all_params.append('_return_http_data_only')
610
624
  all_params.append('_preload_content')
@@ -639,6 +653,8 @@ class LitRegistryServiceApi(object):
639
653
  query_params = []
640
654
  if 'latest_only' in params:
641
655
  query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
656
+ if 'cluster_id' in params:
657
+ query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
642
658
 
643
659
  header_params = {}
644
660
 
@@ -164,7 +164,7 @@ class ModelsStoreApi(object):
164
164
  _request_timeout=params.get('_request_timeout'),
165
165
  collection_formats=collection_formats)
166
166
 
167
- def models_store_complete_multi_part_upload(self, body: 'UploadIdCompleteBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
167
+ def models_store_complete_multi_part_upload(self, body: 'UploadIdCompleteBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
168
168
  """CompleteMultiPartUpload marks the given file-parts as completed/uploaded. Once all parts of a file are completed, the file will materialize in S3 storage # noqa: E501
169
169
 
170
170
  This method makes a synchronous HTTP request by default. To make an
@@ -173,7 +173,7 @@ class ModelsStoreApi(object):
173
173
  >>> result = thread.get()
174
174
 
175
175
  :param async_req bool
176
- :param UploadIdCompleteBody body: (required)
176
+ :param UploadIdCompleteBody1 body: (required)
177
177
  :param str project_id: (required)
178
178
  :param str model_id: (required)
179
179
  :param str version: (required)
@@ -189,7 +189,7 @@ class ModelsStoreApi(object):
189
189
  (data) = self.models_store_complete_multi_part_upload_with_http_info(body, project_id, model_id, version, upload_id, **kwargs) # noqa: E501
190
190
  return data
191
191
 
192
- def models_store_complete_multi_part_upload_with_http_info(self, body: 'UploadIdCompleteBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
192
+ def models_store_complete_multi_part_upload_with_http_info(self, body: 'UploadIdCompleteBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1CompleteMultiPartUploadResponse': # noqa: E501
193
193
  """CompleteMultiPartUpload marks the given file-parts as completed/uploaded. Once all parts of a file are completed, the file will materialize in S3 storage # noqa: E501
194
194
 
195
195
  This method makes a synchronous HTTP request by default. To make an
@@ -198,7 +198,7 @@ class ModelsStoreApi(object):
198
198
  >>> result = thread.get()
199
199
 
200
200
  :param async_req bool
201
- :param UploadIdCompleteBody body: (required)
201
+ :param UploadIdCompleteBody1 body: (required)
202
202
  :param str project_id: (required)
203
203
  :param str model_id: (required)
204
204
  :param str version: (required)
@@ -511,7 +511,7 @@ class ModelsStoreApi(object):
511
511
  _request_timeout=params.get('_request_timeout'),
512
512
  collection_formats=collection_formats)
513
513
 
514
- def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
514
+ def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
515
515
  """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files. # noqa: E501
516
516
 
517
517
  This method makes a synchronous HTTP request by default. To make an
@@ -520,7 +520,7 @@ class ModelsStoreApi(object):
520
520
  >>> result = thread.get()
521
521
 
522
522
  :param async_req bool
523
- :param VersionUploadsBody body: (required)
523
+ :param VersionUploadsBody1 body: (required)
524
524
  :param str project_id: (required)
525
525
  :param str model_id: (required)
526
526
  :param str version: (required)
@@ -535,7 +535,7 @@ class ModelsStoreApi(object):
535
535
  (data) = self.models_store_create_multi_part_upload_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
536
536
  return data
537
537
 
538
- def models_store_create_multi_part_upload_with_http_info(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
538
+ def models_store_create_multi_part_upload_with_http_info(self, body: 'VersionUploadsBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
539
539
  """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files. # noqa: E501
540
540
 
541
541
  This method makes a synchronous HTTP request by default. To make an
@@ -544,7 +544,7 @@ class ModelsStoreApi(object):
544
544
  >>> result = thread.get()
545
545
 
546
546
  :param async_req bool
547
- :param VersionUploadsBody body: (required)
547
+ :param VersionUploadsBody1 body: (required)
548
548
  :param str project_id: (required)
549
549
  :param str model_id: (required)
550
550
  :param str version: (required)
@@ -1052,7 +1052,7 @@ class ModelsStoreApi(object):
1052
1052
  _request_timeout=params.get('_request_timeout'),
1053
1053
  collection_formats=collection_formats)
1054
1054
 
1055
- def models_store_get_model_file_upload_urls(self, body: 'UploadIdPartsBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1055
+ def models_store_get_model_file_upload_urls(self, body: 'UploadIdPartsBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1056
1056
  """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts. # noqa: E501
1057
1057
 
1058
1058
  This method makes a synchronous HTTP request by default. To make an
@@ -1061,7 +1061,7 @@ class ModelsStoreApi(object):
1061
1061
  >>> result = thread.get()
1062
1062
 
1063
1063
  :param async_req bool
1064
- :param UploadIdPartsBody body: (required)
1064
+ :param UploadIdPartsBody1 body: (required)
1065
1065
  :param str project_id: (required)
1066
1066
  :param str model_id: (required)
1067
1067
  :param str version: (required)
@@ -1077,7 +1077,7 @@ class ModelsStoreApi(object):
1077
1077
  (data) = self.models_store_get_model_file_upload_urls_with_http_info(body, project_id, model_id, version, upload_id, **kwargs) # noqa: E501
1078
1078
  return data
1079
1079
 
1080
- def models_store_get_model_file_upload_urls_with_http_info(self, body: 'UploadIdPartsBody', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1080
+ def models_store_get_model_file_upload_urls_with_http_info(self, body: 'UploadIdPartsBody1', project_id: 'str', model_id: 'str', version: 'str', upload_id: 'str', **kwargs) -> 'V1GetModelFileUploadUrlsResponse': # noqa: E501
1081
1081
  """GetModelFileUploadUrls requests pre-signed URLs for a given number of file parts. # noqa: E501
1082
1082
 
1083
1083
  This method makes a synchronous HTTP request by default. To make an
@@ -1086,7 +1086,7 @@ class ModelsStoreApi(object):
1086
1086
  >>> result = thread.get()
1087
1087
 
1088
1088
  :param async_req bool
1089
- :param UploadIdPartsBody body: (required)
1089
+ :param UploadIdPartsBody1 body: (required)
1090
1090
  :param str project_id: (required)
1091
1091
  :param str model_id: (required)
1092
1092
  :param str version: (required)
@@ -1827,7 +1827,7 @@ class ModelsStoreApi(object):
1827
1827
  _request_timeout=params.get('_request_timeout'),
1828
1828
  collection_formats=collection_formats)
1829
1829
 
1830
- def models_store_set_model_default_version(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1830
+ def models_store_set_model_default_version(self, body: 'VersionDefaultBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1831
1831
  """models_store_set_model_default_version # noqa: E501
1832
1832
 
1833
1833
  This method makes a synchronous HTTP request by default. To make an
@@ -1836,7 +1836,7 @@ class ModelsStoreApi(object):
1836
1836
  >>> result = thread.get()
1837
1837
 
1838
1838
  :param async_req bool
1839
- :param VersionDefaultBody body: (required)
1839
+ :param VersionDefaultBody1 body: (required)
1840
1840
  :param str project_id: (required)
1841
1841
  :param str model_id: (required)
1842
1842
  :param str version: (required)
@@ -1851,7 +1851,7 @@ class ModelsStoreApi(object):
1851
1851
  (data) = self.models_store_set_model_default_version_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
1852
1852
  return data
1853
1853
 
1854
- def models_store_set_model_default_version_with_http_info(self, body: 'VersionDefaultBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1854
+ def models_store_set_model_default_version_with_http_info(self, body: 'VersionDefaultBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1Model': # noqa: E501
1855
1855
  """models_store_set_model_default_version # noqa: E501
1856
1856
 
1857
1857
  This method makes a synchronous HTTP request by default. To make an
@@ -1860,7 +1860,7 @@ class ModelsStoreApi(object):
1860
1860
  >>> result = thread.get()
1861
1861
 
1862
1862
  :param async_req bool
1863
- :param VersionDefaultBody body: (required)
1863
+ :param VersionDefaultBody1 body: (required)
1864
1864
  :param str project_id: (required)
1865
1865
  :param str model_id: (required)
1866
1866
  :param str version: (required)
@@ -2061,7 +2061,7 @@ class ModelsStoreApi(object):
2061
2061
  _request_timeout=params.get('_request_timeout'),
2062
2062
  collection_formats=collection_formats)
2063
2063
 
2064
- def models_store_update_model_version(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2064
+ def models_store_update_model_version(self, body: 'VersionsVersionBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2065
2065
  """UpdateModelVersion updates the model version, in particular the model version's name. # noqa: E501
2066
2066
 
2067
2067
  This method makes a synchronous HTTP request by default. To make an
@@ -2070,7 +2070,7 @@ class ModelsStoreApi(object):
2070
2070
  >>> result = thread.get()
2071
2071
 
2072
2072
  :param async_req bool
2073
- :param VersionsVersionBody body: (required)
2073
+ :param VersionsVersionBody1 body: (required)
2074
2074
  :param str project_id: (required)
2075
2075
  :param str model_id: (required)
2076
2076
  :param str version: (required)
@@ -2085,7 +2085,7 @@ class ModelsStoreApi(object):
2085
2085
  (data) = self.models_store_update_model_version_with_http_info(body, project_id, model_id, version, **kwargs) # noqa: E501
2086
2086
  return data
2087
2087
 
2088
- def models_store_update_model_version_with_http_info(self, body: 'VersionsVersionBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2088
+ def models_store_update_model_version_with_http_info(self, body: 'VersionsVersionBody1', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
2089
2089
  """UpdateModelVersion updates the model version, in particular the model version's name. # noqa: E501
2090
2090
 
2091
2091
  This method makes a synchronous HTTP request by default. To make an
@@ -2094,7 +2094,7 @@ class ModelsStoreApi(object):
2094
2094
  >>> result = thread.get()
2095
2095
 
2096
2096
  :param async_req bool
2097
- :param VersionsVersionBody body: (required)
2097
+ :param VersionsVersionBody1 body: (required)
2098
2098
  :param str project_id: (required)
2099
2099
  :param str model_id: (required)
2100
2100
  :param str version: (required)
@@ -0,0 +1,339 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ from __future__ import absolute_import
21
+
22
+ import re # noqa: F401
23
+ from typing import TYPE_CHECKING, Any
24
+
25
+ # python 2 and python 3 compatibility library
26
+ import six
27
+
28
+ from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
29
+
30
+ if TYPE_CHECKING:
31
+ from datetime import datetime
32
+ from lightning_sdk.lightning_cloud.openapi.models import *
33
+
34
+ class PipelineTemplatesServiceApi(object):
35
+ """NOTE: This class is auto generated by the swagger code generator program.
36
+
37
+ Do not edit the class manually.
38
+ Ref: https://github.com/swagger-api/swagger-codegen
39
+ """
40
+
41
+ def __init__(self, api_client=None):
42
+ if api_client is None:
43
+ api_client = ApiClient()
44
+ self.api_client = api_client
45
+
46
+ def pipeline_templates_service_create_pipeline_template(self, body: 'V1CreatePipelineTemplateRequest', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
47
+ """Pipeline templates # noqa: E501
48
+
49
+ This method makes a synchronous HTTP request by default. To make an
50
+ asynchronous HTTP request, please pass async_req=True
51
+ >>> thread = api.pipeline_templates_service_create_pipeline_template(body, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param V1CreatePipelineTemplateRequest body: (required)
56
+ :return: V1PipelineTemplate
57
+ If the method is called asynchronously,
58
+ returns the request thread.
59
+ """
60
+ kwargs['_return_http_data_only'] = True
61
+ if kwargs.get('async_req'):
62
+ return self.pipeline_templates_service_create_pipeline_template_with_http_info(body, **kwargs) # noqa: E501
63
+ else:
64
+ (data) = self.pipeline_templates_service_create_pipeline_template_with_http_info(body, **kwargs) # noqa: E501
65
+ return data
66
+
67
+ def pipeline_templates_service_create_pipeline_template_with_http_info(self, body: 'V1CreatePipelineTemplateRequest', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
68
+ """Pipeline templates # noqa: E501
69
+
70
+ This method makes a synchronous HTTP request by default. To make an
71
+ asynchronous HTTP request, please pass async_req=True
72
+ >>> thread = api.pipeline_templates_service_create_pipeline_template_with_http_info(body, async_req=True)
73
+ >>> result = thread.get()
74
+
75
+ :param async_req bool
76
+ :param V1CreatePipelineTemplateRequest body: (required)
77
+ :return: V1PipelineTemplate
78
+ If the method is called asynchronously,
79
+ returns the request thread.
80
+ """
81
+
82
+ all_params = ['body'] # noqa: E501
83
+ all_params.append('async_req')
84
+ all_params.append('_return_http_data_only')
85
+ all_params.append('_preload_content')
86
+ all_params.append('_request_timeout')
87
+
88
+ params = locals()
89
+ for key, val in six.iteritems(params['kwargs']):
90
+ if key not in all_params:
91
+ raise TypeError(
92
+ "Got an unexpected keyword argument '%s'"
93
+ " to method pipeline_templates_service_create_pipeline_template" % key
94
+ )
95
+ params[key] = val
96
+ del params['kwargs']
97
+ # verify the required parameter 'body' is set
98
+ if ('body' not in params or
99
+ params['body'] is None):
100
+ raise ValueError("Missing the required parameter `body` when calling `pipeline_templates_service_create_pipeline_template`") # noqa: E501
101
+
102
+ collection_formats = {}
103
+
104
+ path_params = {}
105
+
106
+ query_params = []
107
+
108
+ header_params = {}
109
+
110
+ form_params = []
111
+ local_var_files = {}
112
+
113
+ body_params = None
114
+ if 'body' in params:
115
+ body_params = params['body']
116
+ # HTTP header `Accept`
117
+ header_params['Accept'] = self.api_client.select_header_accept(
118
+ ['application/json']) # noqa: E501
119
+
120
+ # HTTP header `Content-Type`
121
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
122
+ ['application/json']) # noqa: E501
123
+
124
+ # Authentication setting
125
+ auth_settings = [] # noqa: E501
126
+
127
+ return self.api_client.call_api(
128
+ '/v1/pipeline-templates', 'POST',
129
+ path_params,
130
+ query_params,
131
+ header_params,
132
+ body=body_params,
133
+ post_params=form_params,
134
+ files=local_var_files,
135
+ response_type='V1PipelineTemplate', # noqa: E501
136
+ auth_settings=auth_settings,
137
+ async_req=params.get('async_req'),
138
+ _return_http_data_only=params.get('_return_http_data_only'),
139
+ _preload_content=params.get('_preload_content', True),
140
+ _request_timeout=params.get('_request_timeout'),
141
+ collection_formats=collection_formats)
142
+
143
+ def pipeline_templates_service_get_pipeline_template(self, id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
144
+ """pipeline_templates_service_get_pipeline_template # noqa: E501
145
+
146
+ This method makes a synchronous HTTP request by default. To make an
147
+ asynchronous HTTP request, please pass async_req=True
148
+ >>> thread = api.pipeline_templates_service_get_pipeline_template(id, async_req=True)
149
+ >>> result = thread.get()
150
+
151
+ :param async_req bool
152
+ :param str id: (required)
153
+ :return: V1PipelineTemplate
154
+ If the method is called asynchronously,
155
+ returns the request thread.
156
+ """
157
+ kwargs['_return_http_data_only'] = True
158
+ if kwargs.get('async_req'):
159
+ return self.pipeline_templates_service_get_pipeline_template_with_http_info(id, **kwargs) # noqa: E501
160
+ else:
161
+ (data) = self.pipeline_templates_service_get_pipeline_template_with_http_info(id, **kwargs) # noqa: E501
162
+ return data
163
+
164
+ def pipeline_templates_service_get_pipeline_template_with_http_info(self, id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
165
+ """pipeline_templates_service_get_pipeline_template # noqa: E501
166
+
167
+ This method makes a synchronous HTTP request by default. To make an
168
+ asynchronous HTTP request, please pass async_req=True
169
+ >>> thread = api.pipeline_templates_service_get_pipeline_template_with_http_info(id, async_req=True)
170
+ >>> result = thread.get()
171
+
172
+ :param async_req bool
173
+ :param str id: (required)
174
+ :return: V1PipelineTemplate
175
+ If the method is called asynchronously,
176
+ returns the request thread.
177
+ """
178
+
179
+ all_params = ['id'] # noqa: E501
180
+ all_params.append('async_req')
181
+ all_params.append('_return_http_data_only')
182
+ all_params.append('_preload_content')
183
+ all_params.append('_request_timeout')
184
+
185
+ params = locals()
186
+ for key, val in six.iteritems(params['kwargs']):
187
+ if key not in all_params:
188
+ raise TypeError(
189
+ "Got an unexpected keyword argument '%s'"
190
+ " to method pipeline_templates_service_get_pipeline_template" % key
191
+ )
192
+ params[key] = val
193
+ del params['kwargs']
194
+ # verify the required parameter 'id' is set
195
+ if ('id' not in params or
196
+ params['id'] is None):
197
+ raise ValueError("Missing the required parameter `id` when calling `pipeline_templates_service_get_pipeline_template`") # noqa: E501
198
+
199
+ collection_formats = {}
200
+
201
+ path_params = {}
202
+ if 'id' in params:
203
+ path_params['id'] = params['id'] # noqa: E501
204
+
205
+ query_params = []
206
+
207
+ header_params = {}
208
+
209
+ form_params = []
210
+ local_var_files = {}
211
+
212
+ body_params = None
213
+ # HTTP header `Accept`
214
+ header_params['Accept'] = self.api_client.select_header_accept(
215
+ ['application/json']) # noqa: E501
216
+
217
+ # Authentication setting
218
+ auth_settings = [] # noqa: E501
219
+
220
+ return self.api_client.call_api(
221
+ '/v1/pipeline-templates/{id}', 'GET',
222
+ path_params,
223
+ query_params,
224
+ header_params,
225
+ body=body_params,
226
+ post_params=form_params,
227
+ files=local_var_files,
228
+ response_type='V1PipelineTemplate', # noqa: E501
229
+ auth_settings=auth_settings,
230
+ async_req=params.get('async_req'),
231
+ _return_http_data_only=params.get('_return_http_data_only'),
232
+ _preload_content=params.get('_preload_content', True),
233
+ _request_timeout=params.get('_request_timeout'),
234
+ collection_formats=collection_formats)
235
+
236
+ def pipeline_templates_service_update_pipeline_template(self, body: 'PipelinetemplatesIdBody', id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
237
+ """pipeline_templates_service_update_pipeline_template # noqa: E501
238
+
239
+ This method makes a synchronous HTTP request by default. To make an
240
+ asynchronous HTTP request, please pass async_req=True
241
+ >>> thread = api.pipeline_templates_service_update_pipeline_template(body, id, async_req=True)
242
+ >>> result = thread.get()
243
+
244
+ :param async_req bool
245
+ :param PipelinetemplatesIdBody body: (required)
246
+ :param str id: (required)
247
+ :return: V1PipelineTemplate
248
+ If the method is called asynchronously,
249
+ returns the request thread.
250
+ """
251
+ kwargs['_return_http_data_only'] = True
252
+ if kwargs.get('async_req'):
253
+ return self.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, **kwargs) # noqa: E501
254
+ else:
255
+ (data) = self.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, **kwargs) # noqa: E501
256
+ return data
257
+
258
+ def pipeline_templates_service_update_pipeline_template_with_http_info(self, body: 'PipelinetemplatesIdBody', id: 'str', **kwargs) -> 'V1PipelineTemplate': # noqa: E501
259
+ """pipeline_templates_service_update_pipeline_template # noqa: E501
260
+
261
+ This method makes a synchronous HTTP request by default. To make an
262
+ asynchronous HTTP request, please pass async_req=True
263
+ >>> thread = api.pipeline_templates_service_update_pipeline_template_with_http_info(body, id, async_req=True)
264
+ >>> result = thread.get()
265
+
266
+ :param async_req bool
267
+ :param PipelinetemplatesIdBody body: (required)
268
+ :param str id: (required)
269
+ :return: V1PipelineTemplate
270
+ If the method is called asynchronously,
271
+ returns the request thread.
272
+ """
273
+
274
+ all_params = ['body', 'id'] # noqa: E501
275
+ all_params.append('async_req')
276
+ all_params.append('_return_http_data_only')
277
+ all_params.append('_preload_content')
278
+ all_params.append('_request_timeout')
279
+
280
+ params = locals()
281
+ for key, val in six.iteritems(params['kwargs']):
282
+ if key not in all_params:
283
+ raise TypeError(
284
+ "Got an unexpected keyword argument '%s'"
285
+ " to method pipeline_templates_service_update_pipeline_template" % key
286
+ )
287
+ params[key] = val
288
+ del params['kwargs']
289
+ # verify the required parameter 'body' is set
290
+ if ('body' not in params or
291
+ params['body'] is None):
292
+ raise ValueError("Missing the required parameter `body` when calling `pipeline_templates_service_update_pipeline_template`") # noqa: E501
293
+ # verify the required parameter 'id' is set
294
+ if ('id' not in params or
295
+ params['id'] is None):
296
+ raise ValueError("Missing the required parameter `id` when calling `pipeline_templates_service_update_pipeline_template`") # noqa: E501
297
+
298
+ collection_formats = {}
299
+
300
+ path_params = {}
301
+ if 'id' in params:
302
+ path_params['id'] = params['id'] # noqa: E501
303
+
304
+ query_params = []
305
+
306
+ header_params = {}
307
+
308
+ form_params = []
309
+ local_var_files = {}
310
+
311
+ body_params = None
312
+ if 'body' in params:
313
+ body_params = params['body']
314
+ # HTTP header `Accept`
315
+ header_params['Accept'] = self.api_client.select_header_accept(
316
+ ['application/json']) # noqa: E501
317
+
318
+ # HTTP header `Content-Type`
319
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
320
+ ['application/json']) # noqa: E501
321
+
322
+ # Authentication setting
323
+ auth_settings = [] # noqa: E501
324
+
325
+ return self.api_client.call_api(
326
+ '/v1/pipeline-templates/{id}', 'PUT',
327
+ path_params,
328
+ query_params,
329
+ header_params,
330
+ body=body_params,
331
+ post_params=form_params,
332
+ files=local_var_files,
333
+ response_type='V1PipelineTemplate', # noqa: E501
334
+ auth_settings=auth_settings,
335
+ async_req=params.get('async_req'),
336
+ _return_http_data_only=params.get('_return_http_data_only'),
337
+ _preload_content=params.get('_preload_content', True),
338
+ _request_timeout=params.get('_request_timeout'),
339
+ collection_formats=collection_formats)
@@ -465,6 +465,7 @@ class PipelinesServiceApi(object):
465
465
  :param str page_token:
466
466
  :param int limit:
467
467
  :param str state:
468
+ :param bool standalone:
468
469
  :return: V1ListPipelinesResponse
469
470
  If the method is called asynchronously,
470
471
  returns the request thread.
@@ -490,12 +491,13 @@ class PipelinesServiceApi(object):
490
491
  :param str page_token:
491
492
  :param int limit:
492
493
  :param str state:
494
+ :param bool standalone:
493
495
  :return: V1ListPipelinesResponse
494
496
  If the method is called asynchronously,
495
497
  returns the request thread.
496
498
  """
497
499
 
498
- all_params = ['project_id', 'parent_pipeline_id', 'page_token', 'limit', 'state'] # noqa: E501
500
+ all_params = ['project_id', 'parent_pipeline_id', 'page_token', 'limit', 'state', 'standalone'] # noqa: E501
499
501
  all_params.append('async_req')
500
502
  all_params.append('_return_http_data_only')
501
503
  all_params.append('_preload_content')
@@ -530,6 +532,8 @@ class PipelinesServiceApi(object):
530
532
  query_params.append(('limit', params['limit'])) # noqa: E501
531
533
  if 'state' in params:
532
534
  query_params.append(('state', params['state'])) # noqa: E501
535
+ if 'standalone' in params:
536
+ query_params.append(('standalone', params['standalone'])) # noqa: E501
533
537
 
534
538
  header_params = {}
535
539