wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-17 03:34:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support for moi in RAG module (#571)

Browse Source 
* feat: add support for moi

* small adjust

* small adjust

* according 2 comments

* add more intro

* add more intro
This commit is contained in:
lele3436
2025-09-16 20:25:59 +08:00
committed by GitHub
parent ea0fe62971
commit 5085bf8ee9
6 changed files with 176 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/rag/builder.py
View File

@@ -3,6 +3,7 @@
from src.config.tools import SELECTED_RAG_PROVIDER, RAGProvider
from src.rag.ragflow import RAGFlowProvider
from src.rag.moi import MOIProvider
from src.rag.retriever import Retriever
from src.rag.vikingdb_knowledge_base import VikingDBKnowledgeBaseProvider
from src.rag.milvus import MilvusProvider
@@ -11,6 +12,8 @@ from src.rag.milvus import MilvusProvider
def build_retriever() -> Retriever | None:
if SELECTED_RAG_PROVIDER == RAGProvider.RAGFLOW.value:
return RAGFlowProvider()
elif SELECTED_RAG_PROVIDER == RAGProvider.MOI.value:
return MOIProvider()
elif SELECTED_RAG_PROVIDER == RAGProvider.VIKINGDB_KNOWLEDGE_BASE.value:
return VikingDBKnowledgeBaseProvider()
elif SELECTED_RAG_PROVIDER == RAGProvider.MILVUS.value: