From 8cf174ac8c10ae15ba56af2102e677392d617607 Mon Sep 17 00:00:00 2001 From: ZanSara Date: Tue, 20 Feb 2024 17:12:17 +0100 Subject: [PATCH] fix test --- integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py b/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py index 574aab5cc..b65cef4ed 100644 --- a/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py +++ b/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py @@ -58,7 +58,7 @@ def test_to_dict(mock_auto_tokenizer, mock_boto3_session, set_env_variables): "model": "anthropic.claude-v2", "generation_kwargs": {"temperature": 0.7}, "stop_words": [], - "streaming_callback": print_streaming_chunk, + "streaming_callback": 'haystack.components.generators.utils.print_streaming_chunk', }, }