15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: check auth when add xinference tti tts model

This commit is contained in:
CaptainB 2024-12-27 11:35:50 +08:00
parent 415b2e1967
commit 75a7630be8
2 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/setting/models_provider/impl/xinference_model_provider/model/tti.py
View File

@ -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)

7
apps/setting/models_provider/impl/xinference_model_provider/model/tts.py
View File

@ -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(