(test) custom_llm_provider in hidden params

2024-01-12 17:09:59 -08:00 · 2024-01-12 17:09:59 -08:00 · 70899521ae
commit 70899521ae
parent fd9bddc71a
1 changed files with 3 additions and 0 deletions
--- 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