refactor: use TranslationProviderError instead of ValueError for consistency
This commit is contained in:
parent
bea7ef5609
commit
1a9c09e93d
1 changed files with 4 additions and 3 deletions
|
|
@ -6,6 +6,7 @@ from cognee.shared.logging_utils import get_logger
|
||||||
|
|
||||||
from .base import TranslationProvider, TranslationResult
|
from .base import TranslationProvider, TranslationResult
|
||||||
from ..config import get_translation_config
|
from ..config import get_translation_config
|
||||||
|
from ..exceptions import TranslationProviderError
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
@ -48,9 +49,9 @@ class AzureTranslationProvider(TranslationProvider):
|
||||||
TranslationResult with translated text and metadata
|
TranslationResult with translated text and metadata
|
||||||
"""
|
"""
|
||||||
if not self.is_available():
|
if not self.is_available():
|
||||||
raise ValueError(
|
raise TranslationProviderError(
|
||||||
"Azure Translator API key not configured. "
|
provider=self.provider_name,
|
||||||
"Set AZURE_TRANSLATOR_KEY environment variable."
|
message="Azure Translator API key not configured. Set AZURE_TRANSLATOR_KEY environment variable.",
|
||||||
)
|
)
|
||||||
|
|
||||||
endpoint = f"{self._config.azure_translator_endpoint}/translate"
|
endpoint = f"{self._config.azure_translator_endpoint}/translate"
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue