From 16f08b657f05a27a180d496d4c6c1beb1f30309f Mon Sep 17 00:00:00 2001 From: Vladimir Blagojevic Date: Fri, 8 Mar 2024 17:22:28 +0100 Subject: [PATCH] Assert non-empty meta --- integrations/amazon_bedrock/tests/test_chat_generator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py index b6b358a7e..8eda7da39 100644 --- a/integrations/amazon_bedrock/tests/test_chat_generator.py +++ b/integrations/amazon_bedrock/tests/test_chat_generator.py @@ -248,6 +248,7 @@ def test_default_inference_params(self, model_name): assert response["replies"][0].content assert ChatMessage.is_from(response["replies"][0], ChatRole.ASSISTANT) assert "paris" in response["replies"][0].content.lower() + assert len(response["replies"][0].meta) > 0 @pytest.mark.parametrize( "model_name", ["anthropic.claude-3-sonnet-20240229-v1:0", "anthropic.claude-v2:1", "meta.llama2-13b-chat-v1"]