starred/codex-proxy-rs
1
0
Fork 0
mirror of https://github.com/zyycn/codex-proxy-rs.git synced 2026-11-06 16:06:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: simplify ops error log snapshot helper

Browse Source 
Co-authored-by: Codex <noreply@openai.com>
This commit is contained in:
Zyy
2026-07-08 14:55:05 +08:00
co-authored by Codex
parent 146776c742
commit 5682d4fbe9
1 changed files with 29 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/proxy/dispatch/chat_upstream/mod.rs
+29 -21
View File
@@ -1443,20 +1443,38 @@ struct ResponseUsageRecordSnapshot {
metadata_json: String,
}
#[derive(sqlx::FromRow)]
struct OpsErrorLogSnapshot {
request_id: Option<String>,
account_id: Option<String>,
route: Option<String>,
model: Option<String>,
status_code: Option<i64>,
response_id: Option<String>,
metadata_json: String,
}
impl From<OpsErrorLogSnapshot> for ResponseUsageRecordSnapshot {
fn from(snapshot: OpsErrorLogSnapshot) -> Self {
Self {
level: "error".to_string(),
request_id: snapshot.request_id,
account_id: snapshot.account_id,
route: snapshot.route,
model: snapshot.model,
status_code: snapshot.status_code,
response_id: snapshot.response_id,
metadata_json: snapshot.metadata_json,
}
}
}
async fn latest_response_usage_record(pool: &SqlitePool) -> ResponseUsageRecordSnapshot {
latest_usage_record(pool, "v1.response").await
}
async fn latest_response_ops_error_log(pool: &SqlitePool) -> ResponseUsageRecordSnapshot {
let row: Option<(
Option<String>,
Option<String>,
Option<String>,
Option<String>,
Option<i64>,
Option<String>,
String,
)> = sqlx::query_as(
let row = sqlx::query_as::<_, OpsErrorLogSnapshot>(
"select request_id, account_id, route, model, status_code, response_id, metadata_json
from ops_error_logs
where kind = ?
@@ -1467,18 +1485,8 @@ async fn latest_response_ops_error_log(pool: &SqlitePool) -> ResponseUsageRecord
.fetch_optional(pool)
.await
.unwrap();
let (request_id, account_id, route, model, status_code, response_id, metadata_json) =
row.unwrap_or_else(|| panic!("expected a v1.response ops error log"));
ResponseUsageRecordSnapshot {
level: "error".to_string(),
request_id,
account_id,
route,
model,
status_code,
response_id,
metadata_json,
}
row.unwrap_or_else(|| panic!("expected a v1.response ops error log"))
.into()
}
async fn latest_usage_record(pool: &SqlitePool, kind: &str) -> ResponseUsageRecordSnapshot {