openaiembedder

Documentation

Index

• type Model
  • func (m Model) String() string
• 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 int

const (
	Unknown Model = iota
	AdaSimilarity
	BabbageSimilarity
	CurieSimilarity
	DavinciSimilarity
	AdaSearchDocument
	AdaSearchQuery
	BabbageSearchDocument
	BabbageSearchQuery
	CurieSearchDocument
	CurieSearchQuery
	DavinciSearchDocument
	DavinciSearchQuery
	AdaCodeSearchCode
	AdaCodeSearchText
	BabbageCodeSearchCode
	BabbageCodeSearchText
	AdaEmbeddingV2
)

func (Model) String

func (m Model) String() string

type OpenAIEmbedder

type OpenAIEmbedder struct {
	// 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