lightning-sdk 2025.7.22__py3-none-any.whl → 2025.7.31__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 (64) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/cloud_account_api.py +58 -8
  3. lightning_sdk/api/job_api.py +74 -11
  4. lightning_sdk/api/llm_api.py +1 -1
  5. lightning_sdk/api/mmt_api.py +35 -5
  6. lightning_sdk/api/studio_api.py +32 -3
  7. lightning_sdk/api/utils.py +6 -2
  8. lightning_sdk/cli/create.py +3 -1
  9. lightning_sdk/cli/deploy/serve.py +3 -1
  10. lightning_sdk/cli/download.py +25 -1
  11. lightning_sdk/cli/entrypoint.py +3 -1
  12. lightning_sdk/cli/list.py +5 -1
  13. lightning_sdk/cli/run.py +3 -1
  14. lightning_sdk/cli/start.py +3 -1
  15. lightning_sdk/cli/switch.py +3 -1
  16. lightning_sdk/job/v2.py +7 -1
  17. lightning_sdk/job/work.py +5 -1
  18. lightning_sdk/lightning_cloud/openapi/__init__.py +6 -1
  19. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +106 -13
  20. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +295 -0
  21. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +93 -0
  22. lightning_sdk/lightning_cloud/openapi/models/__init__.py +6 -1
  23. lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/metricsstream_id_body.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +53 -1
  26. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +53 -1
  27. lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py +29 -3
  28. lightning_sdk/lightning_cloud/openapi/models/user_user_id_body.py +201 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_billing_subscription.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_settings.py +227 -0
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  32. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_function_call.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_health_response.py +149 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/{v1_list_published_managed_endpoint_models_response.py → v1_list_published_managed_endpoints_response.py} +23 -23
  40. lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +95 -17
  42. lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_response_choice.py +29 -3
  44. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_service_health.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py +79 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1_status.py +79 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_tool_call.py +175 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +79 -53
  50. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +1 -0
  51. lightning_sdk/llm/llm.py +41 -7
  52. lightning_sdk/llm/public_assistants.json +32 -8
  53. lightning_sdk/machine.py +139 -43
  54. lightning_sdk/mmt/v2.py +6 -1
  55. lightning_sdk/models.py +1 -1
  56. lightning_sdk/studio.py +12 -5
  57. lightning_sdk/teamspace.py +5 -2
  58. lightning_sdk/utils/resolve.py +8 -0
  59. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/METADATA +7 -5
  60. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/RECORD +64 -59
  61. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/LICENSE +0 -0
  62. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/WHEEL +0 -0
  63. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/entry_points.txt +0 -0
  64. {lightning_sdk-2025.7.22.dist-info → lightning_sdk-2025.7.31.dist-info}/top_level.txt +0 -0
@@ -1166,6 +1166,99 @@ class AssistantsServiceApi(object):
1166
1166
  _request_timeout=params.get('_request_timeout'),
1167
1167
  collection_formats=collection_formats)
1168
1168
 
