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

fix(billing): pass real project_id instead of Uuid::nil() for cost calculation

Browse Source
This commit is contained in:
ZhenYi 2026-04-28 09:57:45 +08:00
parent 211cf0ee3e
commit 7f927a4b6b
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/agent/chat/service.rs
View File

@ -49,6 +49,7 @@ pub struct ProcessResult {
/// Record an AI session with cost calculation. /// Record an AI session with cost calculation.
async fn record_ai_session( async fn record_ai_session(
db: &db::database::AppDatabase, db: &db::database::AppDatabase,
project_id: Uuid,
session_id: Uuid, session_id: Uuid,
room_id: Uuid, room_id: Uuid,
model_id: Uuid, model_id: Uuid,
@ -59,7 +60,7 @@ async fn record_ai_session(
) { ) {
let (cost, currency) = match billing::record_ai_usage( let (cost, currency) = match billing::record_ai_usage(
db, db,
Uuid::nil(), // project_uid not needed for session cost project_id,
model_id, model_id,
input_tokens, input_tokens,
output_tokens, output_tokens,
@ -389,6 +390,7 @@ impl ChatService {
// Record session // Record session
record_ai_session( record_ai_session(
&request.db, &request.db,
request.project.id,
session_id, session_id,
request.room.id, request.room.id,
request.model.id, request.model.id,
@ -410,6 +412,7 @@ impl ChatService {
// Record session // Record session
record_ai_session( record_ai_session(
&request.db, &request.db,
request.project.id,
session_id, session_id,
request.room.id, request.room.id,
request.model.id, request.model.id,
@ -748,6 +751,7 @@ impl ChatService {
// Record session // Record session
record_ai_session( record_ai_session(
&request.db, &request.db,
request.project.id,
session_id, session_id,
request.room.id, request.room.id,
request.model.id, request.model.id,
@ -784,6 +788,7 @@ impl ChatService {
// Record session // Record session
record_ai_session( record_ai_session(
&request.db, &request.db,
request.project.id,
session_id, session_id,
request.room.id, request.room.id,
request.model.id, request.model.id,
@ -1216,6 +1221,7 @@ impl ChatService {
let elapsed_ms = session_start.elapsed().as_millis() as i64; let elapsed_ms = session_start.elapsed().as_millis() as i64;
record_ai_session( record_ai_session(
&request.db, &request.db,
request.project.id,
session_id, session_id,
request.room.id, request.room.id,
request.model.id, request.model.id,