Просмотр исходного кода

fix: ignore spark provider credential validate (#2344)

tags/0.5.3
takatost 1 год назад
Родитель
Сommit
a84e15b8cc
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 2 добавлений и 23 удалений
  1. 2
    23
      api/core/model_runtime/model_providers/spark/spark.py

+ 2
- 23
api/core/model_runtime/model_providers/spark/spark.py Просмотреть файл

@@ -16,26 +16,5 @@ class SparkProvider(ModelProvider):

:param credentials: provider credentials, credentials form defined in `provider_credential_schema`.
"""
try:
model_instance = self.get_model_instance(ModelType.LLM)

model_instance.validate_credentials(
model='spark-1.5',
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:
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
# ignore credentials validation because every model has their own spark quota pool
pass

Загрузка…
Отмена
Сохранить