huggingface

Documentation

Index

• func NewHuggingFace(log logger.Logger, conf config.HuggingFaceConfig) embedding.Embedding
• type HuggingFace
  • func (h HuggingFace) VectorDocs(ctx context.Context, docs []string) ([][]float32, []map[string]interface{}, error)
  • func (h HuggingFace) VectorQuery(ctx context.Context, doc string) ([]float32, map[string]interface{}, error)

Functions

func NewHuggingFace

func NewHuggingFace(log logger.Logger, conf config.HuggingFaceConfig) embedding.Embedding

Types

type HuggingFace

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

func (HuggingFace) VectorDocs

func (h HuggingFace) VectorDocs(ctx context.Context, docs []string) ([][]float32, []map[string]interface{}, error)

func (HuggingFace) VectorQuery

func (h HuggingFace) VectorQuery(ctx context.Context, doc string) ([]float32, map[string]interface{}, error)