training queue

src/service/pg.ts

@@ -1,5 +1,6 @@
 import { Pool } from 'pg';
 import type { QueryResultRow } from 'pg';
+import { ModelDataStatusEnum } from '@/constants/model';
 
 export const connectPg = async () => {
   if (global.pgClient) {
@@ -168,6 +169,7 @@ export const insertKbItem = ({
   userId: string;
   kbId: string;
   data: {
+    vector: number[];
     q: string;
     a: string;
   }[];
@@ -178,7 +180,8 @@ export const insertKbItem = ({
       { key: 'kb_id', value: kbId },
       { key: 'q', value: item.q },
       { key: 'a', value: item.a },
-      { key: 'status', value: 'waiting' }
+      { key: 'vector', value: `[${item.vector}]` },
+      { key: 'status', value: ModelDataStatusEnum.ready }
     ])
   });
 };