huggingfaceembedder

Documentation

Index

• Constants
• type HuggingFaceEmbedder
  • func New() *HuggingFaceEmbedder
  • func (h *HuggingFaceEmbedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
  • func (h *HuggingFaceEmbedder) WithModel(model string) *HuggingFaceEmbedder
  • func (h *HuggingFaceEmbedder) WithToken(token string) *HuggingFaceEmbedder

Constants

const APIBaseURL = "https://api-inference.huggingface.co/pipeline/feature-extraction/"

Types

type HuggingFaceEmbedder

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

func New

func New() *HuggingFaceEmbedder

func (*HuggingFaceEmbedder) Embed

func (h *HuggingFaceEmbedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)

func (*HuggingFaceEmbedder) WithModel

func (h *HuggingFaceEmbedder) WithModel(model string) *HuggingFaceEmbedder

func (*HuggingFaceEmbedder) WithToken

func (h *HuggingFaceEmbedder) WithToken(token string) *HuggingFaceEmbedder