|
|
|
|
|
|
|
|
""" |
|
|
""" |
|
|
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: |