mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
clean up
Signed-off-by: B-Step62 <yuki.watanabe@databricks.com>
This commit is contained in:
parent
53443a8d5a
commit
57f1d436e1
4 changed files with 5 additions and 24 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue