Compare commits

...

3 Commits

Author SHA1 Message Date
chenhe
9b7d839233 remove launch.json 2024-03-16 07:51:20 -07:00
chenhe
370b25f3f8 optionally specify available bedrock model used in validation 2024-03-16 07:44:49 -07:00
chenhe
8e47a37646 create launch.json config 2024-03-16 07:13:43 -07:00
2 changed files with 13 additions and 2 deletions

View File

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

View File

@ -69,3 +69,13 @@ 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: secret-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)