From bdf6c34d26c7fb9d81420e330dff4dcddf2556c6 Mon Sep 17 00:00:00 2001 From: Vladimir Blagojevic Date: Thu, 5 Dec 2024 15:23:32 +0100 Subject: [PATCH] Minot test updates --- .../amazon_bedrock/tests/test_chat_generator.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) 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 """