Corrected type annotation to "Any" from "any" all files in "model_providers" folder (#9135)

This commit is contained in:
ronaksingh27 2024-10-10 08:04:25 +05:30 committed by GitHub
parent 783a6b866b
commit 62051d5171
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 20 additions and 20 deletions

View File

@ -1,7 +1,7 @@
import logging import logging
import re import re
from abc import abstractmethod from abc import abstractmethod
from typing import Optional from typing import Any, Optional
from pydantic import ConfigDict from pydantic import ConfigDict
@ -88,7 +88,7 @@ class TTSModel(AIModel):
else: else:
return [{"name": d["name"], "value": d["mode"]} for d in voices] 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 Get voice for given tts model

View File

@ -1,6 +1,6 @@
import concurrent.futures import concurrent.futures
import copy import copy
from typing import Optional from typing import Any, Optional
from openai import AzureOpenAI from openai import AzureOpenAI
@ -19,7 +19,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel):
def _invoke( def _invoke(
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
) -> any: ) -> Any:
""" """
_invoke text2speech model _invoke text2speech model
@ -56,7 +56,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel):
except Exception as ex: except Exception as ex:
raise CredentialsValidateFailedError(str(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 _tts_invoke_streaming text2speech model
:param model: model name :param model: model name

View File

@ -1,4 +1,4 @@
from typing import Optional from typing import Any, Optional
import httpx import httpx
@ -46,7 +46,7 @@ class FishAudioText2SpeechModel(TTSModel):
content_text: str, content_text: str,
voice: str, voice: str,
user: Optional[str] = None, user: Optional[str] = None,
) -> any: ) -> Any:
""" """
Invoke text2speech model Invoke text2speech model
@ -87,7 +87,7 @@ class FishAudioText2SpeechModel(TTSModel):
except Exception as ex: except Exception as ex:
raise CredentialsValidateFailedError(str(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 Invoke streaming text2speech model
:param model: model name :param model: model name
@ -112,7 +112,7 @@ class FishAudioText2SpeechModel(TTSModel):
except Exception as ex: except Exception as ex:
raise InvokeBadRequestError(str(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 Invoke streaming text2speech model

View File

@ -1,5 +1,5 @@
import concurrent.futures import concurrent.futures
from typing import Optional from typing import Any, Optional
from openai import OpenAI from openai import OpenAI
@ -16,7 +16,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel):
def _invoke( def _invoke(
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
) -> any: ) -> Any:
""" """
_invoke text2speech model _invoke text2speech model
@ -55,7 +55,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel):
except Exception as ex: except Exception as ex:
raise CredentialsValidateFailedError(str(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 _tts_invoke_streaming text2speech model

View File

@ -192,7 +192,7 @@ class SageMakerText2SpeechModel(TTSModel):
InvokeBadRequestError: [InvokeBadRequestError, KeyError, ValueError], 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 "" return ""
def _get_model_word_limit(self, model: str, credentials: dict) -> int: def _get_model_word_limit(self, model: str, credentials: dict) -> int:
@ -225,7 +225,7 @@ class SageMakerText2SpeechModel(TTSModel):
json_obj = json.loads(json_str) json_obj = json.loads(json_str)
return json_obj 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 _tts_invoke_streaming text2speech model

View File

@ -1,6 +1,6 @@
import threading import threading
from queue import Queue from queue import Queue
from typing import Optional from typing import Any, Optional
import dashscope import dashscope
from dashscope import SpeechSynthesizer from dashscope import SpeechSynthesizer
@ -20,7 +20,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel):
def _invoke( def _invoke(
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
) -> any: ) -> Any:
""" """
_invoke text2speech model _invoke text2speech model
@ -58,7 +58,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel):
except Exception as ex: except Exception as ex:
raise CredentialsValidateFailedError(str(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 _tts_invoke_streaming text2speech model

View File

@ -1,5 +1,5 @@
import concurrent.futures import concurrent.futures
from typing import Optional from typing import Any, Optional
from xinference_client.client.restful.restful_client import RESTfulAudioModelHandle from xinference_client.client.restful.restful_client import RESTfulAudioModelHandle
@ -166,7 +166,7 @@ class XinferenceText2SpeechModel(TTSModel):
return self.model_voices["__default"]["all"] 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 "" return ""
def _get_model_word_limit(self, model: str, credentials: dict) -> int: 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: def _get_model_workers_limit(self, model: str, credentials: dict) -> int:
return 5 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 _tts_invoke_streaming text2speech model