beenull/mwmbl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix bug in completions with duplicated terms

Browse source
This commit is contained in:
Daoud Clarke 2022-08-14 22:03:50 +01:00
parent f7660bcd27
commit 480be85cfd
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
mwmbl/tinysearchengine/rank.py
View file

@ -158,8 +158,9 @@ class Ranker:
ordered_results, terms, completions = self.get_results(q)
if len(ordered_results) == 0:
# There are no results so suggest Google searches instead
adjusted_completions = set(completions + [q])
completed = ["search: google.com " + ' '.join(terms[:-1] + [t]) for t in adjusted_completions]
completion_queries = [' '.join(terms[:-1] + [t]) for t in completions]
adjusted_completions = completion_queries if q in completion_queries else [q] + completion_queries
completed = ["search: google.com " + t for t in adjusted_completions]
return [q, completed]
else:
adjusted_completions = [c for c in completions if c != terms[-1]]