pgvector

Documentation

Index

• func Migrate(db *gorm.DB) error
• type Index
  • func (v *Index) From(element *vector.Element) *Index
  • func (v *Index) TableName() string
  • func (v *Index) To() *vector.Doc
  • func (v *Index) ToElement() *vector.Element
  • func (v *Index) Update(vector *Index)
• type PgVectorClient
  • func NewPgVectorClient(log logger.Logger, postgresUrl string) (*PgVectorClient, error)
  • func (p *PgVectorClient) Get(ctx context.Context, oid int64, name string, group int) (*vector.Element, error)
  • func (p *PgVectorClient) Search(ctx context.Context, query vector.VectorDocQuery, vectors []float32, k int) ([]*vector.Doc, error)
  • func (p *PgVectorClient) Store(ctx context.Context, element *vector.Element, extra map[string]any) error

Functions

func Migrate

func Migrate(db *gorm.DB) error

Types

type Index

type Index struct {
	ID        string `gorm:"column:id;primaryKey"`
	Name      string `gorm:"column:name;type:varchar(256);index:index_name"`
	OID       int64  `gorm:"column:oid;index:index_oid"`
	Group     int    `gorm:"column:idx_group;index:index_group"`
	ParentID  int64  `gorm:"column:parent_entry_id;index:index_parent_id"`
	Content   string `gorm:"column:content"`
	Vector    string `gorm:"column:vector;type:vector(1536)"`
	Extra     string `gorm:"column:metadata"`
	CreatedAt int64  `gorm:"column:created_at"`
	ChangedAt int64  `gorm:"column:changed_at"`
}

func (*Index) From

func (v *Index) From(element *vector.Element) *Index

func (*Index) TableName

func (v *Index) TableName() string

func (*Index) To

func (v *Index) To() *vector.Doc

func (*Index) ToElement

func (v *Index) ToElement() *vector.Element

func (*Index) Update

func (v *Index) Update(vector *Index)

type PgVectorClient

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

func NewPgVectorClient

func NewPgVectorClient(log logger.Logger, postgresUrl string) (*PgVectorClient, error)

func (*PgVectorClient) Get

func (p *PgVectorClient) Get(ctx context.Context, oid int64, name string, group int) (*vector.Element, error)

func (*PgVectorClient) Search

func (p *PgVectorClient) Search(ctx context.Context, query vector.VectorDocQuery, vectors []float32, k int) ([]*vector.Doc, error)

func (*PgVectorClient) Store

func (p *PgVectorClient) Store(ctx context.Context, element *vector.Element, extra map[string]any) error