From 5f3fc6ff07b83f9524b0204a46d3da025da67863 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=A6=82=E5=A8=81?= Date: Tue, 29 Jul 2025 09:49:53 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E9=87=8D=E6=8E=92=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- examples/rerank_demo.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/examples/rerank_demo.py b/examples/rerank_demo.py index 873fde2..bd02cc0 100644 --- a/examples/rerank_demo.py +++ b/examples/rerank_demo.py @@ -48,13 +48,9 @@ def main(): "type": "local", "model_name": "sentence-transformers/all-MiniLM-L6-v2", } - + # 重排配置 - rerank_config = { - "enabled": True, - "method": "similarity", - "top_k": 3 - } + rerank_config = {"enabled": True, "method": "cross_encoder", "top_k": 3} # 初始化RAG系统 rag = RerankRAG( @@ -111,21 +107,21 @@ def main(): for i, test_case in enumerate(test_queries, 1): query = test_case["query"] expected = test_case["expected_topic"] - + print(f"=== 测试查询 {i}: {query} ===") print(f"期望结果: {expected}\n") - + # 获取两种搜索结果 docs_no_rerank, docs_with_rerank = rag.query_with_scores(query, k=5) - + print("📍 不使用重排的结果:") for j, doc in enumerate(docs_no_rerank[:3], 1): print(f" {j}. {doc.page_content}") - + print("\n🎯 使用重排的结果:") for j, doc in enumerate(docs_with_rerank[:3], 1): print(f" {j}. {doc.page_content}") - + print("\n" + "="*80 + "\n") print("重排功能测试完成!")