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

ci/cd - anyscale discontinued their API endoints - skip test

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-01 17:58:48 -07:00
parent fe89a613f0
commit 6f79081ea2
3 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_model_alias_map.py
View file

@ -1,19 +1,21 @@
#### What this tests ####
# This tests the model alias mapping - if user passes in an alias, and has set an alias, set it to the actual value
import sys, os
import os
import sys
import traceback
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
import pytest
import litellm
from litellm import completion, embedding
litellm.set_verbose = True
model_alias_map = {"good-model": "anyscale/meta-llama/Llama-2-7b-chat-hf"}
model_alias_map = {"good-model": "groq/llama3-8b-8192"}
def test_model_alias_map(caplog):
@ -33,7 +35,7 @@ def test_model_alias_map(caplog):
for log in captured_logs:
assert "ERROR" not in log
assert "Llama-2-7b-chat-hf" in response.model
assert "groq/llama3-8b-8192" in response.model
except Exception as e:
pytest.fail(f"Error occurred: {e}")