mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-29 07:39:56 +00:00
fix(litellm_logging.py): fix circular reference
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user