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

fix(gateway): route openai image streams with images surface

Browse Source
This commit is contained in:
ZheFox
2026-05-20 18:13:34 +08:00
parent de7be4f15b
commit 2cacf66a37
9 changed files with 425 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/handlers/admin/provider/query/models/model_test.rs
+11 -2
View File
@@ -1798,6 +1798,7 @@ fn provider_query_chatgpt_web_image_internal_url(base_url: &str) -> String {
fn provider_query_openai_image_test_upstream_url(
transport: &AdminGatewayProviderTransportSnapshot,
request_path: Option<&str>,
request_query: Option<&str>,
) -> String {
if transport
@@ -1818,7 +1819,11 @@ fn provider_query_openai_image_test_upstream_url(
crate::provider_transport::GROK_CHAT_PATH,
)
} else {
crate::provider_transport::build_openai_image_upstream_url(transport, request_query)
crate::provider_transport::build_openai_image_upstream_url(
transport,
request_path,
request_query,
)
}
}
@@ -2062,7 +2067,11 @@ async fn provider_query_execute_openai_image_test_candidate(
} else {
normalized_request.summary_json.clone()
};
let request_url = provider_query_openai_image_test_upstream_url(&transport, parts.uri.query());
let request_url = provider_query_openai_image_test_upstream_url(
&transport,
Some(parts.uri.path()),
parts.uri.query(),
);
let upstream_is_stream = provider_request_body
.get("stream")
.and_then(Value::as_bool)
apps/aether-gateway/src/handlers/admin/provider/query/models/model_test/tests.rs
+24 -2
View File
@@ -739,17 +739,39 @@ fn provider_query_grok_image_test_uses_grok_app_chat_upstream_url() {
let transport = sample_openai_image_transport("grok");
assert_eq!(
provider_query_openai_image_test_upstream_url(&transport, Some("trace=1")),
provider_query_openai_image_test_upstream_url(
&transport,
Some("/v1/images/generations"),
Some("trace=1"),
),
"https://grok.com/rest/app-chat/conversations/new"
);
}
#[test]
fn provider_query_custom_image_test_uses_images_upstream_url() {
let transport = sample_openai_image_transport("custom");
assert_eq!(
provider_query_openai_image_test_upstream_url(
&transport,
Some("/v1/images/generations"),
Some("trace=1"),
),
"https://grok.com/v1/images/generations?trace=1"
);
}
#[test]
fn provider_query_chatgpt_web_image_test_uses_internal_upstream_url() {
let transport = sample_openai_image_transport("chatgpt_web");
assert_eq!(
provider_query_openai_image_test_upstream_url(&transport, Some("trace=1")),
provider_query_openai_image_test_upstream_url(
&transport,
Some("/v1/images/generations"),
Some("trace=1"),
),
"https://grok.com/__aether/chatgpt-web-image"
);
}