phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: testing fixes

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-12 10:57:51 -08:00
parent 0cf0c2d6dd
commit fa74447cc8
4 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_model_alias_map.py
View file

@ -17,10 +17,10 @@ model_alias_map = {
"good-model": "anyscale/meta-llama/Llama-2-7b-chat-hf"
}
litellm.model_alias_map = model_alias_map
def test_model_alias_map():
try:
litellm.model_alias_map = model_alias_map
response = completion(
"good-model",
messages=[{"role": "user", "content": "Hey, how's it going?"}],