tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-30 14:25:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

(fix) async custom logger - trigger when stream completed

Browse Source
This commit is contained in:
ishaan-jaff
2023-12-08 17:25:05 -08:00
parent 5fa2b6e5ad
commit 4e8e3ff33a
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+15 -6
View File
@@ -999,12 +999,21 @@ class Logging:
try:
if isinstance(callback, CustomLogger): # custom logger class
print_verbose(f"Async success callbacks: CustomLogger")
await callback.async_log_success_event(
kwargs=self.model_call_details,
response_obj=result,
start_time=start_time,
end_time=end_time,
)
if self.stream:
if "complete_streaming_response" in self.model_call_details:
await callback.async_log_success_event(
kwargs=self.model_call_details,
response_obj=self.model_call_details["complete_streaming_response"],
start_time=start_time,
end_time=end_time,
)
else:
await callback.async_log_success_event(
kwargs=self.model_call_details,
response_obj=result,
start_time=start_time,
end_time=end_time,
)
if callable(callback): # custom logger functions
print_verbose(f"Async success callbacks: async_log_event")
await customLogger.async_log_event(