集成新的向量数据库

本文提供了向 WeKnora 项目添加新向量数据库支持的完整指南。通过实现标准化接口和遵循结构化流程，开发者可以高效地集成自定义向量数据库。

类似的扩展开发模式参见添加网络搜索引擎

集成流程

1. 实现基础检索引擎接口

type RetrieveEngine interface {
    EngineType() types.RetrieverEngineType
    Retrieve(ctx context.Context, params types.RetrieveParams) ([]*types.RetrieveResult, error)
    Support() []types.RetrieverType
}

2. 实现存储层接口

实现 RetrieveEngineRepository 接口，扩展基础检索引擎能力，添加索引管理功能：

Save / BatchSave — 保存索引
EstimateStorageSize — 估算存储空间
DeleteByChunkIDList / DeleteByKnowledgeIDList — 删除索引
CopyIndices — 复制索引

3. 实现服务层接口

创建 RetrieveEngineService 实现，负责索引创建和管理的业务逻辑。

4. 添加环境变量配置

RETRIEVE_DRIVER=postgres,elasticsearch_v8,your_database
YOUR_DATABASE_ADDR=host:port
YOUR_DATABASE_USERNAME=username
YOUR_DATABASE_PASSWORD=password

5. 注册检索引擎

在 internal/container/container.go 的 initRetrieveEngineRegistry 中添加初始化与注册逻辑。

6. 定义检索引擎类型常量

在 internal/types/retriever.go 中添加新的引擎类型常量。

参考实现

建议参考现有实现：

PostgreSQL: internal/application/repository/retriever/postgres/
ElasticsearchV7: internal/application/repository/retriever/elasticsearch/v7/
ElasticsearchV8: internal/application/repository/retriever/elasticsearch/v8/

反向链接

Home — Wiki 首页导航
添加网络搜索引擎 — 同类扩展开发模式
知识图谱 — 另一种知识检索方式（基于图而非向量）
版本路线图 — 路线图中的检索能力扩展方向

集成向量数据库