Skip to content

Commit

Permalink
fix: typo
Browse files Browse the repository at this point in the history
  • Loading branch information
dartpain committed Sep 9, 2024
1 parent fb1fd85 commit a1d3592
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions application/retriever/retriever_creator.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,16 @@


class RetrieverCreator:
retievers = {
retrievers = {
'classic': ClassicRAG,
'duckduck_search': DuckDuckSearch,
'brave_search': BraveRetSearch
'brave_search': BraveRetSearch,
'default': ClassicRAG
}

@classmethod
def create_retriever(cls, type, *args, **kwargs):
retiever_class = cls.retievers.get(type.lower())
retiever_class = cls.retrievers.get(type.lower())
if not retiever_class:
raise ValueError(f"No retievers class found for type {type}")
return retiever_class(*args, **kwargs)

0 comments on commit a1d3592

Please sign in to comment.