diff --git a/client/src/api/plugins/kb.ts b/client/src/api/plugins/kb.ts index 1ddf3cf01d..f01efdb7bd 100644 --- a/client/src/api/plugins/kb.ts +++ b/client/src/api/plugins/kb.ts @@ -42,7 +42,7 @@ export const getKbDataList = (data: GetKbDataListProps) => * 获取导出数据(不分页) */ export const getExportDataList = (kbId: string) => - GET<[string, string][]>( + GET<[string, string, string][]>( `/plugins/kb/data/exportModelData`, { kbId }, { diff --git a/client/src/pages/api/plugins/kb/data/exportModelData.ts b/client/src/pages/api/plugins/kb/data/exportModelData.ts index 7d83191971..3bf6cd5b9e 100644 --- a/client/src/pages/api/plugins/kb/data/exportModelData.ts +++ b/client/src/pages/api/plugins/kb/data/exportModelData.ts @@ -42,16 +42,17 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse< where: [['kb_id', kbId], 'AND', ['user_id', userId]] }); // 从 pg 中获取所有数据 - const pgData = await PgClient.select<{ q: string; a: string }>('modelData', { + const pgData = await PgClient.select<{ q: string; a: string; source: string }>('modelData', { where: [['kb_id', kbId], 'AND', ['user_id', userId]], - fields: ['q', 'a'], + fields: ['q', 'a', 'source'], order: [{ field: 'id', mode: 'DESC' }], limit: count }); - const data: [string, string][] = pgData.rows.map((item) => [ + const data: [string, string, string][] = pgData.rows.map((item) => [ item.q.replace(/\n/g, '\\n'), - item.a.replace(/\n/g, '\\n') + item.a.replace(/\n/g, '\\n'), + item.source ]); // update export time diff --git a/client/src/pages/kb/components/DataCard.tsx b/client/src/pages/kb/components/DataCard.tsx index d4bf9e4044..7296354216 100644 --- a/client/src/pages/kb/components/DataCard.tsx +++ b/client/src/pages/kb/components/DataCard.tsx @@ -95,7 +95,7 @@ const DataCard = ({ kbId }: { kbId: string }) => { onSuccess(res) { try { const text = Papa.unparse({ - fields: ['question', 'answer'], + fields: ['question', 'answer', 'source'], data: res }); fileDownload({ diff --git a/client/src/pages/kb/components/Detail.tsx b/client/src/pages/kb/components/Detail.tsx index 33a3606918..1ea22f1a14 100644 --- a/client/src/pages/kb/components/Detail.tsx +++ b/client/src/pages/kb/components/Detail.tsx @@ -36,7 +36,7 @@ const Detail = ({ kbId }: { kbId: string }) => { }); const { reset } = form; - useQuery([kbId, myKbList], () => getKbDetail(kbId), { + useQuery([kbId], () => getKbDetail(kbId), { onSuccess(res) { kbId && setLastKbId(kbId); if (res) {