scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Address Perplexity attribution test review

Browse source
This commit is contained in:
Saleh Alghusson 2026-08-28 12:15:32 -04:00
parent 9dbb06ca29
commit b4419f2a21
No known key found for this signature in database
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/llms/perplexity/chat/test_perplexity_chat_transformation.py
View file

@ -21,15 +21,21 @@ class TestPerplexityChatTransformation:
("headers", "expected"),
[({}, "litellm"), ({"x-pplx-integration": "custom"}, "custom")],
)
def test_integration_header_default_is_caller_overridable(self, headers, expected):
headers = PerplexityChatConfig().validate_environment(
def test_integration_header_default_is_caller_overridable(
self, headers: dict[str, str], expected: str
):
validated_headers = PerplexityChatConfig().validate_environment(
headers=headers,
model="sonar",
messages=[],
optional_params={},
litellm_params={},
)
request = httpx.Request("POST", "https://api.perplexity.ai/chat/completions", headers=headers)
request = httpx.Request(
"POST",
"https://api.perplexity.ai/chat/completions",
headers=validated_headers,
)
assert request.headers.get_list("x-pplx-integration") == [expected]