From 3c5e0abaf2e2c9dc673f3161027dac17adb76d5e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 14:17:13 -0700 Subject: [PATCH] async_log_success_event --- tests/proxy_unit_tests/test_configs/custom_callbacks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_configs/custom_callbacks.py b/tests/proxy_unit_tests/test_configs/custom_callbacks.py index 42f88b5d19b..c7d66c068e9 100644 --- a/tests/proxy_unit_tests/test_configs/custom_callbacks.py +++ b/tests/proxy_unit_tests/test_configs/custom_callbacks.py @@ -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()