from typing import Optional

from pydantic import Field

from .base import FrigateBaseModel

__all__ = ["SemanticSearchConfig"]


class SemanticSearchConfig(FrigateBaseModel):
    enabled: bool = Field(default=False, title="Enable semantic search.")
    reindex: Optional[bool] = Field(
        default=False, title="Reindex all detections on startup."
    )
    device: str = Field(default="AUTO", title="Device Type")
    model_size: str = Field(
        default="small", title="The size of the embeddings model used."
    )