From 771d09312e8bd26091ac2c1cf51641191deb1701 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 13 Mar 2024 10:30:31 -0700 Subject: [PATCH] (fix) issue with using litellm enterprise license --- litellm/proxy/proxy_server.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index b5b68df0ca7..ebc9a3bfdc5 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -4512,8 +4512,12 @@ async def global_spend_models( dependencies=[Depends(user_api_key_auth)], ) async def global_predict_spend_logs(request: Request): - from litellm.proxy.enterprise.utils import _forecast_daily_cost - + try: + # when using litellm package + from litellm.proxy.enterprise.utils import _forecast_daily_cost + except: + # when using litellm docker image + from enterprise.utils import _forecast_daily_cost data = await request.json() data = data.get("data") return _forecast_daily_cost(data)