Kevin Hu
commited on
Commit
·
4c135d3
1
Parent(s):
9c45d1e
Ensure LIGHTEN work (#3542)
Browse files### What problem does this PR solve?
#3531
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- api/settings.py +4 -1
api/settings.py
CHANGED
|
@@ -61,7 +61,10 @@ kg_retrievaler = None
|
|
| 61 |
|
| 62 |
|
| 63 |
def init_settings():
|
| 64 |
-
global LLM, LLM_FACTORY, LLM_BASE_URL
|
|
|
|
|
|
|
|
|
|
| 65 |
LLM = get_base_config("user_default_llm", {})
|
| 66 |
LLM_FACTORY = LLM.get("factory", "Tongyi-Qianwen")
|
| 67 |
LLM_BASE_URL = LLM.get("base_url")
|
|
|
|
| 61 |
|
| 62 |
|
| 63 |
def init_settings():
|
| 64 |
+
global LLM, LLM_FACTORY, LLM_BASE_URL, LIGHTEN, DATABASE_TYPE, DATABASE
|
| 65 |
+
LIGHTEN = int(os.environ.get('LIGHTEN', "0"))
|
| 66 |
+
DATABASE_TYPE = os.getenv("DB_TYPE", 'mysql')
|
| 67 |
+
DATABASE = decrypt_database_config(name=DATABASE_TYPE)
|
| 68 |
LLM = get_base_config("user_default_llm", {})
|
| 69 |
LLM_FACTORY = LLM.get("factory", "Tongyi-Qianwen")
|
| 70 |
LLM_BASE_URL = LLM.get("base_url")
|