litellm/tests/test_litellm/proxy/hooks
Krrish Dholakia 84874012ad feat(proxy): add max_iterations limiter for agent session loops
Adds a new proxy hook that enforces a per-session cap on the number of
LLM calls an agentic loop can make. Callers send a session_id with each
request, and the hook counts calls per session, returning 429 when the
configured max_iterations limit is exceeded.

- Uses Redis Lua script for atomic increment (multi-instance safe)
- Falls back to in-memory cache when Redis unavailable
- Follows parallel_request_limiter_v3 pattern
- Configurable via key metadata: {"max_iterations": 25}
- Session counters auto-expire via TTL (default 1hr)

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-24 19:37:21 -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_max_iterations_limiter.py feat(proxy): add max_iterations limiter for agent session loops 2026-02-24 19:37:21 -08: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 errors when callbacks are invoked for file operations 2026-02-13 23:30:22 -05:00
test_send_invite_email.py Fix User Invite & Key Generation Email Notification Logic (#18524) 2026-01-06 01:35:52 +05:30