| @@ -17,10 +17,9 @@ class BedrockProvider(ModelProvider): | |||
| """ | |||
| try: | |||
| 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='amazon.titan-text-lite-v1', | |||
| model=bedrock_validate_model_name, | |||
| credentials=credentials | |||
| ) | |||
| except CredentialsValidateFailedError as ex: | |||
| @@ -69,3 +69,12 @@ provider_credential_schema: | |||
| label: | |||
| en_US: AWS GovCloud (US-West) | |||
| zh_Hans: AWS GovCloud (US-West) | |||
| - variable: model_for_validation | |||
| required: false | |||
| label: | |||
| en_US: Available Model Name | |||
| zh_Hans: 可用模型名称 | |||
| type: text-input | |||
| placeholder: | |||
| en_US: A model you have access to (e.g. amazon.titan-text-lite-v1) for validation. | |||
| zh_Hans: 为了进行验证,请输入一个您可用的模型名称 (例如:amazon.titan-text-lite-v1) | |||