Compare commits
3 Commits
main
...
feat/speci
Author | SHA1 | Date | |
---|---|---|---|
![]() |
9b7d839233 | ||
![]() |
370b25f3f8 | ||
![]() |
8e47a37646 |
@ -18,9 +18,10 @@ 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,
|
# 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_instance.validate_credentials(
|
||||||
model='amazon.titan-text-lite-v1',
|
model=model_for_validation,
|
||||||
credentials=credentials
|
credentials=credentials
|
||||||
)
|
)
|
||||||
except CredentialsValidateFailedError as ex:
|
except CredentialsValidateFailedError as ex:
|
||||||
|
@ -69,3 +69,13 @@ provider_credential_schema:
|
|||||||
label:
|
label:
|
||||||
en_US: AWS GovCloud (US-West)
|
en_US: AWS GovCloud (US-West)
|
||||||
zh_Hans: 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)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user