sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-20 06:55:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

Ensure top_k never more than total entries to run symmetric search on

Browse source
This commit is contained in:
debanjum 2021-11-16 11:32:21 -08:00
parent 8c858d1a94
commit 46661b3057
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/search_type/symmetric_ledger.py
View file

@ -153,6 +153,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)
# 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)