| | |
| | | import cc.mrbird.febs.ai.service.AiTalkItemService; |
| | | import cc.mrbird.febs.ai.service.AiTalkService; |
| | | import cc.mrbird.febs.ai.strategy.LlmStrategyFactory; |
| | | import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyContextEnum; |
| | | import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum; |
| | | import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto; |
| | | import cc.mrbird.febs.ai.utils.UUID; |
| | |
| | | import cn.hutool.core.date.DateTime; |
| | | import cn.hutool.core.date.DateUtil; |
| | | import cn.hutool.core.util.StrUtil; |
| | | import cn.hutool.json.JSONUtil; |
| | | import com.alibaba.dashscope.common.Role; |
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
| | | import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
| | |
| | | |
| | | import java.util.ArrayList; |
| | | import java.util.Date; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | |
| | | @Slf4j |
| | |
| | | String talkId = dto.getTalkId(); |
| | | String context = dto.getContext(); |
| | | Integer type = dto.getType(); |
| | | String reasoningContent = dto.getReasoningContent(); |
| | | String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid(); |
| | | |
| | | AiTalk aiTalk = this.getById(talkId); |
| | |
| | | if (StrUtil.isEmpty(talkId)){ |
| | | aiTalk = this.add(memberUuid,context, date); |
| | | } |
| | | |
| | | aiTalkItemService.add(aiTalk.getId(), type, context, memberUuid, date); |
| | | HashMap<String, String> stringStringHashMap = new HashMap<>(); |
| | | stringStringHashMap.put(LlmStrategyContextEnum.THINK.getName(),reasoningContent); |
| | | stringStringHashMap.put(LlmStrategyContextEnum.CONTENT.getName(),context); |
| | | aiTalkItemService.add(aiTalk.getId(), type, JSONUtil.toJsonStr(stringStringHashMap), memberUuid, date); |
| | | |
| | | ApiTalkVo apiTalkVo = new ApiTalkVo(); |
| | | apiTalkVo.setTalkId(aiTalk.getId()); |