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

fix(litellm_logging.py): fix circular reference

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-15 21:26:47 -07:00
parent 13102b0f17
commit 5ba56191c4
2 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_custom_callback_input.py
View file

@ -165,6 +165,11 @@ class CompletionCustomHandler(
def log_success_event(self, kwargs, response_obj, start_time, end_time):
try:
print(f"\n\nkwargs={kwargs}\n\n")
print(
json.dumps(kwargs, default=str)
) # this is a test to confirm no circular references are in the logging object
self.states.append("sync_success")
## START TIME
assert isinstance(start_time, datetime)