"""
Embedding module configuration.

This module is intentionally a plain Python file (no env var parsing, no extra deps).
Edit values here to configure:
- server host/port
- local model settings (paths/devices/batch sizes)
"""

from typing import Optional
import os


class EmbeddingConfig(object):
    # Server
    HOST = os.getenv("EMBEDDING_HOST", "0.0.0.0")
    PORT = int(os.getenv("EMBEDDING_PORT", 6005))

    # Text embeddings (BGE-M3)
    TEXT_MODEL_DIR = "Xorbits/bge-m3"
    TEXT_DEVICE = "cuda"  # "cuda" or "cpu" (model may fall back to CPU if needed)
    TEXT_BATCH_SIZE = 32

    # Image embeddings (CN-CLIP)
    IMAGE_MODEL_NAME = "ViT-H-14"
    IMAGE_DEVICE = None  # type: Optional[str]  # "cuda" / "cpu" / None(auto)

    # Service behavior
    IMAGE_BATCH_SIZE = 8


CONFIG = EmbeddingConfig()