provider

type EngineInfoProvider interface {
	GetVersion(ctx context.Context, resp *types.EngineVersionResponse) (*types.EngineVersionResponse, error)
	GetOperateStatus() int
	SetOperateStatus(status int)
}

type EngineInstaller interface {
	CheckEngine() bool
	InstallEngine(cover bool) error
	InitEnv() error
	UpgradeEngine() error
}

type EngineLifecycleManager interface {
	StartEngine(mode string) error
	StopEngine() error
	HealthCheck() error
	GetConfig() *types.EngineRecommendConfig
}

type EngineManager interface {
	// 启动所有引擎
	StartAllEngines(mode string) error
	// 停止所有引擎
	StopAllEngines() error
	// 启动引擎保活监控
	StartKeepAlive()
	// 停止引擎保活监控
	StopKeepAlive()
	// 获取引擎状态
	GetEngineStatus() map[string]string
	// 注册引擎到管理器
	RegisterEngine(name string, provider ModelServiceProvider)
}

type ModelManager interface {
	PullModel(ctx context.Context, req *types.PullModelRequest, fn types.PullProgressFunc) (*types.ProgressResponse, error)
	PullModelStream(ctx context.Context, req *types.PullModelRequest) (chan []byte, chan error)
	DeleteModel(ctx context.Context, req *types.DeleteRequest) error
	ListModels(ctx context.Context) (*types.ListResponse, error)
	LoadModel(ctx context.Context, req *types.LoadRequest) error
	UnloadModel(ctx context.Context, req *types.UnloadModelRequest) error
	GetRunningModels(ctx context.Context) (*types.ListResponse, error)
}

type ModelServiceProvider interface {
	EngineLifecycleManager
	EngineInstaller
	ModelManager
	EngineInfoProvider
}