from app_types import S3Path

from ..base import SearchLoadIndexBase
from ..constants import TRACK_SEARCH_INDEX_DATA_PATH, TRACK_SEARCH_INDEX_MAPPING, TRACK_SEARCH_INDEX_NAME
from .transform import Transform


__all__ = ["LoadIndex"]


class LoadIndex(SearchLoadIndexBase):
    depends_on = {Transform}
    data_path: S3Path = TRACK_SEARCH_INDEX_DATA_PATH
    base_index_name: str = TRACK_SEARCH_INDEX_NAME
    index_mapping_params: dict = TRACK_SEARCH_INDEX_MAPPING