mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-24 05:45:11 +00:00
chore: ruff format
This commit is contained in:
parent
1c17fec9dd
commit
452ff2a030
1 changed files with 8 additions and 24 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue