AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 21:39:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: spark credentials validate (#1906)

Browse Source
This commit is contained in:
takatost 2024-01-04 13:20:45 +08:00 committed by GitHub
parent 18af84e193
commit 7c7ee633c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
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: 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: 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