scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

test(spend_tracking): cover unresolvable deployment leaving provider empty

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-16 23:50:22 +00:00
parent 9af014d75a
commit 5f1d87911a
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py
View file

@ -4078,6 +4078,24 @@ def test_get_logging_payload_inferred_provider_never_resolves_declared_authentic
assert resolution_attempts == []
def test_get_logging_payload_router_rejected_request_for_unresolvable_deployment_leaves_provider_empty(
monkeypatch,
):
def _router_init_stub(model, custom_llm_provider=None, *args, **kwargs):
return model, custom_llm_provider or "openai", None, None
with monkeypatch.context() as router_init:
router_init.setattr(litellm, "get_llm_provider", _router_init_stub)
llm_router = litellm.Router(
model_list=[{"model_name": "opaque-group", "litellm_params": {"model": "my-unprefixed-model"}}]
)
payload = _router_rejected_failure_payload("opaque-group", llm_router)
assert payload["model_group"] == "opaque-group"
assert payload["custom_llm_provider"] == ""
def test_get_logging_payload_inferred_provider_does_not_rewrite_spend_log_model():
llm_router = litellm.Router(
model_list=[