scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

(test) custom_llm_provider in hidden params

Browse source
This commit is contained in:
ishaan-jaff 2024-01-12 17:09:59 -08:00
parent fd9bddc71a
commit 70899521ae
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

@ -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