scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-08 22:21:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

validate required env vars are present when DATABASE_ENABLE_IAM_TOKEN_AUTH=true

Browse source
This commit is contained in:
Brendan Shanahan 2026-04-10 12:55:09 -04:00
parent ba0066882d
commit 65d718344f
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
backend/open_webui/env.py
View file

@ -337,6 +337,24 @@ if 'postgres://' in DATABASE_URL:
DATABASE_ENABLE_IAM_TOKEN_AUTH = os.environ.get('DATABASE_ENABLE_IAM_TOKEN_AUTH', 'False').lower() == 'true'
if DATABASE_ENABLE_IAM_TOKEN_AUTH:
_iam_missing = [
name for name, val in [
('DATABASE_TYPE', DATABASE_TYPE),
('DATABASE_HOST', os.environ.get('DATABASE_HOST')),
('DATABASE_PORT', os.environ.get('DATABASE_PORT')),
('DATABASE_NAME', os.environ.get('DATABASE_NAME')),
('DATABASE_USER', DATABASE_USER),
]
if not val
]
if _iam_missing:
raise ValueError(
f"DATABASE_ENABLE_IAM_TOKEN_AUTH=true requires individual database connection variables, "
f"but the following are not set: {', '.join(_iam_missing)}. "
f"Set DATABASE_TYPE, DATABASE_HOST, DATABASE_PORT, DATABASE_NAME, and DATABASE_USER."
)
DATABASE_CA_PATH = os.getenv('DATABASE_CA_PATH', None)
DATABASE_SCHEMA = os.environ.get('DATABASE_SCHEMA', None)