nomicembedder

Documentation

Index

• type Embedder
  • func New() *Embedder
  • func (e *Embedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
  • func (e *Embedder) WithAPIKey(apiKey string) *Embedder
  • func (e *Embedder) WithModel(model Model) *Embedder
  • func (e *Embedder) WithTaskType(taskType TaskType) *Embedder
• type Model
• type TaskType
• type Usage

Types

type Embedder

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

func New

func New() *Embedder

func (*Embedder) Embed

func (e *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

func (*Embedder) WithModel

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

func (*Embedder) WithTaskType

func (e *Embedder) WithTaskType(taskType TaskType) *Embedder

type Model

type Model string

const (
	ModelNomicEmbedTextV1 Model = "nomic-embed-text-v1"
	ModelAllMiniLML6V2    Model = "all-MiniLM-L6-v2"
)

type TaskType

type TaskType string

const (
	TaskTypeSearchQuery    TaskType = "search_query"
	TaskTypeSearchDocument TaskType = "search_document"
	TaskTypeClustering     TaskType = "clustering"
	TaskTypeClassification TaskType = "classification"
)

type Usage

type Usage struct {
	TotalTokens int `json:"total_tokens"`
}