fix(litellm_logging.py): fix circular reference

This commit is contained in:
Krrish Dholakia
2024-07-15 21:28:33 -07:00
parent eaee90b02e
commit d136f2b8a7
2 changed files with 10 additions and 7 deletions
@@ -659,7 +659,7 @@ class Logging:
for callback in callbacks:
if isinstance(callback, CustomLogger):
self.model_call_details["input"], result = callback.logging_hook(
self.model_call_details, result = callback.logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
@@ -1316,12 +1316,10 @@ class Logging:
for callback in callbacks:
if isinstance(callback, CustomLogger):
self.model_call_details["input"], result = (
await callback.async_logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
)
self.model_call_details, result = await callback.async_logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
)
for callback in callbacks:
@@ -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)