mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-11 13:10:42 +00:00
19 lines
628 B
Python
19 lines
628 B
Python
from application.retriever.classic_rag import ClassicRAG
|
|
from application.retriever.duckduck_search import DuckDuckSearch
|
|
from application.retriever.brave_search import BraveRetSearch
|
|
|
|
|
|
|
|
class RetrieverCreator:
|
|
retievers = {
|
|
'classic': ClassicRAG,
|
|
'duckduck_search': DuckDuckSearch,
|
|
'brave_search': BraveRetSearch
|
|
}
|
|
|
|
@classmethod
|
|
def create_retriever(cls, type, *args, **kwargs):
|
|
retiever_class = cls.retievers.get(type.lower())
|
|
if not retiever_class:
|
|
raise ValueError(f"No retievers class found for type {type}")
|
|
return retiever_class(*args, **kwargs) |