Corin1998 commited on
Commit
47791a2
·
verified ·
1 Parent(s): 63681be

Create irpr/config.py

Browse files
Files changed (1) hide show
  1. irpr/config.py +15 -0
irpr/config.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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()