Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

embedding

package

Go to main page

Versions in this module

v0

v0.1.5

Mar 15, 2026

Changes in this version

type Provider

+ func WithBEG(modelName, modelPath string) (Provider, error)

+ func WithBERT(modelName, modelPath string) (Provider, error)

v0.1.4

Mar 14, 2026

v0.1.3

Mar 13, 2026

v0.1.2

Mar 13, 2026

Changes in this version

+ type BGEProvider struct

+ func NewBGEProvider(modelPath string) (*BGEProvider, error)

+ func (p *BGEProvider) Dimension() int

+ func (p *BGEProvider) Embed(ctx context.Context, texts []string) ([][]float32, error)

+ type Config struct

+ Dimension int

+ MaxBatchSize int

+ Model EmbeddingModel

+ type DownloadModelInfo struct

+ Description string

+ Name string

+ Size string

+ Type string

+ URLs []string

+ type DownloadProgressCallback func(modelName, fileName string, downloaded, total int64)

+ type Downloader struct

+ func NewDownloader(cacheDir string) *Downloader

+ func (d *Downloader) DownloadModel(modelName string, callback DownloadProgressCallback) (string, error)

+ func (d *Downloader) GetModelInfo() []DownloadModelInfo

+ type EmbeddingModel interface

+ Close func() error

+ Run func(inputs map[string]interface{}) (map[string]interface{}, error)

+ type LocalProvider struct

+ func New(config Config) (*LocalProvider, error)

+ func (p *LocalProvider) Dimension() int

+ func (p *LocalProvider) Embed(ctx context.Context, texts []string) ([][]float32, error)

+ type Model struct

+ func NewModel(dimension int, modelPath string) (*Model, error)

+ func (m *Model) Close() error

+ func (m *Model) Run(inputs map[string]interface{}) (map[string]interface{}, error)

+ type ModelInfo struct

+ Dimension int

+ ModelPath string

+ Name string

+ Type ModelType

+ func NewModelInfo(modelPath string) (*ModelInfo, error)

+ type ModelType string

+ const ModelTypeBERT

+ const ModelTypeBGE

+ const ModelTypeFastText

+ const ModelTypeGPT

+ const ModelTypeGloVe

+ const ModelTypeSentenceBERT

type Provider

+ func NewProvider(modelPath string) (Provider, error)

+ type SentenceBERTProvider struct

+ func NewSentenceBERTProvider(modelPath string) (*SentenceBERTProvider, error)

+ func (p *SentenceBERTProvider) Dimension() int

+ func (p *SentenceBERTProvider) Embed(ctx context.Context, texts []string) ([][]float32, error)

+ type Tokenizer struct

+ func NewTokenizer() (*Tokenizer, error)

+ func (t *Tokenizer) TokenizeBatch(texts []string) ([][]int64, [][]int64, error)

v0.1.1

Mar 12, 2026

Changes in this version

+ type BatchOptions struct

+ MaxBatchSize int

+ MaxConcurrent int

+ type BatchProcessor struct

+ func NewBatchProcessor(provider Provider, options BatchOptions) *BatchProcessor

+ func (bp *BatchProcessor) Process(ctx context.Context, texts []string) ([][]float32, error)

+ func (bp *BatchProcessor) ProcessWithProgress(ctx context.Context, texts []string, callback ProgressCallback) ([][]float32, error)

+ type ProgressCallback func(current, total int, err error) bool

+ type Provider interface

+ Dimension func() int

+ Embed func(ctx context.Context, texts []string) ([][]float32, error)