zhaoqingang
2025-02-08 3c033759200ad7c02dd59521b1aebbbdc35b98fa
models/model.go
@@ -2,17 +2,21 @@
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{} },
   "location":  func() Model { return &LocationModel{} },
   // 添加其他模型
}