修改托管模型

zhangzengfei

2025-02-06 dd39847ba992605ef5fe2254140e8bfd8cf04b25

 models/model.go

@@ -2,17 +2,20 @@

import (
   "fmt"

   "model-engine/db"
)

type Model interface {
   Init(task *db.ModelTask) error
   Run() error
   KeepAlive() error
   Shutdown() error
}

var modelRegistry = map[string]func() Model{
   "gather": func() Model { return &GatherModel{} },
   "gather":    func() Model { return &GatherModel{} },
   "disappear": func() Model { return &DisappearModel{} },
   // 添加其他模型
}