src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkItemServiceImpl.java
@@ -5,7 +5,9 @@
import cc.mrbird.febs.ai.mapper.AiMemberTalkItemMapper;
import cc.mrbird.febs.ai.req.ai.AiMessage;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
import cc.mrbird.febs.ai.req.memberTalk.ApiTalkMemberListDto;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
import cc.mrbird.febs.ai.res.memberTalk.ApiTalkMemberListVo;
import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
import cc.mrbird.febs.ai.utils.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
@@ -38,9 +40,10 @@
    private final AiMemberTalkItemMapper aiMemberTalkItemMapper;
    @Override
    public AiMemberTalkItem add(String memberUuid, String id, int type, String resContext,Date createdTime) {
    public AiMemberTalkItem add(String memberUuid, String id, String companyId, int type, String resContext,Date createdTime) {
        AiMemberTalkItem aiMemberTalkItem = new AiMemberTalkItem();
        aiMemberTalkItem.setId(UUID.getSimpleUUIDString());
        aiMemberTalkItem.setCompanyId(companyId);
        aiMemberTalkItem.setCreatedTime(createdTime);
        aiMemberTalkItem.setMemberId(memberUuid);
        aiMemberTalkItem.setMemberTalkId(id);
@@ -52,8 +55,13 @@
    }
    @Override
    public AiMemberTalkItem getByQuery(LambdaQueryWrapper<AiMemberTalkItem> memberTalkItemQuery) {
    public AiMemberTalkItem getOneByQuery(LambdaQueryWrapper<AiMemberTalkItem> memberTalkItemQuery) {
        return aiMemberTalkItemMapper.selectOne(memberTalkItemQuery);
    }
    @Override
    public List<AiMemberTalkItem> getListByQuery(LambdaQueryWrapper<AiMemberTalkItem> memberTalkItemQuery) {
        return aiMemberTalkItemMapper.selectList(memberTalkItemQuery);
    }
    @Override
@@ -97,4 +105,11 @@
        return aiMessages;
    }
    @Override
    public IPage<ApiTalkMemberListVo> getPageByDto(ApiTalkMemberListDto dto) {
        // 创建分页对象,传入当前页和每页大小
        Page<ApiTalkMemberListVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        return aiMemberTalkItemMapper.getPageByDto(page, dto);
    }
}