feat: index type

src/pages/api/openapi/kb/appKbSearch.ts

@@ -96,9 +96,9 @@ export async function appKbSearch({
     where: [
     where: [
       `kb_id IN (${model.chat.relatedKbs.map((item) => `'${item}'`).join(',')})`,
       `kb_id IN (${model.chat.relatedKbs.map((item) => `'${item}'`).join(',')})`,
       'AND',
       'AND',
-      `vector <=> '[${promptVector[0]}]' < ${similarity}`
+      `vector <#> '[${promptVector[0]}]' < ${similarity}`
     ],
     ],
-    order: [{ field: 'vector', mode: `<=> '[${promptVector[0]}]'` }],
+    order: [{ field: 'vector', mode: `<#> '[${promptVector[0]}]'` }],
     limit: 8
     limit: 8
   });
   });
 
 

src/service/events/generateVector.ts

@@ -133,9 +133,10 @@ export async function generateVector(): Promise<any> {
     }
     }
 
 
     // unlock
     // unlock
-    await TrainingData.findByIdAndUpdate(trainingId, {
+    err.response?.statusText !== 'Too Many Requests' &&
+      (await TrainingData.findByIdAndUpdate(trainingId, {
         lockTime: new Date('2000/1/1')
         lockTime: new Date('2000/1/1')
-    });
+      }));
 
 
     setTimeout(() => {
     setTimeout(() => {
       generateVector();
       generateVector();