程序包 top.aoyudi.rag.impl

类 InMemoryVectorStore

java.lang.Object

top.aoyudi.rag.impl.InMemoryVectorStore

所有已实现的接口:

VectorStore

public class InMemoryVectorStore
extends Object
implements VectorStore

内存实现的向量存储仓库

嵌套类概要

嵌套类

修饰符和类型	类	说明
private static class	InMemoryVectorStore.DocumentScore	辅助类用于存储文档和相似度分数

字段概要

字段

修饰符和类型	字段	说明
private final EmbeddingGenerator	embeddingGenerator
private final Map<String,List<Document>>	repositoryDocuments
private final TextSplitter	textSplitter
private final VectorSimilarityCalc	vectorSimilarityCalc

构造器概要

构造器

构造器	说明
InMemoryVectorStore(EmbeddingGenerator embeddingGenerator, TextSplitter textSplitter, VectorSimilarityCalc vectorSimilarityCalc)

方法概要

修饰符和类型	方法	说明
void	addDocument(String repositoryKey, Document document)	添加文档向量到指定仓库
void	addDocuments(String repositoryKey, List<Document> documents)	将文档添加到指定向量仓库（会自动分割为块）
void	clear(String repositoryKey)	清空指定仓库的所有文档
void	createRepository(String repositoryKey)	创建新仓库（如果不存在）
void	deleteRepository(String repositoryKey)	删除指定仓库及其所有文档
List<String>	listRepositories()	列出所有现有仓库
List<Document>	similaritySearch(String repositoryKey, float[] queryVector)	根据查询向量从指定仓库检索相似文档
List<Document>	similaritySearch(String repositoryKey, String queryText)	根据查询文本从指定仓库检索相似文档（内部会自动生成向量）

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

字段详细资料

repositoryDocuments

private final Map<String,List<Document>> repositoryDocuments

embeddingGenerator

private final EmbeddingGenerator embeddingGenerator

vectorSimilarityCalc

private final VectorSimilarityCalc vectorSimilarityCalc

textSplitter

private final TextSplitter textSplitter

构造器详细资料

InMemoryVectorStore

public InMemoryVectorStore(EmbeddingGenerator embeddingGenerator,
 TextSplitter textSplitter,
 VectorSimilarityCalc vectorSimilarityCalc)

方法详细资料

createRepository

public void createRepository(String repositoryKey)

从接口复制的说明: VectorStore

创建新仓库（如果不存在）

指定者:

createRepository 在接口中 VectorStore

参数:

repositoryKey - 仓库标识

deleteRepository

public void deleteRepository(String repositoryKey)

从接口复制的说明: VectorStore

删除指定仓库及其所有文档

指定者:

deleteRepository 在接口中 VectorStore

参数:

repositoryKey - 仓库标识

listRepositories

public List<String> listRepositories()

从接口复制的说明: VectorStore

列出所有现有仓库

指定者:

listRepositories 在接口中 VectorStore

返回:

仓库标识列表

addDocument

public void addDocument(String repositoryKey,
 Document document)

从接口复制的说明: VectorStore

添加文档向量到指定仓库

指定者:

addDocument 在接口中 VectorStore

参数:

repositoryKey - 仓库标识

document - 文档对象

addDocuments

public void addDocuments(String repositoryKey,
 List<Document> documents)

从接口复制的说明: VectorStore

将文档添加到指定向量仓库（会自动分割为块）

指定者:

addDocuments 在接口中 VectorStore

参数:

repositoryKey - 目标仓库标识

documents - 文档列表

similaritySearch

public List<Document> similaritySearch(String repositoryKey,
 float[] queryVector)

从接口复制的说明: VectorStore

根据查询向量从指定仓库检索相似文档

指定者:

similaritySearch 在接口中 VectorStore

参数:

repositoryKey - 仓库标识

queryVector - 查询向量

返回:

相似文档列表

similaritySearch

public List<Document> similaritySearch(String repositoryKey,
 String queryText)

从接口复制的说明: VectorStore

根据查询文本从指定仓库检索相似文档（内部会自动生成向量）

指定者:

similaritySearch 在接口中 VectorStore

参数:

repositoryKey - 仓库标识

queryText - 查询文本

返回:

相似文档列表

clear

public void clear(String repositoryKey)

从接口复制的说明: VectorStore

清空指定仓库的所有文档

指定者:

clear 在接口中 VectorStore

参数:

repositoryKey - 仓库标识