mlclient

type CacheStorage interface {
	Get(ctx context.Context, key string, v interface{}) error
	Set(ctx context.Context, key string, v interface{}) error
}

type FileStore interface {
	Load(ctx context.Context, path string) ([]byte, error)
	Save(ctx context.Context, path string, content []byte) error
}

type HttpClientMaker interface {
	MakeClient(ctx context.Context) (*http.Client, error)
}

type PersistentStore interface {
	Get(ctx context.Context, kind, key string, v interface{}) ([]data.Property, error)
	Set(ctx context.Context, kind, key string, properties []data.Property, v interface{}) error
	Transact(ctx context.Context, f func(ctx context.Context) error) error
}

type PredictionMaker struct {
	CacheStorage    CacheStorage
	HttpClientMaker HttpClientMaker
}

type PredictionSource interface {
	AllPredictionsSince(ctx context.Context, t time.Time) ([]*predictions.PredictionSummary, error)
	AllPredictionResponses(context.Context, []*predictions.PredictionSummary) ([]*predictions.PredictionSummary, []*predictions.PredictionResponse, error)
}

type Trainer struct {
	PersistentStore  PersistentStore
	FileStore        FileStore
	PredictionSource PredictionSource
	ModelPath        string
	DataPath         string
	TrainPackage     string
	SleepFunc        func(time.Duration)
	HttpClientMaker  HttpClientMaker
}