modrerankernvidia

Documentation

Index

• Constants
• type ReRankerClient
• type RerankerModule
  • func New() *RerankerModule
  • func (m *RerankerModule) AdditionalProperties() map[string]modulecapabilities.AdditionalProperty
  • func (m *RerankerModule) ClassConfigDefaults() map[string]interface{}
  • func (m *RerankerModule) Init(ctx context.Context, params moduletools.ModuleInitParams) error
  • func (m *RerankerModule) MetaInfo() (map[string]interface{}, error)
  • func (m *RerankerModule) Name() string
  • func (m *RerankerModule) PropertyConfigDefaults(dt *schema.DataType) map[string]interface{}
  • func (m *RerankerModule) RootHandler() http.Handler
  • func (m *RerankerModule) Type() modulecapabilities.ModuleType
  • func (m *RerankerModule) ValidateClass(ctx context.Context, class *models.Class, cfg moduletools.ClassConfig) error

Constants

const Name = "reranker-nvidia"

Types

type ReRankerClient

type ReRankerClient interface {
	Rank(ctx context.Context, query string, documents []string, cfg moduletools.ClassConfig) (*ent.RankResult, error)
	MetaInfo() (map[string]interface{}, error)
}

type RerankerModule

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

func New

func New() *RerankerModule

func (*RerankerModule) AdditionalProperties

func (m *RerankerModule) AdditionalProperties() map[string]modulecapabilities.AdditionalProperty

func (*RerankerModule) ClassConfigDefaults

func (m *RerankerModule) ClassConfigDefaults() map[string]interface{}

func (*RerankerModule) Init

func (m *RerankerModule) Init(ctx context.Context,
	params moduletools.ModuleInitParams,
) error

func (*RerankerModule) MetaInfo

func (m *RerankerModule) MetaInfo() (map[string]interface{}, error)

func (*RerankerModule) Name

func (m *RerankerModule) Name() string

func (*RerankerModule) PropertyConfigDefaults

func (m *RerankerModule) PropertyConfigDefaults(
	dt *schema.DataType,
) map[string]interface{}

func (*RerankerModule) RootHandler

func (m *RerankerModule) RootHandler() http.Handler

func (*RerankerModule) Type

func (m *RerankerModule) Type() modulecapabilities.ModuleType

func (*RerankerModule) ValidateClass

func (m *RerankerModule) ValidateClass(ctx context.Context,
	class *models.Class, cfg moduletools.ClassConfig,
) error