lightning-sdk 0.1.30__py3-none-any.whl → 0.1.32__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 (80) hide show
  1. lightning_sdk/__init__.py +3 -1
  2. lightning_sdk/ai_hub.py +29 -0
  3. lightning_sdk/api/__init__.py +2 -0
  4. lightning_sdk/api/ai_hub_api.py +15 -0
  5. lightning_sdk/api/teamspace_api.py +35 -5
  6. lightning_sdk/api/user_api.py +8 -2
  7. lightning_sdk/cli/upload.py +10 -6
  8. lightning_sdk/lightning_cloud/openapi/__init__.py +30 -0
  9. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  10. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +105 -4
  11. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +5 -5
  12. lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py +105 -0
  13. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +674 -0
  14. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +5 -1
  15. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +396 -0
  16. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +118 -1
  17. lightning_sdk/lightning_cloud/openapi/models/__init__.py +29 -0
  18. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +15 -15
  19. lightning_sdk/lightning_cloud/openapi/models/create.py +79 -1
  20. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +121 -17
  22. lightning_sdk/lightning_cloud/openapi/models/id_engage_body.py +3 -29
  23. lightning_sdk/lightning_cloud/openapi/models/id_engage_body1.py +149 -0
  24. lightning_sdk/lightning_cloud/openapi/models/litloggermetrics_id_body.py +175 -0
  25. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +123 -0
  26. lightning_sdk/lightning_cloud/openapi/models/project_id_models_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/update.py +105 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_affiliate_link.py +79 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_app_type.py +104 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +95 -17
  32. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_request.py +201 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_response.py +123 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +79 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +175 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_delete_shared_metrics_stream_response.py +97 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +95 -17
  40. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_engagement_response.py +97 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py +69 -17
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_placement.py +2 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py +1 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_summary.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_efs_folder_data_connection.py +201 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_app.py +227 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_cloud_space.py +149 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_dataset.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_job.py +175 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_slurm_job.py +149 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_snowflake_connection.py +123 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_work.py +149 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_gcs_folder_data_connection.py +123 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_input.py +29 -3
  55. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_spec.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_python_versions_response.py +123 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_apps_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_cloud_spaces_response.py +123 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_datasets_response.py +123 -0
  61. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_jobs_response.py +123 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_slurm_jobs_response.py +123 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_snowflake_response.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +6 -6
  65. lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py +79 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_parameterization_spec.py +227 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py +123 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_transaction.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py +97 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +76 -24
  71. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +81 -3
  72. lightning_sdk/lightning_cloud/openapi/models/validate.py +81 -3
  73. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  74. lightning_sdk/teamspace.py +11 -0
  75. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/METADATA +1 -1
  76. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/RECORD +80 -48
  77. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/LICENSE +0 -0
  78. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/WHEEL +0 -0
  79. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/entry_points.txt +0 -0
  80. {lightning_sdk-0.1.30.dist-info → lightning_sdk-0.1.32.dist-info}/top_level.txt +0 -0
@@ -365,6 +365,7 @@ class JobsServiceApi(object):
365
365
  :param async_req bool
366
366
  :param str project_id: (required)
367
367
  :param str id: (required)
368
+ :param bool delete_templates:
368
369
  :return: V1DeleteDeploymentResponse
369
370
  If the method is called asynchronously,
370
371
  returns the request thread.
@@ -387,12 +388,13 @@ class JobsServiceApi(object):
387
388
  :param async_req bool
388
389
  :param str project_id: (required)
389
390
  :param str id: (required)
391
+ :param bool delete_templates:
390
392
  :return: V1DeleteDeploymentResponse
391
393
  If the method is called asynchronously,
392
394
  returns the request thread.
