tools / config.py
Germano Cavalcante
Fix Openai API for embeddings
8aab0fd
raw
history blame
467 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")
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()