embedder

package

v0.3.1 Latest Latest Go to latest Published: Jan 2, 2026 License: MIT Imports: 9 Imported by: 0

Details

View Source

const (
	EmbeddingDim = 384 // all-MiniLM-L6-v2 output dimension
	MaxSeqLen    = 256 // Maximum sequence length
)

This section is empty.

This section is empty.

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

Embedder generates embeddings using ONNX runtime.

func New(modelPath, onnxLibPath string) (*Embedder, error)

New creates a new Embedder.

func (e *Embedder) Close() error

Close destroys the ONNX session.

func (e *Embedder) Embed(texts []string) ([][]float32, error)

Embed generates embeddings for a batch of texts.