Update litellm/proxy/utils.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Srikanth @adobe 2026-03-13 20:27:18 -07:00 committed by GitHub
parent f023f7ff77
commit 6e89167fd4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2010,6 +2010,12 @@ class ProxyLogging:
)
if getattr(logging_obj, "model_call_details", {}).get("end_time"):
kwargs["llm_end_time"] = logging_obj.model_call_details["end_time"]
if start_time is None:
_kwarg_start = kwargs.get("start_time")
if isinstance(_kwarg_start, datetime):
start_time = _kwarg_start
if getattr(logging_obj, "model_call_details", {}).get("end_time"):
kwargs["llm_end_time"] = logging_obj.model_call_details["end_time"]
for callback in litellm.callbacks:
_callback: Optional[CustomLogger] = None