Browse Source

fix: spark credentials validate (#1906)

takatost 1 year ago
parent
commit
7c7ee633c1
1 changed files with 12 additions and 2 deletions
  1. 12 2
      api/core/model_runtime/model_providers/spark/spark.py

+ 12 - 2
api/core/model_runtime/model_providers/spark/spark.py

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