From 7f927a4b6bb4fce2b4b6bcaffa6dabbcc83cd1e7 Mon Sep 17 00:00:00 2001
From: ZhenYi <434836402@qq.com>
Date: Tue, 28 Apr 2026 09:57:45 +0800
Subject: [PATCH] fix(billing): pass real project_id instead of Uuid::nil() for
 cost calculation

---
 libs/agent/chat/service.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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