fix: default vector

labring · Aug 26, 2023 · 4973d7a · 4973d7a
1 parent defc736
commit 4973d7a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/client/src/pages/api/system/getInitData.ts b/client/src/pages/api/system/getInitData.ts
@@ -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
   }
 ];
 

diff --git a/client/src/service/moduleDispatch/kb/search.ts b/client/src/service/moduleDispatch/kb/search.ts
@@ -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]