agent

Documentation

Index

• func AddPredictor(framework dl.FrameworkManifest, predictor predict.Predictor) error
• func GetPredictor(framework dl.FrameworkManifest) (predict.Predictor, error)
• func Predictors() []string
• func UploadDir() (string, error)
• type Agent
  • func New(predictor predict.Predictor, opts ...Option) (*Agent, error)
  • func (p *Agent) FindFrameworkModel(ctx context.Context, req *dl.PredictRequest) (*dl.FrameworkManifest, *dl.ModelManifest, error)
  • func (p *Agent) Predict(ctx context.Context, req *dl.PredictRequest) (*dl.PredictResponse, error)
  • func (p *Agent) ReadInput(ctx context.Context, req *dl.PredictRequest) (io.ReadCloser, error)
  • func (p *Agent) RegisterManifests() (*grpc.Server, error)
  • func (p *Agent) RegisterPredictor() (*grpc.Server, error)
• type Base
  • func (b *Base) PublishInPredictor(host, prefix string) error
  • func (b *Base) PublishInRegistery(prefix string) error
• type Option
  • func WithHost(host string) Option
  • func WithPort(port int) Option
• type Options
  • func NewOptions() (*Options, error)
• type Registry
  • 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 (c *Registry) PublishInRegistery() error

Functions

func AddPredictor

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

func GetPredictor

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

func Predictors

func Predictors() []string

func UploadDir

func UploadDir() (string, error)

Types

type Agent

type Agent struct {
	Base
	// contains filtered or unexported fields
}

func New

func New(predictor predict.Predictor, opts ...Option) (*Agent, error)

func (*Agent) FindFrameworkModel

func (p *Agent) FindFrameworkModel(ctx context.Context, req *dl.PredictRequest) (*dl.FrameworkManifest, *dl.ModelManifest, error)

func (*Agent) Predict

func (p *Agent) Predict(ctx context.Context, req *dl.PredictRequest) (*dl.PredictResponse, error)

func (*Agent) ReadInput

func (p *Agent) ReadInput(ctx context.Context, req *dl.PredictRequest) (io.ReadCloser, error)

func (*Agent) RegisterManifests

func (p *Agent) RegisterManifests() (*grpc.Server, error)

func (*Agent) RegisterPredictor

func (p *Agent) RegisterPredictor() (*grpc.Server, error)

type Base

type Base struct {
	Framework dl.FrameworkManifest
}

func (*Base) PublishInPredictor

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

func (*Base) PublishInRegistery

func (b *Base) PublishInRegistery(prefix string) error

type Option

type Option func(o *Options) *Options

func WithHost

func WithHost(host string) Option

func WithPort

func WithPort(port int) Option

type Options

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

func NewOptions

func NewOptions() (*Options, error)

type Registry

type Registry struct {
	Base
}

func (*Registry) FrameworkAgents

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

func (*Registry) FrameworkManifests

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

func (*Registry) ModelAgents

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

func (*Registry) ModelManifests

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

func (*Registry) PublishInRegistery

func (c *Registry) PublishInRegistery() error