jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: spark credentials validate (#1906)

Browse source
This commit is contained in:
takatost 2024-01-04 13:20:45 +08:00 • committed by GitHub
commit 7c7ee633c1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
api/core/model_runtime/model_providers/spark/spark.py
View file

@ -19,13 +19,23 @@ class SparkProvider(ModelProvider):
try: try:
model_instance = self.get_model_instance(ModelType.LLM) model_instance = self.get_model_instance(ModelType.LLM)
# Use `claude-instant-1` model for validate,
model_instance.validate_credentials( model_instance.validate_credentials(
model='spark-1.5', model='spark-1.5',
credentials=credentials 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: except CredentialsValidateFailedError as ex:
raise ex raise ex
except Exception as ex: except Exception as ex:
logger.exception(f'{self.get_provider_schema().provider} credentials validate failed') logger.exception(f'{self.get_provider_schema().provider} credentials validate failed')
raise ex raise ex
except Exception as ex:
logger.exception(f'{self.get_provider_schema().provider} credentials validate failed')
raise ex