DocsGPT/application/retriever/base.py

from abc import ABC, abstractmethod


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

    @abstractmethod
    def gen(self, *args, **kwargs):
        pass

    @abstractmethod
    def search(self, *args, **kwargs):
        pass