chore: ruff format

This commit is contained in:
Chenglun Hu 2026-06-05 20:27:33 +08:00
parent 1c17fec9dd
commit 452ff2a030

View file

@ -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()