vectorizer

Documentation

Index

• func NewClassSettings(cfg moduletools.ClassConfig) *classSettings
• type ClassSettings
• type Client
• type Vectorizer
  • func New(client Client) *Vectorizer
  • func (v *Vectorizer) Object(ctx context.Context, object *models.Object, ...) ([]float32, models.AdditionalProperties, error)
  • func (v *Vectorizer) Texts(ctx context.Context, inputs []string, cfg moduletools.ClassConfig) ([]float32, error)
  • func (v *Vectorizer) VectorizeImage(ctx context.Context, id, image string, cfg moduletools.ClassConfig) ([]float32, error)

Functions

func NewClassSettings

func NewClassSettings(cfg moduletools.ClassConfig) *classSettings

Types

type ClassSettings

type ClassSettings interface {
	ImageField(property string) bool
	ImageFieldsWeights() ([]float32, error)
	TextField(property string) bool
	TextFieldsWeights() ([]float32, error)
	InferenceURL() string
}

type Client

type Client interface {
	Vectorize(ctx context.Context,
		texts, images []string, config ent.VectorizationConfig) (*ent.VectorizationResult, error)
}

type Vectorizer

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

func New

func New(client Client) *Vectorizer

func (*Vectorizer) Object

func (v *Vectorizer) Object(ctx context.Context, object *models.Object,
	comp moduletools.VectorizablePropsComparator, cfg moduletools.ClassConfig,
) ([]float32, models.AdditionalProperties, error)

func (*Vectorizer) Texts

func (v *Vectorizer) Texts(ctx context.Context, inputs []string,
	cfg moduletools.ClassConfig,
) ([]float32, error)

func (*Vectorizer) VectorizeImage

func (v *Vectorizer) VectorizeImage(ctx context.Context, id, image string, cfg moduletools.ClassConfig) ([]float32, error)