ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-23 13:03:50 +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: [ 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
}); });

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

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