From 19f684a3503686ed6f64cfcc6ec818c410b83c31 Mon Sep 17 00:00:00 2001
From: tstadel <thomas.stadelmann@deepset.ai>
Date: Wed, 16 Oct 2024 08:57:16 +0200
Subject: [PATCH] add from_dict test

---
 integrations/amazon_bedrock/tests/test_generator.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/integrations/amazon_bedrock/tests/test_generator.py b/integrations/amazon_bedrock/tests/test_generator.py
index 421c28175..2ccd5a3fa 100644
--- a/integrations/amazon_bedrock/tests/test_generator.py
+++ b/integrations/amazon_bedrock/tests/test_generator.py
@@ -58,12 +58,16 @@ def test_from_dict(mock_boto3_session):
                 "aws_profile_name": {"type": "env_var", "env_vars": ["AWS_PROFILE"], "strict": False},
                 "model": "anthropic.claude-v2",
                 "max_length": 99,
+                "boto3_config": {
+                    "read_timeout": 1000,
+                },
             },
         }
     )
 
     assert generator.max_length == 99
     assert generator.model == "anthropic.claude-v2"
+    assert generator.boto3_config == {"read_timeout": 1000}
 
 
 def test_default_constructor(mock_boto3_session, set_env_variables):