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

(fix) use correct value for os.environ dynamo

Browse source
This commit is contained in:
ishaan-jaff 2024-02-12 21:41:13 -08:00
parent c4038e38e5
commit 7d27690203
2 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -1081,14 +1081,19 @@ class ProxyConfig:
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:
value = database_args[k]
verbose_proxy_logger.debug(
f"Loading AWS Web Identity Token from file: {value}"
)
if os.path.exists(value):
with open(value, "r") as file:
token_content = file.read()
database_args[k] = json.loads(token_content)
database_args[k] = token_content
else:
verbose_proxy_logger.info(
f"DynamoDB Loading - {v} is not a valid file path"
f"DynamoDB Loading - {value} is not a valid file path"
)
verbose_proxy_logger.debug(f"database_args: {database_args}")
custom_db_client = DBClient(
custom_db_args=database_args, custom_db_type=database_type
)