From 4ebae83179ca68a78076a4ccb1c2e3c6b91e39ef Mon Sep 17 00:00:00 2001 From: anakin87 Date: Thu, 12 Dec 2024 18:44:19 +0100 Subject: [PATCH] fix bedrock test --- test/components/extractors/test_llm_metadata_extractor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/components/extractors/test_llm_metadata_extractor.py b/test/components/extractors/test_llm_metadata_extractor.py index 92d7deeb..ad8238b8 100644 --- a/test/components/extractors/test_llm_metadata_extractor.py +++ b/test/components/extractors/test_llm_metadata_extractor.py @@ -146,6 +146,7 @@ def test_to_dict_aws_bedrock(self, boto3_session_mock): "strict": False, }, "model": "meta.llama.test", + "model_family": None, "max_length": 100, "truncate": False, "streaming_callback": None,