pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fitin for infinity. (#4722)

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

### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu
2025-02-05 16:47:05 +08:00
committed by GitHub
parent 307717b045
commit 283d036cba
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rag/utils/infinity_conn.py
View File

@ -311,6 +311,7 @@ class InfinityConnection(DocStoreConnection):
if matchExprs:
selectFields.append(score_func)
selectFields.append(PAGERANK_FLD)
selectFields = [f for f in selectFields if f != "_score"]
# Prepare expressions common to all tables
filter_cond = None
@ -405,7 +406,7 @@ class InfinityConnection(DocStoreConnection):
if matchExprs:
res = res.sort(pl.col(score_column) + pl.col(PAGERANK_FLD), descending=True, maintain_order=True)
if score_column and score_column != "SCORE":
res = res.rename({score_column: "SCORE"})
res = res.rename({score_column: "_score"})
res = res.limit(limit)
logger.debug(f"INFINITY search final result: {str(res)}")
return res, total_hits_count