From 152e5f16619f4f603c33f0ce1f2298db765f5036 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Tue, 3 Jan 2023 01:25:34 -0300 Subject: [PATCH] Return the file of each search result in response - Useful for enabling jump to note functionality in interfaces - It will be used in the Khoj plugin for Obsidian --- src/search_type/text_search.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/search_type/text_search.py b/src/search_type/text_search.py index 5bbbdd64..65b234a2 100644 --- a/src/search_type/text_search.py +++ b/src/search_type/text_search.py @@ -186,7 +186,10 @@ def collate_results(hits, entries: list[Entry], count=5) -> list[SearchResponse] return [SearchResponse.parse_obj( { "entry": entries[hit['corpus_id']].raw, - "score": f"{hit['cross-score'] if 'cross-score' in hit else hit['score']:.3f}" + "score": f"{hit['cross-score'] if 'cross-score' in hit else hit['score']:.3f}", + "additional": { + "file": entries[hit['corpus_id']].file, + } }) for hit in hits[0:count]]