1169
+ def assistants_service_get_published_managed_endpoint_by_model_id(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint': # noqa: E501
1170
+ """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata # noqa: E501
1171
+
1172
+ This method makes a synchronous HTTP request by default. To make an
1173
+ asynchronous HTTP request, please pass async_req=True
1174
+ >>> thread = api.assistants_service_get_published_managed_endpoint_by_model_id(id, async_req=True)
1175
+ >>> result = thread.get()
1176
+
1177
+ :param async_req bool
1178
+ :param str id: (required)
1179
+ :return: V1ManagedEndpoint
1180
+ If the method is called asynchronously,
1181
+ returns the request thread.
1182
+ """
1183
+ kwargs['_return_http_data_only'] = True
1184
+ if kwargs.get('async_req'):
1185
+ return self.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, **kwargs) # noqa: E501
1186
+ else:
1187
+ (data) = self.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, **kwargs) # noqa: E501
1188
+ return data
1189
+
1190
+ def assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint': # noqa: E501
1191
+ """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata # noqa: E501
1192
+
1193
+ This method makes a synchronous HTTP request by default. To make an
1194
+ asynchronous HTTP request, please pass async_req=True
1195
+ >>> thread = api.assistants_service_get_published_managed_endpoint_by_model_id_with_http_info(id, async_req=True)
1196
+ >>> result = thread.get()
1197
+
1198
+ :param async_req bool
1199
+ :param str id: (required)
1200
+ :return: V1ManagedEndpoint
1201
+ If the method is called asynchronously,
1202
+ returns the request thread.
1203
+ """
1204
+
1205
+ all_params = ['id'] # noqa: E501
1206
+ all_params.append('async_req')
1207
+ all_params.append('_return_http_data_only')
1208
+ all_params.append('_preload_content')
1209
+ all_params.append('_request_timeout')
1210
+
1211
+ params = locals()
1212
+ for key, val in six.iteritems(params['kwargs']):
1213
+ if key not in all_params:
1214
+ raise TypeError(
1215
+ "Got an unexpected keyword argument '%s'"
1216
+ " to method assistants_service_get_published_managed_endpoint_by_model_id" % key
1217
+ )
1218
+ params[key] = val
1219
+ del params['kwargs']
1220
+ # verify the required parameter 'id' is set
1221
+ if ('id' not in params or
1222
+ params['id'] is None):
1223
+ raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_published_managed_endpoint_by_model_id`") # noqa: E501
1224
+
1225
+ collection_formats = {}
1226
+
1227
+ path_params = {}
1228
+ if 'id' in params:
1229
+ path_params['id'] = params['id'] # noqa: E501
1230
+
1231
+ query_params = []
1232
+
1233
+ header_params = {}
1234
+
1235
+ form_params = []
1236
+ local_var_files = {}
1237
+
1238
+ body_params = None
1239
+ # HTTP header `Accept`
1240
+ header_params['Accept'] = self.api_client.select_header_accept(
1241
+ ['application/json']) # noqa: E501
1242
+
1243
+ # Authentication setting
1244
+ auth_settings = [] # noqa: E501
1245
+
1246
+ return self.api_client.call_api(
1247
+ '/v1/agent-published-managed-model/{id}', 'GET',
1248
+ path_params,
1249
+ query_params,
1250
+ header_params,
1251
+ body=body_params,
1252
+ post_params=form_params,
1253
+ files=local_var_files,
1254
+ response_type='V1ManagedEndpoint', # noqa: E501
1255
+ auth_settings=auth_settings,
1256
+ async_req=params.get('async_req'),
1257
+ _return_http_data_only=params.get('_return_http_data_only'),
1258
+ _preload_content=params.get('_preload_content', True),
1259
+ _request_timeout=params.get('_request_timeout'),
1260
+ collection_formats=collection_formats)
1261
+
1169
1262
  def assistants_service_list_assistant_managed_endpoints(self, **kwargs) -> 'V1ListManagedEndpointsResponse': # noqa: E501
1170
1263
  """ListAssistantManagedEndpoints returns a list of managed endpoint that users can use when creating their own assistant. These are served and managed by Lightning or 3rd parties # noqa: E501
1171
1264
 
@@ -1477,36 +1570,36 @@ class AssistantsServiceApi(object):
1477
1570
  _request_timeout=params.get('_request_timeout'),
1478
1571
  collection_formats=collection_formats)
1479
1572
 
1480
- def assistants_service_list_published_managed_endpoint_models(self, **kwargs) -> 'V1ListPublishedManagedEndpointModelsResponse': # noqa: E501
1481
- """assistants_service_list_published_managed_endpoint_models # noqa: E501
1573
+ def assistants_service_list_published_managed_endpoints(self, **kwargs) -> 'V1ListPublishedManagedEndpointsResponse': # noqa: E501
1574
+ """ListPublishedManagedEndpoints returns a list of all available managed endpoints that are published as Models # noqa: E501
1482
1575
 
1483
1576
  This method makes a synchronous HTTP request by default. To make an
1484
1577
  asynchronous HTTP request, please pass async_req=True
1485
- >>> thread = api.assistants_service_list_published_managed_endpoint_models(async_req=True)
1578
+ >>> thread = api.assistants_service_list_published_managed_endpoints(async_req=True)
1486
1579
  >>> result = thread.get()
1487
1580
 
1488
1581
  :param async_req bool
1489
- :return: V1ListPublishedManagedEndpointModelsResponse
1582
+ :return: V1ListPublishedManagedEndpointsResponse
1490
1583
  If the method is called asynchronously,
1491
1584
  returns the request thread.
1492
1585
  """
1493
1586
  kwargs['_return_http_data_only'] = True
1494
1587
  if kwargs.get('async_req'):
1495
- return self.assistants_service_list_published_managed_endpoint_models_with_http_info(**kwargs) # noqa: E501
1588
+ return self.assistants_service_list_published_managed_endpoints_with_http_info(**kwargs) # noqa: E501
1496
1589
  else:
1497
- (data) = self.assistants_service_list_published_managed_endpoint_models_with_http_info(**kwargs) # noqa: E501
1590
+ (data) = self.assistants_service_list_published_managed_endpoints_with_http_info(**kwargs) # noqa: E501
1498
1591
  return data
1499
1592
 
1500
- def assistants_service_list_published_managed_endpoint_models_with_http_info(self, **kwargs) -> 'V1ListPublishedManagedEndpointModelsResponse': # noqa: E501
1501
- """assistants_service_list_published_managed_endpoint_models # noqa: E501
1593
+ def assistants_service_list_published_managed_endpoints_with_http_info(self, **kwargs) -> 'V1ListPublishedManagedEndpointsResponse': # noqa: E501
1594
+ """ListPublishedManagedEndpoints returns a list of all available managed endpoints that are published as Models # noqa: E501
1502
1595
 
1503
1596
  This method makes a synchronous HTTP request by default. To make an
1504
1597
  asynchronous HTTP request, please pass async_req=True
1505
- >>> thread = api.assistants_service_list_published_managed_endpoint_models_with_http_info(async_req=True)
1598
+ >>> thread = api.assistants_service_list_published_managed_endpoints_with_http_info(async_req=True)
1506
1599
  >>> result = thread.get()
1507
1600
 
1508
1601
  :param async_req bool
1509
- :return: V1ListPublishedManagedEndpointModelsResponse
1602
+ :return: V1ListPublishedManagedEndpointsResponse
1510
1603
  If the method is called asynchronously,
1511
1604
  returns the request thread.
1512
1605
  """
@@ -1522,7 +1615,7 @@ class AssistantsServiceApi(object):
1522
1615
  if key not in all_params:
1523
1616
  raise TypeError(
1524
1617
  "Got an unexpected keyword argument '%s'"
1525
- " to method assistants_service_list_published_managed_endpoint_models" % key
1618
+ " to method assistants_service_list_published_managed_endpoints" % key
1526
1619
  )
1527
1620
  params[key] = val
1528
1621
  del params['kwargs']
@@ -1547,14 +1640,14 @@ class AssistantsServiceApi(object):
1547
1640
  auth_settings = [] # noqa: E501
1548
1641
 
1549
1642
  return self.api_client.call_api(
1550
- '/v1/agents/published-managed-models', 'GET',
1643
+ '/v1/agent-published-managed-endpoints', 'GET',
1551
1644
  path_params,
1552
1645
  query_params,
1553
1646
  header_params,
1554
1647
  body=body_params,
1555
1648
  post_params=form_params,
1556
1649
  files=local_var_files,
1557
- response_type='V1ListPublishedManagedEndpointModelsResponse', # noqa: E501
1650
+ response_type='V1ListPublishedManagedEndpointsResponse', # noqa: E501
1558
1651
  auth_settings=auth_settings,
1559
1652
  async_req=params.get('async_req'),
1560
1653
  _return_http_data_only=params.get('_return_http_data_only'),
@@ -43,6 +43,196 @@ class CloudyServiceApi(object):
43
43
  api_client = ApiClient()
44
44
  self.api_client = api_client
45
45
 
46
+ def cloudy_service_create_user_cloudy_settings(self, body: 'V1CloudySettings', **kwargs) -> 'V1CloudySettings': # noqa: E501
47
+ """cloudy_service_create_user_cloudy_settings # 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_create_user_cloudy_settings(body, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param V1CloudySettings body: (required)
56
+ :return: V1CloudySettings
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.cloudy_service_create_user_cloudy_settings_with_http_info(body, **kwargs) # noqa: E501
63
+ else:
64
+ (data) = self.cloudy_service_create_user_cloudy_settings_with_http_info(body, **kwargs) # noqa: E501
65
+ return data
66
+
67
+ def cloudy_service_create_user_cloudy_settings_with_http_info(self, body: 'V1CloudySettings', **kwargs) -> 'V1CloudySettings': # noqa: E501
68
+ """cloudy_service_create_user_cloudy_settings # 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.cloudy_service_create_user_cloudy_settings_with_http_info(body, async_req=True)
73
+ >>> result = thread.get()
74
+
75
+ :param async_req bool
76
+ :param V1CloudySettings body: (required)
77
+ :return: V1CloudySettings
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 cloudy_service_create_user_cloudy_settings" % 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 `cloudy_service_create_user_cloudy_settings`") # 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/cloudy/settings/user', '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='V1CloudySettings', # 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 cloudy_service_get_user_cloudy_settings(self, user_id: 'str', **kwargs) -> 'V1CloudySettings': # noqa: E501
144
+ """cloudy_service_get_user_cloudy_settings # 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.cloudy_service_get_user_cloudy_settings(user_id, async_req=True)
149
+ >>> result = thread.get()
150
+
151
+ :param async_req bool
152
+ :param str user_id: (required)
153
+ :return: V1CloudySettings
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.cloudy_service_get_user_cloudy_settings_with_http_info(user_id, **kwargs) # noqa: E501
160
+ else:
161
+ (data) = self.cloudy_service_get_user_cloudy_settings_with_http_info(user_id, **kwargs) # noqa: E501
162
+ return data
163
+
164
+ def cloudy_service_get_user_cloudy_settings_with_http_info(self, user_id: 'str', **kwargs) -> 'V1CloudySettings': # noqa: E501
165
+ """cloudy_service_get_user_cloudy_settings # 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.cloudy_service_get_user_cloudy_settings_with_http_info(user_id, async_req=True)
170
+ >>> result = thread.get()
171
+
172
+ :param async_req bool
173
+ :param str user_id: (required)
174
+ :return: V1CloudySettings
175
+ If the method is called asynchronously,
176
+ returns the request thread.
177
+ """
178
+
179
+ all_params = ['user_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 cloudy_service_get_user_cloudy_settings" % key
191
+ )
192
+ params[key] = val
193
+ del params['kwargs']
194
+ # verify the required parameter 'user_id' is set
195
+ if ('user_id' not in params or
196
+ params['user_id'] is None):
197
+ raise ValueError("Missing the required parameter `user_id` when calling `cloudy_service_get_user_cloudy_settings`") # noqa: E501
198
+
199
+ collection_formats = {}
200
+
201
+ path_params = {}
202
+ if 'user_id' in params:
203
+ path_params['userId'] = params['user_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/cloudy/settings/user/{userId}', '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='V1CloudySettings', # 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
+
46
236
  def cloudy_service_list_cloudy_experts(self, **kwargs) -> 'V1ListCloudyExpertsResponse': # noqa: E501
47
237
  """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
238
 
@@ -127,3 +317,108 @@ class CloudyServiceApi(object):
127
317
  _preload_content=params.get('_preload_content', True),
128
318
  _request_timeout=params.get('_request_timeout'),
129
319
  collection_formats=collection_formats)
320
+
321
+ def cloudy_service_update_user_cloudy_settings(self, body: 'UserUserIdBody', user_id: 'str', **kwargs) -> 'V1CloudySettings': # noqa: E501
322
+ """cloudy_service_update_user_cloudy_settings # noqa: E501
323
+
324
+ This method makes a synchronous HTTP request by default. To make an
325
+ asynchronous HTTP request, please pass async_req=True
326
+ >>> thread = api.cloudy_service_update_user_cloudy_settings(body, user_id, async_req=True)
327
+ >>> result = thread.get()
328
+
329
+ :param async_req bool
330
+ :param UserUserIdBody body: (required)
331
+ :param str user_id: (required)
332
+ :return: V1CloudySettings
333
+ If the method is called asynchronously,
334
+ returns the request thread.
335
+ """
336
+ kwargs['_return_http_data_only'] = True
337
+ if kwargs.get('async_req'):
338
+ return self.cloudy_service_update_user_cloudy_settings_with_http_info(body, user_id, **kwargs) # noqa: E501
339
+ else:
340
+ (data) = self.cloudy_service_update_user_cloudy_settings_with_http_info(body, user_id, **kwargs) # noqa: E501
341
+ return data
342
+
343
+ def cloudy_service_update_user_cloudy_settings_with_http_info(self, body: 'UserUserIdBody', user_id: 'str', **kwargs) -> 'V1CloudySettings': # noqa: E501
344
+ """cloudy_service_update_user_cloudy_settings # noqa: E501
345
+
346
+ This method makes a synchronous HTTP request by default. To make an
347
+ asynchronous HTTP request, please pass async_req=True
348
+ >>> thread = api.cloudy_service_update_user_cloudy_settings_with_http_info(body, user_id, async_req=True)
349
+ >>> result = thread.get()
350
+
351
+ :param async_req bool
352
+ :param UserUserIdBody body: (required)
353
+ :param str user_id: (required)
354
+ :return: V1CloudySettings
355
+ If the method is called asynchronously,
356
+ returns the request thread.
357
+ """
358
+
359
+ all_params = ['body', 'user_id'] # noqa: E501
360
+ all_params.append('async_req')
361
+ all_params.append('_return_http_data_only')
362
+ all_params.append('_preload_content')
363
+ all_params.append('_request_timeout')
364
+
365
+ params = locals()
366
+ for key, val in six.iteritems(params['kwargs']):
367
+ if key not in all_params:
368
+ raise TypeError(
369
+ "Got an unexpected keyword argument '%s'"
370
+ " to method cloudy_service_update_user_cloudy_settings" % key
371
+ )
372
+ params[key] = val
373
+ del params['kwargs']
374
+ # verify the required parameter 'body' is set
375
+ if ('body' not in params or
376
+ params['body'] is None):
377
+ raise ValueError("Missing the required parameter `body` when calling `cloudy_service_update_user_cloudy_settings`") # noqa: E501
378
+ # verify the required parameter 'user_id' is set
379
+ if ('user_id' not in params or
380
+ params['user_id'] is None):
381
+ raise ValueError("Missing the required parameter `user_id` when calling `cloudy_service_update_user_cloudy_settings`") # noqa: E501
382
+
383
+ collection_formats = {}
384
+
385
+ path_params = {}
386
+ if 'user_id' in params:
387
+ path_params['userId'] = params['user_id'] # noqa: E501
388
+
389
+ query_params = []
390
+
391
+ header_params = {}
392
+
393
+ form_params = []
394
+ local_var_files = {}
395
+
396
+ body_params = None
397
+ if 'body' in params:
398
+ body_params = params['body']
399
+ # HTTP header `Accept`
400
+ header_params['Accept'] = self.api_client.select_header_accept(
401
+ ['application/json']) # noqa: E501
402
+
403
+ # HTTP header `Content-Type`
404
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
405
+ ['application/json']) # noqa: E501
406
+
407
+ # Authentication setting
408
+ auth_settings = [] # noqa: E501
409
+
410
+ return self.api_client.call_api(
411
+ '/v1/cloudy/settings/user/{userId}', 'PUT',
412
+ path_params,
413
+ query_params,
414
+ header_params,
415
+ body=body_params,
416
+ post_params=form_params,
417
+ files=local_var_files,
418
+ response_type='V1CloudySettings', # noqa: E501
419
+ auth_settings=auth_settings,
420
+ async_req=params.get('async_req'),
421
+ _return_http_data_only=params.get('_return_http_data_only'),
422
+ _preload_content=params.get('_preload_content', True),
423
+ _request_timeout=params.get('_request_timeout'),
424
+ collection_formats=collection_formats)
@@ -2018,6 +2018,99 @@ class ClusterServiceApi(object):
2018
2018
  _request_timeout=params.get('_request_timeout'),
2019
2019
  collection_formats=collection_formats)
2020
2020
 
2021
+ def cluster_service_get_cluster_health(self, id: 'str', **kwargs) -> 'V1GetClusterHealthResponse': # noqa: E501
2022
+ """cluster_service_get_cluster_health # noqa: E501
2023
+
2024
+ This method makes a synchronous HTTP request by default. To make an
2025
+ asynchronous HTTP request, please pass async_req=True
2026
+ >>> thread = api.cluster_service_get_cluster_health(id, async_req=True)
2027
+ >>> result = thread.get()
2028
+
2029
+ :param async_req bool
2030
+ :param str id: (required)
2031
+ :return: V1GetClusterHealthResponse
2032
+ If the method is called asynchronously,
2033
+ returns the request thread.
2034
+ """
2035
+ kwargs['_return_http_data_only'] = True
2036
+ if kwargs.get('async_req'):
2037
+ return self.cluster_service_get_cluster_health_with_http_info(id, **kwargs) # noqa: E501
2038
+ else:
2039
+ (data) = self.cluster_service_get_cluster_health_with_http_info(id, **kwargs) # noqa: E501
2040
+ return data
2041
+
2042
+ def cluster_service_get_cluster_health_with_http_info(self, id: 'str', **kwargs) -> 'V1GetClusterHealthResponse': # noqa: E501
2043
+ """cluster_service_get_cluster_health # noqa: E501
2044
+
2045
+ This method makes a synchronous HTTP request by default. To make an
2046
+ asynchronous HTTP request, please pass async_req=True
2047
+ >>> thread = api.cluster_service_get_cluster_health_with_http_info(id, async_req=True)
2048
+ >>> result = thread.get()
2049
+
2050
+ :param async_req bool
2051
+ :param str id: (required)
2052
+ :return: V1GetClusterHealthResponse
2053
+ If the method is called asynchronously,
2054
+ returns the request thread.
2055
+ """
2056
+
2057
+ all_params = ['id'] # noqa: E501
2058
+ all_params.append('async_req')
2059
+ all_params.append('_return_http_data_only')
2060
+ all_params.append('_preload_content')
2061
+ all_params.append('_request_timeout')
2062
+
2063
+ params = locals()
2064
+ for key, val in six.iteritems(params['kwargs']):
2065
+ if key not in all_params:
2066
+ raise TypeError(
2067
+ "Got an unexpected keyword argument '%s'"
2068
+ " to method cluster_service_get_cluster_health" % key
2069
+ )
2070
+ params[key] = val
2071
+ del params['kwargs']
2072
+ # verify the required parameter 'id' is set
2073
+ if ('id' not in params or
2074
+ params['id'] is None):
2075
+ raise ValueError("Missing the required parameter `id` when calling `cluster_service_get_cluster_health`") # noqa: E501
2076
+
2077
+ collection_formats = {}
2078
+
2079
+ path_params = {}
2080
+ if 'id' in params:
2081
+ path_params['id'] = params['id'] # noqa: E501
2082
+
2083
+ query_params = []
2084
+
2085
+ header_params = {}
2086
+
2087
+ form_params = []
2088
+ local_var_files = {}
2089
+
2090
+ body_params = None
2091
+ # HTTP header `Accept`
2092
+ header_params['Accept'] = self.api_client.select_header_accept(
2093
+ ['application/json']) # noqa: E501
2094
+
2095
+ # Authentication setting
2096
+ auth_settings = [] # noqa: E501
2097
+
2098
+ return self.api_client.call_api(
2099
+ '/v1/core/clusters/{id}/health', 'GET',
2100
+ path_params,
2101
+ query_params,
2102
+ header_params,
2103
+ body=body_params,
2104
+ post_params=form_params,
2105
+ files=local_var_files,
2106
+ response_type='V1GetClusterHealthResponse', # noqa: E501
2107
+ auth_settings=auth_settings,
2108
+ async_req=params.get('async_req'),
2109
+ _return_http_data_only=params.get('_return_http_data_only'),
2110
+ _preload_content=params.get('_preload_content', True),
2111
+ _request_timeout=params.get('_request_timeout'),
2112
+ collection_formats=collection_formats)
2113
+
2021
2114
  def cluster_service_get_project_cluster(self, project_id: 'str', id: 'str', **kwargs) -> 'Externalv1Cluster': # noqa: E501
2022
2115
  """cluster_service_get_project_cluster # noqa: E501
2023
2116
 
@@ -196,6 +196,7 @@ from lightning_sdk.lightning_cloud.openapi.models.user_id_affiliatelinks_body im
196
196
  from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body import UserIdMembershiprolebindingsBody
197
197
  from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body1 import UserIdMembershiprolebindingsBody1
198
198
  from lightning_sdk.lightning_cloud.openapi.models.user_id_upgradetrigger_body import UserIdUpgradetriggerBody
199
+ from lightning_sdk.lightning_cloud.openapi.models.user_user_id_body import UserUserIdBody
199
200
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_cluster_credentials import V1AWSClusterCredentials
200
201
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1 import V1AWSDirectV1
201
202
  from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import V1AWSDirectV1Status
@@ -278,6 +279,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import
278
279
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
279
280
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
280
281
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloudy_expert import V1CloudyExpert
282
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloudy_settings import V1CloudySettings
281
283
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
282
284
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
283
285
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -486,6 +488,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
486
488
  from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
487
489
  from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
488
490
  from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
491
+ from lightning_sdk.lightning_cloud.openapi.models.v1_function_call import V1FunctionCall
489
492
  from lightning_sdk.lightning_cloud.openapi.models.v1_function_tool import V1FunctionTool
490
493
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
491
494
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
@@ -508,6 +511,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_sy
508
511
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
509
512
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
510
513
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
514
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_health_response import V1GetClusterHealthResponse
511
515
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
512
516
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_content_response import V1GetDeploymentRoutingTelemetryContentResponse
513
517
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
@@ -682,7 +686,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_memberships_re
682
686
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_roles_response import V1ListProjectRolesResponse
683
687
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_cloud_spaces_response import V1ListPublishedCloudSpacesResponse
684
688
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_deployment_templates_response import V1ListPublishedDeploymentTemplatesResponse
685
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_managed_endpoint_models_response import V1ListPublishedManagedEndpointModelsResponse
689
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_managed_endpoints_response import V1ListPublishedManagedEndpointsResponse
686
690
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_quests_response import V1ListQuestsResponse
687
691
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_response import V1ListSLURMClusterUsersResponse
688
692
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
@@ -883,6 +887,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetr
883
887
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
884
888
  from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
885
889
  from lightning_sdk.lightning_cloud.openapi.models.v1_tool import V1Tool
890
+ from lightning_sdk.lightning_cloud.openapi.models.v1_tool_call import V1ToolCall
886
891
  from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
887
892
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
888
893
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse