From 92b592dd9851b011af2905d4432d54635c7432d0 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Sat, 3 Jun 2023 09:54:13 +0800
Subject: [PATCH] feat: index type

---
 src/pages/api/openapi/kb/appKbSearch.ts | 4 ++--
 src/service/events/generateVector.ts    | 7 ++++---
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/pages/api/openapi/kb/appKbSearch.ts b/src/pages/api/openapi/kb/appKbSearch.ts
index 0adbcff4d..da4e16e35 100644
--- a/src/pages/api/openapi/kb/appKbSearch.ts
+++ b/src/pages/api/openapi/kb/appKbSearch.ts
@@ -96,9 +96,9 @@ export async function appKbSearch({
     where: [
       `kb_id IN (${model.chat.relatedKbs.map((item) => `'${item}'`).join(',')})`,
       'AND',
-      `vector <=> '[${promptVector[0]}]' < ${similarity}`
+      `vector <#> '[${promptVector[0]}]' < ${similarity}`
     ],
-    order: [{ field: 'vector', mode: `<=> '[${promptVector[0]}]'` }],
+    order: [{ field: 'vector', mode: `<#> '[${promptVector[0]}]'` }],
     limit: 8
   });
 
diff --git a/src/service/events/generateVector.ts b/src/service/events/generateVector.ts
index 376d72bb6..f4728a532 100644
--- a/src/service/events/generateVector.ts
+++ b/src/service/events/generateVector.ts
@@ -133,9 +133,10 @@ export async function generateVector(): Promise<any> {
     }
 
     // unlock
-    await TrainingData.findByIdAndUpdate(trainingId, {
-      lockTime: new Date('2000/1/1')
-    });
+    err.response?.statusText !== 'Too Many Requests' &&
+      (await TrainingData.findByIdAndUpdate(trainingId, {
+        lockTime: new Date('2000/1/1')
+      }));
 
     setTimeout(() => {
       generateVector();