from pydantic_settings import BaseSettings
import os


class Settings(BaseSettings):
    huggingface_key: str = os.environ.get("huggingface_key")
    OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY")
    cache_dir: str = "cache"
    embedding_api: str = "sbert"
    embedding_model: str = "mano-wii/BAAI_bge-base-en-v1.5-tunned-for-blender-issues"
    # embedding_api: str = "openai"
    # embedding_model: str = "text-embedding-ada-002"


settings = Settings()