vectorstore

Documentation

Index

• type Document
• type ErrCollectionNotFound
  • func (e ErrCollectionNotFound) Error() string
• type VectorStore

Types

type Document

type Document struct {
	Content         string         `json:"content"`
	Metadata        map[string]any `json:"metadata"`
	SimilarityScore float32        `json:"similarity_score"`
}

type ErrCollectionNotFound

type ErrCollectionNotFound struct {
	Collection string
}

func (ErrCollectionNotFound) Error

func (e ErrCollectionNotFound) Error() string

type VectorStore

type VectorStore interface {
	CreateCollection(ctx context.Context, collection string) error
	AddDocuments(ctx context.Context, docs []Document, collection string) ([]string, error)                      // @return documentIDs, error
	SimilaritySearch(ctx context.Context, query string, numDocuments int, collection string) ([]Document, error) //nolint:lll
	RemoveCollection(ctx context.Context, collection string) error
	RemoveDocument(ctx context.Context, documentID string, collection string) error
}