add baidu mochow vector vdb

Signed-off-by: chenqingxiang04 <[email protected]>

Author: godchen0212

packages/service/common/vectorStore/constants.ts

@@ -4,3 +4,8 @@ export const DatasetVectorTableName = 'modeldata';
 export const PG_ADDRESS = process.env.PG_URL;
 export const MILVUS_ADDRESS = process.env.MILVUS_ADDRESS;
 export const MILVUS_TOKEN = process.env.MILVUS_TOKEN;
+
+export const MOCHOW_ADDRESS = process.env.MOCHOW_ADDRESS;
+export const MOCHOW_ACCOUNT = process.env.MOCHOW_ACCOUNT;
+export const MOCHOW_APIKEY = process.env.MOCHOW_APIKEY;
+export const MOCHOW_REPLICA_NUM = process.env.MOCHOW_REPLICA_NUM;

packages/service/common/vectorStore/controller.ts

@@ -3,12 +3,14 @@ import { PgVectorCtrl } from './pg/class';
 import { getVectorsByText } from '../../core/ai/embedding';
 import { InsertVectorProps } from './controller.d';
 import { EmbeddingModelItemType } from '@fastgpt/global/core/ai/model.d';
-import { MILVUS_ADDRESS, PG_ADDRESS } from './constants';
+import { MILVUS_ADDRESS, MOCHOW_ADDRESS, PG_ADDRESS } from './constants';
 import { MilvusCtrl } from './milvus/class';
+import { MochowCtrl } from './mochow/class';
 
 const getVectorObj = () => {
   if (PG_ADDRESS) return new PgVectorCtrl();
   if (MILVUS_ADDRESS) return new MilvusCtrl();
+  if (MOCHOW_ADDRESS) return new MochowCtrl();
 
   return new PgVectorCtrl();
 };