chromem

Documentation

Index

• Constants
• 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) ExportCollectionsToFile(ctx context.Context, path string, collections ...string) error
  • func (s *Store) ImportCollectionsFromFile(ctx context.Context, path string, collections ...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)

Constants

const VsChromemEmbeddingParallelThread = "VS_CHROMEM_EMBEDDING_PARALLEL_THREAD"

VsChromemEmbeddingParallelThread can be set as an environment variable to control the number of parallel API calls to create embedding for documents. Default is 100

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

func (s *Store) ExportCollectionsToFile(ctx context.Context, path string, collections ...string) error

func (*Store) ImportCollectionsFromFile

func (s *Store) ImportCollectionsFromFile(ctx context.Context, path string, collections ...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, where map[string]string, whereDocument []chromem.WhereDocument) error

func (*Store) SimilaritySearch

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