from woodwork.components.knowledge_base import knowledge_base

class vector_database(knowledge_base):
    def __init__(self, name, retriever):
        super().__init__(name)
        self.retriever = retriever