|
@@ -17,10 +17,9 @@ class BedrockProvider(ModelProvider):
|
|
"""
|
|
"""
|
|
try:
|
|
try:
|
|
model_instance = self.get_model_instance(ModelType.LLM)
|
|
model_instance = self.get_model_instance(ModelType.LLM)
|
|
-
|
|
|
|
- # Use `gemini-pro` model for validate,
|
|
|
|
|
|
+ bedrock_validate_model_name = credentials.get('model_for_validation', 'amazon.titan-text-lite-v1')
|
|
model_instance.validate_credentials(
|
|
model_instance.validate_credentials(
|
|
- model='amazon.titan-text-lite-v1',
|
|
|
|
|
|
+ model=bedrock_validate_model_name,
|
|
credentials=credentials
|
|
credentials=credentials
|
|
)
|
|
)
|
|
except CredentialsValidateFailedError as ex:
|
|
except CredentialsValidateFailedError as ex:
|