From 95fb77a5755f9f91d1c0f3ed54def077724aca2b Mon Sep 17 00:00:00 2001
From: ziin <hiziying@hotmail.com>
Date: Wed, 17 Dec 2025 19:23:21 +0800
Subject: [PATCH] =?UTF-8?q?fix(chat):=20=E4=BF=9D=E5=AD=98LLM=E5=93=8D?=
 =?UTF-8?q?=E5=BA=94=E7=9A=84=E7=94=9F=E6=88=90ID=E7=94=A8=E4=BA=8E?=
 =?UTF-8?q?=E9=93=BE=E8=B7=AF=E8=BF=BD=E8=B8=AA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/yolo/keyborad/model/entity/KeyboardUserCallLog.java | 3 +++
 .../com/yolo/keyborad/service/impl/ChatServiceImpl.java     | 6 ++++++
 2 files changed, 9 insertions(+)

diff --git a/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserCallLog.java b/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserCallLog.java
index c489d51..c31e1e6 100644
--- a/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserCallLog.java
+++ b/src/main/java/com/yolo/keyborad/model/entity/KeyboardUserCallLog.java
@@ -103,4 +103,7 @@ public class KeyboardUserCallLog {
     @Schema(description="调用记录创建时间")
     private Date createdAt;
 
+    @TableField(value = "gen_id")
+    @Schema(description="生成 Id")
+    private String genId;
 }
\ No newline at end of file
diff --git a/src/main/java/com/yolo/keyborad/service/impl/ChatServiceImpl.java b/src/main/java/com/yolo/keyborad/service/impl/ChatServiceImpl.java
index eeef763..1ede63d 100644
--- a/src/main/java/com/yolo/keyborad/service/impl/ChatServiceImpl.java
+++ b/src/main/java/com/yolo/keyborad/service/impl/ChatServiceImpl.java
@@ -155,6 +155,8 @@ public class ChatServiceImpl implements ChatService {
         AtomicInteger outputTokens = new AtomicInteger(0);
         // 原子引用保存错误代码
         AtomicReference<String> errorCodeRef = new AtomicReference<>();
+        //原子引用保存生成ID
+        AtomicReference<String> genId = new AtomicReference<>();
 
         // ============ 3. 构建LLM流式输出 ============
         Flux<ChatStreamMessage> llmFlux = client
@@ -180,6 +182,9 @@ public class ChatServiceImpl implements ChatService {
                         if (metadata.getModel() != null) {
                             modelRef.set(metadata.getModel());
                         }
+                        if (metadata.getId() != null){
+                            genId.set(metadata.getId());
+                        }
                         // 保存token使用情况
                         if (metadata.getUsage() != null) {
                             var usage = metadata.getUsage();
@@ -281,6 +286,7 @@ public class ChatServiceImpl implements ChatService {
                             });
                             // 设置日志基本信息
                             callLog.setRequestId(requestId);
+                            callLog.setGenId(genId.get());
                             callLog.setFeature("chat_talk");
                             callLog.setModel(modelRef.get());
                             // 设置token使用情况