From 55fc4d43e6c6c60b24fb90ff9153e80a9933fb4c Mon Sep 17 00:00:00 2001 From: laoboli <1293528695@qq.com> Date: Wed, 22 Jul 2026 10:44:41 +0800 Subject: [PATCH] feat: add age phase --- controllers/ai.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/controllers/ai.go b/controllers/ai.go index b1a8640..7e794f8 100644 --- a/controllers/ai.go +++ b/controllers/ai.go @@ -32,6 +32,8 @@ const ( audienceTypePrimarySchool = "primary_school" audienceTypeMiddleSchool = "middle_school" audienceTypeUniversity = "university" + audienceTypeYoungAdult = "young_adult" + audienceTypeMiddleAged = "middle_aged" audienceTypeElderly = "elderly" sourceUpload = "upload" sourceCloud = "cloud" @@ -48,6 +50,10 @@ func resolveAudienceLabel(audienceType string) (string, bool) { return "中学", true case audienceTypeUniversity: return "大学", true + case audienceTypeYoungAdult: + return "青年人", true + case audienceTypeMiddleAged: + return "中年人", true case audienceTypeElderly: return "老年人", true default: @@ -327,7 +333,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 | elderly" +// @Param audience_type formData string false "适用人群: kindergarten(默认) | primary_school | middle_school | university | young_adult | middle_aged | elderly" // @Param audience_detail formData string false "精确班级或年级，例如: 小班、中班、大班、小学一年级" // @Param teaching_plan_source formData string false "教案来源: upload(默认) | cloud | wechat" // @Param regionid formData string false "区域ID" @@ -363,7 +369,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 or elderly"}) + c.JSON(http.StatusBadRequest, gin.H{"error": "invalid audience_type, expected kindergarten, primary_school, middle_school, university, young_adult, middle_aged or elderly"}) return }