393
395
  """
394
396
 
395
- all_params = ['project_id', 'id'] # noqa: E501
397
+ all_params = ['project_id', 'id', 'delete_templates'] # noqa: E501
396
398
  all_params.append('async_req')
397
399
  all_params.append('_return_http_data_only')
398
400
  all_params.append('_preload_content')
@@ -425,6 +427,8 @@ class JobsServiceApi(object):
425
427
  path_params['id'] = params['id'] # noqa: E501
426
428
 
427
429
  query_params = []
430
+ if 'delete_templates' in params:
431
+ query_params.append(('deleteTemplates', params['delete_templates'])) # noqa: E501
428
432
 
429
433
  header_params = {}
430
434
 
@@ -479,6 +479,103 @@ class LitLoggerServiceApi(object):
479
479
  _request_timeout=params.get('_request_timeout'),
480
480
  collection_formats=collection_formats)
481
481
 
482
+ def lit_logger_service_create_shared_metrics_stream(self, body: 'V1CreateSharedMetricsStreamRequest', **kwargs) -> 'V1CreateSharedMetricsStreamResponse': # noqa: E501
483
+ """lit_logger_service_create_shared_metrics_stream # noqa: E501
484
+
485
+ This method makes a synchronous HTTP request by default. To make an
486
+ asynchronous HTTP request, please pass async_req=True
487
+ >>> thread = api.lit_logger_service_create_shared_metrics_stream(body, async_req=True)
488
+ >>> result = thread.get()
489
+
490
+ :param async_req bool
491
+ :param V1CreateSharedMetricsStreamRequest body: (required)
492
+ :return: V1CreateSharedMetricsStreamResponse
493
+ If the method is called asynchronously,
494
+ returns the request thread.
495
+ """
496
+ kwargs['_return_http_data_only'] = True
497
+ if kwargs.get('async_req'):
498
+ return self.lit_logger_service_create_shared_metrics_stream_with_http_info(body, **kwargs) # noqa: E501
499
+ else:
500
+ (data) = self.lit_logger_service_create_shared_metrics_stream_with_http_info(body, **kwargs) # noqa: E501
501
+ return data
502
+
503
+ def lit_logger_service_create_shared_metrics_stream_with_http_info(self, body: 'V1CreateSharedMetricsStreamRequest', **kwargs) -> 'V1CreateSharedMetricsStreamResponse': # noqa: E501
504
+ """lit_logger_service_create_shared_metrics_stream # noqa: E501
505
+
506
+ This method makes a synchronous HTTP request by default. To make an
507
+ asynchronous HTTP request, please pass async_req=True
508
+ >>> thread = api.lit_logger_service_create_shared_metrics_stream_with_http_info(body, async_req=True)
509
+ >>> result = thread.get()
510
+
511
+ :param async_req bool
512
+ :param V1CreateSharedMetricsStreamRequest body: (required)
513
+ :return: V1CreateSharedMetricsStreamResponse
514
+ If the method is called asynchronously,
515
+ returns the request thread.
516
+ """
517
+
518
+ all_params = ['body'] # noqa: E501
519
+ all_params.append('async_req')
520
+ all_params.append('_return_http_data_only')
521
+ all_params.append('_preload_content')
522
+ all_params.append('_request_timeout')
523
+
524
+ params = locals()
525
+ for key, val in six.iteritems(params['kwargs']):
526
+ if key not in all_params:
527
+ raise TypeError(
528
+ "Got an unexpected keyword argument '%s'"
529
+ " to method lit_logger_service_create_shared_metrics_stream" % key
530
+ )
531
+ params[key] = val
532
+ del params['kwargs']
533
+ # verify the required parameter 'body' is set
534
+ if ('body' not in params or
535
+ params['body'] is None):
536
+ raise ValueError("Missing the required parameter `body` when calling `lit_logger_service_create_shared_metrics_stream`") # noqa: E501
537
+
538
+ collection_formats = {}
539
+
540
+ path_params = {}
541
+
542
+ query_params = []
543
+
544
+ header_params = {}
545
+
546
+ form_params = []
547
+ local_var_files = {}
548
+
549
+ body_params = None
550
+ if 'body' in params:
551
+ body_params = params['body']
552
+ # HTTP header `Accept`
553
+ header_params['Accept'] = self.api_client.select_header_accept(
554
+ ['application/json']) # noqa: E501
555
+
556
+ # HTTP header `Content-Type`
557
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
558
+ ['application/json']) # noqa: E501
559
+
560
+ # Authentication setting
561
+ auth_settings = [] # noqa: E501
562
+
563
+ return self.api_client.call_api(
564
+ '/v1/share/litlogger-metrics', 'POST',
565
+ path_params,
566
+ query_params,
567
+ header_params,
568
+ body=body_params,
569
+ post_params=form_params,
570
+ files=local_var_files,
571
+ response_type='V1CreateSharedMetricsStreamResponse', # noqa: E501
572
+ auth_settings=auth_settings,
573
+ async_req=params.get('async_req'),
574
+ _return_http_data_only=params.get('_return_http_data_only'),
575
+ _preload_content=params.get('_preload_content', True),
576
+ _request_timeout=params.get('_request_timeout'),
577
+ collection_formats=collection_formats)
578
+
482
579
  def lit_logger_service_delete_logger_artifact(self, project_id: 'str', metrics_stream_id: 'str', id: 'str', **kwargs) -> 'V1DeleteLoggerArtifactResponse': # noqa: E501
483
580
  """lit_logger_service_delete_logger_artifact # noqa: E501
