Files
FastGPT/test/integrationTest/vectorDB

向量数据库集成测试

对 FastGPT 各向量库控制器(PGVector、后续 Oceanbase/Milvus)做真实环境下的集成测试,保证向量相关操作兼容和稳定。采用工厂模式:同一套数据集(fixtures)和同一套用例(factory)驱动 n 个向量库测试。

环境变量

测试环境变量由 test/.env.test.local 提供(不提交到 git)。请复制模板并填写:

cp test/.env.test.template test/.env.test.local
# 编辑 test/.env.test.local,填入 PG_URL 等

setup.ts 会在测试启动时读取 test/.env.test.local 并注入到 process.env

变量 说明 适用驱动
PG_URL PostgreSQL + pgvector 连接串 PgVectorCtrl
OCEANBASE_URL Oceanbase 连接串(后续) ObVectorCtrl
MILVUS_ADDRESS Milvus 地址(后续) MilvusCtrl

未设置对应环境变量时,该驱动的集成测试会整体跳过,不会报错。

运行方式

在项目根目录执行:

# 仅运行单元测试(未配置 .env.test.local 或未设 PG_URL 时,vectorDB 集成测试会跳过)
pnpm test

# 运行所有向量库测试(包含 vectorDB 集成测试与相关单元测试)
pnpm test:vector

结构说明

  • fixtures.ts:统一测试数据(TEST_TEAM_IDTEST_DATASET_IDTEST_COLLECTION_ID、1536 维 TEST_VECTORS),所有向量库共用。
  • factory.ts:工厂函数 runVectorDBTests(driver),同一套用例(init、insert、getVectorCount、embRecall、getVectorDataByTime、delete)供各驱动复用。
  • integration.test.ts:注册各驱动(PG、后续 Oceanbase/Milvus),按 driver.envKey 决定是否跳过;每个驱动执行同一套 runVectorDBTests(driver)

新增向量库时:在 integration.test.tsdrivers 数组中增加一项(nameenvKeycreateCtrl),无需改 fixtures 或 factory。