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

Merge pull request #879 from Kayphoon/feat/gemini-usage-reasoning-badges

Browse Source 
feat(usage): surface Gemini thinkingConfig as reasoning effort
This commit is contained in:
ZheFox
2026-10-05 12:03:02 +08:00
committed by GitHub
parent 976c3feca6 b296d46e97
commit 8ffd8188e4
5 changed files with 359 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/handlers/public/support/user_me_usage.rs
+23
View File
@@ -1926,6 +1926,29 @@ mod tests {
assert_eq!(active["reasoning_effort"], "max");
}
#[test]
fn user_usage_payloads_expose_gemini_thinking_config_reasoning_mapping() {
let item = StoredRequestUsageAudit {
request_body: Some(json!({
"generationConfig": {
"thinkingConfig": { "includeThoughts": true, "thinkingLevel": "HIGH" }
}
})),
provider_request_body: Some(json!({
"generationConfig": { "thinkingConfig": { "thinkingBudget": 8192 } }
})),
..sample_usage("completed")
};
let record = build_users_me_usage_record_payload(&item, false, &BTreeMap::new(), false);
let active = build_users_me_usage_active_payload(&item);
assert_eq!(record["requested_reasoning_effort"], "high");
assert_eq!(active["requested_reasoning_effort"], "high");
assert_eq!(record["reasoning_effort"], "xhigh");
assert_eq!(active["reasoning_effort"], "xhigh");
}
#[test]
fn user_usage_payloads_expose_websocket_transport() {
let item = StoredRequestUsageAudit {