diff --git a/libs/aws/langchain_aws/chat_models/bedrock.py b/libs/aws/langchain_aws/chat_models/bedrock.py index 52153e39..26259caa 100644 --- a/libs/aws/langchain_aws/chat_models/bedrock.py +++ b/libs/aws/langchain_aws/chat_models/bedrock.py @@ -434,7 +434,9 @@ def _generate( ) if stop: - params["stop_sequences"] = stop + provider = self._get_provider() + if k := self.provider_stop_sequence_key_name_map.get(provider): + params[k] = stop completion, llm_output = self._prepare_input_and_invoke( prompt=prompt,