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

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 eaee90b02e
commit d136f2b8a7
2 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/litellm_core_utils/litellm_logging.py
View file

@ -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: