From 99dcce1e0f45755a12300ce8aa43bd1ca3b1b08c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 12 Jan 2024 17:52:14 -0800 Subject: [PATCH] (test) aembedding includes custom_llm_provider --- litellm/tests/test_embedding.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py index ab01f6f191a..d1f0ee69962 100644 --- a/litellm/tests/test_embedding.py +++ b/litellm/tests/test_embedding.py @@ -319,6 +319,12 @@ def test_aembedding_azure(): input=["good morning from litellm", "this is another item"], ) print(response) + + print( + "hidden params - custom_llm_provider", + response._hidden_params["custom_llm_provider"], + ) + assert response._hidden_params["custom_llm_provider"] == "azure" except Exception as e: pytest.fail(f"Error occurred: {e}")