db

Documentation

Index

• type QuantizationParams
  • func (p QuantizationParams) Dequantize(value float64) float64
  • func (p QuantizationParams) Quantize(value float64) float64
• type Storage
  • func NewStorage(dbPath string) (*Storage, error)
  • func (s *Storage) Close() error
  • func (s *Storage) Delete(id string) error
  • func (s *Storage) Get(id string) (*Vector, error)
  • func (s *Storage) GetAll() ([]*Vector, error)
  • func (s *Storage) Insert(vector *Vector) error
• type Vector
  • func (v *Vector) Add(other Vector) error
  • func (v Vector) Distance(other Vector) (float64, error)
  • func (v *Vector) Normalize()
  • func (v *Vector) Prune(threshold float64)
  • func (v *Vector) Quantize(params QuantizationParams)
  • func (v *Vector) Subtract(other Vector) error
  • func (v *Vector) ToSparse()

Types

type QuantizationParams

type QuantizationParams struct {
	Min  float64
	Max  float64
	Bits int
}

func (QuantizationParams) Dequantize

func (p QuantizationParams) Dequantize(value float64) float64

func (QuantizationParams) Quantize

func (p QuantizationParams) Quantize(value float64) float64

type Storage

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

func NewStorage

func NewStorage(dbPath string) (*Storage, error)

func (*Storage) Close

func (s *Storage) Close() error

func (*Storage) Delete

func (s *Storage) Delete(id string) error

func (*Storage) Get

func (s *Storage) Get(id string) (*Vector, error)

func (*Storage) GetAll

func (s *Storage) GetAll() ([]*Vector, error)

func (*Storage) Insert

func (s *Storage) Insert(vector *Vector) error

type Vector

type Vector struct {
	ID                 string
	Embedding          []float64
	Metadata           map[string]string
	Compressed         bool
	QuantizationParams *QuantizationParams
	PruningMask        []bool
	SparseIndices      []int
}

func (*Vector) Add

func (v *Vector) Add(other Vector) error

func (Vector) Distance

func (v Vector) Distance(other Vector) (float64, error)

func (*Vector) Normalize

func (v *Vector) Normalize()

func (*Vector) Prune

func (v *Vector) Prune(threshold float64)

func (*Vector) Quantize

func (v *Vector) Quantize(params QuantizationParams)

func (*Vector) Subtract

func (v *Vector) Subtract(other Vector) error

func (*Vector) ToSparse

func (v *Vector) ToSparse()