feat: 对话时记录对话角色

2026-03-14 05:03:43 +08:00 · 2025-05-19 16:41:55 +08:00
parent 0ee47a5c00
commit 93ddcd53b1
1 changed files with 3 additions and 0 deletions
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
@@ -63,6 +63,9 @@ public class ChatCostServiceImpl implements IChatCostService {
        Object userId = LocalCache.CACHE.get("userId");
        chatMessageBo.setUserId((Long) userId);

+        // 设置对话角色
+        chatMessageBo.setRole(chatRequest.getRole());
+
        chatMessageBo.setSessionId(chatRequest.getSessionId());
        chatMessageBo.setContent(chatRequest.getPrompt());