lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc1__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 (83) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +9 -2
  3. lightning_sdk/api/deployment_api.py +9 -9
  4. lightning_sdk/api/llm_api.py +23 -13
  5. lightning_sdk/api/pipeline_api.py +31 -11
  6. lightning_sdk/api/studio_api.py +4 -0
  7. lightning_sdk/base_studio.py +22 -6
  8. lightning_sdk/deployment/deployment.py +17 -7
  9. lightning_sdk/lightning_cloud/openapi/__init__.py +18 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +2 -0
  11. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  12. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  13. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +9 -1
  14. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +13 -1
  15. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  17. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  18. lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py +258 -0
  19. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  20. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  23. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +53 -1
  25. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  27. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py +103 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +29 -3
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_volume_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  56. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +149 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py +104 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +1 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_update_volume_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +317 -31
  70. lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.py +123 -0
  71. lightning_sdk/llm/llm.py +118 -115
  72. lightning_sdk/llm/public_assistants.json +8 -0
  73. lightning_sdk/pipeline/pipeline.py +17 -2
  74. lightning_sdk/pipeline/printer.py +11 -10
  75. lightning_sdk/pipeline/steps.py +4 -1
  76. lightning_sdk/pipeline/utils.py +29 -4
  77. lightning_sdk/studio.py +3 -0
  78. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/METADATA +1 -1
  79. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/RECORD +83 -64
  80. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/LICENSE +0 -0
  81. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/WHEEL +0 -0
  82. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/entry_points.txt +0 -0
  83. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/top_level.txt +0 -0
@@ -952,6 +952,111 @@ class AssistantsServiceApi(object):
952
952
  _request_timeout=params.get('_request_timeout'),
953
953
  collection_formats=collection_formats)
954
954
 
