Fix persona document sets

danswer-ai · Dec 20, 2023 · fab767d · fab767d · vercel · Dec 20, 2023
1 parent 7dd70ca
commit fab767d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/danswer/search/request_preprocessing.py b/backend/danswer/search/request_preprocessing.py
@@ -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