From a9c5b02303a1eb6f63ca3285af28631214ca9a9f Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 19 Jan 2024 08:51:14 -0800 Subject: [PATCH] (v0) fix --- litellm/utils.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 2e0650be6e8..c4144267bf4 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2357,12 +2357,16 @@ def client(original_function): print_verbose( f"Async Wrapper: Completed Call, calling async_success_handler: {logging_obj.async_success_handler}" ) - asyncio.create_task( - logging_obj.async_success_handler(result, start_time, end_time) - ) + # asyncio.to_thread( + # logging_obj.async_success_handler(result, start_time, end_time) + # ) threading.Thread( target=logging_obj.success_handler, args=(result, start_time, end_time) ).start() + threading.Thread( + target=logging_obj.async_success_handler, + args=(result, start_time, end_time), + ).start() # RETURN RESULT if hasattr(result, "_hidden_params"): result._hidden_params["model_id"] = kwargs.get("model_info", {}).get(