inmemory

package

v0.1.2 Latest Latest Go to latest Published: Apr 12, 2026 License: MIT Imports: 9 Imported by: 0

Details

Package inmemory provides an in-memory vector store implementation.

This section is empty.

This section is empty.

This section is empty.

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

Store is an in-memory vector store that uses cosine similarity.

func New(embedder embeddings.Embedder) *Store

New creates a new in-memory vector store.

func (s *Store) AddDocuments(ctx context.Context, documents []*core.Document) ([]string, error)

AddDocuments embeds and stores documents.

func (s *Store) Delete(_ context.Context, ids []string) error

Delete removes documents by their IDs.

func (s *Store) GetEmbedder() embeddings.Embedder

GetEmbedder returns the embedder.

func (s *Store) SimilaritySearch(ctx context.Context, query string, k int) ([]*core.Document, error)

SimilaritySearch finds the k most similar documents to the query.

func (s *Store) SimilaritySearchWithScore(ctx context.Context, query string, k int) ([]vectorstores.DocumentWithScore, error)

SimilaritySearchWithScore finds the k most similar documents with scores.