Unverified Commit 7c7ee633 authored by takatost's avatar takatost Committed by GitHub

fix: spark credentials validate (#1906)

parent 18af84e1
...@@ -19,13 +19,23 @@ class SparkProvider(ModelProvider): ...@@ -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
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment