diff --git a/src/searcher.py b/src/searcher.py index 6509dcbfa2bad9f29d6683400bf734b95346fe6b..0d3567935dcbc65f0d79b6bd12e5addad468f882 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"] @@ -78,7 +78,10 @@ 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'] + if 'highlight' in h: + result['description'] = ' '.join(h['highlight']['description']) + results.append(result) return results def autocomplete(es: Elasticsearch, language: str) -> List[str]: