vectorstores

type DocumentWithScore struct {
	Document *core.Document
	Score    float64
}

type VectorStore interface {
	// AddDocuments embeds and adds documents to the store.
	AddDocuments(ctx context.Context, documents []*core.Document) ([]string, error)

	// SimilaritySearch searches for documents similar to the query.
	SimilaritySearch(ctx context.Context, query string, k int) ([]*core.Document, error)

	// SimilaritySearchWithScore searches and returns documents with similarity scores.
	SimilaritySearchWithScore(ctx context.Context, query string, k int) ([]DocumentWithScore, error)

	// Delete removes documents by their IDs.
	Delete(ctx context.Context, ids []string) error

	// GetEmbedder returns the embedder used by this store.
	GetEmbedder() embeddings.Embedder
}