from app_types import S3Path

from ..base import SearchLoadIndexBase
from ..constants import ARTIST_SEARCH_INDEX_DATA_PATH, ARTIST_SEARCH_INDEX_MAPPING, ARTIST_SEARCH_INDEX_NAME
from .transform import Transform


__all__ = ["LoadIndex"]


class LoadIndex(SearchLoadIndexBase):
    depends_on = {Transform}
    data_path: S3Path = ARTIST_SEARCH_INDEX_DATA_PATH
    base_index_name: str = ARTIST_SEARCH_INDEX_NAME
    index_mapping_params: dict = ARTIST_SEARCH_INDEX_MAPPING