浏览代码

feat: AIChatController 支持 draftId 参数的快速分析

iwt 6 天之前
父节点
当前提交
a8344e7f49
共有 1 个文件被更改,包括 7 次插入0 次删除
  1. 7 0
      cfc-backend/src/main/java/com/etotem/cfc/controller/ai/AIChatController.java

+ 7 - 0
cfc-backend/src/main/java/com/etotem/cfc/controller/ai/AIChatController.java

@@ -152,11 +152,18 @@ public class AIChatController {
         if (surveyIdStr != null && !surveyIdStr.trim().isEmpty()) {
         if (surveyIdStr != null && !surveyIdStr.trim().isEmpty()) {
             surveyId = Long.valueOf(surveyIdStr);
             surveyId = Long.valueOf(surveyIdStr);
         }
         }
+        String draftIdStr = params.get("draftId");
+        Long draftId = null;
+        if (draftIdStr != null && !draftIdStr.trim().isEmpty()) {
+            draftId = Long.valueOf(draftIdStr);
+        }
         Map<String, Object> inputs;
         Map<String, Object> inputs;
         if (reportId != null && surveyId != null) {
         if (reportId != null && surveyId != null) {
             inputs = familyContextService.buildContext(userId, reportId, surveyId);
             inputs = familyContextService.buildContext(userId, reportId, surveyId);
         } else if (reportId != null) {
         } else if (reportId != null) {
             inputs = familyContextService.buildContext(userId, reportId);
             inputs = familyContextService.buildContext(userId, reportId);
+        } else if (draftId != null) {
+            inputs = familyContextService.buildContextFromDraft(userId, draftId);
         } else {
         } else {
             inputs = familyContextService.buildContext(userId);
             inputs = familyContextService.buildContext(userId);
         }
         }