gitea/FastGPT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: default vector

Browse Source
This commit is contained in:
archer 2023-08-26 19:32:34 +08:00
parent defc73651b
commit 4973d7ad9c
No known key found for this signature in database
GPG Key ID: 569A5660D2379E28
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client/src/pages/api/system/getInitData.ts
View File

@ -76,11 +76,13 @@ const defaultQAModel = {
price: 0 price: 0
}; };
const defaultVectorModels = [ const defaultVectorModels: VectorModelItemType[] = [
{ {
model: 'text-embedding-ada-002', model: 'text-embedding-ada-002',
name: 'Embedding-2', name: 'Embedding-2',
price: 0 price: 0,
defaultToken: 500,
maxToken: 3000
} }
]; ];

2
client/src/service/moduleDispatch/kb/search.ts
View File

@ -32,7 +32,7 @@ export async function dispatchKBSearch(props: Record<string, any>): Promise<KBSe
} }
// get vector // get vector
const vectorModel = kbList[0]?.vectorModel; const vectorModel = kbList[0]?.vectorModel || global.vectorModels[0];
const { vectors, tokenLen } = await getVector({ const { vectors, tokenLen } = await getVector({
model: vectorModel.model, model: vectorModel.model,
input: [userChatInput] input: [userChatInput]