484
581
 
@@ -693,6 +790,99 @@ class LitLoggerServiceApi(object):
693
790
  _request_timeout=params.get('_request_timeout'),
694
791
  collection_formats=collection_formats)
695
792
 
793
+ def lit_logger_service_delete_shared_metrics_stream(self, id: 'str', **kwargs) -> 'V1DeleteSharedMetricsStreamResponse': # noqa: E501
794
+ """lit_logger_service_delete_shared_metrics_stream # noqa: E501
795
+
796
+ This method makes a synchronous HTTP request by default. To make an
797
+ asynchronous HTTP request, please pass async_req=True
798
+ >>> thread = api.lit_logger_service_delete_shared_metrics_stream(id, async_req=True)
799
+ >>> result = thread.get()
800
+
801
+ :param async_req bool
802
+ :param str id: (required)
803
+ :return: V1DeleteSharedMetricsStreamResponse
804
+ If the method is called asynchronously,
805
+ returns the request thread.
806
+ """
807
+ kwargs['_return_http_data_only'] = True
808
+ if kwargs.get('async_req'):
809
+ return self.lit_logger_service_delete_shared_metrics_stream_with_http_info(id, **kwargs) # noqa: E501
810
+ else:
811
+ (data) = self.lit_logger_service_delete_shared_metrics_stream_with_http_info(id, **kwargs) # noqa: E501
812
+ return data
813
+
814
+ def lit_logger_service_delete_shared_metrics_stream_with_http_info(self, id: 'str', **kwargs) -> 'V1DeleteSharedMetricsStreamResponse': # noqa: E501
815
+ """lit_logger_service_delete_shared_metrics_stream # noqa: E501
816
+
817
+ This method makes a synchronous HTTP request by default. To make an
818
+ asynchronous HTTP request, please pass async_req=True
819
+ >>> thread = api.lit_logger_service_delete_shared_metrics_stream_with_http_info(id, async_req=True)
820
+ >>> result = thread.get()
821
+
822
+ :param async_req bool
823
+ :param str id: (required)
824
+ :return: V1DeleteSharedMetricsStreamResponse
825
+ If the method is called asynchronously,
826
+ returns the request thread.
827
+ """
828
+
829
+ all_params = ['id'] # noqa: E501
830
+ all_params.append('async_req')
831
+ all_params.append('_return_http_data_only')
832
+ all_params.append('_preload_content')
833
+ all_params.append('_request_timeout')
834
+
835
+ params = locals()
836
+ for key, val in six.iteritems(params['kwargs']):
837
+ if key not in all_params:
838
+ raise TypeError(
839
+ "Got an unexpected keyword argument '%s'"
840
+ " to method lit_logger_service_delete_shared_metrics_stream" % key
841
+ )
842
+ params[key] = val
843
+ del params['kwargs']
844
+ # verify the required parameter 'id' is set
845
+ if ('id' not in params or
846
+ params['id'] is None):
847
+ raise ValueError("Missing the required parameter `id` when calling `lit_logger_service_delete_shared_metrics_stream`") # noqa: E501
848
+
849
+ collection_formats = {}
850
+
851
+ path_params = {}
852
+ if 'id' in params:
853
+ path_params['id'] = params['id'] # noqa: E501
854
+
855
+ query_params = []
856
+
857
+ header_params = {}
858
+
859
+ form_params = []
860
+ local_var_files = {}
861
+
862
+ body_params = None
863
+ # HTTP header `Accept`
864
+ header_params['Accept'] = self.api_client.select_header_accept(
865
+ ['application/json']) # noqa: E501
866
+
867
+ # Authentication setting
868
+ auth_settings = [] # noqa: E501
869
+
870
+ return self.api_client.call_api(
871
+ '/v1/share/litlogger-metrics/{id}', 'DELETE',
872
+ path_params,
873
+ query_params,
874
+ header_params,
875
+ body=body_params,
876
+ post_params=form_params,
877
+ files=local_var_files,
878
+ response_type='V1DeleteSharedMetricsStreamResponse', # noqa: E501
879
+ auth_settings=auth_settings,
880
+ async_req=params.get('async_req'),
881
+ _return_http_data_only=params.get('_return_http_data_only'),
882
+ _preload_content=params.get('_preload_content', True),
883
+ _request_timeout=params.get('_request_timeout'),
884
+ collection_formats=collection_formats)
885
+
696
886
  def lit_logger_service_get_logger_metrics(self, project_id: 'str', **kwargs) -> 'V1GetLoggerMetricsResponse': # noqa: E501
697
887
  """lit_logger_service_get_logger_metrics # noqa: E501
698
888
 
@@ -819,6 +1009,107 @@ class LitLoggerServiceApi(object):
819
1009
  _request_timeout=params.get('_request_timeout'),
820
1010
  collection_formats=collection_formats)
821
1011
 
1012
+ def lit_logger_service_get_shared_metrics_stream(self, id: 'str', **kwargs) -> 'V1MetricsStream': # noqa: E501
1013
+ """lit_logger_service_get_shared_metrics_stream # noqa: E501
1014
+
1015
+ This method makes a synchronous HTTP request by default. To make an
1016
+ asynchronous HTTP request, please pass async_req=True
1017
+ >>> thread = api.lit_logger_service_get_shared_metrics_stream(id, async_req=True)
1018
+ >>> result = thread.get()
1019
+
1020
+ :param async_req bool
1021
+ :param str id: (required)
1022
+ :param str org_id:
1023
+ :param str user_id:
1024
+ :return: V1MetricsStream
1025
+ If the method is called asynchronously,
1026
+ returns the request thread.
1027
+ """
1028
+ kwargs['_return_http_data_only'] = True
1029
+ if kwargs.get('async_req'):
1030
+ return self.lit_logger_service_get_shared_metrics_stream_with_http_info(id, **kwargs) # noqa: E501
1031
+ else:
1032
+ (data) = self.lit_logger_service_get_shared_metrics_stream_with_http_info(id, **kwargs) # noqa: E501
1033
+ return data
1034
+
1035
+ def lit_logger_service_get_shared_metrics_stream_with_http_info(self, id: 'str', **kwargs) -> 'V1MetricsStream': # noqa: E501
1036
+ """lit_logger_service_get_shared_metrics_stream # noqa: E501
1037
+
1038
+ This method makes a synchronous HTTP request by default. To make an
1039
+ asynchronous HTTP request, please pass async_req=True
1040
+ >>> thread = api.lit_logger_service_get_shared_metrics_stream_with_http_info(id, async_req=True)
1041
+ >>> result = thread.get()
1042
+
1043
+ :param async_req bool
1044
+ :param str id: (required)
1045
+ :param str org_id:
1046
+ :param str user_id:
1047
+ :return: V1MetricsStream
1048
+ If the method is called asynchronously,
1049
+ returns the request thread.
1050
+ """
1051
+
1052
+ all_params = ['id', 'org_id', 'user_id'] # noqa: E501
1053
+ all_params.append('async_req')
1054
+ all_params.append('_return_http_data_only')
1055
+ all_params.append('_preload_content')
1056
+ all_params.append('_request_timeout')
1057
+
1058
+ params = locals()
1059
+ for key, val in six.iteritems(params['kwargs']):
1060
+ if key not in all_params:
1061
+ raise TypeError(
1062
+ "Got an unexpected keyword argument '%s'"
1063
+ " to method lit_logger_service_get_shared_metrics_stream" % key
1064
+ )
1065
+ params[key] = val
1066
+ del params['kwargs']
1067
+ # verify the required parameter 'id' is set
1068
+ if ('id' not in params or
1069
+ params['id'] is None):
1070
+ raise ValueError("Missing the required parameter `id` when calling `lit_logger_service_get_shared_metrics_stream`") # noqa: E501
1071
+
1072
+ collection_formats = {}
1073
+
1074
+ path_params = {}
1075
+ if 'id' in params:
1076
+ path_params['id'] = params['id'] # noqa: E501
1077
+
1078
+ query_params = []
1079
+ if 'org_id' in params:
1080
+ query_params.append(('orgId', params['org_id'])) # noqa: E501
1081
+ if 'user_id' in params:
1082
+ query_params.append(('userId', params['user_id'])) # noqa: E501
1083
+
1084
+ header_params = {}
1085
+
1086
+ form_params = []
1087
+ local_var_files = {}
1088
+
1089
+ body_params = None
1090
+ # HTTP header `Accept`
1091
+ header_params['Accept'] = self.api_client.select_header_accept(
1092
+ ['application/json']) # noqa: E501
1093
+
1094
+ # Authentication setting
1095
+ auth_settings = [] # noqa: E501
1096
+
1097
+ return self.api_client.call_api(
1098
+ '/v1/share/litlogger-metrics/{id}', 'GET',
1099
+ path_params,
1100
+ query_params,
1101
+ header_params,
1102
+ body=body_params,
1103
+ post_params=form_params,
1104
+ files=local_var_files,
1105
+ response_type='V1MetricsStream', # noqa: E501
1106
+ auth_settings=auth_settings,
1107
+ async_req=params.get('async_req'),
1108
+ _return_http_data_only=params.get('_return_http_data_only'),
1109
+ _preload_content=params.get('_preload_content', True),
1110
+ _request_timeout=params.get('_request_timeout'),
1111
+ collection_formats=collection_formats)
1112
+
822
1113
  def lit_logger_service_list_logger_artifacts(self, project_id: 'str', metrics_stream_id: 'str', **kwargs) -> 'V1ListLoggerArtifactResponse': # noqa: E501
823
1114
  """lit_logger_service_list_logger_artifacts # noqa: E501
