tools / config.py
Germano Cavalcante
Updates
23415c5
raw
history blame contribute delete
438 Bytes
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")
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()