chromem

Documentation

Index

• type Store
  • func New(db *chromem.DB, embeddingFunc chromem.EmbeddingFunc) *Store
  • func (s *Store) AddDocuments(ctx context.Context, docs []vs.Document, collection string) ([]string, error)
  • func (s *Store) CreateCollection(_ context.Context, name string) error
  • func (s *Store) RemoveCollection(_ context.Context, collection string) error
  • func (s *Store) RemoveDocument(ctx context.Context, documentID string, collection string) error
  • func (s *Store) SimilaritySearch(ctx context.Context, query string, numDocuments int, collection string) ([]vs.Document, error)

Types

type Store

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

func New

func New(db *chromem.DB, embeddingFunc chromem.EmbeddingFunc) *Store

New creates a new Chromem vector store.

func (*Store) AddDocuments

func (s *Store) AddDocuments(ctx context.Context, docs []vs.Document, collection string) ([]string, error)

func (*Store) CreateCollection

func (s *Store) CreateCollection(_ context.Context, name string) error

func (*Store) RemoveCollection

func (s *Store) RemoveCollection(_ context.Context, collection string) error

func (*Store) RemoveDocument

func (s *Store) RemoveDocument(ctx context.Context, documentID string, collection string) error

func (*Store) SimilaritySearch

func (s *Store) SimilaritySearch(ctx context.Context, query string, numDocuments int, collection string) ([]vs.Document, error)