Fix ChatBedrock integration with Amazon Bedrock model

libs/aws/langchain_aws/chat_models/bedrock.py

@@ -852,7 +852,7 @@ def _as_converse(self) -> ChatBedrockConverse:
             kwargs["temperature"] = self.temperature
         return ChatBedrockConverse(
             model=self.model_id,
-            region_name=self.region_name,
+            region_name=self.region_name or os.getenv("AWS_REGION"),
             credentials_profile_name=self.credentials_profile_name,
             aws_access_key_id=self.aws_access_key_id,
             aws_secret_access_key=self.aws_secret_access_key,

libs/aws/tests/integration_tests/chat_models/test_bedrock_converse.py

@@ -55,7 +55,7 @@ def chat_model_class(self) -> Type[BaseChatModel]:
 
     @property
     def chat_model_params(self) -> dict:
-        return {"model": "us.amazon.nova-pro-v1:0"}
+        return {"model": "us.amazon.nova-pro-v1:0", "region_name": "us-east-1"}
 
     @property
     def standard_chat_model_params(self) -> dict:
@@ -242,3 +242,22 @@ def test_guardrails() -> None:
     )
     assert response.response_metadata["stopReason"] == "guardrail_intervened"
     assert response.response_metadata["trace"] is not None
+
+
+def test_chat_bedrock_converse_with_region_name() -> None:
+    model = ChatBedrockConverse(
+        model="us.amazon.nova-pro-v1:0", region_name="us-east-1", temperature=0
+    )
+    response = model.invoke("Create a list of 3 pop songs")
+    assert isinstance(response, BaseModel)
+    assert response.content is not None
+
+
+def test_chat_bedrock_converse_with_aws_region_env() -> None:
+    import os
+
+    os.environ["AWS_REGION"] = "us-east-1"
+    model = ChatBedrockConverse(model="us.amazon.nova-pro-v1:0", temperature=0)
+    response = model.invoke("Create a list of 3 pop songs")
+    assert isinstance(response, BaseModel)
+    assert response.content is not None