ZhenYi/gitdataai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: silent AI billing failures — add tracing::warn for billing errors

Browse Source
This commit is contained in:
ZhenYi 2026-04-27 11:15:15 +08:00
parent 1e975c0837
commit 763d47dc45
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/room/src/service/ai_react_streaming.rs
View File

@ -224,14 +224,17 @@ pub async fn process_message_ai_react_streaming(
// Record billing (non-fatal)
// TODO: ReAct agent does not track token counts yet; billing with 0/0
let _ = super::billing::record_ai_usage(
if let Err(e) = super::billing::record_ai_usage(
&db,
project_id_inner,
model_id,
0,
0,
)
.await;
.await
{
tracing::warn!(error = %e, "AI billing recording failed");
}
let msg_event = queue::RoomMessageEvent {
id: streaming_msg_id,

7
libs/room/src/service/ai_streaming.rs
View File

@ -188,14 +188,17 @@ pub async fn process_message_ai_streaming(
}
// Record billing (non-fatal)
let _ = super::billing::record_ai_usage(
if let Err(e) = super::billing::record_ai_usage(
&db,
project_id_inner,
model_id,
result.input_tokens,
result.output_tokens,
)
.await;
.await
{
tracing::warn!(error = %e, "AI billing recording failed");
}
let msg_event = queue::RoomMessageEvent {
id: streaming_msg_id,