zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: check auth when add xinference tti tts model

Browse Source
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 return False
def check_auth(self): def check_auth(self):
chat = OpenAI(api_key=self.api_key, base_url=self.api_base) self.generate_image('生成一个小猫图片')
response_list = chat.models.with_raw_response.list()
# self.generate_image('生成一个小猫图片')
def generate_image(self, prompt: str, negative_prompt: str = None): def generate_image(self, prompt: str, negative_prompt: str = None):
chat = OpenAI(api_key=self.api_key, base_url=self.api_base) 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): def check_auth(self):
client = OpenAI( self.text_to_speech('你好')
base_url=self.api_base,
api_key=self.api_key
)
response_list = client.models.with_raw_response.list()
# print(response_list)
def text_to_speech(self, text): def text_to_speech(self, text):
client = OpenAI( client = OpenAI(