mirror of
https://github.com/labring/FastGPT.git
synced 2026-02-27 01:02:22 +08:00
64f70a41c1
* feat(vectordb): add OceanBase HNSW quantization (HNSW_SQ/HNSW_BQ) (#6348) Support OceanBase vector index quantization via VECTOR_VQ_LEVEL: - 32 (default): hnsw + inner_product - 8: hnsw_sq + inner_product (2-3x memory savings) - 1: hnsw_bq + cosine (~15x memory savings) HNSW_BQ requires cosine distance per OceanBase docs. Tested on OceanBase 4.3.5.5 (BP5). Closes #6202 * feat: add test inclusion for vectorDB tests in vitest configuration (#6358) * feat: add test inclusion for vectorDB tests in vitest configuration * refactor: update vectorDB README and setup for environment configuration - Enhanced README to clarify the use of factory pattern for vectorDB integration tests. - Updated instructions for setting up environment variables from a local file. - Removed obsolete PG integration test file and adjusted test execution instructions. - Improved structure explanation for shared test data and factory functions. * perf: integrationTest * feat: vector integration --------- Co-authored-by: ZHANG Yixin <hi.yixinz@gmail.com> Co-authored-by: Jingchao <alswlx@gmail.com>
向量数据库集成测试
对 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_ID、TEST_DATASET_ID、TEST_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.ts 的 drivers 数组中增加一项（name、envKey、createCtrl），无需改 fixtures 或 factory。