diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 6421d2ca704..2cd2b1d93bd 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -292,7 +292,7 @@ _CALLBACK_ACCEPTS_CALL_INFO: Dict[int, bool] = {} def _accepts_litellm_call_info(cb: CustomLogger) -> bool: - key = id(cb) + key = id(type(cb)) if key not in _CALLBACK_ACCEPTS_CALL_INFO: sig = inspect.signature(cb.async_post_call_response_headers_hook) _CALLBACK_ACCEPTS_CALL_INFO[key] = "litellm_call_info" in sig.parameters