From c7cee264bccb5026fd42a9f0dc83d274aaee40cf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:05:28 +0800
Subject: [PATCH] feat(ai): 新增AI陪练流式对话V2接口
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java | 48 +++++++++++
src/main/java/cc/mrbird/febs/ai/req/memberTalkStream/ApiMemberTalkStreamV2Dto.java | 20 +++++
src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java | 2
src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmApplicationAppIdEnum.java | 2
src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java | 2
src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java | 83 +++++++++++---------
src/main/java/cc/mrbird/febs/ai/res/memberTalkStream/ApiMemberTalkStreamV2Vo.java | 42 ++++++++++
7 files changed, 160 insertions(+), 39 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java b/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java
index 93a7e2b..9f71745 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkStreamController.java
+++ b/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);
+ }
+
}
diff --git a/src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java b/src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java
index f6ed3b1..45d0684 100644
--- a/src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java
+++ b/src/main/java/cc/mrbird/febs/ai/entity/AiProductRole.java
@@ -49,7 +49,7 @@
private String iconImg;
/**
- * 模型ID
+ * 角色的定义内容
*/
private String modelId;
diff --git a/src/main/java/cc/mrbird/febs/ai/req/memberTalkStream/ApiMemberTalkStreamV2Dto.java b/src/main/java/cc/mrbird/febs/ai/req/memberTalkStream/ApiMemberTalkStreamV2Dto.java
new file mode 100644
index 0000000..9f199b0
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/req/memberTalkStream/ApiMemberTalkStreamV2Dto.java
@@ -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;
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/res/memberTalkStream/ApiMemberTalkStreamV2Vo.java b/src/main/java/cc/mrbird/febs/ai/res/memberTalkStream/ApiMemberTalkStreamV2Vo.java
new file mode 100644
index 0000000..4b3c3e0
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/res/memberTalkStream/ApiMemberTalkStreamV2Vo.java
@@ -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;
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java b/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java
index 5e71399..35582c7 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/AiMemberTalkStreamService.java
+++ b/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);
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
index 779cc73..40bba03 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
+++ b/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));
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmApplicationAppIdEnum.java b/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmApplicationAppIdEnum.java
index 1c06bad..02c8bd3 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmApplicationAppIdEnum.java
+++ b/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","通用"),
--
Gitblit v1.9.1