ziin/keyboard_backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(chat): 保存LLM响应的生成ID用于链路追踪

Browse Source
This commit is contained in:
ziin
2025-12-17 19:23:21 +08:00
parent abfac871fd
commit 95fb77a575
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/yolo/keyborad/model/entity/KeyboardUserCallLog.java
View File

@@ -103,4 +103,7 @@ public class KeyboardUserCallLog {
@Schema(description="调用记录创建时间") @Schema(description="调用记录创建时间")
private Date createdAt; private Date createdAt;
@TableField(value = "gen_id")
@Schema(description="生成 Id")
private String genId;
} }

6
src/main/java/com/yolo/keyborad/service/impl/ChatServiceImpl.java
View File

@@ -155,6 +155,8 @@ public class ChatServiceImpl implements ChatService {
AtomicInteger outputTokens = new AtomicInteger(0); AtomicInteger outputTokens = new AtomicInteger(0);
// 原子引用保存错误代码 // 原子引用保存错误代码
AtomicReference<String> errorCodeRef = new AtomicReference<>(); AtomicReference<String> errorCodeRef = new AtomicReference<>();
//原子引用保存生成ID
AtomicReference<String> genId = new AtomicReference<>();
// ============ 3. 构建LLM流式输出 ============ // ============ 3. 构建LLM流式输出 ============
Flux<ChatStreamMessage> llmFlux = client Flux<ChatStreamMessage> llmFlux = client
@@ -180,6 +182,9 @@ public class ChatServiceImpl implements ChatService {
if (metadata.getModel() != null) { if (metadata.getModel() != null) {
modelRef.set(metadata.getModel()); modelRef.set(metadata.getModel());
} }
if (metadata.getId() != null){
genId.set(metadata.getId());
}
// 保存token使用情况 // 保存token使用情况
if (metadata.getUsage() != null) { if (metadata.getUsage() != null) {
var usage = metadata.getUsage(); var usage = metadata.getUsage();
@@ -281,6 +286,7 @@ public class ChatServiceImpl implements ChatService {
}); });
// 设置日志基本信息 // 设置日志基本信息
callLog.setRequestId(requestId); callLog.setRequestId(requestId);
callLog.setGenId(genId.get());
callLog.setFeature("chat_talk"); callLog.setFeature("chat_talk");
callLog.setModel(modelRef.get()); callLog.setModel(modelRef.get());
// 设置token使用情况 // 设置token使用情况