highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-28 05:43:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fixed llm_indices from document search api (#1853)

Browse Source
This commit is contained in:
hagen-danswer
2024-07-17 14:52:49 -07:00
committed by GitHub
parent cdf6089b3e
commit 1c77395503
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
backend/danswer/search/postprocessing/postprocessing.py
View File

@@ -278,17 +278,17 @@ def search_postprocessing(
_log_top_section_links(search_query.search_type.value, reranked_sections)
yield reranked_sections
llm_section_selection = cast(
list[str] | None,
post_processing_results.get(str(llm_filter_task_id))
if llm_filter_task_id
else None,
)
if llm_section_selection is not None:
yield [
index
for index, section in enumerate(reranked_sections or retrieved_sections)
if section.center_chunk.unique_id in llm_section_selection
llm_selected_section_ids = (
[
section.center_chunk.unique_id
for section in post_processing_results.get(str(llm_filter_task_id), [])
]
else:
yield cast(list[int], [])
if llm_filter_task_id
else []
)
yield [
index
for index, section in enumerate(reranked_sections or retrieved_sections)
if section.center_chunk.unique_id in llm_selected_section_ids
]