IR_PR_PilotPro / irpr /config.py
Corin1998's picture
Update irpr/config.py
2e448ce verified
raw
history blame contribute delete
809 Bytes
# irpr/config.py
from __future__ import annotations
from dataclasses import dataclass
import os
@dataclass
class Settings:
# まずは /tmp/irpr を既定に。環境で上書きする場合も、後段で書き込み可否を検証してダメなら自動フォールバック。
DATA_DIR: str = os.environ.get("DATA_DIR", "/tmp/irpr")
# 明示指定があれば使うが、後段で書き込み可否を検証してダメなら自動フォールバック。
INDEX_DIR: str | None = os.environ.get("INDEX_DIR") or None
# OpenAI の既定モデル(環境変数で自由に上書き可)
OPENAI_EMBED_MODEL: str = os.environ.get("OPENAI_EMBED_MODEL", "text-embedding-3-small")
OPENAI_CHAT_MODEL: str = os.environ.get("OPENAI_CHAT_MODEL", "gpt-4o-mini")
settings = Settings()