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

test: stabilize merged PR checks

Browse Source
This commit is contained in:
fawney19
2026-05-19 10:18:25 +08:00
parent d2b42e91d2
commit 052de6b96e
25 changed files with 726 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -680,16 +680,15 @@ impl OpenAiImageStreamTerminalState {
fn observe_image_completed(&mut self, event: &Value) {
self.observed_finish = true;
if self.image_count == 0 {
if event
if self.image_count == 0
&& event
.get("b64_json")
.or_else(|| event.get("result"))
.and_then(Value::as_str)
.map(str::trim)
.is_some_and(|value| !value.is_empty())
{
self.image_count = 1;
}
{
self.image_count = 1;
}
self.usage = event.get("usage").cloned().or_else(|| self.usage.clone());
}
@@ -798,7 +797,7 @@ fn openai_image_stream_standardized_usage(
) -> Option<StandardizedUsage> {
let mut standardized_usage = usage
.and_then(openai_image_usage_to_standardized_usage)
.unwrap_or_else(StandardizedUsage::new);
.unwrap_or_default();
if image_count > 0 {
standardized_usage.request_count = i64::try_from(image_count).unwrap_or(i64::MAX);
standardized_usage

2
crates/aether-ai-formats/src/formats/shared/sync_to_stream.rs
View File

@@ -454,7 +454,7 @@ fn openai_image_standardized_usage(
) -> Option<StandardizedUsage> {
let mut standardized_usage = usage
.and_then(standardized_usage_from_openai_usage)
.unwrap_or_else(StandardizedUsage::new);
.unwrap_or_default();
if image_count > 0 {
standardized_usage.request_count = i64::try_from(image_count).unwrap_or(i64::MAX);
standardized_usage