ask_candid/retrieval/sources/schema.py

from typing import Tuple, Dict, Optional, Any
from dataclasses import dataclass, field


@dataclass
class ElasticSourceConfig:
    index_name: str
    text_fields: Tuple[str]
    excluded_fields: Optional[Tuple[str]] = field(default_factory=tuple)


@dataclass
class ElasticHitsResult:
    """Dataclass for Elasticsearch hits results
    """
    index: str
    id: Any
    score: float
    source: Dict[str, Any]
    inner_hits: Dict[str, Any]