phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) allow using more than 1 custom callback

Browse source
This commit is contained in:
ishaan-jaff 2023-10-19 08:57:10 -07:00
parent 52ff9054e4
commit 5cf3c3dc86
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -333,6 +333,7 @@ class Logging:
messages=self.messages,
kwargs=self.model_call_details,
print_verbose=print_verbose,
callback_func=callback
)
except Exception as e:
traceback.print_exc()
@ -544,6 +545,7 @@ class Logging:
start_time=start_time,
end_time=end_time,
print_verbose=print_verbose,
callback_func=callback
)
except Exception as e:
@ -617,6 +619,7 @@ class Logging:
start_time=start_time,
end_time=end_time,
print_verbose=print_verbose,
callback_func=callback
)
except Exception as e:
print_verbose(
@ -2052,7 +2055,7 @@ def set_callbacks(callback_list, function_id=None):
else:
liteDebuggerClient = LiteDebugger(email=str(uuid.uuid4()))
elif callable(callback):
customLogger = CustomLogger(callback_func=callback)
customLogger = CustomLogger()
except Exception as e:
raise e