Versions in this module Expand all Collapse all v0 v0.1.1 Sep 11, 2026 v0.1.0 Sep 8, 2026 Changes in this version + func EmbedQueries(ctx context.Context, e interface{ ... }, texts []string) ([][]float32, error) + func EnsureModelFiles(cacheDir string, spec ModelSpec, log *slog.Logger) (modelPath, tokenizerPath string, err error) + func EnsureONNXEnvironment(cacheDir string, log *slog.Logger) (string, error) + func NewSessionOptionsForProvider(log *slog.Logger) (*ort.SessionOptions, error) + func OrtProvider() string + type Config struct + BatchSize int + CacheDir string + Log *slog.Logger + ModelName string + type Embedder interface + Close func() error + Dimension func() int + Embed func(ctx context.Context, texts []string) ([][]float32, error) + func Synchronized(e Embedder) Embedder + type ModelSpec struct + Dim int + DocPrefix string + EOSTokenID int64 + InputNames []string + MaxTokens int + ModelSHA256 string + ModelURL string + Name string + Pooling string + QueryPrefix string + TokenizerSHA256 string + TokenizerURL string + func GetModel(name string) (ModelSpec, error) + type OnnxEmbedder struct + func NewOnnxEmbedder(ctx context.Context, cfg Config) (*OnnxEmbedder, error) + func (e *OnnxEmbedder) Close() error + func (e *OnnxEmbedder) Dimension() int + func (e *OnnxEmbedder) Embed(ctx context.Context, texts []string) ([][]float32, error) + func (e *OnnxEmbedder) EmbedQueries(ctx context.Context, texts []string) ([][]float32, error) + func (e *OnnxEmbedder) EmbedTokens(ctx context.Context, texts []string) ([][][]float32, error) + type ZeroEmbedder struct + func NewZeroEmbedder() *ZeroEmbedder + func NewZeroEmbedderWithDim(dim int) *ZeroEmbedder + func (z *ZeroEmbedder) Close() error + func (z *ZeroEmbedder) Dimension() int + func (z *ZeroEmbedder) Embed(_ context.Context, texts []string) ([][]float32, error)
Go to main page