code/modules/retriever/colbert_retriever.py

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