jsdio/voice_assistant
Template
1
0
Fork 0
mirror of https://github.com/ijaric/voice_assistant.git synced 2025-05-24 14:33:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: [#56] add get_all

Browse Source
This commit is contained in:
ksieuk 2023-10-15 07:15:49 +03:00
parent 2349feb108
commit 71387f5e8c
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/assistant/lib/tts/models/protocols.py
View File

@ -4,6 +4,10 @@ import lib.models as models
class TTSRepositoryProtocol(typing.Protocol):
@property
async def voice_models(self) -> models.LIST_VOICE_MODELS_TYPE:
raise NotImplementedError
async def get_audio_as_bytes(self, request: models.TTSCreateRequestModel) -> models.TTSCreateResponseModel:
...

11
src/assistant/lib/tts/services.py
View File

@ -31,3 +31,14 @@ class TTSService:
if voice_models.models:
response_models.extend(voice_models.models)
return response_models
async def get_all_models(self) -> list[_models.AVAILABLE_MODELS_TYPE]:
response_models: list[_models.AVAILABLE_MODELS_TYPE] = []
for repository in self.repositories.values():
response = await repository.voice_models
for model in response.models:
model.languages = [ # type: ignore
_models.BaseLanguageCodesEnum[language.name] for language in model.languages
]
response_models.append(model)
return response_models