From 70899521ae3cbd40f16d79ade794a3bb9050f368 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 12 Jan 2024 17:09:59 -0800 Subject: [PATCH] (test) custom_llm_provider in hidden params --- litellm/tests/test_completion.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index ae2cd06c7..472bc3898 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -849,6 +849,9 @@ def test_completion_azure_key_completion_arg(): max_tokens=10, ) print(f"response: {response}") + + print("Hidden Params", response._hidden_params) + assert response._hidden_params["custom_llm_provider"] == "azure" os.environ["AZURE_API_KEY"] = old_key except Exception as e: os.environ["AZURE_API_KEY"] = old_key