From 38fe80266f680e3b00a3817d8b1e2375aa863d61 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 11 Dec 2023 10:40:01 -0800 Subject: [PATCH] (test) debug proxy logger test --- litellm/tests/test_configs/custom_callbacks.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_configs/custom_callbacks.py b/litellm/tests/test_configs/custom_callbacks.py index 6d3e71ce2e5..71f74bbc2da 100644 --- a/litellm/tests/test_configs/custom_callbacks.py +++ b/litellm/tests/test_configs/custom_callbacks.py @@ -56,7 +56,10 @@ class MyCustomHandler(CustomLogger): print("Value of async success: ", self.async_success) print("\n kwargs: ", kwargs) if kwargs.get("model") == "azure-embedding-model" or kwargs.get("model") == "ada": + print("Got an embedding model", kwargs.get("model")) + print("Setting embedding success to True") self.async_success_embedding = True + print("Value of async success embedding: ", self.async_success_embedding) self.async_embedding_kwargs = kwargs self.async_embedding_response = response_obj if kwargs.get("stream") == True: