Skip to content

Commit 75a7630

Browse files
committed
fix: check auth when add xinference tti tts model
1 parent 415b2e1 commit 75a7630

File tree

2 files changed

+2
-10
lines changed
  • apps/setting/models_provider/impl/xinference_model_provider/model

2 files changed

+2
-10
lines changed

apps/setting/models_provider/impl/xinference_model_provider/model/tti.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,10 +45,7 @@ def is_cache_model(self):
4545
return False
4646

4747
def check_auth(self):
48-
chat = OpenAI(api_key=self.api_key, base_url=self.api_base)
49-
response_list = chat.models.with_raw_response.list()
50-
51-
# self.generate_image('生成一个小猫图片')
48+
self.generate_image('生成一个小猫图片')
5249

5350
def generate_image(self, prompt: str, negative_prompt: str = None):
5451
chat = OpenAI(api_key=self.api_key, base_url=self.api_base)

apps/setting/models_provider/impl/xinference_model_provider/model/tts.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,7 @@ def new_instance(model_type, model_name, model_credential: Dict[str, object], **
4040
)
4141

4242
def check_auth(self):
43-
client = OpenAI(
44-
base_url=self.api_base,
45-
api_key=self.api_key
46-
)
47-
response_list = client.models.with_raw_response.list()
48-
# print(response_list)
43+
self.text_to_speech('你好')
4944

5045
def text_to_speech(self, text):
5146
client = OpenAI(

0 commit comments

Comments
 (0)