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

fix(pattern_match_deployments.py): default to user input if unable to map based on wildcards (#6646)

Browse source
This commit is contained in:
Krish Dholakia 2024-11-07 23:57:37 +05:30 • committed by GitHub
parent c24a7c91c3
commit 8abed99b3c
3 changed files with 41 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/router_unit_tests/test_router_helper_utils.py
View file

@ -960,6 +960,18 @@ def test_replace_model_in_jsonl(model_list):
"openai/gpt-3.5-turbo",
"openai/gpt-3.5-turbo",
),
(
"bedrock/meta.llama3-70b",
"*meta.llama3*",
"bedrock/meta.llama3-*",
"bedrock/meta.llama3-70b",
),
(
"meta.llama3-70b",
"*meta.llama3*",
"bedrock/meta.llama3-*",
"meta.llama3-70b",
),
],
)
def test_pattern_match_deployment_set_model_name(
@ -1000,9 +1012,10 @@ def test_pattern_match_deployment_set_model_name(
for model in updated_models:
assert model["litellm_params"]["model"] == expected_model
@pytest.mark.asyncio
async def test_pass_through_moderation_endpoint_factory(model_list):
router = Router(model_list=model_list)
response = await router._pass_through_moderation_endpoint_factory(
original_function=litellm.amoderation, input="this is valid good text"
)
)