feat: model related kb

2023-05-17 22:09:38 +08:00
parent a79429fdcd
commit 5bf95bd846
16 changed files with 178 additions and 117 deletions
--- a/src/service/plugins/searchKb.ts
+++ b/src/service/plugins/searchKb.ts
@@ -48,7 +48,7 @@ export const searchKb = async ({
          where: [
            ['status', ModelDataStatusEnum.ready],
            'AND',
-            ['model_id', model._id],
+            `kb_id IN (${model.chat.relatedKbs.map((item) => `'${item}'`).join(',')})`,
            'AND',
            `vector <=> '[${promptVector}]' < ${similarity}`
          ],