File size: 267 Bytes
a2ac5f7
 
 
 
 
 
 
 
f2daaee
a2ac5f7
1
2
3
4
5
6
7
8
9
10
11
from .base import BaseRetriever


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

    def return_retriever(self, db, config):
        retriever = db.as_langchain_retriever(k=config["vectorstore"]["search_top_k"])
        return retriever