mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-25 00:05:58 +08:00
13 lines
345 B
Python
13 lines
345 B
Python
from chatchat.server.file_rag.retrievers import (
|
|
BaseRetrieverService,
|
|
VectorstoreRetrieverService,
|
|
EnsembleRetrieverService,
|
|
)
|
|
|
|
Retrivals = {
|
|
"vectorstore": VectorstoreRetrieverService,
|
|
"ensemble": EnsembleRetrieverService,
|
|
}
|
|
|
|
def get_Retriever(type: str = "vectorstore") -> BaseRetrieverService:
|
|
return Retrivals[type] |