embeddings

Documentation

Index

• type Embedder
• type OpenAIClient
  • func NewOpenAIClient(key string) OpenAIClient
  • func NewOpenAIClientWithHTTP(openAIEndpoint, key string, client *http.Client) OpenAIClient
  • func (o *OpenAIClient) EmbedDocuments(_ context.Context, content []string) ([][]float32, error)
  • func (o *OpenAIClient) EmbedQuery(ctx context.Context, content string) ([]float32, error)

Types

type Embedder

type Embedder interface {
	EmbedDocuments(ctx context.Context, texts []string) ([][]float32, error)
	EmbedQuery(ctx context.Context, text string) ([]float32, error)
}

Embedder is an interface matching github.com/tmc/langchaingo embedder used for generating embeddings for a single document and a batch

type OpenAIClient

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

func NewOpenAIClient

func NewOpenAIClient(key string) OpenAIClient

func NewOpenAIClientWithHTTP

func NewOpenAIClientWithHTTP(openAIEndpoint, key string, client *http.Client) OpenAIClient

func (*OpenAIClient) EmbedDocuments

func (o *OpenAIClient) EmbedDocuments(_ context.Context, content []string) ([][]float32, error)

func (*OpenAIClient) EmbedQuery

func (o *OpenAIClient) EmbedQuery(ctx context.Context, content string) ([]float32, error)