pgvector

Oct 23, 2024

Changes in this version

+ const VsPgvectorEmbeddingConcurrency

Oct 16, 2024

Changes in this version

+ var DefaultHNSWIndex = &HNSWIndex

+ var ErrEmbedderWrongNumberVectors = errors.New("number of vectors from embedder does not match number of documents")

+ var ErrInvalidFilters = errors.New("invalid filters")

+ var ErrInvalidScoreThreshold = errors.New("score threshold must be between 0 and 1")

+ var ErrUnsupportedOptions = errors.New("unsupported options")

+ type CloseNoErr interface

+ Close func()

+ type HNSWIndex struct

+ type PGXConn interface

+ Begin func(ctx context.Context) (pgx.Tx, error)

+ Exec func(ctx context.Context, sql string, arguments ...any) (pgconn.CommandTag, error)

+ Ping func(ctx context.Context) error

+ Query func(ctx context.Context, sql string, arguments ...any) (pgx.Rows, error)

+ QueryRow func(ctx context.Context, sql string, arguments ...any) pgx.Row

+ SendBatch func(ctx context.Context, batch *pgx.Batch) pgx.BatchResults

+ type VectorStore struct

+ func New(ctx context.Context, dsn string, embeddingFunc cg.EmbeddingFunc) (*VectorStore, error)

+ func (v VectorStore) AddDocuments(ctx context.Context, docs []vs.Document, collection string) ([]string, error)

+ func (v VectorStore) CreateCollection(ctx context.Context, collection string) error

+ func (v VectorStore) ExportCollectionsToFile(ctx context.Context, path string, collections ...string) error

+ func (v VectorStore) GetDocuments(ctx context.Context, collection string, where map[string]string, ...) ([]vs.Document, error)

+ func (v VectorStore) ImportCollectionsFromFile(ctx context.Context, path string, collections ...string) error

+ func (v VectorStore) RemoveCollection(ctx context.Context, collection string) error

+ func (v VectorStore) RemoveDocument(ctx context.Context, documentID string, collection string, ...) error

+ func (v VectorStore) SimilaritySearch(ctx context.Context, query string, numDocuments int, collection string, ...) ([]vs.Document, error)