feat(rag): add shared retriever for audit pipeline

This commit is contained in:
2026-05-21 15:54:47 +08:00
parent 4847bccdec
commit 6ce1a290ab
6 changed files with 795 additions and 316 deletions
@@ -24,6 +24,7 @@ from leaudit.ocr.base import BaseOCRClient
from leaudit.ocr.models import OcrResult
from fastapi_modules.fastapi_leaudit.leaudit_bridge.storage_adapter import StorageAdapter
from fastapi_modules.fastapi_leaudit.rag_engine.retriever import RagRetriever
log = logging.getLogger(__name__)
@@ -89,10 +90,12 @@ class LauditPipeline:
ocr_client: BaseOCRClient,
llm_client: BaseLLMClient | None = None,
storage_adapter: StorageAdapter | None = None,
rag_retriever: RagRetriever | None = None,
) -> None:
self.ocr_client = ocr_client
self.llm_client = llm_client
self.storage = storage_adapter or StorageAdapter()
self.rag_retriever = rag_retriever or RagRetriever()
async def run(
self,
@@ -219,6 +222,7 @@ class LauditPipeline:
visual_manifest=visual_manifest,
phase=detected_phase,
external_mocks=external_mocks,
retriever=self.rag_retriever,
)
timing["evaluation"] = round(time.time() - t0, 2)
log.info(