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,