..

models/ai_predict_model/ai_predict_model_framework.go

@@ -317,6 +317,7 @@ type AiPredictModelFrameworkItem struct {
 	Nodes                     []*AiPredictModelFrameworkNodeItem `description:"框架节点" gorm:"-"`
 	Button                    AiPredictModelFrameworkItemButton  `description:"操作按钮" gorm:"-"`
 	ClassifyLevels            []string                           `description:"分类层级" gorm:"-"`
+	ClassifyId                int
 }
 
 // AiPredictModelFrameworkItemButton 操作按钮
@@ -345,6 +346,7 @@ func FormatAiPredictModelFramework2Item(origin *AiPredictModelFramework, nodes [
 	item.CreateTime = utils.TimeTransferString(utils.FormatDateTime, origin.CreateTime)
 	item.ModifyTime = utils.TimeTransferString(utils.FormatDateTime, origin.ModifyTime)
 	item.ClassifyLevels = classifyLevels
+	item.ClassifyId = origin.ClassifyId
 	item.Nodes = nodes
 	return
 }