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

fix(gateway): preserve openai image 200 responses and sync success reporting

Browse Source
This commit is contained in:
ZheFox
2026-05-20 19:17:42 +08:00
parent 2cacf66a37
commit cfa32a2b4d
2 changed files with 122 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
crates/aether-ai-formats/src/formats/openai/image/stream.rs
View File

@@ -1146,6 +1146,12 @@ pub fn maybe_build_openai_image_sync_finalize_product(
return Ok(None);
}
if let Some(provider_body_json) = body_json {
if openai_image_response_has_standard_data(provider_body_json) {
return Ok(Some(OpenAiImageSyncFinalizeProduct {
client_body_json: provider_body_json.clone(),
provider_body_json: provider_body_json.clone(),
}));
}
if provider_body_json.get("output").is_some() && provider_body_json.get("data").is_none() {
let Some(client_body_json) = crate::formats::shared::image_bridge::build_openai_image_response_from_response_stream_sync_body(
provider_body_json,
@@ -1279,6 +1285,25 @@ pub fn maybe_build_openai_image_sync_finalize_product(
}))
}
fn openai_image_response_has_standard_data(body_json: &Value) -> bool {
body_json
.get("data")
.and_then(Value::as_array)
.is_some_and(|items| {
items.iter().any(|item| {
item.as_object().is_some_and(|object| {
["b64_json", "url"].iter().any(|field| {
object
.get(*field)
.and_then(Value::as_str)
.map(str::trim)
.is_some_and(|value| !value.is_empty())
})
})
})
})
}
#[cfg(test)]
mod tests {
use base64::Engine as _;
@@ -1487,4 +1512,35 @@ mod tests {
"revised history prompt"
);
}
#[test]
fn sync_finalize_accepts_standard_openai_image_response() {
let provider_body = json!({
"created": 1779273523,
"data": [{
"b64_json": "aGVsbG8=",
"revised_prompt": "draw a small cat"
}]
});
let product = maybe_build_openai_image_sync_finalize_product(
"openai_image_sync_finalize",
200,
Some(&json!({
"client_api_format": "openai:image",
"provider_api_format": "openai:image",
"image_request": {
"operation": "generate",
"response_format": "b64_json"
}
})),
Some(&provider_body),
None,
)
.expect("standard image response should finalize")
.expect("standard image response should match");
assert_eq!(product.client_body_json["created"], 1779273523);
assert_eq!(product.client_body_json["data"][0]["b64_json"], "aGVsbG8=");
assert_eq!(product.provider_body_json, provider_body);
}
}