From 267084a1af7d2b5e40e18f7d527d2573692d5add Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 18 Mar 2025 11:04:59 -0700 Subject: [PATCH] test(test_get_llm_provider.py): cover scenario where xai not in model name --- tests/local_testing/test_get_llm_provider.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_get_llm_provider.py b/tests/local_testing/test_get_llm_provider.py index 5872d4ea07..fa27a8378c 100644 --- a/tests/local_testing/test_get_llm_provider.py +++ b/tests/local_testing/test_get_llm_provider.py @@ -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",