From 5d27ece62e2538d70d7ca3f6533216e1ab3cacba Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Tue, 17 Dec 2024 19:07:00 -0800 Subject: [PATCH] Update libs/aws/langchain_aws/chat_models/bedrock.py --- libs/aws/langchain_aws/chat_models/bedrock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/aws/langchain_aws/chat_models/bedrock.py b/libs/aws/langchain_aws/chat_models/bedrock.py index 2c01779d..11cb01fb 100644 --- a/libs/aws/langchain_aws/chat_models/bedrock.py +++ b/libs/aws/langchain_aws/chat_models/bedrock.py @@ -412,7 +412,7 @@ def get_lc_namespace(cls) -> List[str]: def set_beta_use_converse_api(cls, values: Dict) -> Any: model_id = values.get("model_id", values.get("model")) - if model_id is not None and "beta_use_converse_api" not in values: + if model_id and "beta_use_converse_api" not in values: values["beta_use_converse_api"] = "nova" in model_id return values