root/ailine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化: 重排前增加向量初筛，只让前20个进入重排
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 7m32s
Details

Browse Source
This commit is contained in:
root
2026-05-06 17:08:47 +08:00
parent d09b0d16ce
commit a6813a5ab0
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/app/rag/pipeline.py
View File

@@ -70,6 +70,13 @@ class RAGPipeline:
content_len = len(doc.page_content) content_len = len(doc.page_content)
info(f"[Pipeline] 子文档[{i}] 长度={content_len}字符") info(f"[Pipeline] 子文档[{i}] 长度={content_len}字符")
# Step 1.5: 向量初筛（进入重排前先过滤）
vector_top_n = 20
info(f"[Pipeline] Step 1.5: 向量初筛: 取前 {vector_top_n} 个（当前 {len(child_docs)} 个）")
if len(child_docs) > vector_top_n:
child_docs = child_docs[:vector_top_n]
info(f"[Pipeline] Step 1.5 完成: 向量初筛后 {len(child_docs)} 个")
# Step 2: 重排 # Step 2: 重排
info(f"[Pipeline] Step 2: 开始重排") info(f"[Pipeline] Step 2: 开始重排")
if self.reranker: if self.reranker: