diff --git 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
index 5c51af7c..da2e589b 100644
--- 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());