phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: dynamo_db.py

Browse source 
handle permissions row
This commit is contained in:
Krrish Dholakia 2024-02-15 20:36:59 -08:00
parent cd8d35107b
commit 34fce00960
2 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/proxy_server.py
View file

@ -1013,7 +1013,10 @@ async def update_database(
valid_token.spend = new_spend
user_api_key_cache.set_cache(key=token, value=valid_token)
except Exception as e:
verbose_proxy_logger.info(f"Update Key DB Call failed to execute")
traceback.print_exc()
verbose_proxy_logger.info(
f"Update Key DB Call failed to execute - {str(e)}"
)
### UPDATE SPEND LOGS ###
async def _insert_spend_log_to_db():
@ -1656,9 +1659,7 @@ async def generate_key_helper_fn(
saved_token["config"] = json.loads(saved_token["config"])
if isinstance(saved_token["metadata"], str):
saved_token["metadata"] = json.loads(saved_token["metadata"])
if saved_token["permissions"] is not None and isinstance(
saved_token["permissions"], str
):
if isinstance(saved_token["permissions"], str):
saved_token["permissions"] = json.loads(saved_token["permissions"])
if saved_token.get("expires", None) is not None and isinstance(
saved_token["expires"], datetime