refactor: 重构 rerank 架构,分离服务层和业务逻辑
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Has been cancelled

- rerank_services.py:纯服务层,只负责调用 rerank server
- rag/rerank.py:业务逻辑层,负责文档处理、排序、top_n
- 更新 pipeline.py 使用新架构
- 架构与 embedding_services.py 保持一致
This commit is contained in:
2026-04-26 11:57:42 +08:00
parent 55c910bbe0
commit f63c394fcd
3 changed files with 176 additions and 85 deletions

View File

@@ -7,6 +7,7 @@ from langchain_core.documents import Document
from langchain_core.language_models import BaseLanguageModel
from ..model_services import get_rerank_service
from .rerank import create_document_reranker
from .query_transform import MultiQueryGenerator
from .fusion import reciprocal_rank_fusion
@@ -38,7 +39,7 @@ class RAGPipeline:
# 初始化组件 - 使用统一的重排服务获取接口
self.query_generator = MultiQueryGenerator(llm=llm, num_queries=num_queries)
self.reranker = get_rerank_service()
self.reranker = create_document_reranker()
async def aretrieve(self, query: str) -> List[Document]:
"""