955
+ def assistants_service_get_managed_model_assistant(self, model_name: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
956
+ """Each managed model has a dedicated assistant for direct interaction. By using user_name, org_name, or model_provider as query parameters, this endpoint retrieves that specific assistant only—excluding any other assistants that may use the same model. # noqa: E501
957
+
958
+ This method makes a synchronous HTTP request by default. To make an
959
+ asynchronous HTTP request, please pass async_req=True
960
+ >>> thread = api.assistants_service_get_managed_model_assistant(model_name, async_req=True)
961
+ >>> result = thread.get()
962
+
963
+ :param async_req bool
964
+ :param str model_name: (required)
965
+ :param str user_name:
966
+ :param str org_name:
967
+ :param str model_provider:
968
+ :return: V1Assistant
969
+ If the method is called asynchronously,
970
+ returns the request thread.
971
+ """
972
+ kwargs['_return_http_data_only'] = True
973
+ if kwargs.get('async_req'):
974
+ return self.assistants_service_get_managed_model_assistant_with_http_info(model_name, **kwargs) # noqa: E501
975
+ else:
976
+ (data) = self.assistants_service_get_managed_model_assistant_with_http_info(model_name, **kwargs) # noqa: E501
977
+ return data
978
+
979
+ def assistants_service_get_managed_model_assistant_with_http_info(self, model_name: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
980
+ """Each managed model has a dedicated assistant for direct interaction. By using user_name, org_name, or model_provider as query parameters, this endpoint retrieves that specific assistant only—excluding any other assistants that may use the same model. # noqa: E501
981
+
982
+ This method makes a synchronous HTTP request by default. To make an
983
+ asynchronous HTTP request, please pass async_req=True
984
+ >>> thread = api.assistants_service_get_managed_model_assistant_with_http_info(model_name, async_req=True)
985
+ >>> result = thread.get()
986
+
987
+ :param async_req bool
988
+ :param str model_name: (required)
989
+ :param str user_name:
990
+ :param str org_name:
991
+ :param str model_provider:
992
+ :return: V1Assistant
993
+ If the method is called asynchronously,
994
+ returns the request thread.
995
+ """
996
+
997
+ all_params = ['model_name', 'user_name', 'org_name', 'model_provider'] # noqa: E501
998
+ all_params.append('async_req')
999
+ all_params.append('_return_http_data_only')
1000
+ all_params.append('_preload_content')
1001
+ all_params.append('_request_timeout')
1002
+
1003
+ params = locals()
1004
+ for key, val in six.iteritems(params['kwargs']):
1005
+ if key not in all_params:
1006
+ raise TypeError(
1007
+ "Got an unexpected keyword argument '%s'"
1008
+ " to method assistants_service_get_managed_model_assistant" % key
1009
+ )
1010
+ params[key] = val
1011
+ del params['kwargs']
1012
+ # verify the required parameter 'model_name' is set
1013
+ if ('model_name' not in params or
1014
+ params['model_name'] is None):
1015
+ raise ValueError("Missing the required parameter `model_name` when calling `assistants_service_get_managed_model_assistant`") # noqa: E501
1016
+
1017
+ collection_formats = {}
1018
+
1019
+ path_params = {}
1020
+ if 'model_name' in params:
1021
+ path_params['modelName'] = params['model_name'] # noqa: E501
1022
+
1023
+ query_params = []
1024
+ if 'user_name' in params:
1025
+ query_params.append(('userName', params['user_name'])) # noqa: E501
1026
+ if 'org_name' in params:
1027
+ query_params.append(('orgName', params['org_name'])) # noqa: E501
1028
+ if 'model_provider' in params:
1029
+ query_params.append(('modelProvider', params['model_provider'])) # noqa: E501
1030
+
1031
+ header_params = {}
1032
+
1033
+ form_params = []
1034
+ local_var_files = {}
1035
+
1036
+ body_params = None
1037
+ # HTTP header `Accept`
1038
+ header_params['Accept'] = self.api_client.select_header_accept(
1039
+ ['application/json']) # noqa: E501
1040
+
1041
+ # Authentication setting
1042
+ auth_settings = [] # noqa: E501
1043
+
1044
+ return self.api_client.call_api(
1045
+ '/v1/agents/managed-model/{modelName}', 'GET',
1046
+ path_params,
1047
+ query_params,
1048
+ header_params,
1049
+ body=body_params,
1050
+ post_params=form_params,
1051
+ files=local_var_files,
1052
+ response_type='V1Assistant', # noqa: E501
1053
+ auth_settings=auth_settings,
1054
+ async_req=params.get('async_req'),
1055
+ _return_http_data_only=params.get('_return_http_data_only'),
1056
+ _preload_content=params.get('_preload_content', True),
1057
+ _request_timeout=params.get('_request_timeout'),
1058
+ collection_formats=collection_formats)
1059
+
955
1060
  def assistants_service_get_managed_model_by_name(self, project_id: 'str', managed_endpoint_id: 'str', name: 'str', **kwargs) -> 'V1ManagedModel': # noqa: E501
956
1061
  """assistants_service_get_managed_model_by_name # noqa: E501
957
1062
 
@@ -1169,6 +1274,8 @@ class AssistantsServiceApi(object):
1169
1274
  :param bool published:
1170
1275
  :param str internal_name:
1171
1276
  :param str user_id:
1277
+ :param bool cloudy_compatible:
1278
+ :param bool model_assistants_only:
1172
1279
  :return: V1ListAssistantsResponse
1173
1280
  If the method is called asynchronously,
1174
1281
  returns the request thread.
@@ -1195,12 +1302,14 @@ class AssistantsServiceApi(object):
1195
1302
  :param bool published:
1196
1303
  :param str internal_name:
1197
1304
  :param str user_id:
1305
+ :param bool cloudy_compatible:
1306
+ :param bool model_assistants_only:
1198
1307
  :return: V1ListAssistantsResponse
1199
1308
  If the method is called asynchronously,
1200
1309
  returns the request thread.
1201
1310
  """
1202
1311
 
1203
- all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name', 'user_id'] # noqa: E501
1312
+ all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name', 'user_id', 'cloudy_compatible', 'model_assistants_only'] # noqa: E501
1204
1313
  all_params.append('async_req')
1205
1314
  all_params.append('_return_http_data_only')
1206
1315
  all_params.append('_preload_content')
@@ -1233,6 +1342,10 @@ class AssistantsServiceApi(object):
1233
1342
  query_params.append(('internalName', params['internal_name'])) # noqa: E501
1234
1343
  if 'user_id' in params:
1235
1344
  query_params.append(('userId', params['user_id'])) # noqa: E501
1345
+ if 'cloudy_compatible' in params:
1346
+ query_params.append(('cloudyCompatible', params['cloudy_compatible'])) # noqa: E501
1347
+ if 'model_assistants_only' in params:
1348
+ query_params.append(('modelAssistantsOnly', params['model_assistants_only'])) # noqa: E501
1236
1349
 
1237
1350
  header_params = {}
1238
1351
 
@@ -0,0 +1,129 @@
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 CloudyServiceApi(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 cloudy_service_list_cloudy_experts(self, **kwargs) -> 'V1ListCloudyExpertsResponse': # noqa: E501
47
+ """Lists all available Cloudy Experts This endpoint returns a list of all Cloudy Experts that can be used to generate cloudy responses # 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.cloudy_service_list_cloudy_experts(async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :return: V1ListCloudyExpertsResponse
56
+ If the method is called asynchronously,
57
+ returns the request thread.
58
+ """
59
+ kwargs['_return_http_data_only'] = True
60
+ if kwargs.get('async_req'):
61
+ return self.cloudy_service_list_cloudy_experts_with_http_info(**kwargs) # noqa: E501
62
+ else:
63
+ (data) = self.cloudy_service_list_cloudy_experts_with_http_info(**kwargs) # noqa: E501
64
+ return data
65
+
66
+ def cloudy_service_list_cloudy_experts_with_http_info(self, **kwargs) -> 'V1ListCloudyExpertsResponse': # noqa: E501
67
+ """Lists all available Cloudy Experts This endpoint returns a list of all Cloudy Experts that can be used to generate cloudy responses # noqa: E501
68
+
69
+ This method makes a synchronous HTTP request by default. To make an
70
+ asynchronous HTTP request, please pass async_req=True
71
+ >>> thread = api.cloudy_service_list_cloudy_experts_with_http_info(async_req=True)
72
+ >>> result = thread.get()
73
+
74
+ :param async_req bool
75
+ :return: V1ListCloudyExpertsResponse
76
+ If the method is called asynchronously,
77
+ returns the request thread.
78
+ """
79
+
80
+ all_params = [] # noqa: E501
81
+ all_params.append('async_req')
82
+ all_params.append('_return_http_data_only')
83
+ all_params.append('_preload_content')
84
+ all_params.append('_request_timeout')
85
+
86
+ params = locals()
87
+ for key, val in six.iteritems(params['kwargs']):
88
+ if key not in all_params:
89
+ raise TypeError(
90
+ "Got an unexpected keyword argument '%s'"
91
+ " to method cloudy_service_list_cloudy_experts" % key
92
+ )
93
+ params[key] = val
94
+ del params['kwargs']
95
+
96
+ collection_formats = {}
97
+
98
+ path_params = {}
99
+
100
+ query_params = []
101
+
102
+ header_params = {}
103
+
104
+ form_params = []
105
+ local_var_files = {}
106
+
107
+ body_params = None
108
+ # HTTP header `Accept`
109
+ header_params['Accept'] = self.api_client.select_header_accept(
110
+ ['application/json']) # noqa: E501
111
+
112
+ # Authentication setting
113
+ auth_settings = [] # noqa: E501
114
+
115
+ return self.api_client.call_api(
116
+ '/v1/cloudy/experts', 'GET',
117
+ path_params,
118
+ query_params,
119
+ header_params,
120
+ body=body_params,
121
+ post_params=form_params,
122
+ files=local_var_files,
123
+ response_type='V1ListCloudyExpertsResponse', # noqa: E501
124
+ auth_settings=auth_settings,
125
+ async_req=params.get('async_req'),
126
+ _return_http_data_only=params.get('_return_http_data_only'),
127
+ _preload_content=params.get('_preload_content', True),
128
+ _request_timeout=params.get('_request_timeout'),
129
+ collection_formats=collection_formats)
@@ -2428,6 +2428,8 @@ class ClusterServiceApi(object):
2428
2428
  :param datetime start_time:
2429
2429
  :param datetime end_time:
2430
2430
  :param bool available_only:
2431
+ :param bool from_aggregate:
2432
+ :param str apparent_provider:
2431
2433
  :return: V1ListClusterCapacityReservationsResponse
2432
2434
  If the method is called asynchronously,
2433
2435
  returns the request thread.
@@ -2453,12 +2455,14 @@ class ClusterServiceApi(object):
2453
2455
  :param datetime start_time:
2454
2456
  :param datetime end_time:
2455
2457
  :param bool available_only:
2458
+ :param bool from_aggregate:
2459
+ :param str apparent_provider:
2456
2460
  :return: V1ListClusterCapacityReservationsResponse
2457
2461
  If the method is called asynchronously,
2458
2462
  returns the request thread.
2459
2463
  """
2460
2464
 
2461
- all_params = ['project_id', 'cluster_id', 'start_time', 'end_time', 'available_only'] # noqa: E501
2465
+ all_params = ['project_id', 'cluster_id', 'start_time', 'end_time', 'available_only', 'from_aggregate', 'apparent_provider'] # noqa: E501
2462
2466
  all_params.append('async_req')
2463
2467
  all_params.append('_return_http_data_only')
2464
2468
  all_params.append('_preload_content')
@@ -2497,6 +2501,10 @@ class ClusterServiceApi(object):
2497
2501
  query_params.append(('endTime', params['end_time'])) # noqa: E501
2498
2502
  if 'available_only' in params:
2499
2503
  query_params.append(('availableOnly', params['available_only'])) # noqa: E501
2504
+ if 'from_aggregate' in params:
2505
+ query_params.append(('fromAggregate', params['from_aggregate'])) # noqa: E501
2506
+ if 'apparent_provider' in params:
2507
+ query_params.append(('apparentProvider', params['apparent_provider'])) # noqa: E501
2500
2508
 
2501
2509
  header_params = {}
2502
2510
 
@@ -1329,6 +1329,9 @@ class LitLoggerServiceApi(object):
1329
1329
  :param str user_id:
1330
1330
  :param str cloud_space_id:
1331
1331
  :param str app_id:
1332
+ :param int limit:
1333
+ :param int offset:
1334
+ :param str order_by:
1332
1335
  :return: V1ListMetricsStreamsResponse
1333
1336
  If the method is called asynchronously,
1334
1337
  returns the request thread.
@@ -1353,12 +1356,15 @@ class LitLoggerServiceApi(object):
1353
1356
  :param str user_id:
1354
1357
  :param str cloud_space_id:
1355
1358
  :param str app_id:
1359
+ :param int limit:
1360
+ :param int offset:
1361
+ :param str order_by:
1356
1362
  :return: V1ListMetricsStreamsResponse
1357
1363
  If the method is called asynchronously,
1358
1364
  returns the request thread.
1359
1365
  """
1360
1366
 
1361
- all_params = ['project_id', 'user_id', 'cloud_space_id', 'app_id'] # noqa: E501
1367
+ all_params = ['project_id', 'user_id', 'cloud_space_id', 'app_id', 'limit', 'offset', 'order_by'] # noqa: E501
1362
1368
  all_params.append('async_req')
1363
1369
  all_params.append('_return_http_data_only')
1364
1370
  all_params.append('_preload_content')
@@ -1391,6 +1397,12 @@ class LitLoggerServiceApi(object):
1391
1397
  query_params.append(('cloudSpaceId', params['cloud_space_id'])) # noqa: E501
1392
1398
  if 'app_id' in params:
1393
1399
  query_params.append(('appId', params['app_id'])) # noqa: E501
1400
+ if 'limit' in params:
1401
+ query_params.append(('limit', params['limit'])) # noqa: E501
1402
+ if 'offset' in params:
1403
+ query_params.append(('offset', params['offset'])) # noqa: E501
1404
+ if 'order_by' in params:
1405
+ query_params.append(('orderBy', params['order_by'])) # noqa: E501
1394
1406
 
1395
1407
  header_params = {}
1396
1408
 
@@ -1958,6 +1958,111 @@ class OrganizationsServiceApi(object):
1958
1958
  _request_timeout=params.get('_request_timeout'),
1959
1959
  collection_formats=collection_formats)
1960
1960
 
1961
+ def organizations_service_update_organization_credits_auto_replenish(self, body: 'CreditsAutoreplenishBody', org_id: 'str', **kwargs) -> 'V1UpdateOrganizationCreditsAutoReplenishResponse': # noqa: E501
1962
+ """organizations_service_update_organization_credits_auto_replenish # noqa: E501
1963
+
1964
+ This method makes a synchronous HTTP request by default. To make an
1965
+ asynchronous HTTP request, please pass async_req=True
1966
+ >>> thread = api.organizations_service_update_organization_credits_auto_replenish(body, org_id, async_req=True)
1967
+ >>> result = thread.get()
1968
+
1969
+ :param async_req bool
1970
+ :param CreditsAutoreplenishBody body: (required)
1971
+ :param str org_id: (required)
1972
+ :return: V1UpdateOrganizationCreditsAutoReplenishResponse
1973
+ If the method is called asynchronously,
1974
+ returns the request thread.
1975
+ """
1976
+ kwargs['_return_http_data_only'] = True
1977
+ if kwargs.get('async_req'):
1978
+ return self.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, **kwargs) # noqa: E501
1979
+ else:
1980
+ (data) = self.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, **kwargs) # noqa: E501
1981
+ return data
1982
+
1983
+ def organizations_service_update_organization_credits_auto_replenish_with_http_info(self, body: 'CreditsAutoreplenishBody', org_id: 'str', **kwargs) -> 'V1UpdateOrganizationCreditsAutoReplenishResponse': # noqa: E501
1984
+ """organizations_service_update_organization_credits_auto_replenish # noqa: E501
1985
+
1986
+ This method makes a synchronous HTTP request by default. To make an
1987
+ asynchronous HTTP request, please pass async_req=True
1988
+ >>> thread = api.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, async_req=True)
1989
+ >>> result = thread.get()
1990
+
1991
+ :param async_req bool
1992
+ :param CreditsAutoreplenishBody body: (required)
1993
+ :param str org_id: (required)
1994
+ :return: V1UpdateOrganizationCreditsAutoReplenishResponse
1995
+ If the method is called asynchronously,
1996
+ returns the request thread.
1997
+ """
1998
+
1999
+ all_params = ['body', 'org_id'] # noqa: E501
2000
+ all_params.append('async_req')
2001
+ all_params.append('_return_http_data_only')
2002
+ all_params.append('_preload_content')
2003
+ all_params.append('_request_timeout')
2004
+
2005
+ params = locals()
2006
+ for key, val in six.iteritems(params['kwargs']):
2007
+ if key not in all_params:
2008
+ raise TypeError(
2009
+ "Got an unexpected keyword argument '%s'"
2010
+ " to method organizations_service_update_organization_credits_auto_replenish" % key
2011
+ )
2012
+ params[key] = val
2013
+ del params['kwargs']
2014
+ # verify the required parameter 'body' is set
2015
+ if ('body' not in params or
2016
+ params['body'] is None):
2017
+ raise ValueError("Missing the required parameter `body` when calling `organizations_service_update_organization_credits_auto_replenish`") # noqa: E501
2018
+ # verify the required parameter 'org_id' is set
2019
+ if ('org_id' not in params or
2020
+ params['org_id'] is None):
2021
+ raise ValueError("Missing the required parameter `org_id` when calling `organizations_service_update_organization_credits_auto_replenish`") # noqa: E501
2022
+
2023
+ collection_formats = {}
2024
+
2025
+ path_params = {}
2026
+ if 'org_id' in params:
2027
+ path_params['orgId'] = params['org_id'] # noqa: E501
2028
+
2029
+ query_params = []
2030
+
2031
+ header_params = {}
2032
+
2033
+ form_params = []
2034
+ local_var_files = {}
2035
+
2036
+ body_params = None
2037
+ if 'body' in params:
2038
+ body_params = params['body']
2039
+ # HTTP header `Accept`
2040
+ header_params['Accept'] = self.api_client.select_header_accept(
2041
+ ['application/json']) # noqa: E501
2042
+
2043
+ # HTTP header `Content-Type`
2044
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
2045
+ ['application/json']) # noqa: E501
2046
+
2047
+ # Authentication setting
2048
+ auth_settings = [] # noqa: E501
2049
+
2050
+ return self.api_client.call_api(
2051
+ '/v1/orgs/{orgId}/credits/auto-replenish', 'PUT',
2052
+ path_params,
2053
+ query_params,
2054
+ header_params,
2055
+ body=body_params,
2056
+ post_params=form_params,
2057
+ files=local_var_files,
2058
+ response_type='V1UpdateOrganizationCreditsAutoReplenishResponse', # noqa: E501
2059
+ auth_settings=auth_settings,
2060
+ async_req=params.get('async_req'),
2061
+ _return_http_data_only=params.get('_return_http_data_only'),
2062
+ _preload_content=params.get('_preload_content', True),
2063
+ _request_timeout=params.get('_request_timeout'),
2064
+ collection_formats=collection_formats)
2065
+
1961
2066
  def organizations_service_validate_auto_join_domain(self, body: 'ValidateautojoindomainDomainBody', org_id: 'str', domain: 'str', **kwargs) -> 'V1ValidateAutoJoinDomainResponse': # noqa: E501
