Package com.alibaba.cloud.ai.service.simple

Class SimpleVectorStoreService

java.lang.Object

com.alibaba.cloud.ai.service.base.BaseVectorStoreService

com.alibaba.cloud.ai.service.simple.SimpleVectorStoreService

@Service
@Primary
public class SimpleVectorStoreService
extends BaseVectorStoreService

Constructor Summary

Constructors

Constructor	Description
SimpleVectorStoreService(org.springframework.ai.embedding.EmbeddingModel embeddingModel, com.google.gson.Gson gson, Accessor dbAccessor, DbConfig dbConfig, AgentVectorStoreManager agentVectorStoreManager)

Method Summary

Modifier and Type	Method	Description
org.springframework.ai.document.Document	convertTableToDocument(TableInfoBO tableInfoBO)
org.springframework.ai.document.Document	convertToDocument(TableInfoBO tableInfoBO, ColumnInfoBO columnInfoBO)
Boolean	deleteDocuments(DeleteRequest deleteRequest)	删除指定条件的向量数据
Boolean	deleteDocumentsForAgent(String agentId, DeleteRequest deleteRequest)	为指定智能体删除向量数据
AgentVectorStoreManager	getAgentVectorStoreManager()	获取智能体向量存储管理器（供其他服务使用）
List<org.springframework.ai.document.Document>	getDocumentsForAgent(String agentId, String query, String vectorType)	为指定智能体获取向量库中的文档 重写父类方法，使用智能体特定的向量存储
protected org.springframework.ai.embedding.EmbeddingModel	getEmbeddingModel()	获取嵌入模型
Boolean	schema(SchemaInitRequest schemaInitRequest)	初始化数据库 schema 到向量库
Boolean	schemaForAgent(String agentId, SchemaInitRequest schemaInitRequest)	为指定智能体初始化数据库 schema 到向量库
List<org.springframework.ai.document.Document>	searchTableByNameAndVectorType(SearchRequest searchRequestDTO)	获取表的文档
List<org.springframework.ai.document.Document>	searchWithFilter(SearchRequest searchRequestDTO)	自定义 filter 的搜索接口
List<org.springframework.ai.document.Document>	searchWithVectorType(SearchRequest searchRequestDTO)	默认 filter 的搜索接口
List<org.springframework.ai.document.Document>	searchWithVectorTypeForAgent(String agentId, SearchRequest searchRequestDTO)	为指定智能体搜索向量数据

Methods inherited from class com.alibaba.cloud.ai.service.base.BaseVectorStoreService

convertToDoubleList, convertToFloatList, embedDouble, embedFloat, getDocuments

Methods inherited from class java.lang.Object

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

Constructor Details

SimpleVectorStoreService

@Autowired
public SimpleVectorStoreService(org.springframework.ai.embedding.EmbeddingModel embeddingModel,
 com.google.gson.Gson gson,
 @Qualifier("mysqlAccessor")
 Accessor dbAccessor,
 DbConfig dbConfig,
 AgentVectorStoreManager agentVectorStoreManager)

Method Details

getEmbeddingModel

protected org.springframework.ai.embedding.EmbeddingModel getEmbeddingModel()

Description copied from class: BaseVectorStoreService

获取嵌入模型

Specified by:

getEmbeddingModel in class BaseVectorStoreService

schema

public Boolean schema(SchemaInitRequest schemaInitRequest)
               throws Exception

初始化数据库 schema 到向量库

Overrides:

schema in class BaseVectorStoreService

Parameters:

schemaInitRequest - schema 初始化请求

Throws:

Exception - 如果发生错误

convertToDocument

public org.springframework.ai.document.Document convertToDocument(TableInfoBO tableInfoBO,
 ColumnInfoBO columnInfoBO)

convertTableToDocument

public org.springframework.ai.document.Document convertTableToDocument(TableInfoBO tableInfoBO)

deleteDocuments

public Boolean deleteDocuments(DeleteRequest deleteRequest)
                        throws Exception

删除指定条件的向量数据

Parameters:

deleteRequest - 删除请求

Returns:

是否删除成功

Throws:

Exception

searchWithVectorType

public List<org.springframework.ai.document.Document> searchWithVectorType(SearchRequest searchRequestDTO)

默认 filter 的搜索接口

Specified by:

searchWithVectorType in class BaseVectorStoreService

searchWithFilter

public List<org.springframework.ai.document.Document> searchWithFilter(SearchRequest searchRequestDTO)

自定义 filter 的搜索接口

Specified by:

searchWithFilter in class BaseVectorStoreService

searchTableByNameAndVectorType

public List<org.springframework.ai.document.Document> searchTableByNameAndVectorType(SearchRequest searchRequestDTO)

Description copied from class: BaseVectorStoreService

获取表的文档

Overrides:

searchTableByNameAndVectorType in class BaseVectorStoreService

schemaForAgent

public Boolean schemaForAgent(String agentId,
 SchemaInitRequest schemaInitRequest)
                       throws Exception

为指定智能体初始化数据库 schema 到向量库

Parameters:

agentId - 智能体ID

schemaInitRequest - schema 初始化请求

Throws:

Exception - 如果发生错误

searchWithVectorTypeForAgent

public List<org.springframework.ai.document.Document> searchWithVectorTypeForAgent(String agentId,
 SearchRequest searchRequestDTO)

为指定智能体搜索向量数据

deleteDocumentsForAgent

public Boolean deleteDocumentsForAgent(String agentId,
 DeleteRequest deleteRequest)
                                throws Exception

为指定智能体删除向量数据

Throws:

Exception

getAgentVectorStoreManager

public AgentVectorStoreManager getAgentVectorStoreManager()

获取智能体向量存储管理器（供其他服务使用）

getDocumentsForAgent

public List<org.springframework.ai.document.Document> getDocumentsForAgent(String agentId,
 String query,
 String vectorType)

为指定智能体获取向量库中的文档 重写父类方法，使用智能体特定的向量存储

Overrides:

getDocumentsForAgent in class BaseVectorStoreService