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: [#47] methods to async

Browse Source
This commit is contained in:
ksieuk 2023-10-13 14:56:36 +03:00
parent 5b50d78b13
commit 8eef8b46a7
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/assistant/lib/tts/repositories/base.py
View File

@ -11,24 +11,25 @@ class TTSBaseRepository(abc.ABC):
@property @property
@abc.abstractmethod @abc.abstractmethod
def voice_models(self) -> models.LIST_VOICE_MODELS_TYPE: async def voice_models(self) -> models.LIST_VOICE_MODELS_TYPE:
raise NotImplementedError raise NotImplementedError
@abc.abstractmethod @abc.abstractmethod
def get_audio_as_bytes(self, request: models.TTSCreateRequestModel) -> models.TTSCreateResponseModel: async def get_audio_as_bytes(self, request: models.TTSCreateRequestModel) -> models.TTSCreateResponseModel:
raise NotImplementedError raise NotImplementedError
def get_voice_model_by_name(self, voice_model_name: str) -> models.BaseVoiceModel | None: async def get_voice_model_by_name(self, voice_model_name: str) -> models.BaseVoiceModel | None:
""" """
Search voice model by name Search voice model by name
:param voice_model_name: String name :param voice_model_name: String name
:return: Voice model that match the name :return: Voice model that match the name
""" """
for voice_model in self.voice_models.models: voice_models = await self.voice_models
for voice_model in voice_models.models:
if voice_model.voice_name == voice_model_name: if voice_model.voice_name == voice_model_name:
return voice_model return voice_model
def get_list_voice_models_by_fields( async def get_list_voice_models_by_fields(
self, fields: models.TTSSearchVoiceRequestModel self, fields: models.TTSSearchVoiceRequestModel
) -> list[models.AVAILABLE_MODELS_TYPE]: ) -> list[models.AVAILABLE_MODELS_TYPE]:
""" """
@ -38,7 +39,8 @@ class TTSBaseRepository(abc.ABC):
""" """
fields_dump = fields.model_dump(exclude_none=True) fields_dump = fields.model_dump(exclude_none=True)
voice_models_response = [] voice_models_response = []
for voice_model in self.voice_models.models: voice_models = await self.voice_models
for voice_model in voice_models.models:
for field, field_value in fields_dump.items(): for field, field_value in fields_dump.items():
if field == "languages": # language is a list if field == "languages": # language is a list
language_names: set[str] = {item.name for item in field_value} language_names: set[str] = {item.name for item in field_value}