From 0045706800c46a671d86e863c34c61dfdb8239f9 Mon Sep 17 00:00:00 2001 From: laoboli <1293528695@qq.com> Date: Wed, 5 Aug 2026 11:11:59 +0800 Subject: [PATCH] refactor: army --- controllers/ai.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/controllers/ai.go b/controllers/ai.go index 0d225b9..591e826 100644 --- a/controllers/ai.go +++ b/controllers/ai.go @@ -35,6 +35,7 @@ const ( audienceTypeYoungAdult = "young_adult" audienceTypeMiddleAged = "middle_aged" audienceTypeElderly = "elderly" + audienceTypeArmy = "army" sourceUpload = "upload" sourceCloud = "cloud" sourceWechat = "wechat" @@ -56,6 +57,8 @@ func resolveAudienceLabel(audienceType string) (string, bool) { return "中年人", true case audienceTypeElderly: return "老年人", true + case audienceTypeArmy: + return "18-25岁军人", true default: return "", false } @@ -346,7 +349,7 @@ func callAIForAnalysis(prompt string) (*aiAnalysisResult, error) { // @Param step_data formData file false "步数数据CSV文件(analysis_type为heart_rate_with_steps时必填)" // @Param teaching_plan formData file false "教案DOCX文件(teaching_plan_source为upload/wechat时必填)" // @Param analysis_type formData string false "分析类型: heart_rate_only(默认) | heart_rate_with_steps" -// @Param audience_type formData string false "适用人群: kindergarten(默认) | primary_school | middle_school | university | young_adult | middle_aged | elderly" +// @Param audience_type formData string false "适用人群: kindergarten(默认) | primary_school | middle_school | university | young_adult | middle_aged | elderly | army" // @Param audience_detail formData string false "精确班级或年级，例如: 小班、中班、大班、小学一年级" // @Param teaching_plan_source formData string false "教案来源: upload(默认) | cloud | wechat" // @Param regionid formData string false "区域ID" @@ -382,7 +385,7 @@ func (tc *TrainingController) AnalyzeByAI(c *gin.Context) { } audienceLabel, ok := resolveAudienceDescription(audienceType, audienceDetail) if !ok { - c.JSON(http.StatusBadRequest, gin.H{"error": "invalid audience_type, expected kindergarten, primary_school, middle_school, university, young_adult, middle_aged or elderly"}) + c.JSON(http.StatusBadRequest, gin.H{"error": "invalid audience_type, expected kindergarten, primary_school, middle_school, university, young_adult, middle_aged, elderly or army"}) return }