diff --git a/tests/proxy_unit_tests/test_proxy_custom_logger.py b/tests/proxy_unit_tests/test_proxy_custom_logger.py index 0fbe5a31c33..28de522a4c7 100644 --- a/tests/proxy_unit_tests/test_proxy_custom_logger.py +++ b/tests/proxy_unit_tests/test_proxy_custom_logger.py @@ -80,7 +80,7 @@ def test_embedding(client): my_custom_logger.async_success_embedding is True ) # checks if the status of async_success is True, only the async_log_success_event can set this to true assert ( - my_custom_logger.async_embedding_kwargs["model"] == "azure-embedding-model" + my_custom_logger.async_embedding_kwargs["model"] == "text-embedding-ada-002" ) # checks if kwargs passed to async_log_success_event are correct kwargs = my_custom_logger.async_embedding_kwargs litellm_params = kwargs.get("litellm_params")