feat(ai): 新增AI陪练流式对话V2接口

Administrator

2025-10-30 c7cee264bccb5026fd42a9f0dc83d274aaee40cf

feat(ai): 新增AI陪练流式对话V2接口

- 新增startV2方法支持流式响应对话
- 调整AiProductRole实体类modelId字段注释
- 新增ApiMemberTalkStreamV2Dto和ApiMemberTalkStreamV2Vo类
- 更新Controller接口路径和注释信息
- 添加LlmApplicationAppIdEnum枚举值CONTEXT_TALK

 src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java

@@ -7,6 +7,7 @@
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkListDto;
import cc.mrbird.febs.ai.res.memberTalk.*;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkReloadStreamVo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkStreamV2Vo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkStreamVo;
import cc.mrbird.febs.ai.service.AiMemberTalkStreamService;
import cc.mrbird.febs.common.entity.FebsResponse;
@@ -40,44 +41,13 @@
        return aiMemberTalkStreamService.start(dto);
    }

    @ApiOperation(value = "陪练记录", notes = "陪练记录")
    @ApiOperation("继续回答")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkListVo.class)
            @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamV2Vo.class),
    })
    @PostMapping(value = "/talkList")
    public FebsResponse talkList(@RequestBody @Validated ApiMemberTalkListDto dto) {

        return aiMemberTalkStreamService.talkList(dto);
    }

    @ApiOperation(value = "陪练记录-答题记录", notes = "陪练记录-答题记录")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkMemberListVo.class)
    })
    @PostMapping(value = "/talkMemberList")
    public FebsResponse talkMemberList(@RequestBody @Validated ApiTalkMemberListDto dto) {

        return aiMemberTalkStreamService.talkMemberList(dto);
    }

    @ApiOperation(value = "陪练记录-查看报告", notes = "陪练记录-查看报告")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkReportListVo.class)
    })
    @PostMapping(value = "/talkReportList")
    public Flux<FebsResponse> talkReportList(@RequestBody @Validated ApiTalkReportListDto dto) {

        return aiMemberTalkStreamService.talkReportList(dto);
    }

    @ApiOperation(value = "陪练记录-查看报告", notes = "陪练记录-查看报告")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkReportListVo.class)
    })
    @PostMapping(value = "/talkReportListV2")
    public FebsResponse talkReportListV2(@RequestBody @Validated ApiTalkReportListDto dto) {

        return aiMemberTalkStreamService.talkReportListV2(dto);
    @PostMapping("/startV2")
    public Flux<FebsResponse> startV2(@RequestBody @Validated ApiMemberTalkStreamV2Dto dto) {
        return aiMemberTalkStreamService.startV2(dto);
    }

    @ApiOperation(value = "再练一次（重新回答）", notes = "再练一次（重新回答）")
@@ -160,4 +130,45 @@
        return aiMemberTalkStreamService.saveReport(dto);
    }


    @ApiOperation(value = "陪练记录", notes = "陪练记录")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkListVo.class)
    })
    @PostMapping(value = "/talkList")
    public FebsResponse talkList(@RequestBody @Validated ApiMemberTalkListDto dto) {

        return aiMemberTalkStreamService.talkList(dto);
    }

    @ApiOperation(value = "陪练记录-答题记录", notes = "陪练记录-答题记录")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkMemberListVo.class)
    })
    @PostMapping(value = "/talkMemberList")
    public FebsResponse talkMemberList(@RequestBody @Validated ApiTalkMemberListDto dto) {

        return aiMemberTalkStreamService.talkMemberList(dto);
    }

    @ApiOperation(value = "陪练记录-查看报告", notes = "陪练记录-查看报告")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkReportListVo.class)
    })
    @PostMapping(value = "/talkReportList")
    public Flux<FebsResponse> talkReportList(@RequestBody @Validated ApiTalkReportListDto dto) {

        return aiMemberTalkStreamService.talkReportList(dto);
    }

    @ApiOperation(value = "陪练记录-查看报告", notes = "陪练记录-查看报告")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiTalkReportListVo.class)
    })
    @PostMapping(value = "/talkReportListV2")
    public FebsResponse talkReportListV2(@RequestBody @Validated ApiTalkReportListDto dto) {

        return aiMemberTalkStreamService.talkReportListV2(dto);
    }

}

 src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java

@@ -49,7 +49,7 @@
    private String iconImg;

    /**
     * 模型ID
     * 角色的定义内容
     */
    private String modelId;


 src/main/java/cc/mrbird/febs/ai/req/memberTalkStream/ApiMemberTalkStreamV2Dto.java

New file
@@ -0,0 +1,20 @@
package cc.mrbird.febs.ai.req.memberTalkStream;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import javax.validation.constraints.NotBlank;

@Data
@ApiModel(value = "ApiMemberTalkStreamV2Dto", description = "参数")
public class ApiMemberTalkStreamV2Dto {

    @NotBlank(message = "ID不能为空")
    @ApiModelProperty(value = "产品ID", example = "10")
    private String id;

    @NotBlank(message = "ID不能为空")
    @ApiModelProperty(value = "对话ID", example = "10")
    private String memberTalkId;
}

 src/main/java/cc/mrbird/febs/ai/res/memberTalkStream/ApiMemberTalkStreamV2Vo.java

