From f4b595ce71417432726af2ef3b3fe8234cedea86 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 16 Apr 2024 17:53:28 -0700 Subject: [PATCH] fix(utils.py): return vertex api base for request hanging alerts --- litellm/proxy/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 970ba45c131..598a0732b9b 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -193,8 +193,8 @@ class ProxyLogging: # Convert the timedelta to float (in seconds) time_difference_float = time_difference.total_seconds() litellm_params = kwargs.get("litellm_params", {}) - api_base = litellm_params.get("api_base", "") model = kwargs.get("model", "") + api_base = litellm.get_api_base(model=model, optional_params=litellm_params) messages = kwargs.get("messages", "") return time_difference_float, model, api_base, messages