embedding

package

v0.10.0 Latest Latest Go to latest Published: Jun 16, 2025 License: MIT Imports: 2 Imported by: 0

Details

This section is empty.

View Source

var NewStringEmbedder = func() Embedder {
	return StringEmbedder{}
}

NewStringEmbedder 创建一个新的StringEmbedder实例

This section is empty.

type Embedder interface {
	Embed(ctx context.Context, text string) ([]float32, error)
	Dimension() int
}

Embedder ...

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

OpenAIEmbedder OpenAI embedder

func NewOpenAIEmbedder(apiKey string, model openai.EmbeddingModel) *OpenAIEmbedder

NewOpenAIEmbedder 创建一个新的OpenAI embedder实例

func (e *OpenAIEmbedder) BatchEmbed(ctx context.Context, texts []string) ([][]float32, error)

BatchEmbed 批量将多个文本转换为向量

func (e *OpenAIEmbedder) Embed(ctx context.Context, text string) ([]float32, error)

Embed 将文本转换为向量

type StringEmbedder struct {
}

StringEmbedder ...

func (s StringEmbedder) BatchEmbed(ctx context.Context, texts []string) ([][]float32, error)

BatchEmbed 批量将多个文本转换为向量

func (s StringEmbedder) Dimension() int

Dimension ...

func (s StringEmbedder) Embed(ctx context.Context, text string) ([]float32, error)

Embed 将文本转换为固定维度的向量