Minot test updates

deepset-ai · Dec 5, 2024 · bdf6c34 · bdf6c34
1 parent 88bf808
commit bdf6c34
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py
@@ -1,4 +1,5 @@
 import json
+from typing import Any, Dict, Optional
 
 import pytest
 from haystack.components.generators.utils import print_streaming_chunk
@@ -30,8 +31,16 @@ def chat_messages():
     ]
     return messages
 
-
-def test_to_dict(mock_boto3_session):
+@pytest.mark.parametrize(
+    "boto3_config",
+    [
+        None,
+        {
+            "read_timeout": 1000,
+        },
+    ],
+)
+def test_to_dict(mock_boto3_session, boto3_config):
     """
     Test that the to_dict method returns the correct dictionary without aws credentials
     """