ZhenYi/gitdataai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(billing): remove old billing module, delegate to record_ai_session

Browse Source 
Billing is now handled internally by chat_service.process via record_ai_session.
Remove the old billing.rs file and explicit record_ai_usage calls from all 4
AI streaming modes (nonstreaming, react_nonstreaming, react_streaming, streaming).
This commit is contained in:
ZhenYi 2026-04-28 13:12:25 +08:00
parent 2a9ec6d509
commit bbeaea6614
5 changed files with 5 additions and 99 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/room/src/service/ai_nonstreaming.rs
View File

@ -62,15 +62,7 @@ pub async fn process_message_ai_nonstreaming(
tracing::warn!(error = %e, "Failed to update room_ai call stats");
}
// Record billing (non-fatal)
let _ = super::billing::record_ai_usage(
&db,
project_id,
model_id,
result.input_tokens,
result.output_tokens,
)
.await;
// Billing handled internally by chat_service.process via record_ai_session
}
}
Err(e) => {

14
libs/room/src/service/ai_react_nonstreaming.rs
View File

@ -50,6 +50,7 @@ pub async fn process_message_ai_react_nonstreaming(
{
tracing::error!(error = %e, "Failed to create ReAct AI message");
} else {
// Billing handled internally by chat_service.process_react via record_ai_session
let now = Utc::now();
if let Err(e) = room_ai::Entity::update_many()
.col_expr(
@ -64,19 +65,6 @@ pub async fn process_message_ai_react_nonstreaming(
{
tracing::warn!(error = %e, "Failed to update room_ai call stats");
}
// Record billing (non-fatal)
if let Err(e) = super::billing::record_ai_usage(
&db,
project_id,
model_id,
_input_tokens,
_output_tokens,
)
.await
{
tracing::warn!(error = %e, "AI billing recording failed");
}
}
}
Err(e) => {

16
libs/room/src/service/ai_react_streaming.rs
View File

@ -155,7 +155,7 @@ pub async fn process_message_ai_react_streaming(
})
.await;
let (final_content, input_tokens, output_tokens, err_msg, _should_log) = match result {
let (final_content, _input_tokens, _output_tokens, err_msg, _should_log) = match result {
Ok((content, input, output)) => (content, input, output, None, false),
Err(e) => {
let msg = format!("[Agent error: {}]", e);
@ -250,19 +250,7 @@ pub async fn process_message_ai_react_streaming(
tracing::warn!(error = %e, "Failed to update room_ai call stats");
}
// Record billing (non-fatal)
if let Err(e) = super::billing::record_ai_usage(
&db,
project_id_inner,
model_id,
input_tokens,
output_tokens,
)
.await
{
tracing::warn!(error = %e, "AI billing recording failed");
}
// Billing handled internally by chat_service.process_react via record_ai_session
let msg_event = queue::RoomMessageEvent {
id: streaming_msg_id,
room_id: room_id_inner,

13
libs/room/src/service/ai_streaming.rs
View File

@ -190,18 +190,7 @@ pub async fn process_message_ai_streaming(
tracing::warn!(error = %e, "Failed to update room_ai call stats");
}
// Record billing (non-fatal)
if let Err(e) = super::billing::record_ai_usage(
&db,
project_id_inner,
model_id,
result.input_tokens,
result.output_tokens,
)
.await
{
tracing::warn!(error = %e, "AI billing recording failed");
}
// Billing handled internally by chat_service.process_stream via record_ai_session
let msg_event = queue::RoomMessageEvent {
id: streaming_msg_id,

51
libs/room/src/service/billing.rs
View File

@ -1,51 +0,0 @@
//! AI usage billing helper for room service.
//!
//! Delegates to `agent::billing::record_ai_usage`.
//! Billing is non-fatal — failures are logged but do not block AI responses.
use db::database::AppDatabase;
use uuid::Uuid;
use crate::error::RoomError;
/// Record AI token usage against a project's billing balance.
///
/// Returns `Ok(())` on success. On billing failure (e.g. insufficient balance,
/// missing pricing), returns `Err` but the caller should still complete the AI
/// request — billing is a non-critical side-effect.
pub async fn record_ai_usage(
db: &AppDatabase,
project_id: Uuid,
model_id: Uuid,
input_tokens: i64,
output_tokens: i64,
) -> Result<(), RoomError> {
if input_tokens == 0 && output_tokens == 0 {
return Ok(());
}
match agent::billing::record_ai_usage(db, project_id, model_id, input_tokens, output_tokens).await {
Ok(record) => {
tracing::info!(
project_id = %project_id,
model_id = %model_id,
input_tokens = input_tokens,
output_tokens = output_tokens,
cost_usd = %record.cost,
"ai_usage_recorded"
);
Ok(())
}
Err(e) => {
tracing::warn!(
project_id = %project_id,
model_id = %model_id,
input_tokens = input_tokens,
output_tokens = output_tokens,
error = %e,
"ai_billing_failed_non_fatal"
);
Err(e.into())
}
}
}