824
1115
 
@@ -1137,3 +1428,108 @@ class LitLoggerServiceApi(object):
1137
1428
  _preload_content=params.get('_preload_content', True),
1138
1429
  _request_timeout=params.get('_request_timeout'),
1139
1430
  collection_formats=collection_formats)
1431
+
1432
+ def lit_logger_service_update_shared_metrics_stream(self, body: 'LitloggermetricsIdBody', id: 'str', **kwargs) -> 'V1UpdateSharedMetricsStreamResponse': # noqa: E501
1433
+ """lit_logger_service_update_shared_metrics_stream # noqa: E501
1434
+
1435
+ This method makes a synchronous HTTP request by default. To make an
1436
+ asynchronous HTTP request, please pass async_req=True
1437
+ >>> thread = api.lit_logger_service_update_shared_metrics_stream(body, id, async_req=True)
1438
+ >>> result = thread.get()
1439
+
1440
+ :param async_req bool
1441
+ :param LitloggermetricsIdBody body: (required)
1442
+ :param str id: (required)
1443
+ :return: V1UpdateSharedMetricsStreamResponse
1444
+ If the method is called asynchronously,
1445
+ returns the request thread.
1446
+ """
1447
+ kwargs['_return_http_data_only'] = True
1448
+ if kwargs.get('async_req'):
1449
+ return self.lit_logger_service_update_shared_metrics_stream_with_http_info(body, id, **kwargs) # noqa: E501
1450
+ else:
1451
+ (data) = self.lit_logger_service_update_shared_metrics_stream_with_http_info(body, id, **kwargs) # noqa: E501
1452
+ return data
1453
+
1454
+ def lit_logger_service_update_shared_metrics_stream_with_http_info(self, body: 'LitloggermetricsIdBody', id: 'str', **kwargs) -> 'V1UpdateSharedMetricsStreamResponse': # noqa: E501
1455
+ """lit_logger_service_update_shared_metrics_stream # noqa: E501
1456
+
1457
+ This method makes a synchronous HTTP request by default. To make an
1458
+ asynchronous HTTP request, please pass async_req=True
1459
+ >>> thread = api.lit_logger_service_update_shared_metrics_stream_with_http_info(body, id, async_req=True)
1460
+ >>> result = thread.get()
1461
+
1462
+ :param async_req bool
1463
+ :param LitloggermetricsIdBody body: (required)
1464
+ :param str id: (required)
1465
+ :return: V1UpdateSharedMetricsStreamResponse
1466
+ If the method is called asynchronously,
1467
+ returns the request thread.
1468
+ """
1469
+
1470
+ all_params = ['body', 'id'] # noqa: E501
1471
+ all_params.append('async_req')
1472
+ all_params.append('_return_http_data_only')
1473
+ all_params.append('_preload_content')
1474
+ all_params.append('_request_timeout')
1475
+
1476
+ params = locals()
1477
+ for key, val in six.iteritems(params['kwargs']):
1478
+ if key not in all_params:
1479
+ raise TypeError(
1480
+ "Got an unexpected keyword argument '%s'"
1481
+ " to method lit_logger_service_update_shared_metrics_stream" % key
1482
+ )
1483
+ params[key] = val
1484
+ del params['kwargs']
1485
+ # verify the required parameter 'body' is set
1486
+ if ('body' not in params or
1487
+ params['body'] is None):
1488
+ raise ValueError("Missing the required parameter `body` when calling `lit_logger_service_update_shared_metrics_stream`") # noqa: E501
1489
+ # verify the required parameter 'id' is set
1490
+ if ('id' not in params or
1491
+ params['id'] is None):
1492
+ raise ValueError("Missing the required parameter `id` when calling `lit_logger_service_update_shared_metrics_stream`") # noqa: E501
1493
+
1494
+ collection_formats = {}
1495
+
1496
+ path_params = {}
1497
+ if 'id' in params:
1498
+ path_params['id'] = params['id'] # noqa: E501
1499
+
1500
+ query_params = []
1501
+
1502
+ header_params = {}
1503
+
1504
+ form_params = []
1505
+ local_var_files = {}
1506
+
1507
+ body_params = None
1508
+ if 'body' in params:
1509
+ body_params = params['body']
1510
+ # HTTP header `Accept`
1511
+ header_params['Accept'] = self.api_client.select_header_accept(
1512
+ ['application/json']) # noqa: E501
1513
+
1514
+ # HTTP header `Content-Type`
1515
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1516
+ ['application/json']) # noqa: E501
1517
+
1518
+ # Authentication setting
1519
+ auth_settings = [] # noqa: E501
1520
+
1521
+ return self.api_client.call_api(
1522
+ '/v1/share/litlogger-metrics/{id}', 'PUT',
1523
+ path_params,
1524
+ query_params,
1525
+ header_params,
1526
+ body=body_params,
1527
+ post_params=form_params,
1528
+ files=local_var_files,
1529
+ response_type='V1UpdateSharedMetricsStreamResponse', # noqa: E501
1530
+ auth_settings=auth_settings,
1531
+ async_req=params.get('async_req'),
1532
+ _return_http_data_only=params.get('_return_http_data_only'),
1533
+ _preload_content=params.get('_preload_content', True),
1534
+ _request_timeout=params.get('_request_timeout'),
1535
+ collection_formats=collection_formats)
@@ -398,6 +398,119 @@ class ModelsStoreApi(object):
398
398
  _request_timeout=params.get('_request_timeout'),
