i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-10-12 12:09:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(transport): preserve safe accept encoding

Browse Source
This commit is contained in:
elky
2026-06-10 19:58:57 +08:00
parent ff7ec8575c
commit 0246ba93dd
4 changed files with 296 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-provider-transport/src/standard/mod.rs
+52
View File
@@ -6,6 +6,7 @@ use crate::auth::{
build_claude_passthrough_headers, build_complete_passthrough_headers_with_auth,
build_openai_passthrough_headers, build_passthrough_headers, ensure_upstream_auth_header,
};
use crate::headers::force_identity_accept_encoding;
use crate::rules::{
apply_local_body_rules, apply_local_body_rules_with_request_headers,
apply_local_header_rules_with_request_headers,
@@ -146,6 +147,10 @@ pub fn build_standard_plan_fallback_headers(
}
}
if input.upstream_is_stream {
force_identity_accept_encoding(&mut headers);
}
headers
}
@@ -272,6 +277,7 @@ pub fn build_standard_provider_request_headers(
headers
.entry("accept".to_string())
.or_insert_with(|| "text/event-stream".to_string());
force_identity_accept_encoding(&mut headers);
}
Some(StandardProviderRequestHeaders {
@@ -360,6 +366,10 @@ mod tests {
fn builds_same_format_headers_with_complete_passthrough_and_stream_accept() {
let mut request_headers = HeaderMap::new();
request_headers.insert("x-client", "demo".parse().expect("header"));
request_headers.insert(
http::header::ACCEPT_ENCODING,
"gzip, br".parse().expect("header"),
);
let transport = sample_transport("openai:chat");
let resolved =
build_standard_provider_request_headers(StandardProviderRequestHeadersInput {
@@ -387,9 +397,43 @@ mod tests {
resolved.headers.get("accept"),
Some(&"text/event-stream".to_string())
);
assert_eq!(
resolved.headers.get("accept-encoding"),
Some(&"identity".to_string())
);
assert_eq!(resolved.headers.get("x-client"), Some(&"demo".to_string()));
}
#[test]
fn builds_sync_headers_preserving_supported_accept_encoding() {
let mut request_headers = HeaderMap::new();
request_headers.insert(
http::header::ACCEPT_ENCODING,
"gzip, br".parse().expect("header"),
);
let transport = sample_transport("openai:chat");
let resolved =
build_standard_provider_request_headers(StandardProviderRequestHeadersInput {
transport: &transport,
provider_api_format: "openai:chat",
same_format: true,
headers: &request_headers,
auth_header: "authorization",
auth_value: "Bearer secret",
extra_headers: &BTreeMap::new(),
header_rules: None,
provider_request_body: &json!({"model":"gpt-5"}),
original_request_body: &json!({"model":"gpt-5"}),
upstream_is_stream: false,
})
.expect("headers should build");
assert_eq!(
resolved.headers.get("accept-encoding"),
Some(&"gzip".to_string())
);
}
#[test]
fn applies_header_rules_after_base_headers() {
let transport = sample_transport("claude:messages");
@@ -473,6 +517,10 @@ mod tests {
fn stream_fallback_headers_treat_wildcard_accept_as_absent() {
let mut request_headers = HeaderMap::new();
request_headers.insert(http::header::ACCEPT, "*/*".parse().expect("header"));
request_headers.insert(
http::header::ACCEPT_ENCODING,
"gzip, br".parse().expect("header"),
);
let headers = build_standard_plan_fallback_headers(StandardPlanFallbackHeadersInput {
request_headers: &request_headers,
@@ -492,6 +540,10 @@ mod tests {
headers.get("accept"),
Some(&"text/event-stream".to_string())
);
assert_eq!(
headers.get("accept-encoding"),
Some(&"identity".to_string())
);
}
#[test]