mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-16 08:15:55 +08:00
make sure validation flow works for all model providers in bedrock (#3250)
This commit is contained in:
parent
e635f3dc1d
commit
eb76d7a226
@ -74,7 +74,7 @@ provider_credential_schema:
|
|||||||
label:
|
label:
|
||||||
en_US: Available Model Name
|
en_US: Available Model Name
|
||||||
zh_Hans: 可用模型名称
|
zh_Hans: 可用模型名称
|
||||||
type: secret-input
|
type: text-input
|
||||||
placeholder:
|
placeholder:
|
||||||
en_US: A model you have access to (e.g. amazon.titan-text-lite-v1) for validation.
|
en_US: A model you have access to (e.g. amazon.titan-text-lite-v1) for validation.
|
||||||
zh_Hans: 为了进行验证,请输入一个您可用的模型名称 (例如:amazon.titan-text-lite-v1)
|
zh_Hans: 为了进行验证,请输入一个您可用的模型名称 (例如:amazon.titan-text-lite-v1)
|
||||||
|
@ -402,25 +402,25 @@ class BedrockLargeLanguageModel(LargeLanguageModel):
|
|||||||
:param credentials: model credentials
|
:param credentials: model credentials
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
|
required_params = {}
|
||||||
if "anthropic.claude-3" in model:
|
if "anthropic" in model:
|
||||||
try:
|
required_params = {
|
||||||
self._invoke_claude(model=model,
|
"max_tokens": 32,
|
||||||
credentials=credentials,
|
}
|
||||||
prompt_messages=[{"role": "user", "content": "ping"}],
|
elif "ai21" in model:
|
||||||
model_parameters={},
|
# ValidationException: Malformed input request: #/temperature: expected type: Number, found: Null#/maxTokens: expected type: Integer, found: Null#/topP: expected type: Number, found: Null, please reformat your input and try again.
|
||||||
stop=None,
|
required_params = {
|
||||||
stream=False)
|
"temperature": 0.7,
|
||||||
|
"topP": 0.9,
|
||||||
except Exception as ex:
|
"maxTokens": 32,
|
||||||
raise CredentialsValidateFailedError(str(ex))
|
}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
ping_message = UserPromptMessage(content="ping")
|
ping_message = UserPromptMessage(content="ping")
|
||||||
self._generate(model=model,
|
self._invoke(model=model,
|
||||||
credentials=credentials,
|
credentials=credentials,
|
||||||
prompt_messages=[ping_message],
|
prompt_messages=[ping_message],
|
||||||
model_parameters={},
|
model_parameters=required_params,
|
||||||
stream=False)
|
stream=False)
|
||||||
|
|
||||||
except ClientError as ex:
|
except ClientError as ex:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user