mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-25 23:14:46 +08:00
12 lines
412 B
Python
12 lines
412 B
Python
|
from src.config.tools import SELECTED_RAG_PROVIDER, RAGProvider
|
||
|
|
from src.rag.ragflow import RAGFlowProvider
|
||
|
|
from src.rag.retriever import Retriever
|
||
|
|
|
||
|
|
|
||
|
|
def build_retriever() -> Retriever | None:
|
||
|
|
if SELECTED_RAG_PROVIDER == RAGProvider.RAGFLOW.value:
|
||
|
|
return RAGFlowProvider()
|
||
|
|
elif SELECTED_RAG_PROVIDER:
|
||
|
|
raise ValueError(f"Unsupported RAG provider: {SELECTED_RAG_PROVIDER}")
|
||
|
|
return None
|