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

test(test_get_llm_provider.py): cover scenario where xai not in model name

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-18 11:04:59 -07:00
parent 2c094c2356
commit 267084a1af
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_get_llm_provider.py
View file

@ -218,9 +218,10 @@ def test_bedrock_invoke_anthropic():
assert model == "invoke/anthropic.claude-3-5-sonnet-20240620-v1:0"
def test_xai_api_base():
@pytest.mark.parametrize("model", ["xai/grok-2-vision-latest", "grok-2-vision-latest"])
def test_xai_api_base(model):
args = {
"model": "xai/grok-2-vision-latest",
"model": model,
"custom_llm_provider": "xai",
"api_base": None,
"api_key": "xai-my-specialkey",