yujunjun/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2025-06-26 23:19:56 +00:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats 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)