kb

client/src/service/events/generateVector.ts

@@ -56,7 +56,7 @@ export async function generateVector(): Promise<any> {
     ];
 
     // 生成词向量
-    const vectors = await getVector({
+    const { vectors } = await getVector({
       model: data.model,
       input: dataItems.map((item) => item.q),
       userId

client/src/service/models/kb.ts

@@ -19,6 +19,11 @@ const kbSchema = new Schema({
     type: String,
     required: true
   },
+  model: {
+    type: String,
+    required: true,
+    default: 'text-embedding-ada-002'
+  },
   tags: {
     type: [String],
     default: []