Corin1998 commited on
Commit
ef1019a
·
verified ·
1 Parent(s): fe9b124

Update irpr/config.py

Browse files
Files changed (1) hide show
  1. irpr/config.py +8 -10
irpr/config.py CHANGED
@@ -1,15 +1,13 @@
 
1
  import os
 
2
 
 
3
  class Settings:
4
- EDINET_API_KEY: str = os.getenv("EDINET_API_KEY", "")
5
- TDNET_API_BASE: str = os.getenv("TDNET_API_BASE", "")
6
- TDNET_API_KEY: str = os.getenv("TDNET_API_KEY", "")
7
-
8
- GEN_MODEL: str = os.getenv("GEN_MODEL", "Qwen/Qwen2.5-3B-Instruct")
9
- EMB_MODEL: str = os.getenv("EMB_MODEL", "intfloat/multilingual-e5-base")
10
-
11
- DATA_DIR: str = os.getenv("DATA_DIR", "data")
12
- TARGET_EDINET_CODE: str = os.getenv("TARGET_EDINET_CODE", "")
13
- TARGET_SECCODE: str = os.getenv("TARGET_SECCODE", "")
14
 
15
  settings = Settings()
 
1
+ # irpr/config.py
2
  import os
3
+ from dataclasses import dataclass
4
 
5
+ @dataclass
6
  class Settings:
7
+ DATA_DIR: str = os.environ.get("DATA_DIR", "/data")
8
+ EMB_MODEL: str = os.environ.get("EMB_MODEL", "intfloat/multilingual-e5-base")
9
+ # 既定は空 = LLM無効(抽出フォールバックのみで動作)
10
+ GEN_MODEL: str = os.environ.get("GEN_MODEL", "").strip()
11
+ CHROMA_PATH: str = os.environ.get("CHROMA_PATH", "/data/index")
 
 
 
 
 
12
 
13
  settings = Settings()