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

verify agreement between rds host region, AWS_DEFAULT_REGION

Browse source
This commit is contained in:
Brendan Shanahan 2026-04-10 12:53:10 -04:00
parent ca61f5ae8c
commit ba0066882d
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
backend/open_webui/internal/db.py
View file

@ -74,7 +74,17 @@ class RDSIAMConfig:
else:
log.warning("Unable to verify CA file path; using sslmode=require without certificate verification.")
self.client = boto3.client("rds", region_name=os.getenv("AWS_DEFAULT_REGION", "us-east-1"))
# RDS host format: {db}.{account}.{region}.rds.amazonaws.com
aws_region = os.getenv("AWS_DEFAULT_REGION", "us-east-1")
host_parts = db_host.split(".")
if len(host_parts) >= 4 and host_parts[-3] == "rds" and host_parts[-2] == "amazonaws":
if (host_region := host_parts[-4]) != aws_region:
raise ValueError(
f"AWS_DEFAULT_REGION '{aws_region}' does not match the region inferred from "
f"DATABASE_HOST '{db_host}' ('{host_region}'). "
f"Set AWS_DEFAULT_REGION={host_region} to match your RDS instance."
)
self.client = boto3.client("rds", region_name=aws_region)
self.token = self.get_token()
self.engine = self.create_engine()