This commit is contained in:
archer
2023-08-25 15:45:24 +08:00
parent 6d93059e25
commit 4b088e84c7
6 changed files with 30 additions and 8 deletions

View File

@@ -3,10 +3,8 @@ import type { KbItemType } from '@/types/plugin';
export const defaultKbDetail: KbItemType = {
_id: '',
userId: '',
updateTime: new Date(),
avatar: '/icon/logo.svg',
name: '',
tags: '',
totalData: 0,
model: 'text-embedding-ada-002'
vectorModelName: 'text-embedding-ada-002'
};

View File

@@ -40,7 +40,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
return getVector({
userId,
input: [q],
model: kb.model
model: kb.vectorModel
});
}
return { vectors: [[]] };

View File

@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
import { jsonRes } from '@/service/response';
import { connectToDatabase, KB } from '@/service/mongo';
import { authUser } from '@/service/utils/auth';
import type { KbUpdateParams } from '@/api/plugins/kb';
import type { KbUpdateParams } from '@/api/request/kb';
export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
try {

View File

@@ -56,7 +56,6 @@ const InputDataModal = ({
};
const { insertLen } = await postKbDataFromList({
kbId,
model: vectorModelList[0].model,
mode: TrainingModeEnum.index,
data: [data]
});

View File

@@ -92,9 +92,9 @@ export const sseResponse = ({
/* add logger */
export const addLog = {
info: (msg: string, obj?: Record<string, any>) => {
global.logger.info(msg, { meta: obj });
global.logger?.info(msg, { meta: obj });
},
error: (msg: string, obj?: Record<string, any>) => {
global.logger.error(msg, { meta: obj });
global.logger?.error(msg, { meta: obj });
}
};