mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-10 22:41:41 +00:00
b8df48cd7f
* feat(auto-router): let operators replace the LLM classifier's system prompt The complexity router's LLM classifier has always sent one built-in rubric, so the router could only ever grade difficulty. Operators can now supply their own system prompt, which replaces the rubric outright and repurposes the same tier machinery for whatever taxonomy the prompt defines, data sensitivity being the obvious case. Replacement is total: neither the rubric nor its closing line is appended, since both describe grading difficulty over a "current message" and a prompt grading something else is entitled to contradict them. That closing paragraph is also the classifier's prompt-injection defense, so the config field and the dashboard editor both warn that a replacement omitting it lets a caller ask for a tier and get it. The heuristic fallback still scores complexity, which is meaningless for a repurposed taxonomy, so classifier_fallback now chooses between the heuristic scorer and routing straight to default_model. The default_model path bypasses tier pools, the adaptive bandit, and escalation, because no tier was decided and the point of that fallback is a known destination. It reports itself as default_model_fallback in the spend logs. The dashboard's prompt editor prefills from a new /auto_router/classifier/default_prompt endpoint rather than a copy of the rubric in the frontend, and stores no override when the draft matches the default, so later rubric improvements still reach every router that never customized it. Tier names stay SIMPLE/MEDIUM/COMPLEX/REASONING; a custom prompt redefines what they mean, not what they are called. * fix(complexity-router): don't let the default_model classifier fallback bypass routing plugins * fix(complexity-router): don't pin a session to the default model after a classifier failure * fix(complexity-router): omit the tier from a default-model-fallback routing decision The classifier never answered, so no tier was decided. The record reported the tier whose pool happens to hold default_model, which reads in the spend log and the UI as if the request was classified. Matches how default_fallback already records a route that no tier produced. * fix(proxy): allowlist /auto_router/ on the UI backend component The new GET /auto_router/classifier/default_prompt is a UI-consumed management route, so it belongs on the control plane. Without the prefix it was exposed by neither component and test_gateway_plus_backend_covers_full_app failed. * docs(ui): reword the classifier prompt disclaimer Frames the closing paragraph as a strong recommendation rather than a description of what gets dropped, names prompt injection explicitly, and notes the tier names stay fixed regardless of their display names. * fix(complexity-router): stop logging a fabricated tier on the plugin fallback path The classifier-failed fallback resolves a tier so the routing-plugin pipeline has a pool to filter, but nothing about the request produced that tier. The non-plugin short-circuit already dropped it from the logged decision; the plugin path still reported it, so a spend log claimed a classification the request never received. Record the pool as a plugin-filtered-pool signal instead. Also name the real problem when the resolved tier has no models at all: that raised "No candidate models left after routing-plugin filtering" and sent operators hunting for a policy plugin that never narrowed anything. |
||
|---|---|---|
| .. | ||
| agent_tests | ||
| audio_tests | ||
| base_sdk_tests | ||
| basic_proxy_startup_tests | ||
| batches_tests | ||
| benchmarks | ||
| code_coverage_tests | ||
| documentation_tests | ||
| e2e | ||
| enterprise | ||
| guardrails_tests | ||
| image_gen_tests | ||
| integration | ||
| litellm | ||
| litellm-proxy-extras | ||
| litellm_core_utils | ||
| litellm_utils_tests | ||
| llm_responses_api_testing | ||
| llm_translation | ||
| load_tests | ||
| local_testing | ||
| logging_callback_tests | ||
| mcp_tests | ||
| multi_instance_e2e_tests | ||
| ocr_tests | ||
| old_proxy_tests/tests | ||
| openai_endpoints_tests | ||
| otel_tests | ||
| pass_through_tests | ||
| pass_through_unit_tests | ||
| proxy_admin_ui_tests | ||
| proxy_behavior | ||
| proxy_e2e_anthropic_messages_tests | ||
| proxy_migration_tests | ||
| proxy_security_tests | ||
| proxy_unit_tests | ||
| router_unit_tests | ||
| scim_tests | ||
| search_tests | ||
| spend_tracking_tests | ||
| store_model_in_db_tests | ||
| test_litellm | ||
| unified_google_tests | ||
| vector_store_tests | ||
| windows_tests | ||
| __init__.py | ||
| _fake_openai_endpoint_server.py | ||
| _flush_vcr_cache.py | ||
| _live_test_helpers.py | ||
| _openai_record_replay_proxy.py | ||
| _vcr_conftest_common.py | ||
| _vcr_redis_persister.py | ||
| _ws_vcr.py | ||
| eval_swe_bench.py | ||
| fake_openai_endpoint.py | ||
| gettysburg.wav | ||
| large_text.py | ||
| openai_batch_completions.jsonl | ||
| pyrightconfig.json | ||
| README.MD | ||
| test_anthropic_compaction_usage.py | ||
| test_budget_management.py | ||
| test_callbacks_on_proxy.py | ||
| test_config.py | ||
| test_debug_warning.py | ||
| test_default_encoding_non_root.py | ||
| test_end_users.py | ||
| test_entrypoint.py | ||
| test_fallbacks.py | ||
| test_gpt5_azure_temperature_support.py | ||
| test_health.py | ||
| test_keys.py | ||
| test_litellm_proxy_responses_config.py | ||
| test_logging.conf | ||
| test_models.py | ||
| test_new_vector_store_endpoints.py | ||
| test_openai_endpoints.py | ||
| test_organizations.py | ||
| test_otel_thread_leak.py | ||
| test_passthrough_endpoints.py | ||
| test_presidio_latency.py | ||
| test_proxy_server_non_root.py | ||
| test_ratelimit.py | ||
| test_resource_cleanup.py | ||
| test_service_logger_otel.py | ||
| test_spend_logs.py | ||
| test_team.py | ||
| test_team_logging.py | ||
| test_team_members.py | ||
| test_users.py | ||
In total litellm runs 1000+ tests
[02/20/2025] Update:
To make it easier to contribute and map what behavior is tested,
we've started mapping the litellm directory in tests/test_litellm
This folder can only run mock tests.