litellm/tests/test_litellm/proxy/hooks
yuneng-jiang f4e23c97fd [Fix] Enrich failure spend logs with key/team metadata
Failure spend logs were missing key metadata (key alias, user ID, team ID,
team alias) in two scenarios:

1. Auth errors (401 ProxyException): auth_exception_handler creates a
   minimal UserAPIKeyAuth with only api_key and request_route set — all
   other fields are null. The failure hook now looks up the full key object
   from cache/DB using the key hash to populate the missing fields.

2. Post-auth failures (provider errors, rate limits): key fields are
   present but team_alias is always null because LiteLLM_VerificationTokenView
   SQL view does not include team_alias. The failure hook now looks up the
   team object from cache to populate team_alias.

Both lookups are non-fatal and wrapped in try/except.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-02-24 17:14:37 -08:00
..
test_async_post_call_streaming_iterator_hook.py fix(proxy): async_post_call_streaming_iterator_hook now properly iterates async generators (#17626) 2025-12-07 23:29:42 -08:00
test_dynamic_rate_limiter_v3.py Merge pull request #17707 from raghav-stripe/raghav-fix-responsesapi-rl 2025-12-11 08:57:16 +05:30
test_image_generation_guardrails.py added tests 2026-02-11 15:45:47 +05:30
test_key_management_event_hooks.py feat: hashicorp vault rotate support 2026-01-23 17:32:55 +09:00
test_parallel_request_limiter_v3.py fix(tests): isolate flaky tests - restore global state in setup/teardown (#21791) 2026-02-21 11:30:35 -08:00
test_post_call_failure_hook_integration.py [Feat] Enable async_post_call_failure_hook to transform error responses (#18348) 2025-12-22 11:24:30 -08:00
test_post_call_response_headers_hook.py [Feat] Add async_post_call_response_headers_hook to CustomLogger (#20083) 2026-01-30 12:44:44 -08:00
test_post_call_streaming_hook_integration.py fix(proxy): use return value from CustomLogger.async_post_call_success_hook (#19670) 2026-01-26 08:48:22 -08:00
test_post_call_success_hook_integration.py fix(proxy): use return value from CustomLogger.async_post_call_success_hook (#19670) 2026-01-26 08:48:22 -08:00
test_prompt_injection_detection.py fix: prompt injection not working (#16701) 2025-11-17 20:04:57 -08:00
test_proxy_track_cost_callback.py [Fix] Enrich failure spend logs with key/team metadata 2026-02-24 17:14:37 -08:00
test_send_invite_email.py Fix User Invite & Key Generation Email Notification Logic (#18524) 2026-01-06 01:35:52 +05:30