diff --git a/libs/room/src/service/ai_react_streaming.rs b/libs/room/src/service/ai_react_streaming.rs index ed12c76..fd2d783 100644 --- a/libs/room/src/service/ai_react_streaming.rs +++ b/libs/room/src/service/ai_react_streaming.rs @@ -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, diff --git a/libs/room/src/service/ai_streaming.rs b/libs/room/src/service/ai_streaming.rs index e1254d5..ddf22c1 100644 --- a/libs/room/src/service/ai_streaming.rs +++ b/libs/room/src/service/ai_streaming.rs @@ -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,