向量存储与检索器组件

langchaingo 向量存储接口

1type VectorStore interface {
2	AddDocuments(ctx context.Context, docs []schema.Document, options ...Option) ([]string, error)
3	SimilaritySearch(ctx context.Context, query string, numDocuments int, options ...Option) ([]schema.Document, error) //nolint:lll
4}

检索器接口 —— 向量存储的适配器

1type Retriever struct {
2	CallbacksHandler callbacks.Handler
3	v                VectorStore
4	numDocs          int
5	options          []Option
6}
7
8func (r Retriever) GetRelevantDocuments(ctx context.Context, query string) ([]schema.Document, error) {
9	//...
10}
11
12func ToRetriever(vectorStore VectorStore, numDocuments int, options ...Option) Retriever {
13	//...
14}

使用适配器将向量存储转换为检索器

1store := ... // 向量存储实例；
2retriever := vectorstores.ToRetriever(store, 10) // 获取检索器
3// 在应用其余部分中使用该检索器