Signed-off-by: B-Step62 <yuki.watanabe@databricks.com>
This commit is contained in:
B-Step62 2025-02-03 13:46:21 +09:00
parent 482180cfba
commit f356096772

View file

@ -1083,10 +1083,8 @@ def client(original_function): # noqa: PLR0915
# LOG SUCCESS - handle streaming success logging in the _next_ object, remove `handle_success` once it's deprecated # 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") verbose_logger.info("Wrapper: Completed Call, calling success_handler")
if litellm.sync_logging: if litellm.sync_logging:
print("sync_logging")
logging_obj.success_handler(result, start_time, end_time) logging_obj.success_handler(result, start_time, end_time)
else: else:
print("async_logging")
executor.submit( executor.submit(
logging_obj.success_handler, logging_obj.success_handler,
result, result,