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

updates to litedebugger dashboard

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-23 11:51:14 -07:00
parent 40065c3640
commit d183dc8269
8 changed files with 13 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -899,7 +899,7 @@ def handle_failure(exception, traceback_exception, start_time, end_time, args,
print_verbose("reaches lite_debugger for logging!")
print_verbose(f"liteDebuggerClient: {liteDebuggerClient}")
model = args[0] if len(args) > 0 else kwargs["model"]
messages = args[1] if len(args) > 1 else kwargs.get("messages", {"role": "user", "content": kwargs.get("input", "")})
messages = args[1] if len(args) > 1 else kwargs.get("messages", [{"role": "user", "content": ' '.join(kwargs.get("input", ""))}])
result = {
"model": model,
"created": time.time(),
@ -1031,7 +1031,7 @@ def handle_success(args, kwargs, result, start_time, end_time):
elif callback == "lite_debugger":
print_verbose("reaches lite_debugger for logging!")
print_verbose(f"liteDebuggerClient: {liteDebuggerClient}")
messages = args[1] if len(args) > 1 else kwargs.get("messages", {"role": "user", "content": kwargs.get("input")})
messages = args[1] if len(args) > 1 else kwargs.get("messages", [{"role": "user", "content": ' '.join(kwargs.get("input", ""))}])
liteDebuggerClient.log_event(
model=model,
messages=messages,