Update application files and fix LLM models, create new retriever class

This commit is contained in:
Alex
2024-04-09 14:02:33 +01:00
parent e07df29ab9
commit 391f686173
13 changed files with 202 additions and 185 deletions

View File

@@ -0,0 +1,15 @@
from application.retriever.classic_rag import ClassicRAG
class RetrieverCreator:
retievers = {
'classic': ClassicRAG,
}
@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)