fix(proxy): forward extra_headers in chat
This commit is contained in:
parent
f11c16a0e7
commit
0cb6b58768
@ -329,6 +329,9 @@ class LiteLLMResponsesTransformationHandler(CompletionTransformationBridge):
|
||||
else:
|
||||
request_data[key] = value
|
||||
|
||||
if headers:
|
||||
request_data["extra_headers"] = headers
|
||||
|
||||
return request_data
|
||||
|
||||
@staticmethod
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user