openaiembedder

Documentation

Index

• type Model
• type OpenAIEmbedder
  • func New(model Model) *OpenAIEmbedder
  • func (o *OpenAIEmbedder) Embed(ctx context.Context, texts []string) ([]embedder.Embedding, error)
  • func (o *OpenAIEmbedder) WithClient(client *openai.Client) *OpenAIEmbedder

Types

type Model

type Model = openai.EmbeddingModel

const (
	AdaEmbeddingV2  Model = openai.AdaEmbeddingV2
	SmallEmbedding3 Model = openai.SmallEmbedding3
	LargeEmbedding3 Model = openai.LargeEmbedding3
)

type OpenAIEmbedder

type OpenAIEmbedder struct {
	Name string
	// contains filtered or unexported fields
}

func New

func New(model Model) *OpenAIEmbedder

func (*OpenAIEmbedder) Embed

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

Embed returns the embeddings for the given texts

func (*OpenAIEmbedder) WithClient

func (o *OpenAIEmbedder) WithClient(client *openai.Client) *OpenAIEmbedder

WithClient sets the OpenAI client to use for the embedder