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

test(test_get_llm_provider.py): Minimal repro for https://github.com/BerriAI/litellm/issues/9291

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-18 10:35:50 -07:00
parent c3ee93cd81
commit aeec703c4e
1 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/local_testing/test_get_llm_provider.py
View file

@ -216,3 +216,20 @@ def test_bedrock_invoke_anthropic():
) )
assert custom_llm_provider == "bedrock" assert custom_llm_provider == "bedrock"
assert model == "invoke/anthropic.claude-3-5-sonnet-20240620-v1:0" assert model == "invoke/anthropic.claude-3-5-sonnet-20240620-v1:0"
def test_xai_api_base():
args = {
"model": "xai/grok-2-vision-latest",
"custom_llm_provider": "xai",
"api_base": None,
"api_key": "xai-my-specialkey",
"litellm_params": None,
}
model, custom_llm_provider, dynamic_api_key, api_base = litellm.get_llm_provider(
**args
)
assert custom_llm_provider == "xai"
assert model == "grok-2-vision-latest"
assert api_base == "https://api.x.ai/v1"
assert dynamic_api_key == "xai-my-specialkey"