From ddb5187293ff540b65cd288ea0cb0c59cd9dde6e Mon Sep 17 00:00:00 2001 From: Pang Chun Lam Date: Thu, 5 Jun 2025 17:08:40 +0800 Subject: [PATCH] fix: scope is not passd to vector store query (#747) --- libs/kotaemon/kotaemon/indices/vectorindex.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/kotaemon/kotaemon/indices/vectorindex.py b/libs/kotaemon/kotaemon/indices/vectorindex.py index bc28215..d882d6b 100644 --- a/libs/kotaemon/kotaemon/indices/vectorindex.py +++ b/libs/kotaemon/kotaemon/indices/vectorindex.py @@ -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)