|
|
|
@ -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)
|
|
|
|
|
|
|
|
bedrock_validate_model_name = credentials.get('model_for_validation', 'amazon.titan-text-lite-v1')
|
|
|
|
# Use `gemini-pro` model for validate,
|
|
|
|
|
|
|
|
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:
|
|
|
|
|