1962
2067
  """AutoJoin Domain management # noqa: E501
1963
2068
 
@@ -43,7 +43,7 @@ class PipelinesServiceApi(object):
43
43
  api_client = ApiClient()
44
44
  self.api_client = api_client
45
45
 
46
- def pipelines_service_create_child_pipeline(self, body: 'object', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
46
+ def pipelines_service_create_child_pipeline(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
47
47
  """pipelines_service_create_child_pipeline # noqa: E501
48
48
 
49
49
  This method makes a synchronous HTTP request by default. To make an
@@ -52,7 +52,7 @@ class PipelinesServiceApi(object):
52
52
  >>> result = thread.get()
53
53
 
54
54
  :param async_req bool
55
- :param object body: (required)
55
+ :param PipelinesIdBody1 body: (required)
56
56
  :param str project_id: (required)
57
57
  :param str id: (required)
58
58
  :return: V1Pipeline
@@ -66,7 +66,7 @@ class PipelinesServiceApi(object):
66
66
  (data) = self.pipelines_service_create_child_pipeline_with_http_info(body, project_id, id, **kwargs) # noqa: E501
67
67
  return data
68
68
 
69
- def pipelines_service_create_child_pipeline_with_http_info(self, body: 'object', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
69
+ def pipelines_service_create_child_pipeline_with_http_info(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
70
70
  """pipelines_service_create_child_pipeline # noqa: E501
71
71
 
72
72
  This method makes a synchronous HTTP request by default. To make an
@@ -75,7 +75,7 @@ class PipelinesServiceApi(object):
75
75
  >>> result = thread.get()
76
76
 
77
77
  :param async_req bool
78
- :param object body: (required)
78
+ :param PipelinesIdBody1 body: (required)
79
79
  :param str project_id: (required)
80
80
  :param str id: (required)
81
81
  :return: V1Pipeline
@@ -1284,6 +1284,111 @@ class UserServiceApi(object):
1284
1284
  _request_timeout=params.get('_request_timeout'),
1285
1285
  collection_formats=collection_formats)
1286
1286
 
1287
+ def user_service_update_user_credits_auto_replenish(self, body: 'CreditsAutoreplenishBody1', user_id: 'str', **kwargs) -> 'V1UpdateUserCreditsAutoReplenishResponse': # noqa: E501
1288
+ """user_service_update_user_credits_auto_replenish # noqa: E501
1289
+
1290
+ This method makes a synchronous HTTP request by default. To make an
1291
+ asynchronous HTTP request, please pass async_req=True
1292
+ >>> thread = api.user_service_update_user_credits_auto_replenish(body, user_id, async_req=True)
1293
+ >>> result = thread.get()
1294
+
1295
+ :param async_req bool
1296
+ :param CreditsAutoreplenishBody1 body: (required)
1297
+ :param str user_id: (required)
1298
+ :return: V1UpdateUserCreditsAutoReplenishResponse
1299
+ If the method is called asynchronously,
1300
+ returns the request thread.
1301
+ """
1302
+ kwargs['_return_http_data_only'] = True
1303
+ if kwargs.get('async_req'):
1304
+ return self.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, **kwargs) # noqa: E501
1305
+ else:
1306
+ (data) = self.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, **kwargs) # noqa: E501
1307
+ return data
1308
+
1309
+ def user_service_update_user_credits_auto_replenish_with_http_info(self, body: 'CreditsAutoreplenishBody1', user_id: 'str', **kwargs) -> 'V1UpdateUserCreditsAutoReplenishResponse': # noqa: E501
1310
+ """user_service_update_user_credits_auto_replenish # noqa: E501
1311
+
1312
+ This method makes a synchronous HTTP request by default. To make an
1313
+ asynchronous HTTP request, please pass async_req=True
1314
+ >>> thread = api.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, async_req=True)
1315
+ >>> result = thread.get()
1316
+
1317
+ :param async_req bool
1318
+ :param CreditsAutoreplenishBody1 body: (required)
1319
+ :param str user_id: (required)
1320
+ :return: V1UpdateUserCreditsAutoReplenishResponse
1321
+ If the method is called asynchronously,
1322
+ returns the request thread.
1323
+ """
1324
+
1325
+ all_params = ['body', 'user_id'] # noqa: E501
1326
+ all_params.append('async_req')
1327
+ all_params.append('_return_http_data_only')
1328
+ all_params.append('_preload_content')
1329
+ all_params.append('_request_timeout')
1330
+
1331
+ params = locals()
1332
+ for key, val in six.iteritems(params['kwargs']):
1333
+ if key not in all_params:
1334
+ raise TypeError(
1335
+ "Got an unexpected keyword argument '%s'"
1336
+ " to method user_service_update_user_credits_auto_replenish" % key
1337
+ )
1338
+ params[key] = val
1339
+ del params['kwargs']
1340
+ # verify the required parameter 'body' is set
1341
+ if ('body' not in params or
1342
+ params['body'] is None):
1343
+ raise ValueError("Missing the required parameter `body` when calling `user_service_update_user_credits_auto_replenish`") # noqa: E501
1344
+ # verify the required parameter 'user_id' is set
1345
+ if ('user_id' not in params or
1346
+ params['user_id'] is None):
1347
+ raise ValueError("Missing the required parameter `user_id` when calling `user_service_update_user_credits_auto_replenish`") # noqa: E501
1348
+
1349
+ collection_formats = {}
1350
+
1351
+ path_params = {}
1352
+ if 'user_id' in params:
1353
+ path_params['userId'] = params['user_id'] # noqa: E501
1354
+
1355
+ query_params = []
1356
+
1357
+ header_params = {}
1358
+
1359
+ form_params = []
1360
+ local_var_files = {}
1361
+
1362
+ body_params = None
1363
+ if 'body' in params:
1364
+ body_params = params['body']
1365
+ # HTTP header `Accept`
1366
+ header_params['Accept'] = self.api_client.select_header_accept(
1367
+ ['application/json']) # noqa: E501
1368
+
1369
+ # HTTP header `Content-Type`
1370
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1371
+ ['application/json']) # noqa: E501
1372
+
1373
+ # Authentication setting
1374
+ auth_settings = [] # noqa: E501
1375
+
1376
+ return self.api_client.call_api(
1377
+ '/v1/users/{userId}/credits/auto-replenish', 'PUT',
1378
+ path_params,
1379
+ query_params,
1380
+ header_params,
1381
+ body=body_params,
1382
+ post_params=form_params,
1383
+ files=local_var_files,
1384
+ response_type='V1UpdateUserCreditsAutoReplenishResponse', # noqa: E501
1385
+ auth_settings=auth_settings,
1386
+ async_req=params.get('async_req'),
1387
+ _return_http_data_only=params.get('_return_http_data_only'),
1388
+ _preload_content=params.get('_preload_content', True),
1389
+ _request_timeout=params.get('_request_timeout'),
1390
+ collection_formats=collection_formats)
1391
+
1287
1392
  def user_service_update_user_viewed_new_features(self, body: 'V1UpdateUserViewedNewFeaturesRequest', **kwargs) -> 'V1UpdateUserViewedNewFeaturesResponse': # noqa: E501
1288
1393
  """we use this to update user viewed new features # noqa: E501
1289
1394