fix bedrock test (#156)

deepset-ai · Dec 13, 2024 · 46d3795 · 46d3795
1 parent 7ade6a2
commit 46d3795
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,