mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
Update litellm/proxy/utils.py
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
parent
f023f7ff77
commit
6e89167fd4
1 changed files with 6 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue