phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: bugs in traceloop integration

Browse source
This commit is contained in:
Nir Gazit 2023-10-19 17:23:51 +02:00
parent 00ac43c19a
commit c21e86faec
4 changed files with 144 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -2308,6 +2308,15 @@ def handle_success(args, kwargs, result, start_time, end_time):
end_time=end_time,
print_verbose=print_verbose,
)
elif callback == "traceloop":
traceloopLogger.log_event(
kwargs=kwargs,
response_obj=result,
start_time=start_time,
end_time=end_time,
print_verbose=print_verbose,
)
elif callback == "aispend":
print_verbose("reaches aispend for logging!")