From 6aa2f4bc1d69ec900b755f4adee1f3208f88a28b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 11 Dec 2023 10:24:37 -0800 Subject: [PATCH] (test) custom logger embedding --- litellm/tests/test_configs/custom_callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_configs/custom_callbacks.py b/litellm/tests/test_configs/custom_callbacks.py index 94b89e5893..6d3e71ce2e 100644 --- a/litellm/tests/test_configs/custom_callbacks.py +++ b/litellm/tests/test_configs/custom_callbacks.py @@ -55,7 +55,7 @@ class MyCustomHandler(CustomLogger): self.async_success = True print("Value of async success: ", self.async_success) print("\n kwargs: ", kwargs) - if kwargs.get("model") == "azure-embedding-model": + if kwargs.get("model") == "azure-embedding-model" or kwargs.get("model") == "ada": self.async_success_embedding = True self.async_embedding_kwargs = kwargs self.async_embedding_response = response_obj