scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-05 08:06:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

improve the ES max retrieval items limit to 10000

Browse source
This commit is contained in:
xianzhe.xxz 2024-07-16 18:14:13 +08:00
parent 7546ecdf32
commit 241ffaffb9
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
memory_scope/storage/llama_index_sync_elasticsearch.py
View file

@ -532,11 +532,12 @@ class SyncElasticsearchStore(BasePydanticVectorStore):
filter = [_to_elasticsearch_filter(query.filters)]
else:
filter = es_filter or []
num_candidates = query.similarity_top_k * 10 if query.similarity_top_k <= 1000 else query.similarity_top_k
hits = self._store.search(
query=query.query_str,
query_vector=query.query_embedding,
k=query.similarity_top_k,
num_candidates=100, # query.similarity_top_k * 10,
num_candidates=num_candidates, # query.similarity_top_k * 10,
filter=filter,
custom_query=custom_query,
)