New file
@@ -0,0 +1,42 @@
package cc.mrbird.febs.ai.res.memberTalkStream;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

@Data
@ApiModel(value = "ApiMemberTalkStreamV2Vo", description = "参数")
public class ApiMemberTalkStreamV2Vo {
    /**
     * 用户对话ID (UUID)
     */

    @ApiModelProperty(value = "会话ID")
    private String memberTalkItemId;

    @ApiModelProperty(value = "会话ID")
    private String memberTalkId;

    /**
     * 类型 1-AI提问 2-用户回答 3-AI分析结果
     */

    @ApiModelProperty(value = "类型 1-AI提问 2-用户回答 3-AI分析结果")
    private Integer type;

    /**
     * 内容
     */

    @ApiModelProperty(value = "内容(文本格式)")
    private String context;

    @ApiModelProperty(value = "内容亮点、建议、参考答案、核心知识点雷达图表数据（数据对象）")
    private String report;

    @ApiModelProperty(value = "题目数量")
    private Integer questionCnt;

    @ApiModelProperty(value = "作答数量")
    private Integer doneCnt;
}

 src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java

@@ -20,6 +20,8 @@
     */
    FebsResponse start(ApiMemberTalkStreamDto dto);

    Flux<FebsResponse> startV2(ApiMemberTalkStreamV2Dto dto);

    FebsResponse talkList(ApiMemberTalkListDto dto);

    FebsResponse talkMemberList(ApiTalkMemberListDto dto);

 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java

@@ -10,10 +10,8 @@
import cc.mrbird.febs.ai.req.memberTalkStream.*;
import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkListDto;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkMemberAnswerSavaVo;
import cc.mrbird.febs.ai.res.memberTalk.ApiTalkReportListVo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkReloadStreamVo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkStreamVo;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkListVo;
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;
import cc.mrbird.febs.ai.strategy.enumerates.LlmApplicationAppIdEnum;
@@ -110,6 +108,52 @@
    }

    @Override
    public Flux<FebsResponse> startV2(ApiMemberTalkStreamV2Dto dto) {

        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
        String memberTalkId = dto.getId();
        AiMemberTalk aiMemberTalk = this.getById(memberTalkId);
        if (ObjectUtil.isNull(aiMemberTalk)){
            throw new FebsException("对话不存在");
        }

        LambdaQueryWrapper<AiProductRoleLink> productLinkQuery = Wrappers.lambdaQuery(AiProductRoleLink.class);
        productLinkQuery.eq(AiProductRoleLink::getProductId,aiMemberTalk.getProductId());
        productLinkQuery.last("limit 1");
        AiProductRoleLink aiProductRoleLink = aiProductRoleLinkService.getByQuery(productLinkQuery);
        if(ObjectUtil.isNull(aiProductRoleLink)){
            throw new FebsException("产品没有关联AI陪练");
        }

        String productRoleId = aiProductRoleLink.getProductRoleId();
        AiProductRole aiProductRole = aiProductRoleService.getById(productRoleId);
        if (ObjectUtil.isNull(aiProductRole)){
            throw new FebsException("产品AI陪练不存在");
        }

        String promptHead = aiProductRole.getModelId();

        List<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();
        LlmStrategyDto llmStrategyDto = this.buildLlmStrategyDtoList(promptHead, 1);
        llmStrategyDtoList.add(llmStrategyDto);
        llmStrategyDto = this.buildLlmStrategyDtoList("请按照上下文内容，进行对话的延续，例如提出一个相关的问题或者对内容进行一个延续", 2);
        llmStrategyDtoList.add(llmStrategyDto);

        AiCompanyWorkflow aiCompanyWorkflow = aiCompanyWorkflowService.getByTypeAndCompanyId(LlmApplicationAppIdEnum.CONTEXT_TALK.getCode(),aiMemberTalk.getCompanyId());
        if (ObjectUtil.isNull(aiCompanyWorkflow)){
            throw new FebsException("工作流配置异常，请联系管理员");
        }
        llmStrategyDto = this.buildLlmStrategyDtoList(aiCompanyWorkflow.getCode(), 4);
        llmStrategyDtoList.add(llmStrategyDto);

        LlmStrategyDto llmStrategyDtoMessage = buildMessages(memberTalkId);
        llmStrategyDtoList.add(llmStrategyDtoMessage);
        String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());

        return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreamingNoThink(llmStrategyDtoList);
    }

    @Override
    public FebsResponse talkList(ApiMemberTalkListDto dto) {

        return new FebsResponse().success().data(aiMemberTalkService.getPageByDto(dto));

 src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmApplicationAppIdEnum.java

@@ -5,6 +5,8 @@
@Getter
public enum LlmApplicationAppIdEnum {

    CONTEXT_TALK(7,"48468fdcdfc340f2950e49559a4dbcd3","AI陪练连续对话"),

    REPORT(6,"48468fdcdfc340f2950e49559a4dbcd3","生成报告"),

    NORMAL(5,"8ea7ab1665de4a88b868849318d72f45","通用"),