fix(proxy): forward extra_headers in chat

This commit is contained in:
naaa760 2026-02-04 08:56:50 +05:30
parent f11c16a0e7
commit 0cb6b58768
2 changed files with 25 additions and 0 deletions

View File

@ -329,6 +329,9 @@ class LiteLLMResponsesTransformationHandler(CompletionTransformationBridge):
else:
request_data[key] = value
if headers:
request_data["extra_headers"] = headers
return request_data
@staticmethod

View File

@ -134,3 +134,25 @@ def test_transform_request_with_response_format():
assert result["text"]["format"]["type"] == "json_schema"
assert result["text"]["format"]["name"] == "person_schema"
assert "schema" in result["text"]["format"]
def test_transform_request_includes_extra_headers():
"""Test that transform_request forwards headers as extra_headers for upstream call."""
handler = LiteLLMResponsesTransformationHandler()
messages = [{"role": "user", "content": "Hello"}]
optional_params = {}
litellm_params = {}
class MockLoggingObj:
pass
headers = {"cf-aig-authorization": "secret-token"}
result = handler.transform_request(
model="gpt-5-pro",
messages=messages,
optional_params=optional_params,
litellm_params=litellm_params,
headers=headers,
litellm_logging_obj=MockLoggingObj(),
)
assert result.get("extra_headers") == headers