diff --git a/api/core/llm/provider/azure_provider.py b/api/core/llm/provider/azure_provider.py index 649c64cf7..d87ab761d 100644 --- a/api/core/llm/provider/azure_provider.py +++ b/api/core/llm/provider/azure_provider.py @@ -110,6 +110,8 @@ class AzureProvider(BaseProvider): if missing_model_ids: raise ValidateFailedError("Please add deployments for '{}'.".format(", ".join(missing_model_ids))) + except ValidateFailedError as e: + raise e except AzureAuthenticationError: raise ValidateFailedError('Validation failed, please check your API Key.') except (requests.ConnectionError, requests.RequestException):