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: