fix(proxy_server.py): support reading master key from os environment

This commit is contained in:
Krrish Dholakia 2023-11-28 14:05:17 -08:00
parent 094144de58
commit 4ea52dd571

View file

@ -354,6 +354,9 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str):
if general_settings:
### MASTER KEY ###
master_key = general_settings.get("master_key", None)
if master_key and master_key.startswith("os.environ/"):
master_key_env_name = master_key.replace("os.environ/", "")
master_key = os.getenv(master_key_env_name)
### CONNECT TO DATABASE ###
database_url = general_settings.get("database_url", None)
prisma_setup(database_url=database_url)