From 763d47dc4565a95bce219e09ad66b980f4291b87 Mon Sep 17 00:00:00 2001
From: ZhenYi <434836402@qq.com>
Date: Mon, 27 Apr 2026 11:15:15 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20silent=20AI=20billing=20failures=20?=
 =?UTF-8?q?=E2=80=94=20add=20tracing::warn=20for=20billing=20errors?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 libs/room/src/service/ai_react_streaming.rs | 7 +++++--
 libs/room/src/service/ai_streaming.rs       | 7 +++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

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,