scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix(litellm_logging.py): add debug statements to logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-01 21:11:13 -07:00
parent b761ce9dd1
commit 708e87e265
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/litellm_core_utils/litellm_logging.py
View file

@ -1354,11 +1354,16 @@ class Logging:
end_time=end_time,
)
if isinstance(callback, CustomLogger): # custom logger class
if self.stream == True:
if self.stream is True:
if (
"async_complete_streaming_response"
in self.model_call_details
):
print_verbose(
"Enters async log success event for streaming call for callback={}".format(
callback
)
)
await callback.async_log_success_event(
kwargs=self.model_call_details,
response_obj=self.model_call_details[
@ -1367,6 +1372,11 @@ class Logging:
start_time=start_time,
end_time=end_time,
)
print_verbose(
"Successfully logs async success event for streaming call for callback={}".format(
callback
)
)
else:
await callback.async_log_stream_event( # [TODO]: move this to being an async log stream event function
kwargs=self.model_call_details,