From a84e15b8cceb27fb6c4df64cfb359436d499b7a5 Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 1 Feb 2024 18:04:05 +0800 Subject: [PATCH] fix: ignore spark provider credential validate (#2344) --- .../model_providers/spark/spark.py | 25 ++----------------- 1 file changed, 2 insertions(+), 23 deletions(-) diff --git a/api/core/model_runtime/model_providers/spark/spark.py b/api/core/model_runtime/model_providers/spark/spark.py index 0b606c4e72..c8bea10390 100644 --- a/api/core/model_runtime/model_providers/spark/spark.py +++ b/api/core/model_runtime/model_providers/spark/spark.py @@ -16,26 +16,5 @@ class SparkProvider(ModelProvider): :param credentials: provider credentials, credentials form defined in `provider_credential_schema`. """ - try: - model_instance = self.get_model_instance(ModelType.LLM) - - model_instance.validate_credentials( - model='spark-1.5', - credentials=credentials - ) - except CredentialsValidateFailedError as ex: - try: - model_instance = self.get_model_instance(ModelType.LLM) - - model_instance.validate_credentials( - model='spark-3', - credentials=credentials - ) - except CredentialsValidateFailedError as ex: - raise ex - except Exception as ex: - logger.exception(f'{self.get_provider_schema().provider} credentials validate failed') - raise ex - except Exception as ex: - logger.exception(f'{self.get_provider_schema().provider} credentials validate failed') - raise ex + # ignore credentials validation because every model has their own spark quota pool + pass