server

Documentation

Index

• type S
  • func New(syncer *modelsyncer.S) *S
  • func (s *S) DeleteModel(ctx context.Context, req *v1.DeleteModelRequest) (*emptypb.Empty, error)
  • func (s *S) ListModels(ctx context.Context, req *v1.ListModelsRequest) (*v1.ListModelsResponse, error)
  • func (s *S) PullModel(ctx context.Context, req *v1.PullModelRequest) (*emptypb.Empty, error)
  • func (s *S) Run(port int) error
  • func (s *S) Stop()

Types

type S

type S struct {
	v1.UnimplementedInferenceEngineInternalServiceServer
	// contains filtered or unexported fields
}

S is a server.

func New

func New(syncer *modelsyncer.S) *S

New creates a server.

func (*S) DeleteModel

func (s *S) DeleteModel(ctx context.Context, req *v1.DeleteModelRequest) (*emptypb.Empty, error)

DeleteModel removes a model from the engine.

func (*S) ListModels

func (s *S) ListModels(ctx context.Context, req *v1.ListModelsRequest) (*v1.ListModelsResponse, error)

ListModels lists all downloaded models in the engine.

func (*S) PullModel

func (s *S) PullModel(ctx context.Context, req *v1.PullModelRequest) (*emptypb.Empty, error)

PullModel downloads and registers a model with the engine.

func (*S) Run

func (s *S) Run(port int) error

Run starts the gRPC server.

func (*S) Stop

func (s *S) Stop()

Stop stops the gRPC server.