"""Конфигурация GPU воркера."""

from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
    """Настройки GPU воркера."""

    model_config = SettingsConfigDict(
        env_file=".env",
        env_file_encoding="utf-8",
        case_sensitive=False,
    )

    # PostgreSQL
    POSTGRES_HOST: str = "postgres"
    POSTGRES_PORT: int = 5432
    POSTGRES_DB: str = "antiplagiator"
    POSTGRES_USER: str = "antiplagiator"
    POSTGRES_PASSWORD: str = "changeme"

    # Redis
    REDIS_URL: str = "redis://redis:6379/0"

    # RabbitMQ
    RABBITMQ_URL: str = "amqp://guest:guest@rabbitmq:5672/"

    # MinIO
    MINIO_ENDPOINT: str = "minio:9000"
    MINIO_ACCESS_KEY: str = "minioadmin"
    MINIO_SECRET_KEY: str = "changeme"
    MINIO_BUCKET_DOCS: str = "documents"

    # Elasticsearch
    ELASTICSEARCH_URL: str = "http://elasticsearch:9200"

    # Ollama
    OLLAMA_URL: str = "http://ollama:11434"

    # FAISS / ML
    FAISS_INDEX_PATH: str = "/data/index/faiss.index"
    FAISS_ID_MAP_PATH: str = "/data/index/faiss_id_map.json"
    EMBED_MODEL: str = "paraphrase-multilingual-mpnet-base-v2"
    EMBED_DEVICE: str = "cuda"
    EMBED_BATCH_SIZE: int = 64
    EMBED_DIM: int = 768  # Размерность вектора paraphrase-multilingual-mpnet-base-v2
    FAISS_NLIST: int = 1024  # Количество кластеров для IVFFlat
    FAISS_NPROBE: int = 64  # Количество кластеров для поиска

    # App
    APP_URL: str = "https://academic.jze9.ru"
    ENVIRONMENT: str = "development"
    DEBUG: bool = False

    @property
    def database_url_sync(self) -> str:
        """Синхронный URL для SQLAlchemy (psycopg2)."""
        return (
            f"postgresql+psycopg2://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}"
            f"@{self.POSTGRES_HOST}:{self.POSTGRES_PORT}/{self.POSTGRES_DB}"
        )


settings = Settings()