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))
+            }
+        }
     }
 }