Update irpr/config.py

irpr/config.py

@@ -4,16 +4,18 @@ from dataclasses import dataclass, field
 
 @dataclass
 class Settings:
-    # 既定は ./var（相対パス）。環境変数 DATA_DIR を指定すれば最優先で使います。
+    # 書き込み可能なベースディレクトリ。指定なければ ./var
     DATA_DIR: str = field(default_factory=lambda: os.environ.get("DATA_DIR", "./var"))
-    EMB_MODEL: str = os.environ.get("EMB_MODEL", "intfloat/multilingual-e5-base")
-    # 既定は空 = LLM無効（抽出フォールバックのみで動作）
-    GEN_MODEL: str = os.environ.get("GEN_MODEL", "").strip()
-    # 指定が無ければ DATA_DIR/index を使う
-    CHROMA_PATH: str = field(default_factory=lambda: os.environ.get("CHROMA_PATH", ""))
+
+    # 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")
+
+    # インデックス保存先（シンプル自前ベクタストア）
+    INDEX_DIR: str = field(default_factory=lambda: os.environ.get("INDEX_DIR", ""))
 
     def __post_init__(self):
-        if not self.CHROMA_PATH:
-            self.CHROMA_PATH = os.path.join(self.DATA_DIR, "index")
+        if not self.INDEX_DIR:
+            self.INDEX_DIR = os.path.join(self.DATA_DIR, "simple_index")
 
 settings = Settings()