phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): update dynamodb load config

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-06 08:18:17 -08:00
parent 9c13c65ce2
commit e59aca925a
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -1076,6 +1076,19 @@ class ProxyConfig:
database_type == "dynamo_db" or database_type == "dynamodb"
):
database_args = general_settings.get("database_args", None)
### LOAD FROM os.environ/ ###
for k, v in database_args.items():
if isinstance(v, str) and v.startswith("os.environ/"):
database_args[k] = litellm.get_secret(v)
if isinstance(k, str) and k == "aws_web_identity_token":
if os.path.exists(v):
with open(v, "r") as file:
token_content = file.read()
database_args[k] = json.loads(token_content)
else:
verbose_proxy_logger.info(
f"DynamoDB Loading - {v} is not a valid file path"
)
custom_db_client = DBClient(
custom_db_args=database_args, custom_db_type=database_type
)