cohereembedder

Documentation

Index

• Variables
• type Embedder
  • func New() *Embedder
  • func (h *Embedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
  • func (e *Embedder) WithAPIKey(apiKey string) *Embedder
  • func (e *Embedder) WithModel(model EmbedderModel) *Embedder
• type EmbedderModel

Variables

var EmbedderModelsSize = map[EmbedderModel]int{
	EmbedderModelEnglishLightV20: 1024,
	EmbedderModelEnglishV20:      4096,
	EmbedderModelMultilingualV20: 768,
}

Types

type Embedder

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

func New

func New() *Embedder

func (*Embedder) Embed

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

Embed returns the embeddings for the given texts

func (*Embedder) WithAPIKey

func (e *Embedder) WithAPIKey(apiKey string) *Embedder

WithAPIKey sets the API key to use for the embedder

func (*Embedder) WithModel

func (e *Embedder) WithModel(model EmbedderModel) *Embedder

WithModel sets the model to use for the embedder

type EmbedderModel

type EmbedderModel = model.EmbedModel

const (
	EmbedderModelEnglishV20      EmbedderModel = model.EmbedModelEnglishV20
	EmbedderModelEnglishLightV20 EmbedderModel = model.EmbedModelEnglishLightV20
	EmbedderModelMultilingualV20 EmbedderModel = model.EmbedModelMultilingualV20
)