diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py index 6913ddc9a..ed2b80250 100644 --- 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 """