File size: 414 Bytes
f2daaee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from .helpers import VectorStoreRetrieverScore
from .base import BaseRetriever


class RaptorRetriever(BaseRetriever):
    def __init__(self):
        pass

    def return_retriever(self, db, config):
        retriever = VectorStoreRetrieverScore(
            vectorstore=db,
            search_kwargs={
                "k": config["vectorstore"]["search_top_k"],
            },
        )
        return retriever