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