From a8bc10170ac16288c52360b68db38f7a0b8dc75c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 6 Mar 2024 21:27:41 -0800 Subject: [PATCH] fix(proxy_server.py): support cost tracking if general_settings is none works if database_url is in env --- litellm/proxy/proxy_server.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 95570d7fa..54791e62c 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1797,8 +1797,6 @@ class ProxyConfig: custom_db_client = DBClient( custom_db_args=database_args, custom_db_type=database_type ) - ## COST TRACKING ## - cost_tracking() ## ADMIN UI ACCESS ## ui_access_mode = general_settings.get( "ui_access_mode", "all" @@ -2350,6 +2348,10 @@ async def startup_event(): # if not, assume it's a json string worker_config = json.loads(os.getenv("WORKER_CONFIG")) await initialize(**worker_config) + + ## COST TRACKING ## + cost_tracking() + proxy_logging_obj._init_litellm_callbacks() # INITIALIZE LITELLM CALLBACKS ON SERVER STARTUP <- do this to catch any logging errors on startup, not when calls are being made if use_background_health_checks: