scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

async_log_success_event

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-27 14:17:13 -07:00
parent b97e56252d
commit 3c5e0abaf2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/proxy_unit_tests/test_configs/custom_callbacks.py
View file

@ -61,7 +61,7 @@ class testCustomCallbackProxy(CustomLogger):
print("Value of async success: ", self.async_success)
print("\n kwargs: ", kwargs)
if (
kwargs.get("model") == "azure-embedding-model"
kwargs.get("model") == "text-embedding-ada-002"
or kwargs.get("model") == "ada"
):
print("Got an embedding model", kwargs.get("model"))
@ -118,4 +118,5 @@ class testCustomCallbackProxy(CustomLogger):
self.async_completion_kwargs_fail = kwargs
# Global instance that can be accessed from tests
my_custom_logger = testCustomCallbackProxy()