feat: add rag retriever bridge

This commit is contained in:
wren
2026-05-22 18:49:57 +08:00
parent 14d1199675
commit ba8e93c0d3
8 changed files with 836 additions and 326 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(