diff --git a/tests/integration/authorization/test_tool_search_replay_guardrails.py b/tests/integration/authorization/test_tool_search_replay_guardrails.py index 14f647ecf94..2f0ca383201 100644 --- a/tests/integration/authorization/test_tool_search_replay_guardrails.py +++ b/tests/integration/authorization/test_tool_search_replay_guardrails.py @@ -67,7 +67,12 @@ def _config(directory: Path, model: str, upstream: str, shell: str) -> Path: "router_settings": {"disable_cooldowns": True}, "litellm_settings": { "cache": True, - "cache_params": {"type": "redis", "host": "os.environ/REDIS_HOST", "port": "os.environ/REDIS_PORT"}, + "cache_params": { + "type": "redis", + "host": "os.environ/REDIS_HOST", + "port": "os.environ/REDIS_PORT", + "max_messages": None, + }, }, "guardrails": [ {