From 6ed4928ced6e9f5a041df48eaa42443b09e09ce9 Mon Sep 17 00:00:00 2001 From: ksieuk Date: Fri, 13 Oct 2023 17:04:12 +0300 Subject: [PATCH] fix: [#47] check not is none --- src/assistant/lib/models/tts/voice/base.py | 2 ++ src/assistant/lib/models/tts/voice/yandex.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/assistant/lib/models/tts/voice/base.py b/src/assistant/lib/models/tts/voice/base.py index 45c3960..1682b02 100644 --- a/src/assistant/lib/models/tts/voice/base.py +++ b/src/assistant/lib/models/tts/voice/base.py @@ -20,6 +20,8 @@ class BaseVoiceModel(pydantic.BaseModel): @pydantic.model_validator(mode="before") @classmethod def check_voice_name_exists(cls, data: typing.Any) -> typing.Any: + if not data: + return data voice_id = data.get("voice_id") voice_name = data.get("voice_name") if not voice_name and voice_id: diff --git a/src/assistant/lib/models/tts/voice/yandex.py b/src/assistant/lib/models/tts/voice/yandex.py index 476a1f0..90f8c12 100644 --- a/src/assistant/lib/models/tts/voice/yandex.py +++ b/src/assistant/lib/models/tts/voice/yandex.py @@ -16,6 +16,8 @@ class YandexVoiceModel(models_tts_base.BaseVoiceModel): @pydantic.model_validator(mode="before") @classmethod def check_voice_name_exists(cls, data: typing.Any) -> typing.Any: + if not data: + return data voice_id = data.get("voice_id") voice_name = data.get("voice_name") role = data.get("role")