From 29ef0504d736e384210fa66d8665d291530140cf Mon Sep 17 00:00:00 2001 From: Yucheng Zhu Date: Thu, 30 Jul 2026 10:21:34 -0700 Subject: [PATCH] test(rate-limits): add direct litellm_metadata lookup coverage The new _stash_value_in_internal_metadata helper routes stash writes to the proxy-internal metadata bucket, which can be litellm_metadata on Responses routes. The lookup logic was updated to read from both channels, but the test only verified the legacy metadata path. Add explicit assertion testing lookup when values live in litellm_params["litellm_metadata"] to close the coverage gap Greptile flagged. Co-Authored-By: Claude Haiku 4.5 --- .../proxy/hooks/test_parallel_request_limiter_v3.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py index ee4eaa4decd..118af8dd654 100644 --- a/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py +++ b/tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py @@ -3247,6 +3247,13 @@ async def test_pre_call_hook_does_not_touch_provider_metadata_on_litellm_metadat key=key, ) == litellm_metadata.get(key) + for key in _LITELLM_STASH_KEYS: + assert handler._lookup_stashed_value( + kwargs={"litellm_params": {"litellm_metadata": litellm_metadata}}, + standard_logging_metadata=None, + key=key, + ) == litellm_metadata.get(key) + @pytest.mark.asyncio async def test_pre_call_hook_rejects_caller_supplied_stash_values():