ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-21 11:43:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: index type

Browse Source
This commit is contained in:
archer
2023-06-03 09:54:13 +08:00
parent 8cafebe26c
commit 92b592dd98
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/pages/api/openapi/kb/appKbSearch.ts
View File

@@ -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
});

7
src/service/events/generateVector.ts
View File

@@ -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();