Просмотр исходного кода

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

iwt 1 неделя назад
Родитель
Сommit
a8344e7f49

+ 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()) {
             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;
         if (reportId != null && surveyId != null) {
             inputs = familyContextService.buildContext(userId, reportId, surveyId);
         } else if (reportId != null) {
             inputs = familyContextService.buildContext(userId, reportId);
+        } else if (draftId != null) {
+            inputs = familyContextService.buildContextFromDraft(userId, draftId);
         } else {
             inputs = familyContextService.buildContext(userId);
         }