From 4571d4d04256f3f7257b4c4ae1a8348b7d5ae744 Mon Sep 17 00:00:00 2001
From: ZhenYi <434836402@qq.com>
Date: Tue, 28 Apr 2026 19:59:17 +0800
Subject: [PATCH] =?UTF-8?q?fix(service):=20=E4=BF=AE=E5=A4=8D=E6=89=A3?=
 =?UTF-8?q?=E8=B4=B9=E7=BB=93=E6=9E=9C=E7=B1=BB=E5=9E=8B=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- service/agent/billing.rs: 适配新的 BillingResult 枚举类型
- 将 InsufficientBalance 错误转换为 AppError::BadRequest
---
 libs/service/agent/billing.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/libs/service/agent/billing.rs b/libs/service/agent/billing.rs
index ce55850..f1a9d24 100644
--- a/libs/service/agent/billing.rs
+++ b/libs/service/agent/billing.rs
@@ -12,13 +12,21 @@ impl AppService {
         input_tokens: i64,
         output_tokens: i64,
     ) -> Result<agent::billing::BillingRecord, AppError> {
-        Ok(agent::billing::record_ai_usage(
+        use agent::billing::BillingResult;
+
+        match agent::billing::record_ai_usage(
             &self.db,
             project_uid,
             model_id,
             input_tokens,
             output_tokens,
         )
-        .await?)
+        .await?
+        {
+            BillingResult::Success(record) => Ok(record),
+            BillingResult::InsufficientBalance { message } => {
+                Err(AppError::BadRequest(message))
+            }
+        }
     }
 }