ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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