bolt

Documentation

Index

• Constants
• type Option
  • func WithBucket(bucket string) Option
  • func WithPath(path string) Option
  • func WithPermission(permission fs.FileMode) Option
  • func WithTimeout(timeout time.Duration) Option
• type VectorStore
  • func NewVectorStore(embeddings llmcomposer.Embeddings, opts ...Option) (*VectorStore, func(), error)
  • func (s *VectorStore) AddDocuments(ctx context.Context, documents ...llmcomposer.Document) error
  • func (s *VectorStore) SimilaritySearch(ctx context.Context, query string, k int) ([]llmcomposer.Document, error)
  • func (s *VectorStore) SimilaritySearchVectorWithScore(_ context.Context, query []float32, k int) ([]llmcomposer.ScoredDocument, error)

Constants

const (
	DefaultPath       = "vector_store.db"
	DefaultBucket     = "embeddings"
	DefaultPermission = 0600
)

Types

type Option

type Option func(*VectorStore)

func WithBucket

func WithBucket(bucket string) Option

func WithPath

func WithPath(path string) Option

func WithPermission

func WithPermission(permission fs.FileMode) Option

func WithTimeout

func WithTimeout(timeout time.Duration) Option

type VectorStore

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

func NewVectorStore

func NewVectorStore(embeddings llmcomposer.Embeddings, opts ...Option) (*VectorStore, func(), error)

func (*VectorStore) AddDocuments

func (s *VectorStore) AddDocuments(ctx context.Context, documents ...llmcomposer.Document) error

func (*VectorStore) SimilaritySearch

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

func (*VectorStore) SimilaritySearchVectorWithScore

func (s *VectorStore) SimilaritySearchVectorWithScore(_ context.Context, query []float32, k int) ([]llmcomposer.ScoredDocument, error)