diff --git a/unit_tests/destinations/vector_db_based/writer_test.py b/unit_tests/destinations/vector_db_based/writer_test.py index 8a2afd42..b02cf969 100644 --- a/unit_tests/destinations/vector_db_based/writer_test.py +++ b/unit_tests/destinations/vector_db_based/writer_test.py @@ -91,7 +91,13 @@ def test_write(omit_raw_text: bool): mock_indexer.post_sync.return_value = [post_sync_log_message] # Create the DestinationLangchain instance - writer = Writer(config_model, mock_indexer, mock_embedder, BATCH_SIZE, omit_raw_text) + writer = Writer( + config_model, + mock_indexer, + mock_embedder, + BATCH_SIZE, + omit_raw_text=omit_raw_text, + ) output_messages = writer.write(configured_catalog, input_messages) output_message = next(output_messages)