diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py index c20e3365..ea866819 100644 --- a/libs/aws/langchain_aws/llms/bedrock.py +++ b/libs/aws/langchain_aws/llms/bedrock.py @@ -412,7 +412,7 @@ class BedrockBase(BaseLanguageModel, ABC): "amazon": "stopSequences", "ai21": "stop_sequences", "cohere": "stop_sequences", - "mistral": "stop_sequences", + "mistral": "stop", } provider_stop_reason_key_map: Mapping[str, str] = {