Corin1998 commited on
Commit
2e448ce
·
verified ·
1 Parent(s): 07d86ef

Update irpr/config.py

Browse files
Files changed (1) hide show
  1. irpr/config.py +7 -11
irpr/config.py CHANGED
@@ -1,21 +1,17 @@
1
  # irpr/config.py
 
 
2
  import os
3
- from dataclasses import dataclass, field
4
 
5
  @dataclass
6
  class Settings:
7
- # 書き込み可能なベースディレクトリ。指定なければ ./var
8
- DATA_DIR: str = field(default_factory=lambda: os.environ.get("DATA_DIR", "./var"))
 
 
9
 
10
- # OpenAI
11
  OPENAI_EMBED_MODEL: str = os.environ.get("OPENAI_EMBED_MODEL", "text-embedding-3-small")
12
  OPENAI_CHAT_MODEL: str = os.environ.get("OPENAI_CHAT_MODEL", "gpt-4o-mini")
13
 
14
- # インデックス保存先(シンプル自前ベクタストア)
15
- INDEX_DIR: str = field(default_factory=lambda: os.environ.get("INDEX_DIR", ""))
16
-
17
- def __post_init__(self):
18
- if not self.INDEX_DIR:
19
- self.INDEX_DIR = os.path.join(self.DATA_DIR, "simple_index")
20
-
21
  settings = Settings()
 
1
  # irpr/config.py
2
+ from __future__ import annotations
3
+ from dataclasses import dataclass
4
  import os
 
5
 
6
  @dataclass
7
  class Settings:
8
+ # まずは /tmp/irpr を既定に。環境で上書きする場合も、後段で書き込み可否を検証してダメなら自動フォールバック。
9
+ DATA_DIR: str = os.environ.get("DATA_DIR", "/tmp/irpr")
10
+ # 明示指定があれば使うが、後段で書き込み可否を検証してダメなら自動フォールバック。
11
+ INDEX_DIR: str | None = os.environ.get("INDEX_DIR") or None
12
 
13
+ # OpenAI の既定モデル(環境変数で自由に上書き可)
14
  OPENAI_EMBED_MODEL: str = os.environ.get("OPENAI_EMBED_MODEL", "text-embedding-3-small")
15
  OPENAI_CHAT_MODEL: str = os.environ.get("OPENAI_CHAT_MODEL", "gpt-4o-mini")
16
 
 
 
 
 
 
 
 
17
  settings = Settings()