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