lightning-sdk 0.1.38__py3-none-any.whl → 0.1.40__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.
- lightning_sdk/__init__.py +1 -1
- lightning_sdk/_mmt/__init__.py +3 -0
- lightning_sdk/_mmt/base.py +180 -0
- lightning_sdk/_mmt/mmt.py +161 -0
- lightning_sdk/_mmt/v1.py +69 -0
- lightning_sdk/_mmt/v2.py +141 -0
- lightning_sdk/api/deployment_api.py +0 -2
- lightning_sdk/api/job_api.py +4 -0
- lightning_sdk/api/mmt_api.py +147 -0
- lightning_sdk/api/teamspace_api.py +4 -11
- lightning_sdk/api/utils.py +6 -3
- lightning_sdk/cli/download.py +3 -5
- lightning_sdk/cli/mmt.py +137 -0
- lightning_sdk/cli/run.py +16 -0
- lightning_sdk/cli/upload.py +3 -10
- lightning_sdk/job/base.py +24 -3
- lightning_sdk/job/job.py +10 -1
- lightning_sdk/job/v1.py +7 -1
- lightning_sdk/job/v2.py +18 -9
- lightning_sdk/lightning_cloud/openapi/__init__.py +7 -3
- lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +90 -284
- lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +6 -1
- lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +235 -1
- lightning_sdk/lightning_cloud/openapi/models/__init__.py +7 -3
- lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/{project_id_agentmanagedmodels_body.py → v1_body.py} +21 -47
- lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +201 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -53
- lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_model.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +1 -2
- lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/{v1_list_managed_models_response.py → v1_resource_visibility.py} +23 -23
- lightning_sdk/lightning_cloud/openapi/models/{v1_delete_managed_model_response.py → v1_update_model_visibility_response.py} +6 -6
- lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -1
- lightning_sdk/models.py +153 -0
- lightning_sdk/teamspace.py +15 -11
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/METADATA +1 -1
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/RECORD +52 -41
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/entry_points.txt +1 -0
- lightning_sdk/cli/models.py +0 -68
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/LICENSE +0 -0
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/WHEEL +0 -0
- {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/top_level.txt +0 -0
|
@@ -253,111 +253,6 @@ class AssistantsServiceApi(object):
|
|
|
253
253
|
_request_timeout=params.get('_request_timeout'),
|
|
254
254
|
collection_formats=collection_formats)
|
|
255
255
|
|
|
256
|
-
def assistants_service_create_managed_model(self, body: 'ProjectIdAgentmanagedmodelsBody', project_id: 'str', **kwargs) -> 'V1ManagedModel': # noqa: E501
|
|
257
|
-
"""assistants_service_create_managed_model # noqa: E501
|
|
258
|
-
|
|
259
|
-
This method makes a synchronous HTTP request by default. To make an
|
|
260
|
-
asynchronous HTTP request, please pass async_req=True
|
|
261
|
-
>>> thread = api.assistants_service_create_managed_model(body, project_id, async_req=True)
|
|
262
|
-
>>> result = thread.get()
|
|
263
|
-
|
|
264
|
-
:param async_req bool
|
|
265
|
-
:param ProjectIdAgentmanagedmodelsBody body: (required)
|
|
266
|
-
:param str project_id: (required)
|
|
267
|
-
:return: V1ManagedModel
|
|
268
|
-
If the method is called asynchronously,
|
|
269
|
-
returns the request thread.
|
|
270
|
-
"""
|
|
271
|
-
kwargs['_return_http_data_only'] = True
|
|
272
|
-
if kwargs.get('async_req'):
|
|
273
|
-
return self.assistants_service_create_managed_model_with_http_info(body, project_id, **kwargs) # noqa: E501
|
|
274
|
-
else:
|
|
275
|
-
(data) = self.assistants_service_create_managed_model_with_http_info(body, project_id, **kwargs) # noqa: E501
|
|
276
|
-
return data
|
|
277
|
-
|
|
278
|
-
def assistants_service_create_managed_model_with_http_info(self, body: 'ProjectIdAgentmanagedmodelsBody', project_id: 'str', **kwargs) -> 'V1ManagedModel': # noqa: E501
|
|
279
|
-
"""assistants_service_create_managed_model # noqa: E501
|
|
280
|
-
|
|
281
|
-
This method makes a synchronous HTTP request by default. To make an
|
|
282
|
-
asynchronous HTTP request, please pass async_req=True
|
|
283
|
-
>>> thread = api.assistants_service_create_managed_model_with_http_info(body, project_id, async_req=True)
|
|
284
|
-
>>> result = thread.get()
|
|
285
|
-
|
|
286
|
-
:param async_req bool
|
|
287
|
-
:param ProjectIdAgentmanagedmodelsBody body: (required)
|
|
288
|
-
:param str project_id: (required)
|
|
289
|
-
:return: V1ManagedModel
|
|
290
|
-
If the method is called asynchronously,
|
|
291
|
-
returns the request thread.
|
|
292
|
-
"""
|
|
293
|
-
|
|
294
|
-
all_params = ['body', 'project_id'] # noqa: E501
|
|
295
|
-
all_params.append('async_req')
|
|
296
|
-
all_params.append('_return_http_data_only')
|
|
297
|
-
all_params.append('_preload_content')
|
|
298
|
-
all_params.append('_request_timeout')
|
|
299
|
-
|
|
300
|
-
params = locals()
|
|
301
|
-
for key, val in six.iteritems(params['kwargs']):
|
|
302
|
-
if key not in all_params:
|
|
303
|
-
raise TypeError(
|
|
304
|
-
"Got an unexpected keyword argument '%s'"
|
|
305
|
-
" to method assistants_service_create_managed_model" % key
|
|
306
|
-
)
|
|
307
|
-
params[key] = val
|
|
308
|
-
del params['kwargs']
|
|
309
|
-
# verify the required parameter 'body' is set
|
|
310
|
-
if ('body' not in params or
|
|
311
|
-
params['body'] is None):
|
|
312
|
-
raise ValueError("Missing the required parameter `body` when calling `assistants_service_create_managed_model`") # noqa: E501
|
|
313
|
-
# verify the required parameter 'project_id' is set
|
|
314
|
-
if ('project_id' not in params or
|
|
315
|
-
params['project_id'] is None):
|
|
316
|
-
raise ValueError("Missing the required parameter `project_id` when calling `assistants_service_create_managed_model`") # noqa: E501
|
|
317
|
-
|
|
318
|
-
collection_formats = {}
|
|
319
|
-
|
|
320
|
-
path_params = {}
|
|
321
|
-
if 'project_id' in params:
|
|
322
|
-
path_params['projectId'] = params['project_id'] # noqa: E501
|
|
323
|
-
|
|
324
|
-
query_params = []
|
|
325
|
-
|
|
326
|
-
header_params = {}
|
|
327
|
-
|
|
328
|
-
form_params = []
|
|
329
|
-
local_var_files = {}
|
|
330
|
-
|
|
331
|
-
body_params = None
|
|
332
|
-
if 'body' in params:
|
|
333
|
-
body_params = params['body']
|
|
334
|
-
# HTTP header `Accept`
|
|
335
|
-
header_params['Accept'] = self.api_client.select_header_accept(
|
|
336
|
-
['application/json']) # noqa: E501
|
|
337
|
-
|
|
338
|
-
# HTTP header `Content-Type`
|
|
339
|
-
header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
|
|
340
|
-
['application/json']) # noqa: E501
|
|
341
|
-
|
|
342
|
-
# Authentication setting
|
|
343
|
-
auth_settings = [] # noqa: E501
|
|
344
|
-
|
|
345
|
-
return self.api_client.call_api(
|
|
346
|
-
'/v1/projects/{projectId}/agent-managed-models', 'POST',
|
|
347
|
-
path_params,
|
|
348
|
-
query_params,
|
|
349
|
-
header_params,
|
|
350
|
-
body=body_params,
|
|
351
|
-
post_params=form_params,
|
|
352
|
-
files=local_var_files,
|
|
353
|
-
response_type='V1ManagedModel', # noqa: E501
|
|
354
|
-
auth_settings=auth_settings,
|
|
355
|
-
async_req=params.get('async_req'),
|
|
356
|
-
_return_http_data_only=params.get('_return_http_data_only'),
|
|
357
|
-
_preload_content=params.get('_preload_content', True),
|
|
358
|
-
_request_timeout=params.get('_request_timeout'),
|
|
359
|
-
collection_formats=collection_formats)
|
|
360
|
-
|
|
361
256
|
def assistants_service_delete_assistant(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteAssistantResponse': # noqa: E501
|
|
362
257
|
"""assistants_service_delete_assistant # noqa: E501
|
|
363
258
|
|
|
@@ -665,45 +560,43 @@ class AssistantsServiceApi(object):
|
|
|
665
560
|
_request_timeout=params.get('_request_timeout'),
|
|
666
561
|
collection_formats=collection_formats)
|
|
667
562
|
|
|
668
|
-
def
|
|
669
|
-
"""
|
|
563
|
+
def assistants_service_get_assistant(self, id: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
|
|
564
|
+
"""assistants_service_get_assistant # noqa: E501
|
|
670
565
|
|
|
671
566
|
This method makes a synchronous HTTP request by default. To make an
|
|
672
567
|
asynchronous HTTP request, please pass async_req=True
|
|
673
|
-
>>> thread = api.
|
|
568
|
+
>>> thread = api.assistants_service_get_assistant(id, async_req=True)
|
|
674
569
|
>>> result = thread.get()
|
|
675
570
|
|
|
676
571
|
:param async_req bool
|
|
677
|
-
:param str project_id: (required)
|
|
678
572
|
:param str id: (required)
|
|
679
|
-
:return:
|
|
573
|
+
:return: V1Assistant
|
|
680
574
|
If the method is called asynchronously,
|
|
681
575
|
returns the request thread.
|
|
682
576
|
"""
|
|
683
577
|
kwargs['_return_http_data_only'] = True
|
|
684
578
|
if kwargs.get('async_req'):
|
|
685
|
-
return self.
|
|
579
|
+
return self.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
|
|
686
580
|
else:
|
|
687
|
-
(data) = self.
|
|
581
|
+
(data) = self.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
|
|
688
582
|
return data
|
|
689
583
|
|
|
690
|
-
def
|
|
691
|
-
"""
|
|
584
|
+
def assistants_service_get_assistant_with_http_info(self, id: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
|
|
585
|
+
"""assistants_service_get_assistant # noqa: E501
|
|
692
586
|
|
|
693
587
|
This method makes a synchronous HTTP request by default. To make an
|
|
694
588
|
asynchronous HTTP request, please pass async_req=True
|
|
695
|
-
>>> thread = api.
|
|
589
|
+
>>> thread = api.assistants_service_get_assistant_with_http_info(id, async_req=True)
|
|
696
590
|
>>> result = thread.get()
|
|
697
591
|
|
|
698
592
|
:param async_req bool
|
|
699
|
-
:param str project_id: (required)
|
|
700
593
|
:param str id: (required)
|
|
701
|
-
:return:
|
|
594
|
+
:return: V1Assistant
|
|
702
595
|
If the method is called asynchronously,
|
|
703
596
|
returns the request thread.
|
|
704
597
|
"""
|
|
705
598
|
|
|
706
|
-
all_params = ['
|
|
599
|
+
all_params = ['id'] # noqa: E501
|
|
707
600
|
all_params.append('async_req')
|
|
708
601
|
all_params.append('_return_http_data_only')
|
|
709
602
|
all_params.append('_preload_content')
|
|
@@ -714,24 +607,18 @@ class AssistantsServiceApi(object):
|
|
|
714
607
|
if key not in all_params:
|
|
715
608
|
raise TypeError(
|
|
716
609
|
"Got an unexpected keyword argument '%s'"
|
|
717
|
-
" to method
|
|
610
|
+
" to method assistants_service_get_assistant" % key
|
|
718
611
|
)
|
|
719
612
|
params[key] = val
|
|
720
613
|
del params['kwargs']
|
|
721
|
-
# verify the required parameter 'project_id' is set
|
|
722
|
-
if ('project_id' not in params or
|
|
723
|
-
params['project_id'] is None):
|
|
724
|
-
raise ValueError("Missing the required parameter `project_id` when calling `assistants_service_delete_managed_model`") # noqa: E501
|
|
725
614
|
# verify the required parameter 'id' is set
|
|
726
615
|
if ('id' not in params or
|
|
727
616
|
params['id'] is None):
|
|
728
|
-
raise ValueError("Missing the required parameter `id` when calling `
|
|
617
|
+
raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_assistant`") # noqa: E501
|
|
729
618
|
|
|
730
619
|
collection_formats = {}
|
|
731
620
|
|
|
732
621
|
path_params = {}
|
|
733
|
-
if 'project_id' in params:
|
|
734
|
-
path_params['projectId'] = params['project_id'] # noqa: E501
|
|
735
622
|
if 'id' in params:
|
|
736
623
|
path_params['id'] = params['id'] # noqa: E501
|
|
737
624
|
|
|
@@ -751,14 +638,14 @@ class AssistantsServiceApi(object):
|
|
|
751
638
|
auth_settings = [] # noqa: E501
|
|
752
639
|
|
|
753
640
|
return self.api_client.call_api(
|
|
754
|
-
'/v1/
|
|
641
|
+
'/v1/agents/{id}', 'GET',
|
|
755
642
|
path_params,
|
|
756
643
|
query_params,
|
|
757
644
|
header_params,
|
|
758
645
|
body=body_params,
|
|
759
646
|
post_params=form_params,
|
|
760
647
|
files=local_var_files,
|
|
761
|
-
response_type='
|
|
648
|
+
response_type='V1Assistant', # noqa: E501
|
|
762
649
|
auth_settings=auth_settings,
|
|
763
650
|
async_req=params.get('async_req'),
|
|
764
651
|
_return_http_data_only=params.get('_return_http_data_only'),
|
|
@@ -766,38 +653,38 @@ class AssistantsServiceApi(object):
|
|
|
766
653
|
_request_timeout=params.get('_request_timeout'),
|
|
767
654
|
collection_formats=collection_formats)
|
|
768
655
|
|
|
769
|
-
def
|
|
770
|
-
"""
|
|
656
|
+
def assistants_service_get_assistant_knowledge_status(self, id: 'str', **kwargs) -> 'V1AssistantKnowledgeStatus': # noqa: E501
|
|
657
|
+
"""GetAssistantKnowledgeStatus get the status of current uploads. For new uploads (direct binary uploads ) there is a dedicated HTTP handler registered on POST /v1/agents/{id}/knowledge # noqa: E501
|
|
771
658
|
|
|
772
659
|
This method makes a synchronous HTTP request by default. To make an
|
|
773
660
|
asynchronous HTTP request, please pass async_req=True
|
|
774
|
-
>>> thread = api.
|
|
661
|
+
>>> thread = api.assistants_service_get_assistant_knowledge_status(id, async_req=True)
|
|
775
662
|
>>> result = thread.get()
|
|
776
663
|
|
|
777
664
|
:param async_req bool
|
|
778
665
|
:param str id: (required)
|
|
779
|
-
:return:
|
|
666
|
+
:return: V1AssistantKnowledgeStatus
|
|
780
667
|
If the method is called asynchronously,
|
|
781
668
|
returns the request thread.
|
|
782
669
|
"""
|
|
783
670
|
kwargs['_return_http_data_only'] = True
|
|
784
671
|
if kwargs.get('async_req'):
|
|
785
|
-
return self.
|
|
672
|
+
return self.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
|
|
786
673
|
else:
|
|
787
|
-
(data) = self.
|
|
674
|
+
(data) = self.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
|
|
788
675
|
return data
|
|
789
676
|
|
|
790
|
-
def
|
|
791
|
-
"""
|
|
677
|
+
def assistants_service_get_assistant_knowledge_status_with_http_info(self, id: 'str', **kwargs) -> 'V1AssistantKnowledgeStatus': # noqa: E501
|
|
678
|
+
"""GetAssistantKnowledgeStatus get the status of current uploads. For new uploads (direct binary uploads ) there is a dedicated HTTP handler registered on POST /v1/agents/{id}/knowledge # noqa: E501
|
|
792
679
|
|
|
793
680
|
This method makes a synchronous HTTP request by default. To make an
|
|
794
681
|
asynchronous HTTP request, please pass async_req=True
|
|
795
|
-
>>> thread = api.
|
|
682
|
+
>>> thread = api.assistants_service_get_assistant_knowledge_status_with_http_info(id, async_req=True)
|
|
796
683
|
>>> result = thread.get()
|
|
797
684
|
|
|
798
685
|
:param async_req bool
|
|
799
686
|
:param str id: (required)
|
|
800
|
-
:return:
|
|
687
|
+
:return: V1AssistantKnowledgeStatus
|
|
801
688
|
If the method is called asynchronously,
|
|
802
689
|
returns the request thread.
|
|
803
690
|
"""
|
|
@@ -813,14 +700,14 @@ class AssistantsServiceApi(object):
|
|
|
813
700
|
if key not in all_params:
|
|
814
701
|
raise TypeError(
|
|
815
702
|
"Got an unexpected keyword argument '%s'"
|
|
816
|
-
" to method
|
|
703
|
+
" to method assistants_service_get_assistant_knowledge_status" % key
|
|
817
704
|
)
|
|
818
705
|
params[key] = val
|
|
819
706
|
del params['kwargs']
|
|
820
707
|
# verify the required parameter 'id' is set
|
|
821
708
|
if ('id' not in params or
|
|
822
709
|
params['id'] is None):
|
|
823
|
-
raise ValueError("Missing the required parameter `id` when calling `
|
|
710
|
+
raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_assistant_knowledge_status`") # noqa: E501
|
|
824
711
|
|
|
825
712
|
collection_formats = {}
|
|
826
713
|
|
|
@@ -844,14 +731,14 @@ class AssistantsServiceApi(object):
|
|
|
844
731
|
auth_settings = [] # noqa: E501
|
|
845
732
|
|
|
846
733
|
return self.api_client.call_api(
|
|
847
|
-
'/v1/agents/{id}', 'GET',
|
|
734
|
+
'/v1/agents/{id}/knowledge', 'GET',
|
|
848
735
|
path_params,
|
|
849
736
|
query_params,
|
|
850
737
|
header_params,
|
|
851
738
|
body=body_params,
|
|
852
739
|
post_params=form_params,
|
|
853
740
|
files=local_var_files,
|
|
854
|
-
response_type='
|
|
741
|
+
response_type='V1AssistantKnowledgeStatus', # noqa: E501
|
|
855
742
|
auth_settings=auth_settings,
|
|
856
743
|
async_req=params.get('async_req'),
|
|
857
744
|
_return_http_data_only=params.get('_return_http_data_only'),
|
|
@@ -859,43 +746,45 @@ class AssistantsServiceApi(object):
|
|
|
859
746
|
_request_timeout=params.get('_request_timeout'),
|
|
860
747
|
collection_formats=collection_formats)
|
|
861
748
|
|
|
862
|
-
def
|
|
863
|
-
"""
|
|
749
|
+
def assistants_service_get_conversation(self, assistant_id: 'str', id: 'str', **kwargs) -> 'V1Conversation': # noqa: E501
|
|
750
|
+
"""assistants_service_get_conversation # noqa: E501
|
|
864
751
|
|
|
865
752
|
This method makes a synchronous HTTP request by default. To make an
|
|
866
753
|
asynchronous HTTP request, please pass async_req=True
|
|
867
|
-
>>> thread = api.
|
|
754
|
+
>>> thread = api.assistants_service_get_conversation(assistant_id, id, async_req=True)
|
|
868
755
|
>>> result = thread.get()
|
|
869
756
|
|
|
870
757
|
:param async_req bool
|
|
758
|
+
:param str assistant_id: (required)
|
|
871
759
|
:param str id: (required)
|
|
872
|
-
:return:
|
|
760
|
+
:return: V1Conversation
|
|
873
761
|
If the method is called asynchronously,
|
|
874
762
|
returns the request thread.
|
|
875
763
|
"""
|
|
876
764
|
kwargs['_return_http_data_only'] = True
|
|
877
765
|
if kwargs.get('async_req'):
|
|
878
|
-
return self.
|
|
766
|
+
return self.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
|
|
879
767
|
else:
|
|
880
|
-
(data) = self.
|
|
768
|
+
(data) = self.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
|
|
881
769
|
return data
|
|
882
770
|
|
|
883
|
-
def
|
|
884
|
-
"""
|
|
771
|
+
def assistants_service_get_conversation_with_http_info(self, assistant_id: 'str', id: 'str', **kwargs) -> 'V1Conversation': # noqa: E501
|
|
772
|
+
"""assistants_service_get_conversation # noqa: E501
|
|
885
773
|
|
|
886
774
|
This method makes a synchronous HTTP request by default. To make an
|
|
887
775
|
asynchronous HTTP request, please pass async_req=True
|
|
888
|
-
>>> thread = api.
|
|
776
|
+
>>> thread = api.assistants_service_get_conversation_with_http_info(assistant_id, id, async_req=True)
|
|
889
777
|
>>> result = thread.get()
|
|
890
778
|
|
|
891
779
|
:param async_req bool
|
|
780
|
+
:param str assistant_id: (required)
|
|
892
781
|
:param str id: (required)
|
|
893
|
-
:return:
|
|
782
|
+
:return: V1Conversation
|
|
894
783
|
If the method is called asynchronously,
|
|
895
784
|
returns the request thread.
|
|
896
785
|
"""
|
|
897
786
|
|
|
898
|
-
all_params = ['id'] # noqa: E501
|
|
787
|
+
all_params = ['assistant_id', 'id'] # noqa: E501
|
|
899
788
|
all_params.append('async_req')
|
|
900
789
|
all_params.append('_return_http_data_only')
|
|
901
790
|
all_params.append('_preload_content')
|
|
@@ -906,18 +795,24 @@ class AssistantsServiceApi(object):
|
|
|
906
795
|
if key not in all_params:
|
|
907
796
|
raise TypeError(
|
|
908
797
|
"Got an unexpected keyword argument '%s'"
|
|
909
|
-
" to method
|
|
798
|
+
" to method assistants_service_get_conversation" % key
|
|
910
799
|
)
|
|
911
800
|
params[key] = val
|
|
912
801
|
del params['kwargs']
|
|
802
|
+
# verify the required parameter 'assistant_id' is set
|
|
803
|
+
if ('assistant_id' not in params or
|
|
804
|
+
params['assistant_id'] is None):
|
|
805
|
+
raise ValueError("Missing the required parameter `assistant_id` when calling `assistants_service_get_conversation`") # noqa: E501
|
|
913
806
|
# verify the required parameter 'id' is set
|
|
914
807
|
if ('id' not in params or
|
|
915
808
|
params['id'] is None):
|
|
916
|
-
raise ValueError("Missing the required parameter `id` when calling `
|
|
809
|
+
raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_conversation`") # noqa: E501
|
|
917
810
|
|
|
918
811
|
collection_formats = {}
|
|
919
812
|
|
|
920
813
|
path_params = {}
|
|
814
|
+
if 'assistant_id' in params:
|
|
815
|
+
path_params['assistantId'] = params['assistant_id'] # noqa: E501
|
|
921
816
|
if 'id' in params:
|
|
922
817
|
path_params['id'] = params['id'] # noqa: E501
|
|
923
818
|
|
|
@@ -937,14 +832,14 @@ class AssistantsServiceApi(object):
|
|
|
937
832
|
auth_settings = [] # noqa: E501
|
|
938
833
|
|
|
939
834
|
return self.api_client.call_api(
|
|
940
|
-
'/v1/agents/{id}
|
|
835
|
+
'/v1/agents/{assistantId}/conversations/{id}', 'GET',
|
|
941
836
|
path_params,
|
|
942
837
|
query_params,
|
|
943
838
|
header_params,
|
|
944
839
|
body=body_params,
|
|
945
840
|
post_params=form_params,
|
|
946
841
|
files=local_var_files,
|
|
947
|
-
response_type='
|
|
842
|
+
response_type='V1Conversation', # noqa: E501
|
|
948
843
|
auth_settings=auth_settings,
|
|
949
844
|
async_req=params.get('async_req'),
|
|
950
845
|
_return_http_data_only=params.get('_return_http_data_only'),
|
|
@@ -952,45 +847,47 @@ class AssistantsServiceApi(object):
|
|
|
952
847
|
_request_timeout=params.get('_request_timeout'),
|
|
953
848
|
collection_formats=collection_formats)
|
|
954
849
|
|
|
955
|
-
def
|
|
956
|
-
"""
|
|
850
|
+
def assistants_service_get_managed_model_by_name(self, project_id: 'str', managed_endpoint_id: 'str', name: 'str', **kwargs) -> 'V1ManagedModel': # noqa: E501
|
|
851
|
+
"""assistants_service_get_managed_model_by_name # noqa: E501
|
|
957
852
|
|
|
958
853
|
This method makes a synchronous HTTP request by default. To make an
|
|
959
854
|
asynchronous HTTP request, please pass async_req=True
|
|
960
|
-
>>> thread = api.
|
|
855
|
+
>>> thread = api.assistants_service_get_managed_model_by_name(project_id, managed_endpoint_id, name, async_req=True)
|
|
961
856
|
>>> result = thread.get()
|
|
962
857
|
|
|
963
858
|
:param async_req bool
|
|
964
|
-
:param str
|
|
965
|
-
:param str
|
|
966
|
-
:
|
|
859
|
+
:param str project_id: (required)
|
|
860
|
+
:param str managed_endpoint_id: (required)
|
|
861
|
+
:param str name: (required)
|
|
862
|
+
:return: V1ManagedModel
|
|
967
863
|
If the method is called asynchronously,
|
|
968
864
|
returns the request thread.
|
|
969
865
|
"""
|
|
970
866
|
kwargs['_return_http_data_only'] = True
|
|
971
867
|
if kwargs.get('async_req'):
|
|
972
|
-
return self.
|
|
868
|
+
return self.assistants_service_get_managed_model_by_name_with_http_info(project_id, managed_endpoint_id, name, **kwargs) # noqa: E501
|
|
973
869
|
else:
|
|
974
|
-
(data) = self.
|
|
870
|
+
(data) = self.assistants_service_get_managed_model_by_name_with_http_info(project_id, managed_endpoint_id, name, **kwargs) # noqa: E501
|
|
975
871
|
return data
|
|
976
872
|
|
|
977
|
-
def
|
|
978
|
-
"""
|
|
873
|
+
def assistants_service_get_managed_model_by_name_with_http_info(self, project_id: 'str', managed_endpoint_id: 'str', name: 'str', **kwargs) -> 'V1ManagedModel': # noqa: E501
|
|
874
|
+
"""assistants_service_get_managed_model_by_name # noqa: E501
|
|
979
875
|
|
|
980
876
|
This method makes a synchronous HTTP request by default. To make an
|
|
981
877
|
asynchronous HTTP request, please pass async_req=True
|
|
982
|
-
>>> thread = api.
|
|
878
|
+
>>> thread = api.assistants_service_get_managed_model_by_name_with_http_info(project_id, managed_endpoint_id, name, async_req=True)
|
|
983
879
|
>>> result = thread.get()
|
|
984
880
|
|
|
985
881
|
:param async_req bool
|
|
986
|
-
:param str
|
|
987
|
-
:param str
|
|
988
|
-
:
|
|
882
|
+
:param str project_id: (required)
|
|
883
|
+
:param str managed_endpoint_id: (required)
|
|
884
|
+
:param str name: (required)
|
|
885
|
+
:return: V1ManagedModel
|
|
989
886
|
If the method is called asynchronously,
|
|
990
887
|
returns the request thread.
|
|
991
888
|
"""
|
|
992
889
|
|
|
993
|
-
all_params = ['
|
|
890
|
+
all_params = ['project_id', 'managed_endpoint_id', 'name'] # noqa: E501
|
|
994
891
|
all_params.append('async_req')
|
|
995
892
|
all_params.append('_return_http_data_only')
|
|
996
893
|
all_params.append('_preload_content')
|
|
@@ -1001,26 +898,32 @@ class AssistantsServiceApi(object):
|
|
|
1001
898
|
if key not in all_params:
|
|
1002
899
|
raise TypeError(
|
|
1003
900
|
"Got an unexpected keyword argument '%s'"
|
|
1004
|
-
" to method
|
|
901
|
+
" to method assistants_service_get_managed_model_by_name" % key
|
|
1005
902
|
)
|
|
1006
903
|
params[key] = val
|
|
1007
904
|
del params['kwargs']
|
|
1008
|
-
# verify the required parameter '
|
|
1009
|
-
if ('
|
|
1010
|
-
params['
|
|
1011
|
-
raise ValueError("Missing the required parameter `
|
|
1012
|
-
# verify the required parameter '
|
|
1013
|
-
if ('
|
|
1014
|
-
params['
|
|
1015
|
-
raise ValueError("Missing the required parameter `
|
|
905
|
+
# verify the required parameter 'project_id' is set
|
|
906
|
+
if ('project_id' not in params or
|
|
907
|
+
params['project_id'] is None):
|
|
908
|
+
raise ValueError("Missing the required parameter `project_id` when calling `assistants_service_get_managed_model_by_name`") # noqa: E501
|
|
909
|
+
# verify the required parameter 'managed_endpoint_id' is set
|
|
910
|
+
if ('managed_endpoint_id' not in params or
|
|
911
|
+
params['managed_endpoint_id'] is None):
|
|
912
|
+
raise ValueError("Missing the required parameter `managed_endpoint_id` when calling `assistants_service_get_managed_model_by_name`") # noqa: E501
|
|
913
|
+
# verify the required parameter 'name' is set
|
|
914
|
+
if ('name' not in params or
|
|
915
|
+
params['name'] is None):
|
|
916
|
+
raise ValueError("Missing the required parameter `name` when calling `assistants_service_get_managed_model_by_name`") # noqa: E501
|
|
1016
917
|
|
|
1017
918
|
collection_formats = {}
|
|
1018
919
|
|
|
1019
920
|
path_params = {}
|
|
1020
|
-
if '
|
|
1021
|
-
path_params['
|
|
1022
|
-
if '
|
|
1023
|
-
path_params['
|
|
921
|
+
if 'project_id' in params:
|
|
922
|
+
path_params['projectId'] = params['project_id'] # noqa: E501
|
|
923
|
+
if 'managed_endpoint_id' in params:
|
|
924
|
+
path_params['managedEndpointId'] = params['managed_endpoint_id'] # noqa: E501
|
|
925
|
+
if 'name' in params:
|
|
926
|
+
path_params['name'] = params['name'] # noqa: E501
|
|
1024
927
|
|
|
1025
928
|
query_params = []
|
|
1026
929
|
|
|
@@ -1038,14 +941,14 @@ class AssistantsServiceApi(object):
|
|
|
1038
941
|
auth_settings = [] # noqa: E501
|
|
1039
942
|
|
|
1040
943
|
return self.api_client.call_api(
|
|
1041
|
-
'/v1/
|
|
944
|
+
'/v1/projects/{projectId}/agent-managed-endpoints/{managedEndpointId}/model/{name}', 'GET',
|
|
1042
945
|
path_params,
|
|
1043
946
|
query_params,
|
|
1044
947
|
header_params,
|
|
1045
948
|
body=body_params,
|
|
1046
949
|
post_params=form_params,
|
|
1047
950
|
files=local_var_files,
|
|
1048
|
-
response_type='
|
|
951
|
+
response_type='V1ManagedModel', # noqa: E501
|
|
1049
952
|
auth_settings=auth_settings,
|
|
1050
953
|
async_req=params.get('async_req'),
|
|
1051
954
|
_return_http_data_only=params.get('_return_http_data_only'),
|
|
@@ -1348,103 +1251,6 @@ class AssistantsServiceApi(object):
|
|
|
1348
1251
|
_request_timeout=params.get('_request_timeout'),
|
|
1349
1252
|
collection_formats=collection_formats)
|
|
1350
1253
|
|
|
1351
|
-
def assistants_service_list_managed_models(self, project_id: 'str', **kwargs) -> 'V1ListManagedModelsResponse': # noqa: E501
|
|
1352
|
-
"""ManagedModels # noqa: E501
|
|
1353
|
-
|
|
1354
|
-
This method makes a synchronous HTTP request by default. To make an
|
|
1355
|
-
asynchronous HTTP request, please pass async_req=True
|
|
1356
|
-
>>> thread = api.assistants_service_list_managed_models(project_id, async_req=True)
|
|
1357
|
-
>>> result = thread.get()
|
|
1358
|
-
|
|
1359
|
-
:param async_req bool
|
|
1360
|
-
:param str project_id: (required)
|
|
1361
|
-
:param str org_id:
|
|
1362
|
-
:return: V1ListManagedModelsResponse
|
|
1363
|
-
If the method is called asynchronously,
|
|
1364
|
-
returns the request thread.
|
|
1365
|
-
"""
|
|
1366
|
-
kwargs['_return_http_data_only'] = True
|
|
1367
|
-
if kwargs.get('async_req'):
|
|
1368
|
-
return self.assistants_service_list_managed_models_with_http_info(project_id, **kwargs) # noqa: E501
|
|
1369
|
-
else:
|
|
1370
|
-
(data) = self.assistants_service_list_managed_models_with_http_info(project_id, **kwargs) # noqa: E501
|
|
1371
|
-
return data
|
|
1372
|
-
|
|
1373
|
-
def assistants_service_list_managed_models_with_http_info(self, project_id: 'str', **kwargs) -> 'V1ListManagedModelsResponse': # noqa: E501
|
|
1374
|
-
"""ManagedModels # noqa: E501
|
|
1375
|
-
|
|
1376
|
-
This method makes a synchronous HTTP request by default. To make an
|
|
1377
|
-
asynchronous HTTP request, please pass async_req=True
|
|
1378
|
-
>>> thread = api.assistants_service_list_managed_models_with_http_info(project_id, async_req=True)
|
|
1379
|
-
>>> result = thread.get()
|
|
1380
|
-
|
|
1381
|
-
:param async_req bool
|
|
1382
|
-
:param str project_id: (required)
|
|
1383
|
-
:param str org_id:
|
|
1384
|
-
:return: V1ListManagedModelsResponse
|
|
1385
|
-
If the method is called asynchronously,
|
|
1386
|
-
returns the request thread.
|
|
1387
|
-
"""
|
|
1388
|
-
|
|
1389
|
-
all_params = ['project_id', 'org_id'] # noqa: E501
|
|
1390
|
-
all_params.append('async_req')
|
|
1391
|
-
all_params.append('_return_http_data_only')
|
|
1392
|
-
all_params.append('_preload_content')
|
|
1393
|
-
all_params.append('_request_timeout')
|
|
1394
|
-
|
|
1395
|
-
params = locals()
|
|
1396
|
-
for key, val in six.iteritems(params['kwargs']):
|
|
1397
|
-
if key not in all_params:
|
|
1398
|
-
raise TypeError(
|
|
1399
|
-
"Got an unexpected keyword argument '%s'"
|
|
1400
|
-
" to method assistants_service_list_managed_models" % key
|
|
1401
|
-
)
|
|
1402
|
-
params[key] = val
|
|
1403
|
-
del params['kwargs']
|
|
1404
|
-
# verify the required parameter 'project_id' is set
|
|
1405
|
-
if ('project_id' not in params or
|
|
1406
|
-
params['project_id'] is None):
|
|
1407
|
-
raise ValueError("Missing the required parameter `project_id` when calling `assistants_service_list_managed_models`") # noqa: E501
|
|
1408
|
-
|
|
1409
|
-
collection_formats = {}
|
|
1410
|
-
|
|
1411
|
-
path_params = {}
|
|
1412
|
-
if 'project_id' in params:
|
|
1413
|
-
path_params['projectId'] = params['project_id'] # noqa: E501
|
|
1414
|
-
|
|
1415
|
-
query_params = []
|
|
1416
|
-
if 'org_id' in params:
|
|
1417
|
-
query_params.append(('orgId', params['org_id'])) # noqa: E501
|
|
1418
|
-
|
|
1419
|
-
header_params = {}
|
|
1420
|
-
|
|
1421
|
-
form_params = []
|
|
1422
|
-
local_var_files = {}
|
|
1423
|
-
|
|
1424
|
-
body_params = None
|
|
1425
|
-
# HTTP header `Accept`
|
|
1426
|
-
header_params['Accept'] = self.api_client.select_header_accept(
|
|
1427
|
-
['application/json']) # noqa: E501
|
|
1428
|
-
|
|
1429
|
-
# Authentication setting
|
|
1430
|
-
auth_settings = [] # noqa: E501
|
|
1431
|
-
|
|
1432
|
-
return self.api_client.call_api(
|
|
1433
|
-
'/v1/projects/{projectId}/agent-managed-models', 'GET',
|
|
1434
|
-
path_params,
|
|
1435
|
-
query_params,
|
|
1436
|
-
header_params,
|
|
1437
|
-
body=body_params,
|
|
1438
|
-
post_params=form_params,
|
|
1439
|
-
files=local_var_files,
|
|
1440
|
-
response_type='V1ListManagedModelsResponse', # noqa: E501
|
|
1441
|
-
auth_settings=auth_settings,
|
|
1442
|
-
async_req=params.get('async_req'),
|
|
1443
|
-
_return_http_data_only=params.get('_return_http_data_only'),
|
|
1444
|
-
_preload_content=params.get('_preload_content', True),
|
|
1445
|
-
_request_timeout=params.get('_request_timeout'),
|
|
1446
|
-
collection_formats=collection_formats)
|
|
1447
|
-
|
|
1448
1254
|
def assistants_service_start_conversation(self, body: 'AssistantIdConversationsBody', assistant_id: 'str', **kwargs) -> 'StreamResultOfV1ConversationResponseChunk': # noqa: E501
|
|
1449
1255
|
"""assistants_service_start_conversation # noqa: E501
|
|
1450
1256
|
|
|
@@ -716,6 +716,7 @@ class DataConnectionServiceApi(object):
|
|
|
716
716
|
:param async_req bool
|
|
717
717
|
:param str project_id: (required)
|
|
718
718
|
:param str cluster_id:
|
|
719
|
+
:param list[str] state:
|
|
719
720
|
:return: V1ListDataConnectionsResponse
|
|
720
721
|
If the method is called asynchronously,
|
|
721
722
|
returns the request thread.
|
|
@@ -738,12 +739,13 @@ class DataConnectionServiceApi(object):
|
|
|
738
739
|
:param async_req bool
|
|
739
740
|
:param str project_id: (required)
|
|
740
741
|
:param str cluster_id:
|
|
742
|
+
:param list[str] state:
|
|
741
743
|
:return: V1ListDataConnectionsResponse
|
|
742
744
|
If the method is called asynchronously,
|
|
743
745
|
returns the request thread.
|
|
744
746
|
"""
|
|
745
747
|
|
|
746
|
-
all_params = ['project_id', 'cluster_id'] # noqa: E501
|
|
748
|
+
all_params = ['project_id', 'cluster_id', 'state'] # noqa: E501
|
|
747
749
|
all_params.append('async_req')
|
|
748
750
|
all_params.append('_return_http_data_only')
|
|
749
751
|
all_params.append('_preload_content')
|
|
@@ -772,6 +774,9 @@ class DataConnectionServiceApi(object):
|
|
|
772
774
|
query_params = []
|
|
773
775
|
if 'cluster_id' in params:
|
|
774
776
|
query_params.append(('clusterId', params['cluster_id'])) # noqa: E501
|
|
777
|
+
if 'state' in params:
|
|
778
|
+
query_params.append(('state', params['state'])) # noqa: E501
|
|
779
|
+
collection_formats['state'] = 'multi' # noqa: E501
|
|
775
780
|
|
|
776
781
|
header_params = {}
|
|
777
782
|
|