agent

package

v1.4.4 Latest Latest Go to latest Published: Nov 15, 2021 License: NCSA Imports: 24 Imported by: 5

Details

This section is empty.

This section is empty.

func AddPredictor(framework dl.FrameworkManifest, pred predictor.Predictor) error

func GetPredictor(framework dl.FrameworkManifest) (predictor.Predictor, error)

func GetPredictors(framework dl.FrameworkManifest) ([]predictor.Predictor, error)

func PredictorFrameworks() []dl.FrameworkManifest

func Predictors() []string

func UploadDir() (string, error)

type Option func(o *Options) *Options

func WithHost(host string) Option

func WithPort(port int) Option

func WithPortString(port string) Option

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

func NewOptions(opts ...Option) (*Options, error)

type PredictorLifetime struct {
	Predictor      *predictor.Predictor
	ReferenceCount uint64
}

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

func (c *Registry) FrameworkAgents(context.Context, *dl.FrameworkRequest) (*dl.Agents, error)

func (c *Registry) FrameworkManifests(context.Context, *dl.FrameworkRequest) (*dl.FrameworkManifestsResponse, error)

func (c *Registry) ModelAgents(context.Context, *dl.ModelRequest) (*dl.Agents, error)

func (c *Registry) ModelManifests(context.Context, *dl.ModelRequest) (*dl.ModelManifestsResponse, error)

func (b *Registry) PublishInPredictor(host, prefix string) error

func (c *Registry) PublishInRegistery() error