Testys commited on
Commit
d7bc2ed
·
1 Parent(s): 2dec497

Update search_utils.py

Browse files
Files changed (1) hide show
  1. search_utils.py +1 -3
search_utils.py CHANGED
@@ -168,9 +168,7 @@ class SemanticSearch:
168
  results['similarity'] = 1 - (distances / 2)
169
 
170
  # Deduplicate and sort results
171
- results = results.drop_duplicates(subset=["title", "source"])
172
- .sort_values("similarity", ascending=False)
173
- .head(top_k)
174
 
175
  return results.reset_index(drop=True)
176
 
 
168
  results['similarity'] = 1 - (distances / 2)
169
 
170
  # Deduplicate and sort results
171
+ results = results.drop_duplicates(subset=["title", "source"]).sort_values("similarity", ascending=False).head(top_k)
 
 
172
 
173
  return results.reset_index(drop=True)
174