From 65f10033448a07535d610ad56a628d7b3f137480 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 27 Aug 2025 17:22:14 +0800
Subject: [PATCH] feat(ai): 优化对话项添加逻辑并增强接口返回值

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java |   13 +++++++------
 1 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java
index ac1a089..2b93ccc 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java
@@ -1,7 +1,6 @@
 package cc.mrbird.febs.ai.service.impl;
 
-import cc.mrbird.febs.ai.controller.enumerates.AiTypeEnum;
-import cc.mrbird.febs.ai.entity.AiMemberTalk;
+import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
 import cc.mrbird.febs.ai.entity.AiMemberTalkItem;
 import cc.mrbird.febs.ai.mapper.AiMemberTalkItemMapper;
 import cc.mrbird.febs.ai.req.ai.AiMessage;
@@ -12,6 +11,7 @@
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cn.hutool.core.collection.CollUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -38,7 +38,7 @@
     private final AiMemberTalkItemMapper aiMemberTalkItemMapper;
 
     @Override
-    public void add(String memberUuid, String id, int type, String resContext,Date createdTime) {
+    public AiMemberTalkItem add(String memberUuid, String id, int type, String resContext,Date createdTime) {
         AiMemberTalkItem aiMemberTalkItem = new AiMemberTalkItem();
         aiMemberTalkItem.setId(UUID.getSimpleUUIDString());
         aiMemberTalkItem.setCreatedTime(createdTime);
@@ -47,6 +47,8 @@
         aiMemberTalkItem.setType(type);
         aiMemberTalkItem.setContext(resContext);
         aiMemberTalkItemMapper.insert(aiMemberTalkItem);
+
+        return aiMemberTalkItem;
     }
 
     @Override
@@ -55,11 +57,10 @@
     }
 
     @Override
-    public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {
+    public IPage<ApiMemberTalkItemVo> historyPage(ApiMemberTalkItemPageDto dto) {
         // 创建分页对象,传入当前页和每页大小
         Page<ApiMemberTalkItemVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
-        Page<ApiMemberTalkItemVo> pageListByQuery = aiMemberTalkItemMapper.getPageListByQuery(page, dto);
-        return new FebsResponse().success().data(pageListByQuery);
+        return aiMemberTalkItemMapper.getPageListByQuery(page, dto);
     }
 
     @Override

--
Gitblit v1.9.1