diff --git a/src/main/java/cn/teammodel/controller/frontend/AiController.java b/src/main/java/cn/teammodel/controller/frontend/AiController.java index d58a2fa..18501cc 100644 --- a/src/main/java/cn/teammodel/controller/frontend/AiController.java +++ b/src/main/java/cn/teammodel/controller/frontend/AiController.java @@ -38,8 +38,8 @@ public class AiController { @PostMapping("session/create") @ApiOperation("创建聊天会话") public R createSession() { - chatSessionService.createSession(); - return R.success("创建会话成功"); + String sessionId = chatSessionService.createSession(); + return R.success(sessionId); } @PostMapping("session/remove") diff --git a/src/main/java/cn/teammodel/service/ChatSessionService.java b/src/main/java/cn/teammodel/service/ChatSessionService.java index 17ea615..861e9e7 100644 --- a/src/main/java/cn/teammodel/service/ChatSessionService.java +++ b/src/main/java/cn/teammodel/service/ChatSessionService.java @@ -11,7 +11,7 @@ import java.util.List; */ public interface ChatSessionService { - void createSession(); + String createSession(); List listMySession(); diff --git a/src/main/java/cn/teammodel/service/impl/ChatSessionServiceImpl.java b/src/main/java/cn/teammodel/service/impl/ChatSessionServiceImpl.java index 80c1d4e..dfd4489 100644 --- a/src/main/java/cn/teammodel/service/impl/ChatSessionServiceImpl.java +++ b/src/main/java/cn/teammodel/service/impl/ChatSessionServiceImpl.java @@ -35,7 +35,7 @@ public class ChatSessionServiceImpl implements ChatSessionService { private ChatSessionRepository chatSessionRepository; @Override - public void createSession() { + public String createSession() { User user = SecurityUtil.getLoginUser(); String userId = user.getId(); // 初始化欢迎语 @@ -49,7 +49,7 @@ public class ChatSessionServiceImpl implements ChatSessionService { chatSession.setCreateTime(Instant.now().toEpochMilli()); chatSession.setUpdateTime(Instant.now().toEpochMilli()); chatSession.setHistory(history); - chatSessionRepository.save(chatSession); + return chatSessionRepository.save(chatSession).getId(); } @Override