fix(service): 修复扣费结果类型处理

- service/agent/billing.rs: 适配新的 BillingResult 枚举类型
- 将 InsufficientBalance 错误转换为 AppError::BadRequest
This commit is contained in:
ZhenYi 2026-04-28 19:59:17 +08:00
parent c6bb72682b
commit 4571d4d042

View File

@ -12,13 +12,21 @@ impl AppService {
input_tokens: i64, input_tokens: i64,
output_tokens: i64, output_tokens: i64,
) -> Result<agent::billing::BillingRecord, AppError> { ) -> Result<agent::billing::BillingRecord, AppError> {
Ok(agent::billing::record_ai_usage( use agent::billing::BillingResult;
match agent::billing::record_ai_usage(
&self.db, &self.db,
project_uid, project_uid,
model_id, model_id,
input_tokens, input_tokens,
output_tokens, output_tokens,
) )
.await?) .await?
{
BillingResult::Success(record) => Ok(record),
BillingResult::InsufficientBalance { message } => {
Err(AppError::BadRequest(message))
}
}
} }
} }