perf: export source

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 },
     {

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

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({

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) {