Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func CosineSimilarity ¶
CosineSimilarity calculates cosine similarity between two vectors
Types ¶
type EmbeddingGenerator ¶
type EmbeddingGenerator struct {
// contains filtered or unexported fields
}
EmbeddingGenerator generates embeddings for text using BGE model
func NewEmbeddingGenerator ¶
func NewEmbeddingGenerator(enabled bool) *EmbeddingGenerator
NewEmbeddingGenerator creates a new embedding generator
func (*EmbeddingGenerator) Close ¶
func (e *EmbeddingGenerator) Close() error
Close releases resources
func (*EmbeddingGenerator) Dimension ¶
func (e *EmbeddingGenerator) Dimension() int
Dimension returns the embedding dimension
func (*EmbeddingGenerator) Generate ¶
func (e *EmbeddingGenerator) Generate(text string) ([]float32, error)
Generate creates an embedding vector for the given text
func (*EmbeddingGenerator) Initialize ¶
func (e *EmbeddingGenerator) Initialize(modelsDir string) error
Initialize loads the ONNX model and tokenizer
Source Files
Click to show internal directories.
Click to hide internal directories.