|
|
@@ -221,7 +221,7 @@ class RetrievalManager:
|
|
|
self.logger.info(f"[async_bfp_recall] 开始召回, top_k={top_k}, 实体数量={len(entity_list)}, 背景='{background[:50]}...'")
|
|
|
|
|
|
# 异步并发召回编制依据
|
|
|
- collection_name = "first_bfp_collection_test"
|
|
|
+ collection_name = "rag_children_hybrid"
|
|
|
|
|
|
gather_start = time.time()
|
|
|
# 优化:降低hybrid_top_k参数从50到20,减少混合搜索时间
|
|
|
@@ -284,11 +284,13 @@ class RetrievalManager:
|
|
|
# 处理二次重排序的高分文档
|
|
|
for rerank_item in bfp_rerank_result:
|
|
|
text = rerank_item.get('text', '')
|
|
|
+ parent_id = rerank_item.get('parent_id', '')
|
|
|
score = rerank_item.get('score', 0.0)
|
|
|
|
|
|
if text in text_to_metadata:
|
|
|
original_item = text_to_metadata[text].copy()
|
|
|
original_item['bfp_rerank_score'] = score
|
|
|
+ original_item['bfp_rerank_parent_id'] = parent_id
|
|
|
final_results.append(original_item)
|
|
|
|
|
|
reorganize_end = time.time()
|