db

Documentation

Index

• Variables
• type DistributedStorage
  • func NewDistributedStorage(nodeAddresses []string) (*DistributedStorage, error)
  • func (ds *DistributedStorage) Close() error
  • func (ds *DistributedStorage) DeleteObject(id string) error
  • func (ds *DistributedStorage) DeleteVector(id string) error
  • func (ds *DistributedStorage) GetAllObjects() ([]*Object, error)
  • func (ds *DistributedStorage) GetAllVectors() ([]*Vector, error)
  • func (ds *DistributedStorage) GetObject(id string) (*Object, error)
  • func (ds *DistributedStorage) GetVector(id string) (*Vector, error)
  • func (ds *DistributedStorage) InsertObject(object *Object) error
  • func (ds *DistributedStorage) InsertVector(vector *Vector) error
  • func (ds *DistributedStorage) UpdateObjectMetadata(id string, metadata map[string]string) error
  • func (ds *DistributedStorage) UpdateVectorMetadata(id string, metadata map[string]string) error
• type Object
• 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) DeleteObject(id string) error
  • func (s *Storage) DeleteVector(id string) error
  • func (s *Storage) GetAllObjects() ([]*Object, error)
  • func (s *Storage) GetAllVectors() ([]*Vector, error)
  • func (s *Storage) GetObject(id string) (*Object, error)
  • func (s *Storage) GetVector(id string) (*Vector, error)
  • func (s *Storage) InsertObject(object *Object) error
  • func (s *Storage) InsertVector(vector *Vector) error
  • func (s *Storage) UpdateObjectMetadata(id string, metadata map[string]string) error
  • func (s *Storage) UpdateVectorMetadata(id string, metadata map[string]string) 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()

Variables

var ErrNotFound = errors.New("object not found")

var ErrObjectNotFound = errors.New("object not found")

var ErrVectorNotFound = errors.New("vector not found")

Types

type DistributedStorage

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

func NewDistributedStorage

func NewDistributedStorage(nodeAddresses []string) (*DistributedStorage, error)

func (*DistributedStorage) Close

func (ds *DistributedStorage) Close() error

func (*DistributedStorage) DeleteObject

func (ds *DistributedStorage) DeleteObject(id string) error

func (*DistributedStorage) DeleteVector

func (ds *DistributedStorage) DeleteVector(id string) error

func (*DistributedStorage) GetAllObjects

func (ds *DistributedStorage) GetAllObjects() ([]*Object, error)

func (*DistributedStorage) GetAllVectors

func (ds *DistributedStorage) GetAllVectors() ([]*Vector, error)

func (*DistributedStorage) GetObject

func (ds *DistributedStorage) GetObject(id string) (*Object, error)

func (*DistributedStorage) GetVector

func (ds *DistributedStorage) GetVector(id string) (*Vector, error)

func (*DistributedStorage) InsertObject

func (ds *DistributedStorage) InsertObject(object *Object) error

func (*DistributedStorage) InsertVector

func (ds *DistributedStorage) InsertVector(vector *Vector) error

func (*DistributedStorage) UpdateObjectMetadata

func (ds *DistributedStorage) UpdateObjectMetadata(id string, metadata map[string]string) error

func (*DistributedStorage) UpdateVectorMetadata

func (ds *DistributedStorage) UpdateVectorMetadata(id string, metadata map[string]string) error

type Object

type Object struct {
	ID       string            `json:"id"`
	Object   []byte            `json:"object"`
	Metadata map[string]string `json:"metadata"`
}

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) DeleteObject

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

func (*Storage) DeleteVector

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

func (*Storage) GetAllObjects

func (s *Storage) GetAllObjects() ([]*Object, error)

func (*Storage) GetAllVectors

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

func (*Storage) GetObject

func (s *Storage) GetObject(id string) (*Object, error)

func (*Storage) GetVector

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

func (*Storage) InsertObject

func (s *Storage) InsertObject(object *Object) error

func (*Storage) InsertVector

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

func (*Storage) UpdateObjectMetadata

func (s *Storage) UpdateObjectMetadata(id string, metadata map[string]string) error

func (*Storage) UpdateVectorMetadata

func (s *Storage) UpdateVectorMetadata(id string, metadata map[string]string) error

type Vector

type Vector struct {
	ID                 string
	Embedding          []float64
	Metadata           map[string]string
	Text               string
	Object             []byte
	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()