uni

package
v0.0.0-...-8a3d196 Latest Latest
Warning

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

Go to latest
Published: Feb 12, 2024 License: BSD-3-Clause Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Float32ToFloat64

func Float32ToFloat64(float32Slice []float32) []float64

func Float64ToFloat32

func Float64ToFloat32(float64Slice []float64) []float32

func WithCohere

func WithCohere() providerIden

func WithCohereEmbed

func WithCohereEmbed(model, truncate, inputType, apikeyOptional string) *withCohereOption

func WithOpenAI

func WithOpenAI() providerIden

func WithOpenAIEmbed

func WithOpenAIEmbed(model openai.Model, apikeyOptional string) *withOpenAIOption

Types

type Embedder

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

func NewEmbedder

func NewEmbedder(opts ...EmbedderOption) *Embedder

func (*Embedder) BatchEmbed

func (m *Embedder) BatchEmbed(texts []string, opts ...WithProviderOption) ([]*Embedding, error)

func (*Embedder) Embed

func (m *Embedder) Embed(text string, opts ...WithProviderOption) (*Embedding, error)

type EmbedderOption

type EmbedderOption interface {
	EmbedderOption()
}

type Embedding

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

func EmbeddingFrom

func EmbeddingFrom(provider WithProviderOption, vector []float64) *Embedding

func EmbeddingFrom32

func EmbeddingFrom32(provider WithProviderOption, vector []float32) *Embedding

func GetMinMaxConcatenatedEmbedding

func GetMinMaxConcatenatedEmbedding(embeddings []*Embedding) (*Embedding, error)

embeddings must all have the same providers

func NewEmbedding

func NewEmbedding() *Embedding

func (*Embedding) GetByProvider

func (em *Embedding) GetByProvider(provider providerIden) ([]float64, error)

func (*Embedding) GetByProvider32

func (em *Embedding) GetByProvider32(provider providerIden) ([]float32, error)

func (*Embedding) Range

func (emb *Embedding) Range(fn func(provider WithProviderOption, vec []float64) error) error

func (*Embedding) Range32

func (emb *Embedding) Range32(fn func(provider WithProviderOption, vec []float32) error) error

func (*Embedding) SetByProvider

func (emb *Embedding) SetByProvider(provider providerIden, vector []float64) error

func (*Embedding) SetByProvider32

func (emb *Embedding) SetByProvider32(provider providerIden, vector []float32) error

func (*Embedding) ToSingleProvider

func (emb *Embedding) ToSingleProvider() (*SingleProviderEmbedding, error)

type SingleProviderEmbedder

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

func NewSingleProviderEmbedder

func NewSingleProviderEmbedder(opts ...EmbedderOption) *SingleProviderEmbedder

func (*SingleProviderEmbedder) BatchEmbed

func (m *SingleProviderEmbedder) BatchEmbed(texts []string, opts ...WithProviderOption) ([]*SingleProviderEmbedding, error)

func (*SingleProviderEmbedder) Embed

type SingleProviderEmbedding

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

func GetMinMaxConcatenatedSingleProviderEmbedding

func GetMinMaxConcatenatedSingleProviderEmbedding(embeddings []*SingleProviderEmbedding) (*SingleProviderEmbedding, error)

embeddings' vectors must all have the same length

func NewSingleProviderEmbedding

func NewSingleProviderEmbedding() *SingleProviderEmbedding

func SingleProviderEmbeddingFrom

func SingleProviderEmbeddingFrom(v []float64) *SingleProviderEmbedding

func SingleProviderEmbeddingFrom32

func SingleProviderEmbeddingFrom32(v []float32) *SingleProviderEmbedding

func (*SingleProviderEmbedding) Get

func (em *SingleProviderEmbedding) Get() []float64

func (*SingleProviderEmbedding) Get32

func (em *SingleProviderEmbedding) Get32() []float32

func (*SingleProviderEmbedding) Set

func (em *SingleProviderEmbedding) Set(vector []float64)

func (*SingleProviderEmbedding) Set32

func (em *SingleProviderEmbedding) Set32(vector []float32)

func (*SingleProviderEmbedding) ToEmbedding

func (s *SingleProviderEmbedding) ToEmbedding(provider WithProviderOption) *Embedding

type WithProviderOption

type WithProviderOption interface {
	WithProviderOption()
}

Jump to

Keyboard shortcuts

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