litellm-mirror/litellm/proxy/auth/rds_iam_token.py
Krrish Dholakia 24e20718e4 feat(proxy_cli.py): support iam-based auth to rds
Initial pr for iam-based auth support for rds
2024-08-05 17:35:48 -07:00

11 lines
327 B
Python

def generate_iam_auth_token(db_host, db_port, db_user) -> str:
from urllib.parse import quote
import boto3
client = boto3.client("rds")
token = client.generate_db_auth_token(
DBHostname=db_host, Port=db_port, DBUsername=db_user
)
cleaned_token = quote(token, safe="")
return cleaned_token