| @@ -1,7 +1,7 @@ | |||
| import logging | |||
| import re | |||
| from abc import abstractmethod | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| from pydantic import ConfigDict | |||
| @@ -88,7 +88,7 @@ class TTSModel(AIModel): | |||
| else: | |||
| return [{"name": d["name"], "value": d["mode"]} for d in voices] | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> any: | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> Any: | |||
| """ | |||
| Get voice for given tts model | |||
| @@ -1,6 +1,6 @@ | |||
| import concurrent.futures | |||
| import copy | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| from openai import AzureOpenAI | |||
| @@ -19,7 +19,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel): | |||
| def _invoke( | |||
| self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None | |||
| ) -> any: | |||
| ) -> Any: | |||
| """ | |||
| _invoke text2speech model | |||
| @@ -56,7 +56,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel): | |||
| except Exception as ex: | |||
| raise CredentialsValidateFailedError(str(ex)) | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any: | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any: | |||
| """ | |||
| _tts_invoke_streaming text2speech model | |||
| :param model: model name | |||
| @@ -1,4 +1,4 @@ | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| import httpx | |||
| @@ -46,7 +46,7 @@ class FishAudioText2SpeechModel(TTSModel): | |||
| content_text: str, | |||
| voice: str, | |||
| user: Optional[str] = None, | |||
| ) -> any: | |||
| ) -> Any: | |||
| """ | |||
| Invoke text2speech model | |||
| @@ -87,7 +87,7 @@ class FishAudioText2SpeechModel(TTSModel): | |||
| except Exception as ex: | |||
| raise CredentialsValidateFailedError(str(ex)) | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any: | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any: | |||
| """ | |||
| Invoke streaming text2speech model | |||
| :param model: model name | |||
| @@ -112,7 +112,7 @@ class FishAudioText2SpeechModel(TTSModel): | |||
| except Exception as ex: | |||
| raise InvokeBadRequestError(str(ex)) | |||
| def _tts_invoke_streaming_sentence(self, credentials: dict, content_text: str, voice: Optional[str] = None) -> any: | |||
| def _tts_invoke_streaming_sentence(self, credentials: dict, content_text: str, voice: Optional[str] = None) -> Any: | |||
| """ | |||
| Invoke streaming text2speech model | |||
| @@ -1,5 +1,5 @@ | |||
| import concurrent.futures | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| from openai import OpenAI | |||
| @@ -16,7 +16,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel): | |||
| def _invoke( | |||
| self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None | |||
| ) -> any: | |||
| ) -> Any: | |||
| """ | |||
| _invoke text2speech model | |||
| @@ -55,7 +55,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel): | |||
| except Exception as ex: | |||
| raise CredentialsValidateFailedError(str(ex)) | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any: | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any: | |||
| """ | |||
| _tts_invoke_streaming text2speech model | |||
| @@ -192,7 +192,7 @@ class SageMakerText2SpeechModel(TTSModel): | |||
| InvokeBadRequestError: [InvokeBadRequestError, KeyError, ValueError], | |||
| } | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> any: | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> Any: | |||
| return "" | |||
| def _get_model_word_limit(self, model: str, credentials: dict) -> int: | |||
| @@ -225,7 +225,7 @@ class SageMakerText2SpeechModel(TTSModel): | |||
| json_obj = json.loads(json_str) | |||
| return json_obj | |||
| def _tts_invoke_streaming(self, model_type: str, payload: dict, sagemaker_endpoint: str) -> any: | |||
| def _tts_invoke_streaming(self, model_type: str, payload: dict, sagemaker_endpoint: str) -> Any: | |||
| """ | |||
| _tts_invoke_streaming text2speech model | |||
| @@ -1,6 +1,6 @@ | |||
| import threading | |||
| from queue import Queue | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| import dashscope | |||
| from dashscope import SpeechSynthesizer | |||
| @@ -20,7 +20,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel): | |||
| def _invoke( | |||
| self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None | |||
| ) -> any: | |||
| ) -> Any: | |||
| """ | |||
| _invoke text2speech model | |||
| @@ -58,7 +58,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel): | |||
| except Exception as ex: | |||
| raise CredentialsValidateFailedError(str(ex)) | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any: | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any: | |||
| """ | |||
| _tts_invoke_streaming text2speech model | |||
| @@ -1,5 +1,5 @@ | |||
| import concurrent.futures | |||
| from typing import Optional | |||
| from typing import Any, Optional | |||
| from xinference_client.client.restful.restful_client import RESTfulAudioModelHandle | |||
| @@ -166,7 +166,7 @@ class XinferenceText2SpeechModel(TTSModel): | |||
| return self.model_voices["__default"]["all"] | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> any: | |||
| def _get_model_default_voice(self, model: str, credentials: dict) -> Any: | |||
| return "" | |||
| def _get_model_word_limit(self, model: str, credentials: dict) -> int: | |||
| @@ -178,7 +178,7 @@ class XinferenceText2SpeechModel(TTSModel): | |||
| def _get_model_workers_limit(self, model: str, credentials: dict) -> int: | |||
| return 5 | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any: | |||
| def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any: | |||
| """ | |||
| _tts_invoke_streaming text2speech model | |||