From bbd3eb2ff73606274903611b55e4ba30eacbf5d2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manoel=20Ver=C3=ADssimo=20dos=20Santos=20Neto?= Date: Sun, 21 Jul 2024 11:51:30 -0300 Subject: [PATCH] fix: Adjustments in _prepare_input_and_invoke_stream for meta models - Correct the condition to avoid fetching provider_stop_sequence_key_name_map for meta models. --- libs/aws/langchain_aws/llms/bedrock.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py index 4f7da276..f3250b34 100644 --- a/libs/aws/langchain_aws/llms/bedrock.py +++ b/libs/aws/langchain_aws/llms/bedrock.py @@ -835,11 +835,7 @@ async def _aprepare_input_and_invoke_stream( _model_kwargs = self.model_kwargs or {} provider = self._get_provider() - if stop: - if provider not in self.provider_stop_sequence_key_name_map: - raise ValueError( - f"Stop sequence key name for {provider} is not supported." - ) + if stop and provider in self.provider_stop_sequence_key_name_map: if k := self.provider_stop_sequence_key_name_map.get(provider): _model_kwargs[k] = stop