perf: insert kb data

src/service/events/generateQA.ts

@@ -4,7 +4,7 @@ import { OpenAiChatEnum } from '@/constants/model';
 import { pushSplitDataBill } from '@/service/events/pushBill';
 import { generateVector } from './generateVector';
 import { openaiError2 } from '../errorCode';
-import { PgClient } from '@/service/pg';
+import { insertKbItem } from '@/service/pg';
 import { SplitDataSchema } from '@/types/mongoSchema';
 import { modelServiceToolMap } from '../utils/chat';
 import { ChatRoleEnum } from '@/constants/chat';
@@ -133,14 +133,10 @@ A2:
         textList: dataItem.textList.slice(0, -5)
       }),
       // 生成的内容插入 pg
-      PgClient.insert('modelData', {
-        values: resultList.map((item) => [
-          { key: 'user_id', value: dataItem.userId },
-          { key: 'kb_id', value: dataItem.kbId },
-          { key: 'q', value: item.q },
-          { key: 'a', value: item.a },
-          { key: 'status', value: 'waiting' }
-        ])
+      insertKbItem({
+        userId: dataItem.userId,
+        kbId: dataItem.kbId,
+        data: resultList
       })
     ]);
     console.log('生成QA成功，time:', `${(Date.now() - startTime) / 1000}s`);

src/service/pg.ts

@@ -156,3 +156,29 @@ class Pg {
 }
 
 export const PgClient = new Pg();
+
+/**
+ * data insert kb
+ */
+export const insertKbItem = ({
+  userId,
+  kbId,
+  data
+}: {
+  userId: string;
+  kbId: string;
+  data: {
+    q: string;
+    a: string;
+  }[];
+}) => {
+  return PgClient.insert('modelData', {
+    values: data.map((item) => [
+      { key: 'user_id', value: userId },
+      { key: 'kb_id', value: kbId },
+      { key: 'q', value: item.q },
+      { key: 'a', value: item.a },
+      { key: 'status', value: 'waiting' }
+    ])
+  });
+};