scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(proxy_cli.py): check for env var for IAM_TOKEN_DB_AUTH

Browse source
This commit is contained in:
Krrish Dholakia 2025-11-11 11:56:49 -08:00
parent be05324645
commit a987e3ca46
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_cli.py
View file

@ -13,6 +13,7 @@ import httpx
from dotenv import load_dotenv
from litellm.constants import DEFAULT_NUM_WORKERS_LITELLM_PROXY
from litellm.secret_managers.main import get_secret_bool
if TYPE_CHECKING:
from fastapi import FastAPI
@ -615,7 +616,7 @@ def run_server( # noqa: PLR0915
general_settings = {}
### GET DB TOKEN FOR IAM AUTH ###
if iam_token_db_auth:
if iam_token_db_auth or get_secret_bool("IAM_TOKEN_DB_AUTH"):
from litellm.proxy.auth.rds_iam_token import generate_iam_auth_token
db_host = os.getenv("DATABASE_HOST")
@ -693,15 +694,14 @@ def run_server( # noqa: PLR0915
litellm._key_management_settings = KeyManagementSettings(
**key_management_settings
)
if general_settings:
### LOAD SECRET MANAGER ###
key_management_system = general_settings.get(
"key_management_system", None
)
proxy_config.initialize_secret_manager(
key_management_system=key_management_system,
config_file_path=config
key_management_system=key_management_system, config_file_path=config
)
database_url = general_settings.get("database_url", None)
if database_url is None and os.getenv("DATABASE_URL") is None: