diff --git a/tests/test_pgvector_filters.py b/tests/test_pgvector_filters.py index e670daf..e63b6a1 100644 --- a/tests/test_pgvector_filters.py +++ b/tests/test_pgvector_filters.py @@ -6,7 +6,14 @@ def test_whitelist_includes_expected_columns() -> None: - assert {"project", "file_id", "entity_id", "parent_chunk", "domain"} == pvs._ALLOWED_FILTER_COLS + assert { + "project", + "file_id", + "entity_id", + "parent_id", + "parent_chunk", + "domain", + } == pvs._ALLOWED_FILTER_COLS def test_injection_shape_dropped() -> None: