klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ensure top_k never more than total entries to search from

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2021-11-17 00:56:31 +05:30
parent 4d37ace3d6
commit 8cf2465e8e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/search_type/asymmetric.py
View File

@@ -161,6 +161,7 @@ def setup(config: TextSearchConfig, regenerate: bool) -> TextSearchModel:
# Extract Entries # Extract Entries
entries = extract_entries(config.compressed_jsonl, config.verbose) entries = extract_entries(config.compressed_jsonl, config.verbose)
top_k = min(len(entries), top_k) # top_k hits can't be more than the total entries in corpus
# Compute or Load Embeddings # Compute or Load Embeddings
corpus_embeddings = compute_embeddings(entries, bi_encoder, config.embeddings_file, regenerate=regenerate, verbose=config.verbose) corpus_embeddings = compute_embeddings(entries, bi_encoder, config.embeddings_file, regenerate=regenerate, verbose=config.verbose)