agent

Documentation

Index

• Variables
• func StartPuller(downloader Downloader, commands <-chan ModelOp, logger *zap.SugaredLogger)
• func SyncModelDir(modelDir string, logger *zap.SugaredLogger) (map[string]modelWrapper, error)
• type Downloader
  • func (d *Downloader) DownloadModel(modelName string, modelSpec *v1alpha1.ModelSpec) error
• type FileError
• type ModelChannel
• type ModelOp
• type OpType
• type Puller
• type Watcher
  • func NewWatcher(configDir string, modelDir string, logger *zap.SugaredLogger) Watcher
  • func (w *Watcher) Start()

Variables

var SupportedProtocols = []storage.Protocol{storage.S3, storage.GCS}

Functions

func StartPuller

func StartPuller(downloader Downloader, commands <-chan ModelOp, logger *zap.SugaredLogger)

func SyncModelDir

func SyncModelDir(modelDir string, logger *zap.SugaredLogger) (map[string]modelWrapper, error)

Types

type Downloader

type Downloader struct {
	ModelDir  string
	Providers map[storage.Protocol]storage.Provider
	Logger    *zap.SugaredLogger
}

func (*Downloader) DownloadModel

func (d *Downloader) DownloadModel(modelName string, modelSpec *v1alpha1.ModelSpec) error

type FileError

type FileError error

var NoSuccessFile FileError = fmt.Errorf("no success file can be found")

type ModelChannel

type ModelChannel struct {
	// contains filtered or unexported fields
}

type ModelOp

type ModelOp struct {
	ModelName string
	Op        OpType
	Spec      *v1.ModelSpec
}

type OpType

type OpType string

const (
	Add    OpType = "Add"
	Remove OpType = "Remove"
)

type Puller

type Puller struct {
	Downloader Downloader
	// contains filtered or unexported fields
}

type Watcher

type Watcher struct {
	ModelEvents chan ModelOp
	// contains filtered or unexported fields
}

func NewWatcher

func NewWatcher(configDir string, modelDir string, logger *zap.SugaredLogger) Watcher

func (*Watcher) Start

func (w *Watcher) Start()