agent

Documentation

Index

• func StartPullerAndProcessModels(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 WaitGroupWrapper
• type Watcher
  • func NewWatcher(configDir string, modelDir string, logger *zap.SugaredLogger) Watcher
  • func (w *Watcher) Start()

Functions

func StartPullerAndProcessModels

func StartPullerAndProcessModels(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
	// contains filtered or unexported fields
}

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 {
	OnStartup bool
	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 WaitGroupWrapper

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

type Watcher

type Watcher struct {
	ModelTracker map[string]modelWrapper
	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()