xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-08 09:17:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 对话时记录对话角色

Browse Source
This commit is contained in:
ageerle
2025-05-19 16:41:55 +08:00
parent 0ee47a5c00
commit 93ddcd53b1
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
View File

@@ -63,6 +63,9 @@ public class ChatCostServiceImpl implements IChatCostService {
Object userId = LocalCache.CACHE.get("userId"); Object userId = LocalCache.CACHE.get("userId");
chatMessageBo.setUserId((Long) userId); chatMessageBo.setUserId((Long) userId);
// 设置对话角色
chatMessageBo.setRole(chatRequest.getRole());
chatMessageBo.setSessionId(chatRequest.getSessionId()); chatMessageBo.setSessionId(chatRequest.getSessionId());
chatMessageBo.setContent(chatRequest.getPrompt()); chatMessageBo.setContent(chatRequest.getPrompt());