Fix persona document sets

This commit is contained in:
Weves 2023-12-20 15:22:47 -08:00 committed by Chris Weaver
parent 7dd70ca4c0
commit fab767d794

View File

@ -43,6 +43,10 @@ def retrieval_preprocessing(
"""
preset_filters = retrieval_details.filters or BaseFilters()
if persona and persona.document_sets and preset_filters.document_set is None:
preset_filters.document_set = [
document_set.name for document_set in persona.document_sets
]
time_filter = preset_filters.time_cutoff
source_filter = preset_filters.source_type