From 452ff2a03069e04a9a49b11df92c39122ae1f0fb Mon Sep 17 00:00:00 2001 From: Chenglun Hu Date: Fri, 5 Jun 2026 20:27:33 +0800 Subject: [PATCH] chore: ruff format --- .../common_utils/test_http_parsing_utils.py | 32 +++++-------------- 1 file changed, 8 insertions(+), 24 deletions(-) diff --git a/tests/test_litellm/proxy/common_utils/test_http_parsing_utils.py b/tests/test_litellm/proxy/common_utils/test_http_parsing_utils.py index efcd55b0b8a..ca9871ce0fd 100644 --- a/tests/test_litellm/proxy/common_utils/test_http_parsing_utils.py +++ b/tests/test_litellm/proxy/common_utils/test_http_parsing_utils.py @@ -8,9 +8,7 @@ import pytest from fastapi import Request from fastapi.testclient import TestClient -sys.path.insert( - 0, os.path.abspath("../../../..") -) # Adds the parent directory to the system path +sys.path.insert(0, os.path.abspath("../../../..")) # Adds the parent directory to the system path import litellm @@ -356,9 +354,7 @@ async def test_circular_reference_handling(): # Second parse using the same request - will use the modified cached value result2 = await _read_request_body(mock_request) - assert ( - "proxy_server_request" not in result2 - ) # This will pass, showing the cache pollution + assert "proxy_server_request" not in result2 # This will pass, showing the cache pollution @pytest.mark.asyncio @@ -469,9 +465,7 @@ async def test_surrogate_repair_skipped_above_size_limit(monkeypatch): import litellm.proxy.common_utils.http_parsing_utils as http_parsing_utils # Cap the repair at ~100 bytes so the test stays fast and independent of the default. - monkeypatch.setattr( - http_parsing_utils, "MAX_REQUEST_BODY_SIZE_TO_REPAIR_MB", 100 / (1024 * 1024) - ) + monkeypatch.setattr(http_parsing_utils, "MAX_REQUEST_BODY_SIZE_TO_REPAIR_MB", 100 / (1024 * 1024)) small_body = b'{"model":"gpt-4o","x":"\\ud83d"}' assert len(small_body) <= 100 @@ -479,9 +473,7 @@ async def test_surrogate_repair_skipped_above_size_limit(monkeypatch): assert repaired["model"] == "gpt-4o" padding = "a" * 200 - large_body = ( - b'{"model":"gpt-4o","pad":"' + padding.encode() + b'","x":"\\ud83d"}' - ) + large_body = b'{"model":"gpt-4o","pad":"' + padding.encode() + b'","x":"\\ud83d"}' assert len(large_body) > 100 with pytest.raises(ProxyException) as exc_info: await _read_request_body(_make_json_request(large_body)) @@ -490,9 +482,7 @@ async def test_surrogate_repair_skipped_above_size_limit(monkeypatch): # Disabling the cap (0) restores repair for the same large body, proving the cap # — not the malformed content — is what short-circuits the repair. - monkeypatch.setattr( - http_parsing_utils, "MAX_REQUEST_BODY_SIZE_TO_REPAIR_MB", 0 - ) + monkeypatch.setattr(http_parsing_utils, "MAX_REQUEST_BODY_SIZE_TO_REPAIR_MB", 0) repaired_large = await _read_request_body(_make_json_request(large_body)) assert repaired_large["model"] == "gpt-4o" @@ -796,9 +786,7 @@ def test_populate_request_with_path_params_does_not_overwrite_existing_values(): # Verify existing values were NOT overwritten assert result["model"] == "gpt-4" # Should keep original, not "gpt-3.5-turbo" - assert ( - result["organization_id"] == "org-existing" - ) # Should keep original, not "org-query-param" + assert result["organization_id"] == "org-existing" # Should keep original, not "org-query-param" # Verify other data is preserved assert result["messages"] == [{"role": "user", "content": "Hello"}] @@ -966,9 +954,7 @@ class TestGetTagsFromRequestBodyStringCoerce: ) # Must not raise; must yield no metadata tags but keep root tags - tags = get_tags_from_request_body( - {"metadata": "not-json", "tags": ["root-only"]} - ) + tags = get_tags_from_request_body({"metadata": "not-json", "tags": ["root-only"]}) assert tags == ["root-only"] def test_dict_metadata_still_works(self): @@ -1025,9 +1011,7 @@ class TestReadRequestBodyNonCanonicalContentType: "multiform/anything", ], ) - async def test_json_body_with_formlike_content_type_parses_as_json( - self, content_type - ): + async def test_json_body_with_formlike_content_type_parses_as_json(self, content_type): payload = {"user_config": {"model_list": []}, "model": "x"} mock_request = MagicMock()