File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
libs/aws/langchain_aws/chat_models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -598,10 +598,10 @@ def validate_environment(self) -> Self:
598
598
"""Validate that AWS credentials to and python package exists in environment."""
599
599
600
600
# As of 12/03/24:
601
- # only claude-3, mistral-large, and nova models support tool choice:
601
+ # only claude-3/4 , mistral-large, and nova models support tool choice:
602
602
# https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_ToolChoice.html
603
603
if self .supports_tool_choice_values is None :
604
- if "claude-3 " in self ._get_base_model ():
604
+ if "claude" in self ._get_base_model ():
605
605
# Tool choice not supported when thinking is enabled
606
606
thinking_params = (self .additional_model_request_fields or {}).get (
607
607
"thinking" , {}
You can’t perform that action at this time.
0 commit comments