From a04d4bf08ffaee40fd4b6bf88e257180f22ce629 Mon Sep 17 00:00:00 2001 From: Israel Yago Pereira Date: Mon, 29 Nov 2021 10:52:13 -0300 Subject: [PATCH 1/2] Use highlighted body for search results --- src/searcher.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/searcher.py b/src/searcher.py index 6509dcb..aab8fd0 100644 --- a/src/searcher.py +++ b/src/searcher.py @@ -78,7 +78,9 @@ def search_query(es: Elasticsearch, user_query: str, language: str): res = es.search(index=index_name, body=query) results = [] for h in res['hits']['hits']: - results.append(h['_source']) + result = h['_source'] + result['body']= h['highlight']['body'] + results.append(result) return results def autocomplete(es: Elasticsearch, language: str) -> List[str]: -- GitLab From 2e144caf7357f47f354fc70360922bd96d0d366b Mon Sep 17 00:00:00 2001 From: Israel Yago Pereira Date: Mon, 29 Nov 2021 12:01:43 -0300 Subject: [PATCH 2/2] Using description instead of body --- src/searcher.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/searcher.py b/src/searcher.py index aab8fd0..0d35679 100644 --- a/src/searcher.py +++ b/src/searcher.py @@ -69,7 +69,7 @@ def search_query(es: Elasticsearch, user_query: str, language: str): }, "highlight": { "fields" : { - "body" : {} + "description" : {} } }, "_source": ["title", "subtitles", "url", "description", "lang", "body"] @@ -79,7 +79,8 @@ def search_query(es: Elasticsearch, user_query: str, language: str): results = [] for h in res['hits']['hits']: result = h['_source'] - result['body']= h['highlight']['body'] + if 'highlight' in h: + result['description'] = ' '.join(h['highlight']['description']) results.append(result) return results -- GitLab