From 7c7ee633c14237428b613ed9bbbc94084f14017d Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 4 Jan 2024 13:20:45 +0800 Subject: [PATCH] fix: spark credentials validate (#1906) --- .../model_runtime/model_providers/spark/spark.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/api/core/model_runtime/model_providers/spark/spark.py b/api/core/model_runtime/model_providers/spark/spark.py index 51b066ad3e..0b606c4e72 100644 --- a/api/core/model_runtime/model_providers/spark/spark.py +++ b/api/core/model_runtime/model_providers/spark/spark.py @@ -19,13 +19,23 @@ class SparkProvider(ModelProvider): try: model_instance = self.get_model_instance(ModelType.LLM) - # Use `claude-instant-1` model for validate, model_instance.validate_credentials( model='spark-1.5', credentials=credentials ) except CredentialsValidateFailedError as ex: - raise 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