phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(v0) fix

Browse source
This commit is contained in:
ishaan-jaff 2024-01-19 08:51:14 -08:00
parent 9aaa2c417d
commit a9c5b02303
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/utils.py
View file

@ -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(