lightning-sdk 0.1.39__py3-none-any.whl → 0.1.41__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 (48) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/_mmt/__init__.py +3 -0
  3. lightning_sdk/_mmt/base.py +180 -0
  4. lightning_sdk/_mmt/mmt.py +161 -0
  5. lightning_sdk/_mmt/v1.py +69 -0
  6. lightning_sdk/_mmt/v2.py +141 -0
  7. lightning_sdk/api/mmt_api.py +148 -0
  8. lightning_sdk/api/teamspace_api.py +0 -9
  9. lightning_sdk/api/utils.py +2 -1
  10. lightning_sdk/cli/mmt.py +137 -0
  11. lightning_sdk/job/base.py +2 -3
  12. lightning_sdk/job/v1.py +2 -1
  13. lightning_sdk/job/v2.py +6 -9
  14. lightning_sdk/lightning_cloud/openapi/__init__.py +8 -3
  15. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +90 -284
  16. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +118 -1
  17. lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py +5 -1
  18. lightning_sdk/lightning_cloud/openapi/models/__init__.py +8 -3
  19. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  20. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
  22. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/{project_id_agentmanagedmodels_body.py → v1_body.py} +21 -47
  24. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +227 -0
  26. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_spec.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_header.py +175 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +55 -3
  31. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +29 -3
  33. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
  36. lightning_sdk/lightning_cloud/openapi/models/{v1_list_managed_models_response.py → v1_resource_visibility.py} +23 -23
  37. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_secret_type.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_managed_model_response.py → v1_update_model_visibility_response.py} +6 -6
  40. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +41 -15
  41. lightning_sdk/models.py +29 -8
  42. lightning_sdk/teamspace.py +17 -15
  43. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.dist-info}/METADATA +1 -1
  44. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.dist-info}/RECORD +48 -36
  45. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.dist-info}/entry_points.txt +1 -0
  46. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.dist-info}/LICENSE +0 -0
  47. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.dist-info}/WHEEL +0 -0
  48. {lightning_sdk-0.1.39.dist-info → lightning_sdk-0.1.41.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 assistants_service_delete_managed_model(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteManagedModelResponse': # noqa: E501
669
- """assistants_service_delete_managed_model # noqa: E501
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.assistants_service_delete_managed_model(project_id, id, async_req=True)
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: V1DeleteManagedModelResponse
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.assistants_service_delete_managed_model_with_http_info(project_id, id, **kwargs) # noqa: E501
579
+ return self.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
686
580
  else:
687
- (data) = self.assistants_service_delete_managed_model_with_http_info(project_id, id, **kwargs) # noqa: E501
581
+ (data) = self.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
688
582
  return data
689
583
 
690
- def assistants_service_delete_managed_model_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteManagedModelResponse': # noqa: E501
691
- """assistants_service_delete_managed_model # noqa: E501
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.assistants_service_delete_managed_model_with_http_info(project_id, id, async_req=True)
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: V1DeleteManagedModelResponse
594
+ :return: V1Assistant
702
595
  If the method is called asynchronously,
703
596
  returns the request thread.
704
597
  """
705
598
 
706
- all_params = ['project_id', 'id'] # noqa: E501
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 assistants_service_delete_managed_model" % key
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 `assistants_service_delete_managed_model`") # noqa: E501
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/projects/{projectId}/agent-managed-models/{id}', 'DELETE',
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='V1DeleteManagedModelResponse', # noqa: E501
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 assistants_service_get_assistant(self, id: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
770
- """assistants_service_get_assistant # noqa: E501
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.assistants_service_get_assistant(id, async_req=True)
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: V1Assistant
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.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
672
+ return self.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
786
673
  else:
787
- (data) = self.assistants_service_get_assistant_with_http_info(id, **kwargs) # noqa: E501
674
+ (data) = self.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
788
675
  return data
789
676
 
790
- def assistants_service_get_assistant_with_http_info(self, id: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
791
- """assistants_service_get_assistant # noqa: E501
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.assistants_service_get_assistant_with_http_info(id, async_req=True)
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: V1Assistant
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 assistants_service_get_assistant" % key
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 `assistants_service_get_assistant`") # noqa: E501
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='V1Assistant', # noqa: E501
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 assistants_service_get_assistant_knowledge_status(self, id: 'str', **kwargs) -> 'V1AssistantKnowledgeStatus': # noqa: E501
863
- """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
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.assistants_service_get_assistant_knowledge_status(id, async_req=True)
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: V1AssistantKnowledgeStatus
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.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
766
+ return self.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
879
767
  else:
880
- (data) = self.assistants_service_get_assistant_knowledge_status_with_http_info(id, **kwargs) # noqa: E501
768
+ (data) = self.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
881
769
  return data
882
770
 
883
- def assistants_service_get_assistant_knowledge_status_with_http_info(self, id: 'str', **kwargs) -> 'V1AssistantKnowledgeStatus': # noqa: E501
884
- """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
+ 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.assistants_service_get_assistant_knowledge_status_with_http_info(id, async_req=True)
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: V1AssistantKnowledgeStatus
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 assistants_service_get_assistant_knowledge_status" % key
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 `assistants_service_get_assistant_knowledge_status`") # noqa: E501
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}/knowledge', 'GET',
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='V1AssistantKnowledgeStatus', # noqa: E501
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 assistants_service_get_conversation(self, assistant_id: 'str', id: 'str', **kwargs) -> 'V1Conversation': # noqa: E501
956
- """assistants_service_get_conversation # noqa: E501
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.assistants_service_get_conversation(assistant_id, id, async_req=True)
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 assistant_id: (required)
965
- :param str id: (required)
966
- :return: V1Conversation
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.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
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.assistants_service_get_conversation_with_http_info(assistant_id, id, **kwargs) # noqa: E501
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 assistants_service_get_conversation_with_http_info(self, assistant_id: 'str', id: 'str', **kwargs) -> 'V1Conversation': # noqa: E501
978
- """assistants_service_get_conversation # noqa: E501
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.assistants_service_get_conversation_with_http_info(assistant_id, id, async_req=True)
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 assistant_id: (required)
987
- :param str id: (required)
988
- :return: V1Conversation
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 = ['assistant_id', 'id'] # noqa: E501
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 assistants_service_get_conversation" % key
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 'assistant_id' is set
1009
- if ('assistant_id' not in params or
1010
- params['assistant_id'] is None):
1011
- raise ValueError("Missing the required parameter `assistant_id` when calling `assistants_service_get_conversation`") # noqa: E501
1012
- # verify the required parameter 'id' is set
1013
- if ('id' not in params or
1014
- params['id'] is None):
1015
- raise ValueError("Missing the required parameter `id` when calling `assistants_service_get_conversation`") # noqa: E501
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 'assistant_id' in params:
1021
- path_params['assistantId'] = params['assistant_id'] # noqa: E501
1022
- if 'id' in params:
1023
- path_params['id'] = params['id'] # noqa: E501
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/agents/{assistantId}/conversations/{id}', 'GET',
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='V1Conversation', # noqa: E501
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