File size: 809 Bytes
ef1019a
2e448ce
 
47791a2
 
ef1019a
47791a2
2e448ce
 
 
 
b35f8db
2e448ce
b35f8db
 
 
47791a2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# 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()