Signed-off-by: B-Step62 <yuki.watanabe@databricks.com>
This commit is contained in:
B-Step62 2025-03-20 02:06:34 +09:00
parent 53443a8d5a
commit 57f1d436e1
4 changed files with 5 additions and 24 deletions

View file

@ -1161,13 +1161,7 @@ def client(original_function): # noqa: PLR0915
# LOG SUCCESS - handle streaming success logging in the _next_ object, remove `handle_success` once it's deprecated
verbose_logger.info("Wrapper: Completed Call, calling success_handler")
if litellm.sync_logging:
logging_obj.success_handler(result, start_time, end_time)
else:
executor.submit(
# NB: We already run this in a TPE so the handler itself should run sync
logging_obj.success_handler, result, start_time, end_time, synchronous=True,
)
logging_obj.success_handler(result, start_time, end_time)
# RETURN RESULT
update_response_metadata(