phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(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
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, max_tokens=10,
) )
print(f"response: {response}") 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 os.environ["AZURE_API_KEY"] = old_key
except Exception as e: except Exception as e:
os.environ["AZURE_API_KEY"] = old_key os.environ["AZURE_API_KEY"] = old_key