From 75a7630be8e53b9b54f5d925dc99de0c65afb635 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Fri, 27 Dec 2024 11:35:50 +0800 Subject: [PATCH] fix: check auth when add xinference tti tts model --- .../impl/xinference_model_provider/model/tti.py | 5 +---- .../impl/xinference_model_provider/model/tts.py | 7 +------ 2 files changed, 2 insertions(+), 10 deletions(-) diff --git a/apps/setting/models_provider/impl/xinference_model_provider/model/tti.py b/apps/setting/models_provider/impl/xinference_model_provider/model/tti.py index ee5b655f4..e050c6bf7 100644 --- a/apps/setting/models_provider/impl/xinference_model_provider/model/tti.py +++ b/apps/setting/models_provider/impl/xinference_model_provider/model/tti.py @@ -45,10 +45,7 @@ class XinferenceTextToImage(MaxKBBaseModel, BaseTextToImage): return False def check_auth(self): - chat = OpenAI(api_key=self.api_key, base_url=self.api_base) - response_list = chat.models.with_raw_response.list() - - # self.generate_image('生成一个小猫图片') + self.generate_image('生成一个小猫图片') def generate_image(self, prompt: str, negative_prompt: str = None): chat = OpenAI(api_key=self.api_key, base_url=self.api_base) diff --git a/apps/setting/models_provider/impl/xinference_model_provider/model/tts.py b/apps/setting/models_provider/impl/xinference_model_provider/model/tts.py index 3a9be27db..cf529fb48 100644 --- a/apps/setting/models_provider/impl/xinference_model_provider/model/tts.py +++ b/apps/setting/models_provider/impl/xinference_model_provider/model/tts.py @@ -40,12 +40,7 @@ class XInferenceTextToSpeech(MaxKBBaseModel, BaseTextToSpeech): ) def check_auth(self): - client = OpenAI( - base_url=self.api_base, - api_key=self.api_key - ) - response_list = client.models.with_raw_response.list() - # print(response_list) + self.text_to_speech('你好') def text_to_speech(self, text): client = OpenAI(