Administrator
2025-09-03 a1d01d25e2c5f48bc51557da55f40e8748d94532
feat(ai): 优化对话输入方式并添加思考内容

- 将 AiPromptEnum 中的 STREAM_NORMAL 提示语从"回答我提出的问题"改为"回答我输入的问题"- 在 AiTalkServiceImpl 中添加 reasoningContent 字段处理思考内容
- 更新 ApiTalkDto,增加 reasoningContent 参数用于接收思考内容
- 在 TestController 中添加对 reasoningContent 的处理逻辑
4 files modified
17 ■■■■ changed files
src/main/java/cc/mrbird/febs/ai/controller/TestController.java 2 ●●●●● patch | view | raw | blame | history
src/main/java/cc/mrbird/febs/ai/enumerates/AiPromptEnum.java 2 ●●● patch | view | raw | blame | history
src/main/java/cc/mrbird/febs/ai/req/talk/ApiTalkDto.java 3 ●●●●● patch | view | raw | blame | history
src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java 10 ●●●● patch | view | raw | blame | history
src/main/java/cc/mrbird/febs/ai/controller/TestController.java
@@ -278,6 +278,8 @@
        }
        Flux.from(resultStream)
                .doOnNext(message -> {
                    String reasoningContent = message.getOutput().getChoices().get(0).getMessage().getReasoningContent();
                    System.out.print(reasoningContent);
                    String content = message.getOutput().getChoices().get(0).getMessage().getContent();
                    System.out.print(content);
                })
src/main/java/cc/mrbird/febs/ai/enumerates/AiPromptEnum.java
@@ -11,7 +11,7 @@
    HIGH_LIGHT(2,"请输出亮点"),
    STREAM_NORMAL(1,"你是人工智能,请回答我提出的问题");
    STREAM_NORMAL(1,"你是人工智能,请回答我输入的问题");
    private final int code;
    private final String prompt;
src/main/java/cc/mrbird/febs/ai/req/talk/ApiTalkDto.java
@@ -14,6 +14,9 @@
@ApiModel(value = "ApiTalkDto", description = "参数")
public class ApiTalkDto {
    @ApiModelProperty(value = "思考内容", example = "10")
    private String reasoningContent;
    @NotBlank(message = "内容不能为空")
    @ApiModelProperty(value = "内容", example = "10")
    private String context;
src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java
@@ -16,6 +16,7 @@
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;
@@ -25,6 +26,7 @@
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;
@@ -37,6 +39,7 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
@Slf4j
@@ -71,6 +74,7 @@
        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);
@@ -78,8 +82,10 @@
        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());