Signed-off-by: B-Step62 <yuki.watanabe@databricks.com>
This commit is contained in:
B-Step62 2025-02-03 21:55:00 +09:00
parent f356096772
commit 77de86e0e7
8 changed files with 78 additions and 99 deletions

View file

@ -1082,15 +1082,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(
logging_obj.success_handler,
result,
start_time,
end_time,
)
logging_obj.success_handler(result, start_time, end_time)
# RETURN RESULT
update_response_metadata(