llamacppembedder

Documentation

Index

• type LlamaCppEmbedder
  • func New() *LlamaCppEmbedder
  • func (l *LlamaCppEmbedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
  • func (l *LlamaCppEmbedder) WithArgs(llamacppArgs []string) *LlamaCppEmbedder
  • func (l *LlamaCppEmbedder) WithLlamaCppPath(llamacppPath string) *LlamaCppEmbedder
  • func (l *LlamaCppEmbedder) WithModel(modelPath string) *LlamaCppEmbedder

Types

type LlamaCppEmbedder

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

func New

func New() *LlamaCppEmbedder

func (*LlamaCppEmbedder) Embed

func (l *LlamaCppEmbedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)

Embed returns the embeddings for the given texts

func (*LlamaCppEmbedder) WithArgs

func (l *LlamaCppEmbedder) WithArgs(llamacppArgs []string) *LlamaCppEmbedder

WithArgs sets the args to pass to the llamacpp binary

func (*LlamaCppEmbedder) WithLlamaCppPath

func (l *LlamaCppEmbedder) WithLlamaCppPath(llamacppPath string) *LlamaCppEmbedder

WithLlamaCppPath sets the path to the llamacpp binary

func (*LlamaCppEmbedder) WithModel

func (l *LlamaCppEmbedder) WithModel(modelPath string) *LlamaCppEmbedder

WithModel sets the model to use for the embedder