From 1a9c09e93de8f8c50ff40fb6c557265885128bcd Mon Sep 17 00:00:00 2001 From: andikarachman Date: Sat, 3 Jan 2026 09:43:56 +0700 Subject: [PATCH] refactor: use TranslationProviderError instead of ValueError for consistency --- cognee/tasks/translation/providers/azure_provider.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cognee/tasks/translation/providers/azure_provider.py b/cognee/tasks/translation/providers/azure_provider.py index 349445ca1..455f57e3a 100644 --- a/cognee/tasks/translation/providers/azure_provider.py +++ b/cognee/tasks/translation/providers/azure_provider.py @@ -6,6 +6,7 @@ from cognee.shared.logging_utils import get_logger from .base import TranslationProvider, TranslationResult from ..config import get_translation_config +from ..exceptions import TranslationProviderError logger = get_logger(__name__) @@ -48,9 +49,9 @@ class AzureTranslationProvider(TranslationProvider): TranslationResult with translated text and metadata """ if not self.is_available(): - raise ValueError( - "Azure Translator API key not configured. " - "Set AZURE_TRANSLATOR_KEY environment variable." + raise TranslationProviderError( + provider=self.provider_name, + message="Azure Translator API key not configured. Set AZURE_TRANSLATOR_KEY environment variable.", ) endpoint = f"{self._config.azure_translator_endpoint}/translate"