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
|