From 8fd162136f4cd8235e27ac64909bec8439ea539d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 12 Jun 2024 20:03:46 -0700 Subject: [PATCH] feat - litellm_model_cost_map --- litellm/proxy/proxy_server.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 1bba35c82a..1ab6ade0ec 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -14738,6 +14738,22 @@ async def cache_flushall(): ) +@router.get( + "/get/litellm_model_cost_map", + include_in_schema=False, + dependencies=[Depends(user_api_key_auth)], +) +async def get_litellm_model_cost_map(): + try: + _model_cost_map = litellm.model_cost + return _model_cost_map + except Exception as e: + raise HTTPException( + status_code=500, + detail=f"Internal Server Error ({str(e)})", + ) + + @router.get("/", dependencies=[Depends(user_api_key_auth)]) async def home(request: Request): return "LiteLLM: RUNNING"