From 7807866a37c89f7e88994db17db0311e3b579f8c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=A6=82=E5=A8=81?= Date: Thu, 10 Jul 2025 00:41:44 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20rerank=20=E6=A8=A1=E5=9E=8B=E4=BD=BF?= =?UTF-8?q?=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/rag_service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/services/rag_service.py b/services/rag_service.py index a5298d9..0f301bd 100644 --- a/services/rag_service.py +++ b/services/rag_service.py @@ -26,7 +26,7 @@ class AsyncRAGService: ) self.rerank_llm = ChatOpenAI( - model="deepseek-r1:8b", + model="dengcao/Qwen3-Reranker-8B:Q3_K_M", temperature=0.7, openai_api_key=self.openai_api_key, openai_api_base=self.openai_api_base, @@ -233,6 +233,7 @@ class AsyncRAGService: ) try: response = await asyncio.to_thread(self.rerank_llm.invoke, prompt) + self.logger.info(f"rerank 评分: {response.content.strip()} for {result['metadata']['filename']}") score = float(response.content.strip()) return max(0.0, min(score, 1.0)) except Exception as e: