add from_dict test

deepset-ai · Oct 16, 2024 · 19f684a · 19f684a
1 parent ca46113
commit 19f684a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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):