|
@@ -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
|