399
399
  collection_formats=collection_formats)
400
400
 
401
+ def models_store_create_model_version(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
402
+ """models_store_create_model_version # noqa: E501
403
+
404
+ This method makes a synchronous HTTP request by default. To make an
405
+ asynchronous HTTP request, please pass async_req=True
406
+ >>> thread = api.models_store_create_model_version(body, project_id, model_id, async_req=True)
407
+ >>> result = thread.get()
408
+
409
+ :param async_req bool
410
+ :param ModelIdVersionsBody body: (required)
411
+ :param str project_id: (required)
412
+ :param str model_id: (required)
413
+ :return: V1ModelVersionArchive
414
+ If the method is called asynchronously,
415
+ returns the request thread.
416
+ """
417
+ kwargs['_return_http_data_only'] = True
418
+ if kwargs.get('async_req'):
419
+ return self.models_store_create_model_version_with_http_info(body, project_id, model_id, **kwargs) # noqa: E501
420
+ else:
421
+ (data) = self.models_store_create_model_version_with_http_info(body, project_id, model_id, **kwargs) # noqa: E501
422
+ return data
423
+
424
+ def models_store_create_model_version_with_http_info(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive': # noqa: E501
425
+ """models_store_create_model_version # noqa: E501
426
+
427
+ This method makes a synchronous HTTP request by default. To make an
428
+ asynchronous HTTP request, please pass async_req=True
429
+ >>> thread = api.models_store_create_model_version_with_http_info(body, project_id, model_id, async_req=True)
430
+ >>> result = thread.get()
431
+
432
+ :param async_req bool
433
+ :param ModelIdVersionsBody body: (required)
434
+ :param str project_id: (required)
435
+ :param str model_id: (required)
436
+ :return: V1ModelVersionArchive
437
+ If the method is called asynchronously,
438
+ returns the request thread.
439
+ """
440
+
441
+ all_params = ['body', 'project_id', 'model_id'] # noqa: E501
442
+ all_params.append('async_req')
443
+ all_params.append('_return_http_data_only')
444
+ all_params.append('_preload_content')
445
+ all_params.append('_request_timeout')
446
+
447
+ params = locals()
448
+ for key, val in six.iteritems(params['kwargs']):
449
+ if key not in all_params:
450
+ raise TypeError(
451
+ "Got an unexpected keyword argument '%s'"
452
+ " to method models_store_create_model_version" % key
453
+ )
454
+ params[key] = val
455
+ del params['kwargs']
456
+ # verify the required parameter 'body' is set
457
+ if ('body' not in params or
458
+ params['body'] is None):
459
+ raise ValueError("Missing the required parameter `body` when calling `models_store_create_model_version`") # noqa: E501
460
+ # verify the required parameter 'project_id' is set
461
+ if ('project_id' not in params or
462
+ params['project_id'] is None):
463
+ raise ValueError("Missing the required parameter `project_id` when calling `models_store_create_model_version`") # noqa: E501
464
+ # verify the required parameter 'model_id' is set
465
+ if ('model_id' not in params or
466
+ params['model_id'] is None):
467
+ raise ValueError("Missing the required parameter `model_id` when calling `models_store_create_model_version`") # noqa: E501
468
+
469
+ collection_formats = {}
470
+
471
+ path_params = {}
472
+ if 'project_id' in params:
473
+ path_params['projectId'] = params['project_id'] # noqa: E501
474
+ if 'model_id' in params:
475
+ path_params['modelId'] = params['model_id'] # noqa: E501
476
+
477
+ query_params = []
478
+
479
+ header_params = {}
480
+
481
+ form_params = []
482
+ local_var_files = {}
483
+
484
+ body_params = None
485
+ if 'body' in params:
486
+ body_params = params['body']
487
+ # HTTP header `Accept`
488
+ header_params['Accept'] = self.api_client.select_header_accept(
489
+ ['application/json']) # noqa: E501
490
+
491
+ # HTTP header `Content-Type`
492
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
493
+ ['application/json']) # noqa: E501
494
+
495
+ # Authentication setting
496
+ auth_settings = [] # noqa: E501
497
+
498
+ return self.api_client.call_api(
499
+ '/v1/projects/{projectId}/models/{modelId}/versions', 'POST',
500
+ path_params,
501
+ query_params,
502
+ header_params,
503
+ body=body_params,
504
+ post_params=form_params,
505
+ files=local_var_files,
506
+ response_type='V1ModelVersionArchive', # noqa: E501
507
+ auth_settings=auth_settings,
508
+ async_req=params.get('async_req'),
509
+ _return_http_data_only=params.get('_return_http_data_only'),
510
+ _preload_content=params.get('_preload_content', True),
511
+ _request_timeout=params.get('_request_timeout'),
512
+ collection_formats=collection_formats)
513
+
401
514
  def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse': # noqa: E501
402
515
  """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files. # noqa: E501
403
516
 
@@ -1280,6 +1393,7 @@ class ModelsStoreApi(object):
1280
1393
 
1281
1394
  :param async_req bool
1282
1395
  :param str project_id: (required)
1396
+ :param str name:
1283
1397
  :return: V1ListModelsResponse
1284
1398
  If the method is called asynchronously,
1285
1399
  returns the request thread.
@@ -1301,12 +1415,13 @@ class ModelsStoreApi(object):
1301
1415
 
1302
1416
  :param async_req bool
1303
1417
  :param str project_id: (required)
1418
+ :param str name:
1304
1419
  :return: V1ListModelsResponse
1305
1420
  If the method is called asynchronously,
1306
1421
  returns the request thread.
1307
1422
  """
1308
1423
 
1309
- all_params = ['project_id'] # noqa: E501
1424
+ all_params = ['project_id', 'name'] # noqa: E501
1310
1425
  all_params.append('async_req')
1311
1426
  all_params.append('_return_http_data_only')
1312
1427
  all_params.append('_preload_content')
@@ -1333,6 +1448,8 @@ class ModelsStoreApi(object):
1333
1448
  path_params['projectId'] = params['project_id'] # noqa: E501
1334
1449
 
1335
1450
  query_params = []
1451
+ if 'name' in params:
1452
+ query_params.append(('name', params['name'])) # noqa: E501
1336
1453
 
1337
1454
  header_params = {}
1338
1455