scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(grayswan): normalize the client user agent in the generic body assert

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-30 01:26:24 +00:00
parent 1cfac8e8c1
commit 428cb58892
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/integration/observability/test_grayswan_wire.py
View file

@ -146,7 +146,9 @@ def _chat_provider(message: dict[str, JsonValue]):
def _normalized_generic_body(body: dict[str, JsonValue]) -> dict[str, JsonValue]:
headers: Final = body.get("request_headers")
normalized_headers: Final = (
{**headers, "host": "<host>", "content-length": "<length>"} if isinstance(headers, dict) else headers
{**headers, "host": "<host>", "content-length": "<length>", "user-agent": "<user-agent>"}
if isinstance(headers, dict)
else headers
)
return {
**body,
@ -1167,7 +1169,7 @@ def test_post_call_generic_guardrail_inputs_unchanged(gateway: Gateway, tmp_path
"content-length": "<length>",
"content-type": "application/json",
"host": "<host>",
"user-agent": "python-httpx/0.28.1",
"user-agent": "<user-agent>",
},
"structured_messages": None,
"texts": [response_text],