AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-13 14:09:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix disabled doc is still retreivalable (#695)

Browse Source 
### What problem does this PR solve?

Fix that disabled doc is still retreivalable

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
KevinHuSh 2024-05-09 15:32:24 +08:00 committed by GitHub
parent 9392b8bc8f
commit 648a2baaa9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rag/nlp/search.py
View File

@ -52,6 +52,8 @@ class Dealer:
def search(self, req, idxnm, emb_mdl=None): def search(self, req, idxnm, emb_mdl=None):
qst = req.get("question", "") qst = req.get("question", "")
bqry, keywords = self.qryr.question(qst) bqry, keywords = self.qryr.question(qst)
def add_filters(bqry):
nonlocal req
if req.get("kb_ids"): if req.get("kb_ids"):
bqry.filter.append(Q("terms", kb_id=req["kb_ids"])) bqry.filter.append(Q("terms", kb_id=req["kb_ids"]))
if req.get("doc_ids"): if req.get("doc_ids"):
@ -62,6 +64,9 @@ class Dealer:
else: else:
bqry.filter.append( bqry.filter.append(
Q("bool", must_not=Q("range", available_int={"lt": 1}))) Q("bool", must_not=Q("range", available_int={"lt": 1})))
return bqry
bqry = add_filters(bqry)
bqry.boost = 0.05 bqry.boost = 0.05
s = Search() s = Search()
@ -117,8 +122,7 @@ class Dealer:
es_logger.info("TOTAL: {}".format(self.es.getTotal(res))) es_logger.info("TOTAL: {}".format(self.es.getTotal(res)))
if self.es.getTotal(res) == 0 and "knn" in s: if self.es.getTotal(res) == 0 and "knn" in s:
bqry, _ = self.qryr.question(qst, min_match="10%") bqry, _ = self.qryr.question(qst, min_match="10%")
if req.get("kb_ids"): bqry = add_filters(bqry)
bqry.filter.append(Q("terms", kb_id=req["kb_ids"]))
s["query"] = bqry.to_dict() s["query"] = bqry.to_dict()
s["knn"]["filter"] = bqry.to_dict() s["knn"]["filter"] = bqry.to_dict()
s["knn"]["similarity"] = 0.17 s["knn"]["similarity"] = 0.17