Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: scope is not passd to vector store query (#747)

Browse Source
This commit is contained in:
Pang Chun Lam
2025-06-05 17:08:40 +08:00
committed by GitHub
parent 5132288386
commit ddb5187293
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/kotaemon/kotaemon/indices/vectorindex.py
View File

@@ -168,7 +168,7 @@ class VectorRetrieval(BaseRetrieval):
if self.retrieval_mode == "vector":
emb = self.embedding(text)[0].embedding
_, scores, ids = self.vector_store.query(
embedding=emb, top_k=top_k_first_round, **kwargs
embedding=emb, top_k=top_k_first_round, doc_ids=scope, **kwargs
)
docs = self.doc_store.get(ids)
result = [
@@ -197,7 +197,7 @@ class VectorRetrieval(BaseRetrieval):
assert self.doc_store is not None
_, vs_scores, vs_ids = self.vector_store.query(
embedding=emb, top_k=top_k_first_round, **kwargs
embedding=emb, top_k=top_k_first_round, doc_ids=scope, **kwargs
)
if vs_ids:
vs_docs = self.doc_store.get(vs_ids)