scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-15 09:12:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 311

test(rate_limiter): drop explanatory docstrings from v3 dict response tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-09 21:20:12 +00:00
parent cb5d40deb3
commit fbdd5d7be6
2 changed files with 0 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/proxy/hooks/test_dynamic_rate_limiter_v3.py
View file

@ -1865,11 +1865,6 @@ async def test_tpm_only_model_enforces_priority_and_model_capacity(monkeypatch):
@pytest.mark.asyncio
async def test_post_call_success_hook_attaches_priority_headers_to_dict_response():
"""
Non-streaming /v1/messages hands the success hook a plain dict, not a ModelResponse.
The router has already stamped ``_hidden_params`` on it, so both the v3 descriptor
headers and the priority headers must land next to the router's headers there.
"""
from litellm.proxy.hooks.parallel_request_limiter_v3 import (
RateLimitResponse,
RateLimitStatus,
@ -1913,10 +1908,6 @@ async def test_post_call_success_hook_attaches_priority_headers_to_dict_response
@pytest.mark.asyncio
async def test_post_call_success_hook_leaves_raw_provider_dict_untouched():
"""
Passthrough routes hand the hook the raw provider JSON body and serialize whatever
comes back, so a dict without ``_hidden_params`` must not have one injected.
"""
handler = DynamicRateLimitHandler(internal_usage_cache=DualCache())
response = {"id": "msg_123", "type": "message", "role": "assistant", "content": []}

9
tests/test_litellm/proxy/hooks/test_parallel_request_limiter_v3.py
View file

@ -6175,11 +6175,6 @@ async def test_success_hook_leaves_stash_untouched_for_non_batch_responses():
@pytest.mark.asyncio
async def test_post_call_success_hook_attaches_ratelimit_headers_to_dict_response():
"""
Non-streaming /v1/messages responses are plain dicts carrying the router's
``_hidden_params`` as a key, not an attribute, so the descriptor headers used to
be dropped for that route.
"""
from litellm.proxy.hooks.parallel_request_limiter_v3 import RateLimitResponse, RateLimitStatus
handler = _PROXY_MaxParallelRequestsHandler(internal_usage_cache=InternalUsageCache(DualCache()))
@ -6217,10 +6212,6 @@ async def test_post_call_success_hook_attaches_ratelimit_headers_to_dict_respons
@pytest.mark.asyncio
async def test_post_call_success_hook_leaves_raw_provider_dict_untouched():
"""
Passthrough routes hand the hook the raw provider JSON body and serialize whatever
comes back, so a dict without ``_hidden_params`` must not have one injected.
"""
from litellm.proxy.hooks.parallel_request_limiter_v3 import RateLimitResponse, RateLimitStatus
handler = _PROXY_MaxParallelRequestsHandler(internal_usage_cache=InternalUsageCache(DualCache()))