embedding

Documentation

Index

• type CohereEmbeddingProvider
  • func NewCohereEmbeddingProvider(subType string, inputType string, secretKey string) (*CohereEmbeddingProvider, error)
  • func (p *CohereEmbeddingProvider) GetPricing() string
  • func (p *CohereEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *EmbeddingResult, error)
• type EmbeddingProvider
  • func GetEmbeddingProvider(typ string, subType string, clientId string, clientSecret string, ...) (EmbeddingProvider, error)
• type EmbeddingResult
  • func GetDefaultEmbeddingResult(modelSubType string, text string) (*EmbeddingResult, error)
• type ErnieEmbeddingProvider
  • func NewErnieEmbeddingProvider(subType string, apiKey string, secretKey string) (*ErnieEmbeddingProvider, error)
  • func (e *ErnieEmbeddingProvider) GetPricing() string
  • func (p *ErnieEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *EmbeddingResult, error)
• type GeminiEmbeddingProvider
  • func NewGeminiEmbeddingProvider(subType string, secretKey string) (*GeminiEmbeddingProvider, error)
  • func (p *GeminiEmbeddingProvider) GetPricing() string
  • func (p *GeminiEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *EmbeddingResult, error)
• type HuggingFaceEmbeddingProvider
  • func NewHuggingFaceEmbeddingProvider(subType string, secretKey string) (*HuggingFaceEmbeddingProvider, error)
  • func (p *HuggingFaceEmbeddingProvider) GetPricing() string
  • func (p *HuggingFaceEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *EmbeddingResult, error)
• type LocalEmbeddingProvider
  • func NewAzureEmbeddingProvider(typ string, subType string, deploymentName string, secretKey string, ...) (*LocalEmbeddingProvider, error)
  • func NewLocalEmbeddingProvider(typ string, subType string, secretKey string, providerUrl string) (*LocalEmbeddingProvider, error)
  • func NewOpenAiEmbeddingProvider(typ string, subType string, secretKey string) (*LocalEmbeddingProvider, error)
  • func (p *LocalEmbeddingProvider) GetPricing() string
  • func (p *LocalEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *EmbeddingResult, error)

Types

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) ([]float32, *EmbeddingResult, error)

type EmbeddingProvider

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

func GetEmbeddingProvider

func GetEmbeddingProvider(typ string, subType string, clientId string, clientSecret string, providerUrl string, apiVersion string) (EmbeddingProvider, error)

type EmbeddingResult

type EmbeddingResult struct {
	TokenCount int
	Price      float64
	Currency   string
}

func GetDefaultEmbeddingResult

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

type ErnieEmbeddingProvider

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

func NewErnieEmbeddingProvider

func NewErnieEmbeddingProvider(subType string, apiKey string, secretKey string) (*ErnieEmbeddingProvider, error)

func (*ErnieEmbeddingProvider) GetPricing

func (e *ErnieEmbeddingProvider) GetPricing() string

func (*ErnieEmbeddingProvider) QueryVector

func (p *ErnieEmbeddingProvider) QueryVector(text string, ctx context.Context) ([]float32, *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) ([]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) ([]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) (*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) ([]float32, *EmbeddingResult, error)