embedding

package
v1.779.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 29, 2026 License: Apache-2.0 Imports: 24 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type AlibabacloudEmbeddingProvider

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

func NewAlibabacloudEmbeddingProvider

func NewAlibabacloudEmbeddingProvider(typ string, subType string, secretKey string, providerUrl string) (*AlibabacloudEmbeddingProvider, error)

func (*AlibabacloudEmbeddingProvider) GetPricing

func (p *AlibabacloudEmbeddingProvider) GetPricing() string

func (*AlibabacloudEmbeddingProvider) QueryVector

func (p *AlibabacloudEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type BaiduCloudEmbeddingProvider

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

func NewBaiduCloudEmbeddingProvider

func NewBaiduCloudEmbeddingProvider(subType string, apiKey string, secretKey string) (*BaiduCloudEmbeddingProvider, error)

func (*BaiduCloudEmbeddingProvider) GetPricing

func (p *BaiduCloudEmbeddingProvider) GetPricing() string

func (*BaiduCloudEmbeddingProvider) QueryVector

func (p *BaiduCloudEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type CohereEmbeddingProvider

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

func NewCohereEmbeddingProvider

func NewCohereEmbeddingProvider(subType string, inputType string, secretKey string) (*CohereEmbeddingProvider, error)

func (*CohereEmbeddingProvider) GetPricing

func (p *CohereEmbeddingProvider) GetPricing() string

func (*CohereEmbeddingProvider) QueryVector

func (p *CohereEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type DummyEmbeddingProvider

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

func NewDummyEmbeddingProvider

func NewDummyEmbeddingProvider(subType string) (*DummyEmbeddingProvider, error)

func (*DummyEmbeddingProvider) GetPricing

func (p *DummyEmbeddingProvider) GetPricing() string

func (*DummyEmbeddingProvider) QueryVector

func (p *DummyEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type EmbeddingProvider

type EmbeddingProvider interface {
	GetPricing() string
	QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)
}

func GetEmbeddingProvider

func GetEmbeddingProvider(typ string, subType string, clientId string, clientSecret string, providerUrl string, apiVersion string, pricePerThousandTokens float64, currency string, lang string) (EmbeddingProvider, error)

type EmbeddingRequest

type EmbeddingRequest struct {
	Texts []string `json:"texts"`
	Model string   `json:"model"`
	Type  string   `json:"type"`
}

type EmbeddingResponse

type EmbeddingResponse struct {
	Vectors      [][]float32 `json:"vectors"`
	Total_tokens int64       `json:"total_tokens"`
	Base_resp    base_resp   `json:"base_resp"`
}

type EmbeddingResult

type EmbeddingResult struct {
	TokenCount int
	Price      float64
	Currency   string
}

func GetDefaultEmbeddingResult

func GetDefaultEmbeddingResult(modelSubType string, text string) (*EmbeddingResult, error)

type GeminiEmbeddingProvider

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

func NewGeminiEmbeddingProvider

func NewGeminiEmbeddingProvider(subType string, secretKey string) (*GeminiEmbeddingProvider, error)

func (*GeminiEmbeddingProvider) GetPricing

func (p *GeminiEmbeddingProvider) GetPricing() string

func (*GeminiEmbeddingProvider) QueryVector

func (p *GeminiEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type HuggingFaceEmbeddingProvider

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

func NewHuggingFaceEmbeddingProvider

func NewHuggingFaceEmbeddingProvider(subType string, secretKey string) (*HuggingFaceEmbeddingProvider, error)

func (*HuggingFaceEmbeddingProvider) GetPricing

func (p *HuggingFaceEmbeddingProvider) GetPricing() string

func (*HuggingFaceEmbeddingProvider) QueryVector

func (p *HuggingFaceEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type JinaEmbeddingProvider

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

func NewJinaEmbeddingProvider

func NewJinaEmbeddingProvider(subType string, apiKey string) (*JinaEmbeddingProvider, error)

func (*JinaEmbeddingProvider) GetPricing

func (p *JinaEmbeddingProvider) GetPricing() string

func (*JinaEmbeddingProvider) QueryVector

func (p *JinaEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type LocalEmbeddingProvider

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

func NewAzureEmbeddingProvider

func NewAzureEmbeddingProvider(typ string, subType string, deploymentName string, secretKey string, providerUrl string, apiVersion string) (*LocalEmbeddingProvider, error)

func NewLocalEmbeddingProvider

func NewLocalEmbeddingProvider(typ string, subType string, secretKey string, providerUrl string, compatibleProvider string, pricePerThousandTokens float64, currency string) (*LocalEmbeddingProvider, error)

func NewOpenAiEmbeddingProvider

func NewOpenAiEmbeddingProvider(typ string, subType string, secretKey string) (*LocalEmbeddingProvider, error)

func (*LocalEmbeddingProvider) GetPricing

func (p *LocalEmbeddingProvider) GetPricing() string

func (*LocalEmbeddingProvider) QueryVector

func (p *LocalEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type MiniMaxEmbeddingProvider

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

func NewMiniMaxEmbeddingProvider

func NewMiniMaxEmbeddingProvider(typ string, subType string, apiKey string, providerUrl string) (*MiniMaxEmbeddingProvider, error)

func (*MiniMaxEmbeddingProvider) GetPricing

func (p *MiniMaxEmbeddingProvider) GetPricing() string

func (*MiniMaxEmbeddingProvider) QueryVector

func (p *MiniMaxEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type TencentCloudEmbeddingProvider

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

func NewTencentCloudEmbeddingProvider

func NewTencentCloudEmbeddingProvider(clientId, clientSecret string, lang string) (*TencentCloudEmbeddingProvider, error)

func (*TencentCloudEmbeddingProvider) GetPricing

func (p *TencentCloudEmbeddingProvider) GetPricing() string

func (*TencentCloudEmbeddingProvider) QueryVector

func (p *TencentCloudEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

type Word2VecEmbeddingProvider

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

func NewWord2VecEmbeddingProvider

func NewWord2VecEmbeddingProvider(typ string, subType string, lang string) (*Word2VecEmbeddingProvider, error)

func (*Word2VecEmbeddingProvider) GetPricing

func (p *Word2VecEmbeddingProvider) GetPricing() string

func (*Word2VecEmbeddingProvider) QueryVector

func (p *Word2VecEmbeddingProvider) QueryVector(text string, ctx context.Context, lang string) ([]float32, *EmbeddingResult, error)

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL