ask_candid/base/retrieval/schemas.py

from typing import Any
from dataclasses import dataclass, field


@dataclass
class ElasticSourceConfig:
    index_name: str
    semantic_fields: tuple[str,...] = field(default_factory=tuple)
    text_fields: tuple[str,...] | None = field(default_factory=tuple)
    highlight_fields: tuple[str,...] | None = field(default_factory=tuple)
    excluded_fields: tuple[str,...] | None = 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] | None
    highlight: dict[str, list[str]] | None