From 4b281170556c6fe82278fd5e2e995b4235b7d491 Mon Sep 17 00:00:00 2001 From: ksieuk Date: Fri, 13 Oct 2023 14:56:58 +0300 Subject: [PATCH] fix: [#47] rewrite from_api --- src/assistant/lib/models/tts/voice/eleven_labs.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/assistant/lib/models/tts/voice/eleven_labs.py b/src/assistant/lib/models/tts/voice/eleven_labs.py index 80804b4..50b6fbb 100644 --- a/src/assistant/lib/models/tts/voice/eleven_labs.py +++ b/src/assistant/lib/models/tts/voice/eleven_labs.py @@ -71,5 +71,13 @@ class ElevenLabsListVoiceModelsModel(pydantic.BaseModel): @classmethod def from_api(cls, voice_models_from_api: list[dict[str, typing.Any]]) -> typing.Self: - voice_models = [ElevenLabsVoiceModel.model_validate(voice_model) for voice_model in voice_models_from_api] + voice_models = [] + for voice_model in voice_models_from_api: + voice_model["voice_id"] = voice_model.pop("model_id") + voice_model["voice_name"] = voice_model.pop("name") + voice_model["languages"] = [ + models_tts_languages.ElevenLabsLanguageCodesEnum(item.get("language_id")) + for item in voice_model.pop("languages") + ] + voice_models.append(ElevenLabsVoiceModel.model_validate(voice_model)) return ElevenLabsListVoiceModelsModel(models=voice_models)