python-services/Retrieve/search_service/base_search.py

from typing import List

from bio_requests.rag_request import RagRequest
from dto.bio_document import BaseBioDocument


class BaseSearchService:
    _registry = []

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        BaseSearchService._registry.append(cls)

    @classmethod
    def get_subclasses(cls):
        return cls._registry

    def __init__(self):
        self.data_source = "Base"
        pass

    async def filter_search(self, rag_request: RagRequest) -> List[BaseBioDocument]:
        if self.data_source in rag_request.data_source:
            return await self.search(rag_request)
        return []

    async def search(self, rag_request: RagRequest) -> List[BaseBioDocument]:
        return []