Procházet zdrojové kódy

fix: fish audio wrong validate credentials interface (#11019)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
yihong před 5 měsíci
rodič
revize
448a19bf54

+ 2 - 1
api/core/model_runtime/model_providers/fishaudio/fishaudio.py

@@ -18,7 +18,8 @@ class FishAudioProvider(ModelProvider):
         """
         try:
             model_instance = self.get_model_instance(ModelType.TTS)
-            model_instance.validate_credentials(credentials=credentials)
+            # FIXME fish tts do not have model for now, so set it to empty string instead
+            model_instance.validate_credentials(model="", credentials=credentials)
         except CredentialsValidateFailedError as ex:
             raise ex
         except Exception as ex:

+ 2 - 2
api/core/model_runtime/model_providers/fishaudio/tts/tts.py

@@ -66,7 +66,7 @@ class FishAudioText2SpeechModel(TTSModel):
             voice=voice,
         )
 
-    def validate_credentials(self, credentials: dict, user: Optional[str] = None) -> None:
+    def validate_credentials(self, model: str, credentials: dict, user: Optional[str] = None) -> None:
         """
         Validate credentials for text2speech model
 
@@ -76,7 +76,7 @@ class FishAudioText2SpeechModel(TTSModel):
 
         try:
             self.get_tts_model_voices(
-                None,
+                "",
                 credentials={
                     "api_key": credentials["api_key"],
                     "api_base": credentials["api_base"],