fix: default vector

This commit is contained in:
archer
2023-08-26 19:32:34 +08:00
parent defc73651b
commit 4973d7ad9c
2 changed files with 5 additions and 3 deletions

View File

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

View File

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